from discord import (
    ApplicationContext,
    SlashCommandGroup,
    User,
    option,
)
from discord.ext.commands import Cog
from libbot.i18n import _, in_every_locale

from classes import PycordUser
from classes.pycord_bot import PycordBot
from modules.utils import is_operation_confirmed


class CogUser(Cog):
    """Cog with user management commands."""

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

    command_group: SlashCommandGroup = SlashCommandGroup(
        "user",
        description=_("description", "commands", "user"),
        description_localizations=in_every_locale("description", "commands", "user"),
    )

    # TODO Implement the command
    # @command_group.command(
    #     name="create_channel",
    #     description="Create channel for the user",
    # )
    # @option(
    #     "user",
    #     description="Selected user",
    # )
    # async def command_user_create_channel(self, ctx: ApplicationContext, user: User) -> None:
    #     await ctx.respond("Not implemented.")

    # TODO Implement the command
    # @command_group.command(
    #     name="update_channel",
    #     description="Update user's channel",
    # )
    # @option(
    #     "user",
    #     description="Selected user",
    # )
    # async def command_user_update_channel(self, ctx: ApplicationContext, user: User) -> None:
    #     await ctx.respond("Not implemented.")

    # TODO Implement the command
    # @command_group.command(
    #     name="delete_channel",
    #     description="Delete user's channel",
    # )
    # @option(
    #     "user",
    #     description="Selected user",
    # )
    # @option("confirm", description="Confirmation of the operation", required=False)
    # async def command_user_delete_channel(
    #     self, ctx: ApplicationContext, user: User, confirm: bool = False
    # ) -> None:
    #     await ctx.respond("Not implemented.")

    @command_group.command(
        name="jail",
        description=_("description", "commands", "user_jail"),
        description_localizations=in_every_locale("description", "commands", "user_jail"),
    )
    @option(
        "user",
        description=_("description", "commands", "user_jail", "options", "user"),
        description_localizations=in_every_locale(
            "description", "commands", "user_jail", "options", "user"
        ),
    )
    @option(
        "confirm",
        description=_("description", "commands", "user_jail", "options", "confirm"),
        description_localizations=in_every_locale(
            "description", "commands", "user_jail", "options", "confirm"
        ),
        required=False,
    )
    async def command_user_jail(self, ctx: ApplicationContext, user: User, confirm: bool = False) -> None:
        if not (await is_operation_confirmed(ctx, confirm)):
            return

        pycord_user: PycordUser = await self.bot.find_user(user, ctx.guild)

        if pycord_user.is_jailed:
            await ctx.respond(
                self.bot._("user_jail_already_jailed", "messages", locale=ctx.locale).format(
                    display_name=user.display_name
                ),
                ephemeral=True,
            )
            return

        await pycord_user.jail(self.bot.cache)

        await ctx.respond(
            self.bot._("user_jail_successful", "messages", locale=ctx.locale).format(
                display_name=user.display_name
            )
        )

    @command_group.command(
        name="unjail",
        description=_("description", "commands", "user_unjail"),
        description_localizations=in_every_locale("description", "commands", "user_unjail"),
    )
    @option(
        "user",
        description=_("description", "commands", "user_unjail", "options", "user"),
        description_localizations=in_every_locale(
            "description", "commands", "user_unjail", "options", "user"
        ),
    )
    @option(
        "confirm",
        description=_("description", "commands", "user_unjail", "options", "confirm"),
        description_localizations=in_every_locale(
            "description", "commands", "user_unjail", "options", "confirm"
        ),
        required=False,
    )
    async def command_user_unjail(self, ctx: ApplicationContext, user: User, confirm: bool = False) -> None:
        if not (await is_operation_confirmed(ctx, confirm)):
            return

        pycord_user: PycordUser = await self.bot.find_user(user, ctx.guild)

        if not pycord_user.is_jailed:
            await ctx.respond(
                self.bot._("user_unjail_not_jailed", "messages", locale=ctx.locale).format(
                    display_name=user.display_name
                ),
                ephemeral=True,
            )
            return

        await pycord_user.unjail(self.bot.cache)

        await ctx.respond(
            self.bot._("user_unjail_successful", "messages", locale=ctx.locale).format(
                display_name=user.display_name
            )
        )


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