complete restructure

vibe_bot/commands/admin.py

@@ -0,0 +1,135 @@
+"""Admin and debug commands: wipe history, debug menu, chat history."""
+
+from __future__ import annotations
+
+import asyncio
+import logging
+from typing import TYPE_CHECKING
+
+from discord.ext import commands
+
+from vibe_bot.commands._state import require_app, set_app
+from vibe_bot.prompts import get_user_info
+from vibe_bot.textutil import split_message
+from vibe_bot.tools import get_channel_members, get_channel_members_impl
+
+if TYPE_CHECKING:
+    from discord.ext.commands import Bot
+    from discord.ext.commands import Context as CommandsContext
+
+    from vibe_bot.app import App
+
+logger = logging.getLogger(__name__)
+
+
+def register(bot: commands.Bot, app: App) -> None:
+    """Register the admin and debug commands."""
+    set_app(app)
+    bot.command(name="lobotomize")(lobotomize)
+    bot.command(name="debug")(debug)
+    bot.command(name="history")(history)
+
+
+async def lobotomize(ctx: CommandsContext[Bot]) -> None:
+    """Clear all conversation history and memory for all bots."""
+    app = require_app()
+    logger.info("Lobotomize command triggered by user %s", ctx.author.id)
+    await asyncio.to_thread(app.db.clear_all_messages)
+    await ctx.send("All conversation history and memory has been cleared. 🧠✨")
+
+
+async def debug(
+    ctx: CommandsContext[Bot],
+    *,
+    subcommand: str | None = None,
+) -> None:
+    """Debug menu for various debugging sub-commands.
+
+    Usage: !debug <subcommand>
+    Available sub-commands:
+      - members: List all members in the current channel
+      - whoami: Show all information the bot has about you
+      - tools: Show the LLM's available tools
+    """
+    logger.info(
+        "Debug command triggered by user %s with subcommand %r",
+        ctx.author.id,
+        subcommand,
+    )
+
+    if not subcommand:
+        menu = "Debug Menu:\n\n"
+        menu += "Available sub-commands:\n"
+        menu += "- `members` - List all members in the current channel\n"
+        menu += "- `whoami` - Show all information the bot has about you\n"
+        menu += "- `tools` - Show the LLM's available tools"
+        await ctx.send(menu)
+        return
+
+    if subcommand == "members":
+        result = get_channel_members_impl(ctx.channel)
+        for chunk in split_message(result):
+            await ctx.send(chunk)
+        return
+
+    if subcommand == "whoami":
+        user_info = get_user_info(ctx.author)
+        for chunk in split_message(user_info):
+            await ctx.send(chunk)
+        return
+
+    if subcommand == "tools":
+        tool_list = "LLM Tools:\n\n"
+        tool_list += f"- `{get_channel_members.name}`\n"
+        tool_list += f"  Description: {get_channel_members.description}\n"
+        args_schema = get_channel_members.args_schema
+        if isinstance(args_schema, type):
+            tool_list += f"  Parameters: {args_schema.model_json_schema()}"
+        for chunk in split_message(tool_list):
+            await ctx.send(chunk)
+        return
+
+    await ctx.send(
+        f"Unknown debug sub-command: `{subcommand}`\n\n"
+        f"Use `!debug` to see available sub-commands.",
+    )
+
+
+async def history(ctx: CommandsContext[Bot], bot_name: str) -> None:
+    """View the chat history of a custom bot.
+
+    Usage: !history <bot_name>
+    """
+    app = require_app()
+    logger.info(
+        "History command triggered by user %s for bot %r",
+        ctx.author.id,
+        bot_name,
+    )
+
+    bot_info = await asyncio.to_thread(app.manager.get_custom_bot, bot_name)
+
+    if not bot_info:
+        await ctx.send(f"Custom bot '{bot_name}' not found.")
+        return
+
+    history = await asyncio.to_thread(
+        app.db.get_bot_history, bot_name=bot_name, limit=20
+    )
+
+    if not history:
+        await ctx.send(f"No chat history found for **{bot_name}**. ")
+        return
+
+    history.reverse()
+
+    formatted_history: list[str] = []
+    for user_msg, bot_resp in history:
+        formatted_history.append(user_msg)
+        formatted_history.append(f"{bot_name}: {bot_resp}")
+
+    header = f"Chat History for **{bot_name}**:\n\n"
+    full_text = header + "\n---\n".join(formatted_history)
+
+    for chunk in split_message(full_text):
+        await ctx.send(chunk)