import discord
from discord import app_commands
from discord.ext import commands
from lib.quiz_helper import *
from lib.leaderboard_helper import *

class Select(discord.ui.Select):
    def __init__(self):
        self.theme = None
        options = [
            discord.SelectOption(label="History", emoji="1️⃣", description="Do you know your past?"),
            discord.SelectOption(label="Geography", emoji="2️⃣", description="Do you know your geography?"),
            discord.SelectOption(label="Pop Culture", emoji="3️⃣", description="What about pop culture?"),
            discord.SelectOption(label="Math", emoji="4️⃣", description="Do you know your math?"),
            discord.SelectOption(label="Riddles", emoji="5️⃣", description="Can you answer these riddles?")
        ]
        super().__init__(custom_id="quizSelect", placeholder="Please select only one set of questions", max_values=1, min_values=1, options=options)

    async def callback(self, interaction: discord.Interaction):
        if self.values[0] == "History":
            self.theme = "History"
        elif self.values[0] == "Geography":
            self.theme = "Geography"
        elif self.values[0] == "Pop Culture":
            self.theme = "Pop Culture"
        elif self.values[0] == "Math":
            self.theme = "Math"
        elif self.values[0] == "Riddles":
            self.theme = "Riddles"

        questions = load_questions(self.theme)
        if self.theme != "Riddles":
            for i in range(len(questions)):
                em = discord.Embed(title=f"Question {i + 1}", description=questions[i])
                await interaction.response.send_message(embed=em, view=load_view(i, questions, self.theme))
        else:
            correct = 0
            for i in range(len(questions)):
                answer = load_answers(i, self.theme)
                em = discord.Embed(title=f"Question {i + 1}", description=f"{questions[i]}\nPlease type your answer")
                try:
                    await interaction.response.send_message(embed=em)
                except discord.InteractionResponded:
                    await interaction.followup.send(embed=em)

                try:
                    message = await interaction.client.wait_for(
                        'message',
                        timeout=30.0,
                        check=lambda m: m.author == interaction.user and m.channel == interaction.channel
                    )

                    if message.author.bot:
                        return

                    # Check if the answer is correct (case-insensitive)
                    if message.content.lower() == answer.lower():
                        correct += 1
                        increment_correct(interaction.user.id, self.theme)
                        await interaction.followup.send("Answer recorded..")
                
                except TimeoutError:
                    await interaction.followup.send(f"Time's up! Please try again.\nYou got {correct} answers correct!")
                    return

                if i == len(questions) - 1:
                    await interaction.followup.send(f"You got {correct} answers correct!")

class SelectView(discord.ui.View):
    def __init__(self):
        super().__init__(timeout=None)
        self.add_item(Select())

class Quiz(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.Cog.listener()
    async def on_ready(self):
        print("Quiz online")

    #quiz = app_commands.Group(name="quiz", description="All the commands related to quiz")

    # command might not work as of now but if it does have an issue ill fix and remove comment
    # if the comment is still here just assume it worked first try
    @app_commands.command(name="leaderboard", description="See who has the highest amount of questions correct")
    async def leaderboard(self, interaction: discord.Interaction, theme: str):
        coll = db.leaderboard
        guild = coll.find_one({"guild_id": interaction.guild.id})

        if guild is None:
            await interaction.response.send_message("An error has occurred..")
            return

        if coll.count_documents({"guild_id": interaction.guild.id}) < 10:
            await interaction.response.send_message("Not enough users to make a leaderboard")
            return

        leaderboard = coll.find({"guild_id": interaction.guild.id}).sort(f"{theme}", pymongo.DESCENDING).limit(10)

        em = discord.Embed(title="Quiz Leaderboard", description="Top 10", color=interaction.user.color)
        for i, user in enumerate(leaderboard):
            em.add_field(name=f"{i}. {self.bot.get_user(user['_id'])}", value=f"Correct: {user[f"{theme}"]["amount_correct"]}")

        await interaction.response.send_message(embed=em)


    @app_commands.command(name="quiz", description="Test your knowledge")
    async def quiz(self, interaction: discord.Interaction):
        em = discord.Embed(title="Quiz", description="Are you ready to test your knowledge? Select a theme from the dropdown below", color=interaction.user.color)
        await interaction.response.send_message(embed=em, view=SelectView())

async def setup(bot):
    await bot.add_cog(Quiz(bot))