feat: add bootstrapping command (#100)

Author: JacobCoffee

byte_bot/byte/bot.py

@@ -14,6 +14,7 @@
 
 from byte_bot.byte.lib import settings
 from byte_bot.byte.lib.log import get_logger
+from byte_bot.server.lib.settings import ServerSettings
 
 __all__ = [
     "Byte",
@@ -22,6 +23,7 @@
 
 logger = get_logger()
 load_dotenv()
+server_settings = ServerSettings()
 
 
 class Byte(Bot):
@@ -115,7 +117,7 @@ async def on_guild_join(self, guild: discord.Guild) -> None:
             guild: Guild object.
         """
         await self.tree.sync(guild=guild)
-        api_url = f"http://0.0.0.0:8000/api/guilds/create?guild_id={guild.id}&guild_name={guild.name}"
+        api_url = f"http://{server_settings.HOST}:{server_settings.PORT}/api/guilds/create?guild_id={guild.id}&guild_name={guild.name}"
 
         try:
             async with httpx.AsyncClient() as client:

byte_bot/byte/plugins/admin.py

@@ -3,14 +3,23 @@
 .. todo:: add an unload cog command.
 """
 
+import discord
+import httpx
 from discord import Interaction
 from discord.app_commands import command as app_command
 from discord.ext import commands
 from discord.ext.commands import Bot, Cog, Context, command, group, is_owner
+from httpx import ConnectError
 
 __all__ = ("AdminCommands", "setup")
 
 from byte_bot.byte.lib.checks import is_byte_dev
+from byte_bot.byte.lib import settings
+from byte_bot.byte.lib.log import get_logger
+from byte_bot.server.lib.settings import ServerSettings
+
+logger = get_logger()
+server_settings = ServerSettings()
 
 
 class AdminCommands(Cog):
@@ -88,6 +97,101 @@ async def tree_sync(self, interaction: Interaction) -> None:
         results = await self.bot.tree.sync()
         await interaction.response.send_message("\n".join(i.name for i in results), ephemeral=True)
 
+    @command(name="bootstrap-guild", help="Bootstrap existing guild to database (dev only).", hidden=True)
+    @is_byte_dev()
+    async def bootstrap_guild(self, ctx: Context, guild_id: int | None = None) -> None:
+        """Bootstrap an existing guild to the database.
+        
+        Args:
+            ctx: Context object.
+            guild_id: Guild ID to bootstrap. If not provided, uses current guild.
+        """
+        guild = await self._get_target_guild(ctx, guild_id)
+        if not guild:
+            return
+            
+        await ctx.send(f"🔄 Bootstrapping guild {guild.name} (ID: {guild.id})...")
+        
+        await self._sync_guild_commands(guild)
+        await self._register_guild_in_database(ctx, guild)
+
+    async def _get_target_guild(self, ctx: Context, guild_id: int | None) -> discord.Guild | None:
+        """Get the target guild for bootstrapping."""
+        target_guild_id = guild_id or (ctx.guild.id if ctx.guild else None)
+        
+        if not target_guild_id:
+            await ctx.send("❌ No guild ID provided and command not used in a guild.")
+            return None
+            
+        guild = self.bot.get_guild(target_guild_id)
+        if not guild:
+            await ctx.send(f"❌ Bot is not in guild with ID {target_guild_id}")
+            return None
+            
+        return guild
+
+    async def _sync_guild_commands(self, guild: discord.Guild) -> None:
+        """Sync commands to the guild."""
+        try:
+            await self.bot.tree.sync(guild=guild)
+            logger.info("Commands synced to guild %s (id: %s)", guild.name, guild.id)
+        except Exception as e:
+            logger.error("Failed to sync commands to guild %s: %s", guild.name, e)
+
+    async def _register_guild_in_database(self, ctx: Context, guild: discord.Guild) -> None:
+        """Register guild in database via API."""
+        api_url = f"http://{server_settings.HOST}:{server_settings.PORT}/api/guilds/create?guild_id={guild.id}&guild_name={guild.name}"
+        
+        try:
+            async with httpx.AsyncClient() as client:
+                response = await client.post(api_url)
+                await self._handle_api_response(ctx, guild, response)
+        except ConnectError:
+            error_msg = f"Failed to connect to API to bootstrap guild {guild.name} (id: {guild.id})"
+            logger.exception(error_msg)
+            await ctx.send(f"❌ {error_msg}")
+
+    async def _handle_api_response(self, ctx: Context, guild: discord.Guild, response: httpx.Response) -> None:
+        """Handle API response for guild registration."""
+        if response.status_code == httpx.codes.CREATED:
+            await self._send_success_message(ctx, guild)
+            await self._notify_dev_channel(guild)
+        elif response.status_code == httpx.codes.CONFLICT:
+            await ctx.send(f"⚠️ Guild {guild.name} already exists in database")
+        else:
+            error_msg = f"Failed to add guild to database (status: {response.status_code})"
+            logger.error(error_msg)
+            await ctx.send(f"❌ {error_msg}")
+
+    async def _send_success_message(self, ctx: Context, guild: discord.Guild) -> None:
+        """Send success message to user."""
+        logger.info("Successfully bootstrapped guild %s (id: %s)", guild.name, guild.id)
+        embed = discord.Embed(
+            title="Guild Bootstrapped",
+            description=f"Successfully bootstrapped guild {guild.name} (ID: {guild.id})",
+            color=discord.Color.green(),
+        )
+        embed.add_field(name="Commands Synced", value="✅", inline=True)
+        embed.add_field(name="Database Entry", value="✅", inline=True)
+        await ctx.send(embed=embed)
+
+    async def _notify_dev_channel(self, guild: discord.Guild) -> None:
+        """Notify dev channel about guild bootstrap."""
+        dev_guild = self.bot.get_guild(settings.discord.DEV_GUILD_ID)
+        if not dev_guild:
+            return
+            
+        dev_channel = dev_guild.get_channel(settings.discord.DEV_GUILD_INTERNAL_ID)
+        if not dev_channel or not hasattr(dev_channel, "send"):
+            return
+            
+        embed = discord.Embed(
+            title="Guild Bootstrapped",
+            description=f"Guild {guild.name} (ID: {guild.id}) was manually bootstrapped",
+            color=discord.Color.blue(),
+        )
+        await dev_channel.send(embed=embed)  # type: ignore[attr-defined]
+
 
 async def setup(bot: Bot) -> None:
     """Add cog to bot.