code-puppy 0.0.134__py3-none-any.whl → 0.0.136__py3-none-any.whl

code_puppy/command_line/mcp/handler.py

@@ -0,0 +1,133 @@
+"""
+MCP Command Handler - Main router for MCP server management commands.
+
+This module provides the MCPCommandHandler class that routes MCP commands
+to their respective command modules.
+"""
+
+import logging
+import shlex
+
+from code_puppy.messaging import emit_info
+
+from .add_command import AddCommand
+from .base import MCPCommandBase
+from .help_command import HelpCommand
+from .install_command import InstallCommand
+
+# Import all command modules
+from .list_command import ListCommand
+from .logs_command import LogsCommand
+from .remove_command import RemoveCommand
+from .restart_command import RestartCommand
+from .search_command import SearchCommand
+from .start_all_command import StartAllCommand
+from .start_command import StartCommand
+from .status_command import StatusCommand
+from .stop_all_command import StopAllCommand
+from .stop_command import StopCommand
+from .test_command import TestCommand
+
+# Configure logging
+logger = logging.getLogger(__name__)
+
+
+class MCPCommandHandler(MCPCommandBase):
+    """
+    Main command handler for MCP server management operations.
+
+    Routes MCP commands to their respective command modules.
+    Each command is implemented in its own module for better maintainability.
+
+    Example usage:
+        handler = MCPCommandHandler()
+        handler.handle_mcp_command("/mcp list")
+        handler.handle_mcp_command("/mcp start filesystem")
+        handler.handle_mcp_command("/mcp status filesystem")
+    """
+
+    def __init__(self):
+        """Initialize the MCP command handler."""
+        super().__init__()
+
+        # Initialize command handlers
+        self._commands = {
+            "list": ListCommand(),
+            "start": StartCommand(),
+            "start-all": StartAllCommand(),
+            "stop": StopCommand(),
+            "stop-all": StopAllCommand(),
+            "restart": RestartCommand(),
+            "status": StatusCommand(),
+            "test": TestCommand(),
+            "add": AddCommand(),
+            "remove": RemoveCommand(),
+            "logs": LogsCommand(),
+            "search": SearchCommand(),
+            "install": InstallCommand(),
+            "help": HelpCommand(),
+        }
+
+        logger.info("MCPCommandHandler initialized with all command modules")
+
+    def handle_mcp_command(self, command: str) -> bool:
+        """
+        Handle MCP commands and route to appropriate handler.
+
+        Args:
+            command: The full command string (e.g., "/mcp list", "/mcp start server")
+
+        Returns:
+            True if command was handled successfully, False otherwise
+        """
+        group_id = self.generate_group_id()
+
+        try:
+            # Remove /mcp prefix and parse arguments
+            command = command.strip()
+            if not command.startswith("/mcp"):
+                return False
+
+            # Remove the /mcp prefix
+            args_str = command[4:].strip()
+
+            # If no subcommand, show status dashboard
+            if not args_str:
+                self._commands["list"].execute([], group_id=group_id)
+                return True
+
+            # Parse arguments using shlex for proper handling of quoted strings
+            try:
+                args = shlex.split(args_str)
+            except ValueError as e:
+                emit_info(
+                    f"[red]Invalid command syntax: {e}[/red]", message_group=group_id
+                )
+                return True
+
+            if not args:
+                self._commands["list"].execute([], group_id=group_id)
+                return True
+
+            subcommand = args[0].lower()
+            sub_args = args[1:] if len(args) > 1 else []
+
+            # Route to appropriate command handler
+            command_handler = self._commands.get(subcommand)
+            if command_handler:
+                command_handler.execute(sub_args, group_id=group_id)
+                return True
+            else:
+                emit_info(
+                    f"[yellow]Unknown MCP subcommand: {subcommand}[/yellow]",
+                    message_group=group_id,
+                )
+                emit_info(
+                    "Type '/mcp help' for available commands", message_group=group_id
+                )
+                return True
+
+        except Exception as e:
+            logger.error(f"Error handling MCP command '{command}': {e}")
+            emit_info(f"Error executing MCP command: {e}", message_group=group_id)
+            return True

code_puppy/command_line/mcp/help_command.py

@@ -0,0 +1,146 @@
+"""
+MCP Help Command - Shows help for all MCP commands.
+"""
+
+import logging
+from typing import List, Optional
+
+from rich.text import Text
+
+from code_puppy.messaging import emit_info
+
+from .base import MCPCommandBase
+
+# Configure logging
+logger = logging.getLogger(__name__)
+
+
+class HelpCommand(MCPCommandBase):
+    """
+    Command handler for showing MCP command help.
+
+    Displays comprehensive help information for all available MCP commands.
+    """
+
+    def execute(self, args: List[str], group_id: Optional[str] = None) -> None:
+        """
+        Show help for MCP commands.
+
+        Args:
+            args: Command arguments (unused)
+            group_id: Optional message group ID for grouping related messages
+        """
+        if group_id is None:
+            group_id = self.generate_group_id()
+
+        try:
+            # Build help text programmatically to avoid markup conflicts
+            help_lines = []
+
+            # Title
+            help_lines.append(
+                Text("MCP Server Management Commands", style="bold magenta")
+            )
+            help_lines.append(Text(""))
+
+            # Registry Commands
+            help_lines.append(Text("Registry Commands:", style="bold cyan"))
+            help_lines.append(
+                Text("/mcp search", style="cyan")
+                + Text(" [query]     Search 30+ pre-configured servers")
+            )
+            help_lines.append(
+                Text("/mcp install", style="cyan")
+                + Text(" <id>       Install server from registry")
+            )
+            help_lines.append(Text(""))
+
+            # Core Commands
+            help_lines.append(Text("Core Commands:", style="bold cyan"))
+            help_lines.append(
+                Text("/mcp", style="cyan")
+                + Text("                    Show server status dashboard")
+            )
+            help_lines.append(
+                Text("/mcp list", style="cyan")
+                + Text("               List all registered servers")
+            )
+            help_lines.append(
+                Text("/mcp start", style="cyan")
+                + Text(" <name>       Start a specific server")
+            )
+            help_lines.append(
+                Text("/mcp start-all", style="cyan")
+                + Text("          Start all servers")
+            )
+            help_lines.append(
+                Text("/mcp stop", style="cyan")
+                + Text(" <name>        Stop a specific server")
+            )
+            help_lines.append(
+                Text("/mcp stop-all", style="cyan")
+                + Text(" [group_id]  Stop all running servers")
+            )
+            help_lines.append(
+                Text("/mcp restart", style="cyan")
+                + Text(" <name>     Restart a specific server")
+            )
+            help_lines.append(Text(""))
+
+            # Management Commands
+            help_lines.append(Text("Management Commands:", style="bold cyan"))
+            help_lines.append(
+                Text("/mcp status", style="cyan")
+                + Text(" [name]      Show detailed status (all servers or specific)")
+            )
+            help_lines.append(
+                Text("/mcp test", style="cyan")
+                + Text(" <name>        Test connectivity to a server")
+            )
+            help_lines.append(
+                Text("/mcp logs", style="cyan")
+                + Text(" <name> [limit] Show recent events (default limit: 10)")
+            )
+            help_lines.append(
+                Text("/mcp add", style="cyan")
+                + Text(" [json]         Add new server (JSON or wizard)")
+            )
+            help_lines.append(
+                Text("/mcp remove", style="cyan")
+                + Text(" <name>      Remove/disable a server")
+            )
+            help_lines.append(
+                Text("/mcp help", style="cyan")
+                + Text("               Show this help message")
+            )
+            help_lines.append(Text(""))
+
+            # Status Indicators
+            help_lines.append(Text("Status Indicators:", style="bold"))
+            help_lines.append(
+                Text("✓ Running    ✗ Stopped    ⚠ Error    ⏸ Quarantined    ⭐ Popular")
+            )
+            help_lines.append(Text(""))
+
+            # Examples
+            help_lines.append(Text("Examples:", style="bold"))
+            examples_text = """/mcp search database     # Find database servers
+/mcp install postgres    # Install PostgreSQL server
+/mcp start filesystem    # Start a specific server
+/mcp start-all           # Start all servers at once
+/mcp stop-all            # Stop all running servers
+/mcp add {"name": "test", "type": "stdio", "command": "echo"}"""
+            help_lines.append(Text(examples_text, style="dim"))
+
+            # Combine all lines
+            final_text = Text()
+            for i, line in enumerate(help_lines):
+                if i > 0:
+                    final_text.append("\n")
+                final_text.append_text(line)
+
+            emit_info(final_text, message_group=group_id)
+
+        except Exception as e:
+            logger.error(f"Error showing help: {e}")
+            emit_info(f"[red]Error showing help: {e}[/red]", message_group=group_id)

code_puppy/command_line/mcp/install_command.py

@@ -0,0 +1,176 @@
+"""
+MCP Install Command - Installs pre-configured MCP servers from the registry.
+"""
+
+import logging
+from typing import List, Optional
+
+from code_puppy.messaging import emit_info
+from code_puppy.state_management import is_tui_mode
+
+from .base import MCPCommandBase
+from .wizard_utils import run_interactive_install_wizard
+
+# Configure logging
+logger = logging.getLogger(__name__)
+
+
+class InstallCommand(MCPCommandBase):
+    """
+    Command handler for installing MCP servers from registry.
+
+    Installs pre-configured MCP servers with optional interactive wizard.
+    """
+
+    def execute(self, args: List[str], group_id: Optional[str] = None) -> None:
+        """
+        Install a pre-configured MCP server from the registry.
+
+        Args:
+            args: Server ID and optional custom name
+            group_id: Optional message group ID for grouping related messages
+        """
+        if group_id is None:
+            group_id = self.generate_group_id()
+
+        try:
+            # If in TUI mode, show message to use Ctrl+T
+            if is_tui_mode():
+                emit_info(
+                    "In TUI mode, use Ctrl+T to open the MCP Install Wizard",
+                    message_group=group_id,
+                )
+                return
+
+            # In interactive mode, use the comprehensive installer
+            if not args:
+                # No args - launch interactive wizard
+                success = run_interactive_install_wizard(self.manager, group_id)
+                if success:
+                    try:
+                        from code_puppy.agent import reload_mcp_servers
+
+                        reload_mcp_servers()
+                    except ImportError:
+                        pass
+                return
+
+            # Has args - install directly from catalog
+            server_id = args[0]
+            success = self._install_from_catalog(server_id, group_id)
+            if success:
+                try:
+                    from code_puppy.agent import reload_mcp_servers
+
+                    reload_mcp_servers()
+                except ImportError:
+                    pass
+            return
+
+        except ImportError:
+            emit_info("Server registry not available", message_group=group_id)
+        except Exception as e:
+            logger.error(f"Error installing server: {e}")
+            emit_info(f"Installation failed: {e}", message_group=group_id)
+
+    def _install_from_catalog(self, server_name_or_id: str, group_id: str) -> bool:
+        """Install a server directly from the catalog by name or ID."""
+        try:
+            from code_puppy.mcp.server_registry_catalog import catalog
+            from code_puppy.messaging import emit_prompt
+
+            from .utils import find_server_id_by_name
+            from .wizard_utils import install_server_from_catalog
+
+            # Try to find server by ID first, then by name/search
+            selected_server = catalog.get_by_id(server_name_or_id)
+
+            if not selected_server:
+                # Try searching by name
+                results = catalog.search(server_name_or_id)
+                if not results:
+                    emit_info(
+                        f"❌ No server found matching '{server_name_or_id}'",
+                        message_group=group_id,
+                    )
+                    emit_info(
+                        "Try '/mcp install' to browse available servers",
+                        message_group=group_id,
+                    )
+                    return False
+                elif len(results) == 1:
+                    selected_server = results[0]
+                else:
+                    # Multiple matches, show them
+                    emit_info(
+                        f"🔍 Multiple servers found matching '{server_name_or_id}':",
+                        message_group=group_id,
+                    )
+                    for i, server in enumerate(results[:5]):
+                        indicators = []
+                        if server.verified:
+                            indicators.append("✓")
+                        if server.popular:
+                            indicators.append("⭐")
+
+                        indicator_str = ""
+                        if indicators:
+                            indicator_str = " " + "".join(indicators)
+
+                        emit_info(
+                            f"  {i + 1}. {server.display_name}{indicator_str}",
+                            message_group=group_id,
+                        )
+                        emit_info(f"     ID: {server.id}", message_group=group_id)
+
+                    emit_info(
+                        "Please use the exact server ID: '/mcp install <server_id>'",
+                        message_group=group_id,
+                    )
+                    return False
+
+            # Show what we're installing
+            emit_info(
+                f"📦 Installing: {selected_server.display_name}", message_group=group_id
+            )
+            description = (
+                selected_server.description
+                if selected_server.description
+                else "No description available"
+            )
+            emit_info(f"Description: {description}", message_group=group_id)
+            emit_info("", message_group=group_id)
+
+            # Get custom name (default to server name)
+            server_name = emit_prompt(
+                f"Enter custom name for this server [{selected_server.name}]: "
+            ).strip()
+            if not server_name:
+                server_name = selected_server.name
+
+            # Check if name already exists
+            existing_server = find_server_id_by_name(self.manager, server_name)
+            if existing_server:
+                override = emit_prompt(
+                    f"Server '{server_name}' already exists. Override it? [y/N]: "
+                )
+                if not override.lower().startswith("y"):
+                    emit_info("Installation cancelled", message_group=group_id)
+                    return False
+
+            # Install with default configuration (simplified)
+            env_vars = {}
+            cmd_args = {}
+
+            # Install the server
+            return install_server_from_catalog(
+                self.manager, selected_server, server_name, env_vars, cmd_args, group_id
+            )
+
+        except ImportError:
+            emit_info("Server catalog not available", message_group=group_id)
+            return False
+        except Exception as e:
+            logger.error(f"Error installing from catalog: {e}")
+            emit_info(f"[red]Installation error: {e}[/red]", message_group=group_id)
+            return False

code_puppy/command_line/mcp/list_command.py

@@ -0,0 +1,94 @@
+"""
+MCP List Command - Lists all registered MCP servers in a formatted table.
+"""
+
+import logging
+from typing import List, Optional
+
+from rich.table import Table
+from rich.text import Text
+
+from code_puppy.mcp.managed_server import ServerState
+from code_puppy.messaging import emit_info
+
+from .base import MCPCommandBase
+from .utils import format_state_indicator, format_uptime
+
+# Configure logging
+logger = logging.getLogger(__name__)
+
+
+class ListCommand(MCPCommandBase):
+    """
+    Command handler for listing MCP servers.
+
+    Displays all registered MCP servers in a formatted table with status information.
+    """
+
+    def execute(self, args: List[str], group_id: Optional[str] = None) -> None:
+        """
+        List all registered MCP servers in a formatted table.
+
+        Args:
+            args: Command arguments (unused for list command)
+            group_id: Optional message group ID for grouping related messages
+        """
+        if group_id is None:
+            group_id = self.generate_group_id()
+
+        try:
+            servers = self.manager.list_servers()
+
+            if not servers:
+                emit_info("No MCP servers registered", message_group=group_id)
+                return
+
+            # Create table for server list
+            table = Table(title="🔌 MCP Server Status Dashboard")
+            table.add_column("Name", style="cyan", no_wrap=True)
+            table.add_column("Type", style="dim", no_wrap=True)
+            table.add_column("State", justify="center")
+            table.add_column("Enabled", justify="center")
+            table.add_column("Uptime", style="dim")
+            table.add_column("Status", style="dim")
+
+            for server in servers:
+                # Format state with appropriate color and icon
+                state_display = format_state_indicator(server.state)
+
+                # Format enabled status
+                enabled_display = "✓" if server.enabled else "✗"
+                enabled_style = "green" if server.enabled else "red"
+
+                # Format uptime
+                uptime_display = format_uptime(server.uptime_seconds)
+
+                # Format status message
+                status_display = server.error_message or "OK"
+                if server.quarantined:
+                    status_display = "Quarantined"
+
+                table.add_row(
+                    server.name,
+                    server.type.upper(),
+                    state_display,
+                    Text(enabled_display, style=enabled_style),
+                    uptime_display,
+                    status_display,
+                )
+
+            emit_info(table, message_group=group_id)
+
+            # Show summary
+            total = len(servers)
+            running = sum(
+                1 for s in servers if s.state == ServerState.RUNNING and s.enabled
+            )
+            emit_info(
+                f"\n📊 Summary: {running}/{total} servers running",
+                message_group=group_id,
+            )
+
+        except Exception as e:
+            logger.error(f"Error listing MCP servers: {e}")
+            emit_info(f"[red]Error listing servers: {e}[/red]", message_group=group_id)

code_puppy/command_line/mcp/logs_command.py

@@ -0,0 +1,126 @@
+"""
+MCP Logs Command - Shows recent events/logs for a server.
+"""
+
+import logging
+from datetime import datetime
+from typing import List, Optional
+
+from rich.table import Table
+from rich.text import Text
+
+from code_puppy.messaging import emit_info
+
+from .base import MCPCommandBase
+from .utils import find_server_id_by_name, suggest_similar_servers
+
+# Configure logging
+logger = logging.getLogger(__name__)
+
+
+class LogsCommand(MCPCommandBase):
+    """
+    Command handler for showing MCP server logs.
+
+    Shows recent events/logs for a specific MCP server with configurable limit.
+    """
+
+    def execute(self, args: List[str], group_id: Optional[str] = None) -> None:
+        """
+        Show recent events/logs for a server.
+
+        Args:
+            args: Command arguments, expects [server_name] and optional [limit]
+            group_id: Optional message group ID for grouping related messages
+        """
+        if group_id is None:
+            group_id = self.generate_group_id()
+
+        if not args:
+            emit_info("Usage: /mcp logs <server_name> [limit]", message_group=group_id)
+            return
+
+        server_name = args[0]
+        limit = 10  # Default limit
+
+        if len(args) > 1:
+            try:
+                limit = int(args[1])
+                if limit <= 0 or limit > 100:
+                    emit_info(
+                        "Limit must be between 1 and 100, using default: 10",
+                        message_group=group_id,
+                    )
+                    limit = 10
+            except ValueError:
+                emit_info(
+                    f"Invalid limit '{args[1]}', using default: 10",
+                    message_group=group_id,
+                )
+
+        try:
+            # Find server by name
+            server_id = find_server_id_by_name(self.manager, server_name)
+            if not server_id:
+                emit_info(f"Server '{server_name}' not found", message_group=group_id)
+                suggest_similar_servers(self.manager, server_name, group_id=group_id)
+                return
+
+            # Get server status which includes recent events
+            status = self.manager.get_server_status(server_id)
+
+            if not status.get("exists", True):
+                emit_info(
+                    f"Server '{server_name}' status not available",
+                    message_group=group_id,
+                )
+                return
+
+            recent_events = status.get("recent_events", [])
+
+            if not recent_events:
+                emit_info(
+                    f"No recent events for server: {server_name}",
+                    message_group=group_id,
+                )
+                return
+
+            # Show events in a table
+            table = Table(title=f"📋 Recent Events for {server_name} (last {limit})")
+            table.add_column("Time", style="dim", no_wrap=True)
+            table.add_column("Event", style="cyan")
+            table.add_column("Details", style="dim")
+
+            # Take only the requested number of events
+            events_to_show = (
+                recent_events[-limit:] if len(recent_events) > limit else recent_events
+            )
+
+            for event in reversed(events_to_show):  # Show newest first
+                timestamp = datetime.fromisoformat(event["timestamp"])
+                time_str = timestamp.strftime("%H:%M:%S")
+                event_type = event["event_type"]
+
+                # Format details
+                details = event.get("details", {})
+                details_str = details.get("message", "")
+                if not details_str and "error" in details:
+                    details_str = str(details["error"])
+
+                # Color code event types
+                event_style = "cyan"
+                if "error" in event_type.lower():
+                    event_style = "red"
+                elif event_type in ["started", "enabled", "registered"]:
+                    event_style = "green"
+                elif event_type in ["stopped", "disabled"]:
+                    event_style = "yellow"
+
+                table.add_row(
+                    time_str, Text(event_type, style=event_style), details_str or "-"
+                )
+            emit_info(table, message_group=group_id)
+
+        except Exception as e:
+            logger.error(f"Error getting logs for server '{server_name}': {e}")
+            emit_info(f"[red]Error getting logs: {e}[/red]", message_group=group_id)