import discord
from discord.ext import commands
from discord import app_commands
import sqlite3

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

    @app_commands.choices(
        leaderboard_type=[
            app_commands.Choice(name="Accuracy", value="accuracy"),
            app_commands.Choice(name="Perfect Games", value="perfect")
        ],
        period=[
            app_commands.Choice(name="Monthly", value="monthly"),
            app_commands.Choice(name="Yearly", value="yearly"),
            app_commands.Choice(name="All Time", value="alltime")
        ]
    )
    @app_commands.command(
        name="leaderboard",
        description="Show a leaderboard"
    )
    async def leaderboard(
        self,
        interaction: discord.Interaction,
        leaderboard_type: str,
        period: str
    ):
        try:
            if leaderboard_type == "accuracy":
                msg = get_accuracy_leaderboard(
                    period,
                    interaction.guild
                )

            elif leaderboard_type == "perfect":
                msg = get_perfect_score_leaderboard(
                    period,
                    interaction.guild
                )

            await interaction.response.send_message(msg)

        except Exception as e:
            print("Leaderboard error:", e)
            await interaction.response.send_message(
                "Error loading leaderboard."
            )

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

def get_accuracy_leaderboard(period, guild):
    conn = sqlite3.connect("Goonerdle.db")
    cursor = conn.cursor()

    cursor.execute("""
        SELECT
            User_ID,
            COUNT(*) AS games_played,
            ROUND(
                (SUM(Result) * 100.0) /
                (COUNT(*) * 10),
                2
            ) AS accuracy
        FROM Result
        WHERE
            ? = 'alltime'
            OR (
                ? = 'monthly'
                AND DateScore >= date('now', 'start of month')
            )
            OR (
                ? = 'yearly'
                AND DateScore >= date('now', 'start of year')
            )
        GROUP BY User_ID
        HAVING COUNT(*) >= 5
        ORDER BY accuracy DESC
        LIMIT 10
    """, (period, period, period))

    rows = cursor.fetchall()
    conn.close()

    lines = []

    for position, (user_id, games_played, accuracy) in enumerate(rows, start=1):

        member = guild.get_member(user_id)

        if member:
            name = member.display_name
        else:
            name = f"User {user_id}"

        lines.append(
            f"{position}. {name} — **{accuracy:.2f}%** ({games_played} games)"
        )

    return "🏆 **Accuracy Leaderboard** 🏆\n\n" + "\n".join(lines)

def get_perfect_score_leaderboard(period, guild):
    conn = sqlite3.connect("Goonerdle.db")
    cursor = conn.cursor()

    cursor.execute("""
        SELECT
            User_ID,
            COUNT(*) AS perfect_games
        FROM Result
        WHERE
            Result = 10
            AND (
                ? = 'alltime'
                OR (
                    ? = 'monthly'
                    AND DateScore >= date('now', 'start of month')
                )
                OR (
                    ? = 'yearly'
                    AND DateScore >= date('now', 'start of year')
                )
            )
        GROUP BY User_ID
        ORDER BY perfect_games DESC
        LIMIT 10
    """, (period, period, period))

    rows = cursor.fetchall()
    conn.close()

    lines = []

    for position, (user_id, perfect_games) in enumerate(rows, start=1):

        member = guild.get_member(user_id)

        if member:
            name = member.display_name
        else:
            name = f"User {user_id}"

        lines.append(
            f"{position}. {name} — **{perfect_games}** perfect games"
        )

    return "🏆 **Perfect Game Leaderboard** 🏆\n\n" + "\n".join(lines)