feat: big refactor

bot/commands.py

@@ -0,0 +1,63 @@
+from typing import Literal
+
+import discord
+from discord import app_commands
+
+from bot.utils import compile_stats, fetch_game_stats, perform_search
+
+GUILD_ID = 1110531063161299074
+
+
+async def setup(bot):
+    async def on_tree_error(
+        interaction: discord.Interaction, error: app_commands.AppCommandError
+    ):
+        if isinstance(error, app_commands.CommandOnCooldown):
+            return await interaction.response.send_message(
+                f"Command is currently on cooldown! Try again in **{error.retry_after:.2f}** seconds!"
+            )
+        elif isinstance(error, app_commands.MissingPermissions):
+            return await interaction.response.send_message(
+                "You are missing permissions to use that"
+            )
+        else:
+            raise error
+
+    bot.tree.on_error = on_tree_error
+
+    @bot.tree.command(
+        name="search",
+        description="Search for servers by hostname or IP.",
+        guild=discord.Object(id=GUILD_ID),
+    )
+    async def slash_search(interaction: discord.Interaction, query: str):
+        results = await perform_search(query)
+        await interaction.response.send_message(results)
+
+    @app_commands.checks.cooldown(1, 60, key=lambda i: (i.guild_id, i.user.id))
+    @bot.tree.command(
+        name="stats",
+        description="Get stats for a specific game or all games",
+        guild=discord.Object(id=GUILD_ID),
+    )
+    async def stats(
+        interaction: discord.Interaction, game: Literal["iw4", "s1", "iw6", "t7", "all"]
+    ):
+        if game == "all":
+            stats_message = await compile_stats()
+        else:
+            data = await fetch_game_stats(game)
+            if data:
+                stats_message = f"**Stats for {game.upper()}:**\n"
+                count_servers = data.get("countServers", "N/A")
+                count_players = data.get("countPlayers", "N/A")
+                stats_message += f"Total Servers: {count_servers}\n"
+                stats_message += f"Total Players: {count_players}\n"
+            else:
+                stats_message = "Failed to fetch game stats. Please try again later."
+
+        await interaction.response.send_message(stats_message, ephemeral=True)
+
+    await bot.tree.sync(guild=discord.Object(id=GUILD_ID))  # Force sync
+
+    print("Commands extension loaded!")