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: # TODO Introduce i18n await ctx.respond(f"User **{user.display_name}** is already jailed.") return await pycord_user.jail(self.bot.cache) # TODO Introduce i18n await ctx.respond( f"User **{user.display_name}** has been jailed and cannot interact with events anymore." ) @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: # TODO Introduce i18n await ctx.respond(f"User **{user.display_name}** is not jailed.") return await pycord_user.unjail(self.bot.cache) # TODO Introduce i18n await ctx.respond( f"User **{user.display_name}** has been unjailed and can interact with events again." ) def setup(bot: PycordBot) -> None: bot.add_cog(CogUser(bot))