from discord import option
from discord.ext import commands, tasks
import discord
import requests
from bs4 import BeautifulSoup


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

    palworld = discord.SlashCommandGroup("palworld", "Palworld related commands")

    async def get_all_pals(ctx: discord.AutocompleteContext):
        """
        returns a list of all pals in the game, which can then be passed to the /ship command for auto complete
        """
        import requests
        from bs4 import BeautifulSoup

        all_pals = []

        url = "https://paldex.io/palworld/pals/"
        response = requests.get(url)

        soup = BeautifulSoup(response.text, "html.parser")

        x = soup.find_all(
            "p",
            class_="text-base lg:text-lg group-hover:underline underline-offset-4 text-center",
        )
        for pal in x:
            all_pals.append(pal.text)

        return set(all_pals)

    @palworld.command(
        guild_ids=None,
        name="paldeck",
        description="Looks up data about a pal",
    )
    async def pal_lookup(
        self,
        ctx: commands.Context,
        pal: discord.Option(
            str,
            autocomplete=discord.utils.basic_autocomplete(get_all_pals),
            description="Pal to look up",
        ),
    ):
        pal_url = "https://paldex.io/palworld/pals/%s/" % pal.lower().replace(" ", "-")
        response = requests.get(pal_url)
        soup = BeautifulSoup(response.text, "html.parser")

        elements = {}

        pal_elements = soup.find("div", class_="flex flex-row flex-wrap gap-3")
        for div in pal_elements:
            pal_element_badge = div.find("img", class_="rounded-full w-12 px-2 py-2")
            element_type = pal_element_badge.find_next("p").text
            element_icon = "https://paldex.io" + pal_element_badge["src"]
            elements[element_type] = element_icon

        primary_element = list(elements.keys())[0]

        # https://discordpy.readthedocs.io/en/latest/api.html?highlight=colour#discord.Colour
        color_lookup = {
            "Dark": discord.Color.dark_purple(),
            "Dragon": discord.Color.purple(),
            "Earth": discord.Color.greyple(),
            "Electricity": discord.Color.yellow(),
            "Fire": discord.Color.red(),
            "Leaf": discord.Color.green(),
            "Ice": discord.Color.blue(),
            "Normal": discord.Color.lighter_grey(),
            "Water": discord.Color.dark_blue(),
        }

        embed = discord.Embed(
            description="-------",
            color=color_lookup[primary_element],
            type="rich",
            title=pal,
        )

        embed.set_author(
            name=", ".join(elements.keys()),
            icon_url=elements[primary_element],
            url=pal_url,
        )

        embed.set_thumbnail(
            url=soup.find(
                "img", class_="rounded-full w-32 sm:w-auto border-2 border-amber-500"
            )["src"]
        )

        pal_description = soup.find("h3", class_="text-lg mt-2 text-slate-300").text
        embed.add_field(name="**Description**", value=pal_description, inline=False)

        embed.add_field(name="Work Skills", value="-----", inline=False)

        # work skills
        skills_emojis = {
            "Cooling": "❄️",
            "Farming": "🚜",
            "Gathering": "🍃",
            "Generating Electricity": "⚡️",
            "Handwork": "🤚",
            "Kindling": "🔥",
            "Lumbering": "🪵",
            "Medicine Production": "💉",
            "Mining": "⛏️",
            "Planting": "🌱",
            "Transporting": "📦",
            "Watering": "💦",
        }

        specialties = soup.find("div", class_="grid grid-cols-2 grid-rows-6 gap-2")
        for work_skill in specialties.find_all(
            "div",
            class_="inline-flex flex-row gap-1 items-center bg-black-300 border border-amber-300 break-all",
        ):
            skill_name = work_skill.text.split("Lv")[0]
            skill_level = work_skill.text.split("Lv")[1]

            embed.add_field(
                name=skills_emojis[skill_name] + " " + skill_name,
                value="%s" % (int(skill_level) * "⭐️"),
                inline=True,
            )

        # Partner Skill
        embed.add_field(name="Partner Skill", value="-----", inline=False)
        skill_name = soup.find("h2", string="Partner Skill").find_next(
            "p", class_="text-xl"
        )
        skill_description = skill_name.find_next("p", class_="text-lg mr-4").text
        embed.add_field(
            name=skill_name.text,
            value=skill_description,
            inline=False,
        )
        # Drops
        embed.add_field(name="Drops", value="-----", inline=False)
        for div in soup.find("div", class_="flex flex-col gap-2"):
            embed.add_field(
                name=div.text.split("(")[0],
                value="",
                inline=False,
            )
        await ctx.defer()
        await ctx.send_followup(embed=embed)

    @palworld.command(
        guild_ids=None,
        name="elements",
        description="Posts an infographic about which Pal elements are weak/strong against which",
    )
    async def post_medpen_guide(self, ctx: commands.Context):
        await ctx.respond(
            "https://img.game8.co/3822502/5ae8382d16bd390dd19f343e87680d51.png/show"
        )


def setup(bot):
    bot.add_cog(PalWorld(bot))