from bson.errors import InvalidId
from discord import ApplicationContext, Cog, option, slash_command
from discord.utils import basic_autocomplete

from classes import PycordEvent, PycordGuild, PycordUser
from classes.pycord_bot import PycordBot
from modules.utils import autocomplete_active_events


class Register(Cog):
    """Cog with the event registration command."""

    def __init__(self, bot: PycordBot):
        self.bot: PycordBot = bot

    # TODO Introduce i18n
    @slash_command(
        name="register",
        description="Enter the selected event",
    )
    @option(
        "event",
        description="Name of the event",
        autocomplete=basic_autocomplete(autocomplete_active_events),
    )
    async def command_register(self, ctx: ApplicationContext, event: str) -> None:
        guild: PycordGuild = await self.bot.find_guild(ctx.guild.id)

        try:
            pycord_event: PycordEvent = await self.bot.find_event(event_id=event)
        except (InvalidId, RuntimeError):
            # TODO Make a nice message
            await ctx.respond("Event was not found.")
            return

        if not guild.is_configured():
            # TODO Make a nice message
            await ctx.respond("Guild is not configured.")
            return

        user: PycordUser = await self.bot.find_user(ctx.author)

        if pycord_event._id in user.registered_event_ids:
            # TODO Make a nice message
            await ctx.respond("You are already registered for this event.")
            return

        await user.event_register(pycord_event._id, cache=self.bot.cache)

        await ctx.respond("Ok.")


def setup(bot: PycordBot) -> None:
    bot.add_cog(Register(bot))