import discord
from discord import app_commands
from discord.ext import commands
from discord.ext.commands import Context
import aiohttp


def dictionary_command():
    async def send_embed(
        context, embed: discord.Embed, *, ephemeral: bool = False
    ) -> None:
        interaction = getattr(context, "interaction", None)
        if interaction is not None:
            if interaction.response.is_done():
                await interaction.followup.send(embed=embed, ephemeral=ephemeral)
            else:
                await interaction.response.send_message(
                    embed=embed, ephemeral=ephemeral
                )
        else:
            await context.send(embed=embed)

    async def fetch_definition(word: str) -> dict:
        try:
            url = f"https://api.dictionaryapi.dev/api/v2/entries/en/{word.lower()}"

            async with aiohttp.ClientSession() as session:
                async with session.get(url) as response:
                    if response.status == 200:
                        data = await response.json()
                        return {"success": True, "data": data}
                    elif response.status == 404:
                        return {"success": False, "error": "Word not found"}
                    else:
                        return {
                            "success": False,
                            "error": f"API returned status code {response.status}",
                        }
        except aiohttp.ClientError:
            return {"success": False, "error": "Network error occurred"}
        except Exception as e:
            return {"success": False, "error": "An unexpected error occurred"}

    @commands.hybrid_command(
        name="dictionary",
        description="Get the definition of a word",
    )
    @app_commands.describe(word="The word to look up")
    async def dictionary(self, context, word: str = None):
        if not word or not word.strip():
            if (
                context.message
                and context.message.reference
                and context.message.reference.resolved
            ):
                replied_message = context.message.reference.resolved
                if hasattr(replied_message, "content") and replied_message.content:
                    word = replied_message.content.strip().split()[0]
                else:
                    embed = discord.Embed(
                        title="Error",
                        description="The replied message has no text content to look up.",
                        color=0xE02B2B,
                    ).set_author(
                        name="Utility",
                        icon_url="https://yes.nighty.works/raw/8VLDcg.webp",
                    )
                    await send_embed(context, embed, ephemeral=True)
                    return
            else:
                embed = discord.Embed(
                    title="Error",
                    description="Please provide a word to look up or reply to a message with a word.",
                    color=0xE02B2B,
                ).set_author(
                    name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
                )
                await send_embed(context, embed, ephemeral=True)
                return

        word = word.strip().split()[0]

        interaction = getattr(context, "interaction", None)
        if interaction is not None and not interaction.response.is_done():
            await interaction.response.defer()

        result = await fetch_definition(word)

        if not result["success"]:
            error_message = result.get("error", "Unknown error")

            if error_message == "Word not found":
                description = f"Could not find a definition for **{word}**.\nPlease check the spelling and try again."
            else:
                description = f"Failed to fetch definition: {error_message}"

            embed = discord.Embed(
                title="Error",
                description=description,
                color=0xE02B2B,
            ).set_author(
                name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
            )
            await send_embed(context, embed, ephemeral=True)
            return

        data = result["data"]

        if not data or len(data) == 0:
            embed = discord.Embed(
                title="Error",
                description=f"No definition found for **{word}**.",
                color=0xE02B2B,
            ).set_author(
                name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
            )
            await send_embed(context, embed, ephemeral=True)
            return

        entry = data[0]
        word_title = entry.get("word", word)
        phonetic = entry.get("phonetic", "")
        origin = entry.get("origin", "")
        meanings = entry.get("meanings", [])

        embed = discord.Embed(
            title=f"Dictionary",
            description=f"**```{word_title}```**",
            color=0x7289DA,
        )
        embed.set_author(
            name="Utilities", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
        )

        if phonetic:
            embed.add_field(name="Pronunciation", value=f"`{phonetic}`", inline=False)

        max_meanings = 3
        for idx, meaning in enumerate(meanings[:max_meanings]):
            part_of_speech = meaning.get("partOfSpeech", "").capitalize()
            definitions = meaning.get("definitions", [])

            if definitions:
                def_text = ""
                examples = []

                for def_idx, definition in enumerate(definitions[:2], 1):
                    def_line = definition.get("definition", "")
                    example = definition.get("example", "")

                    if def_line:
                        def_text += f"{def_idx}. {def_line}\n"
                        if example:
                            examples.append(f"{def_idx}. {example}")

                if def_text:
                    field_name = (
                        f"{part_of_speech}"
                        if part_of_speech
                        else f"Definition {idx + 1}"
                    )
                    embed.add_field(
                        name=field_name, value=def_text.strip(), inline=False
                    )

                    if examples:
                        example_text = "\n".join(examples)
                        embed.add_field(
                            name="Examples", value=example_text, inline=False
                        )

                        if idx < len(meanings[:max_meanings]) - 1:
                            embed.add_field(name="────────", value="", inline=False)

        if origin and len(origin) < 1000:
            embed.add_field(name="Origin", value=origin, inline=False)

        synonyms = []
        antonyms = []

        for meaning in meanings:
            for definition in meaning.get("definitions", []):
                synonyms.extend(definition.get("synonyms", []))
                antonyms.extend(definition.get("antonyms", []))

        if synonyms:
            synonym_text = ", ".join(synonyms[:10])
            embed.add_field(name="Synonyms", value=synonym_text, inline=True)

        if antonyms:
            antonym_text = ", ".join(antonyms[:10])
            embed.add_field(name="Antonyms", value=antonym_text, inline=True)

        await send_embed(context, embed)

    return dictionary