Allow users to view and delete tournaments

exclamation_mark_charity/modules/tournament/cog.py

@@ -1,12 +1,19 @@
+import asyncio
 import logging
 
 import interactions
+import table2ascii
+from table2ascii import PresetStyle
 
 from exclamation_mark_charity.constants import GUILD_ID, OWNER_IDS
 from exclamation_mark_charity.database import Database
 from exclamation_mark_charity.logger_factory import LoggerFactory
 
 
+def create_button(style: interactions.ButtonStyle, label: str, custom_id: str) -> interactions.Button:
+    return interactions.Button(style=style, label=label, custom_id=custom_id)
+
+
 class Tournament(interactions.Extension):
     def __init__(self, bot: interactions.Client):
         self.bot: interactions.Client = bot
@@ -17,45 +24,121 @@ class Tournament(interactions.Extension):
         description="Tournament Command",
         scope=GUILD_ID,
         options=[
-            interactions.Option(
-                name="update",
-                description="Update The Tournament Details",
-                type=interactions.OptionType.SUB_COMMAND,
-                options=[]
-            ),
             interactions.Option(
                 name="create",
-                description="Create A !charity Tournament",
+                description="Create a !charity tournament",
                 type=interactions.OptionType.SUB_COMMAND,
                 options=[
                     interactions.Option(
-                        name="name",
-                        description="The name of the tournament",
+                        name="title",
+                        description="Title of the tournament",
                         type=interactions.OptionType.STRING,
                         required=True,
                     )
                 ]
+            ),
+            interactions.Option(
+                name="all",
+                description="View all existing tournaments",
+                type=interactions.OptionType.SUB_COMMAND,
+                options=[]
+            ),
+            interactions.Option(
+                name="update",
+                description="Update the tournament details",
+                type=interactions.OptionType.SUB_COMMAND,
+                options=[]
+            ),
+            interactions.Option(
+                name="delete",
+                description="Delete a tournament",
+                type=interactions.OptionType.SUB_COMMAND,
+                options=[]
             )
         ]
     )
-    async def tournament_command(self, ctx: interactions.CommandContext, sub_command, name: str = None):
+    async def tournament_command(self, ctx: interactions.CommandContext, sub_command, title: str = None):
         # TODO: Remove when interactions.py implement application_command_permissions
         if int(ctx.author.id) not in OWNER_IDS:
             await ctx.send("**[ERROR 0069]** -> Not Allowed!")
             return
 
-        await ctx.defer()
+        # Check function to ensure inputs are from the original user who invoked slash command
+        async def check(button_ctx):
+            if int(button_ctx.author.user.id) == int(ctx.author.user.id):
+                return True
+            await button_ctx.send("I wasn't asking you!", ephemeral=True)
+            return False
 
         if sub_command == "create":
-            success, tournament_id = Database.create_tournament(name)
+            row = interactions.ActionRow(
+                components=[
+                    create_button(interactions.ButtonStyle.PRIMARY, "Yes", "success"),
+                    create_button(interactions.ButtonStyle.DANGER, "Cancel", "cancel")
+                ]
+            )
+
+            message: interactions.Message = await ctx.send(
+                f"Create A Tournament With The Name **{title}**?", components=row)
+
+            try:
+                button_ctx: interactions.ComponentContext = await self.bot.wait_for_component(
+                    components=row, check=check, timeout=25
+                )
+                if button_ctx.label == "Yes":
+                    success, tournament_id = Database.create_tournament(title)
+                    if success:
+                        await message.edit(f"Tournament **{title}** Successfully Created!", components=None)
+                        self.logger.info(f"Tournament '{title}' Created Successfully With ID '{tournament_id}'!")
+                    else:
+                        await message.edit("Tournament Creation Failed! **Check Logs!**", components=None)
+                else:
+                    await message.edit("Tournament Creation Aborted!", components=None)
+            except asyncio.TimeoutError:
+                await message.edit("You didn't click it :(", components=None)
+
+        elif sub_command == "all":
+            success, all_tournaments = Database.view_all_tournaments()
             if success:
-                await ctx.send(f"Tournament **{name}** Successfully Created!")
-                self.logger.info(f"Tournament '{name}' Created Successfully With ID '{tournament_id}'!")
+                output = table2ascii.table2ascii(
+                    header=all_tournaments[0],
+                    body=all_tournaments[1:],
+                    style=PresetStyle.thick_compact
+                )
+                await ctx.send(f"```\n{output}\n```")
+                self.logger.info("All Tournaments Successfully Retrieved!")
             else:
-                await ctx.send(f"Tournament Creation Failed! Check Logs!")
+                await ctx.send("Could Not List All Tournaments! **Check Logs!**")
+
         elif sub_command == "update":
             await ctx.send("**[INFO 0069]** Not Implemented Yet!")
 
+        elif sub_command == "delete":
+            success, all_tournaments = Database.view_all_tournaments()
+            tables = [table[0] for table in all_tournaments[1:]]
+
+            menu = interactions.SelectMenu(
+                custom_id="tournament_deletes_menu",
+                options=[interactions.SelectOption(label=table, value=table) for table in tables],
+                placeholder="Select Row(s)",
+                max_values=len(tables)
+            )
+
+            message: interactions.Message = await ctx.send(
+                f"Select Tournaments To **DELETE FOREVER**", components=menu)
+
+            try:
+                menu_ctx: interactions.ComponentContext = await self.bot.wait_for_component(
+                    components=menu, check=check, timeout=25
+                )
+                success = Database.delete_many_tournaments([tuple([row]) for row in menu_ctx.data.values])
+                if success:
+                    await message.edit("Chosen Table(s) Deleted", components=None)
+                else:
+                    await message.edit("Table(s) Could Not Be Deleted! **Check Logs!**", components=None)
+            except asyncio.TimeoutError:
+                await message.edit("You didn't choose anything :(", components=None)
+
 
 def setup(bot):
     Tournament(bot)