from discord.ext import commands
import discord
import requests
from bs4 import BeautifulSoup


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

    @staticmethod
    def get_tracks():
        """
        returns a dictionary of all tracks and their URLs
        """
        url = "https://www.trackpinata.com/"
        response = requests.get(url).text
        soup = BeautifulSoup(response, "html.parser")

        return {
            x.find("h2", class_="thumbnail-header").text: x["href"]
            for x in soup.find("div", class_="track-list w-dyn-items").find_all("a")
        }

    async def get_all_tracks(ctx: discord.AutocompleteContext):
        """
        returns a list of all the cali tracks for use in auto-complete
        """
        return TrackDays.get_tracks().keys()

    @commands.slash_command(
        guild_ids=None,
        name="trackdays",
        description="Look up upcoming trackdays",
        contexts={
            discord.InteractionContextType.guild,
            discord.InteractionContextType.bot_dm,
            discord.InteractionContextType.private_channel,
        },
        integration_types={
            discord.IntegrationType.guild_install,
            discord.IntegrationType.user_install,
        },
    )
    async def trackdays_lookup(
        self,
        ctx: discord.ApplicationContext,
        track: discord.Option(
            str,
            autocomplete=discord.utils.basic_autocomplete(get_all_tracks),
            description="Track to look up days for",
        ),
    ):
        tracks = self.get_tracks()
        track_url = tracks.get(track)
        if not track_url:
            await ctx.respond(f"Track {track} not found")
            return

        base_url = "https://www.trackpinata.com"
        full_url = f"{base_url}{track_url}"

        await ctx.defer()

        data = {}
        response = requests.get(full_url).text
        soup = BeautifulSoup(response, "html.parser")

        months = soup.find_all("h4", class_="month")

        for month in months:
            month_name = month.text
            data[month_name] = {"events": []}

            events = month.find_next(
                "div", class_="collection-list-wrapper w-dyn-list"
            ).find_all("a", class_="list-item w-inline-block")

            for event in events:
                data[month_name]["events"].append(
                    {
                        "reg_url": base_url + event["href"],
                        "provider": event.find("div", class_="text-block").text,
                        "day": event.find("div", class_="text-block-5").text
                        + " "
                        + month_name
                        + " "
                        + event.find("div", class_="text-block-6").text,
                    }
                )

        # Create a month order mapping
        month_order = {
            "January": 1,
            "February": 2,
            "March": 3,
            "April": 4,
            "May": 5,
            "June": 6,
            "July": 7,
            "August": 8,
            "September": 9,
            "October": 10,
            "November": 11,
            "December": 12,
        }

        embed = discord.Embed(description="", color=discord.Color.blue(), type="rich")

        embed.set_author(name=track)
        embed.set_thumbnail(
            url="https://t4.ftcdn.net/jpg/02/80/57/05/360_F_280570531_y52gDRp2ce9YSno3tfuIqKoRcEbn5Eau.jpg"
        )

        for month in sorted(data.keys(), key=lambda x: month_order[x]):
            embed.add_field(
                name=f"🏁 {month} 🏁\n---------",
                value="\n".join(
                    f"[{event['provider']}: {event['day']}]({event['reg_url']})"
                    for event in data[month]["events"]
                ),
                inline=False,
            )

        embed.set_footer("Pulled from https://www.trackpinata.com")
        await ctx.send_followup(embed=embed)


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