centaurus-cli 3.0.1 → 3.1.0

package/dist/mcp/mcp-command-handler.js

@@ -1,283 +1,320 @@
-import { createMCPToolWrappers } from './mcp-tool-wrapper.js';
-import { logError, logWarning } from '../utils/logger.js';
-export class MCPCommandHandler {
-    mcpConfigManager;
-    mcpServerManager;
-    toolRegistry;
-    onShowMCPAddScreen;
-    onShowMCPRemoveScreen;
-    onShowMCPEnableScreen;
-    onShowMCPDisableScreen;
-    onShowMCPListScreen;
-    constructor(mcpConfigManager, mcpServerManager, toolRegistry) {
-        this.mcpConfigManager = mcpConfigManager;
-        this.mcpServerManager = mcpServerManager;
-        this.toolRegistry = toolRegistry;
-    }
-    // Callback setters for UI screens
-    setOnShowMCPAddScreen(callback) {
-        this.onShowMCPAddScreen = callback;
+import { createMCPToolWrappers } from "./mcp-tool-wrapper.js";
+import { logError, logWarning } from "../utils/logger.js";
+class MCPCommandHandler {
+  constructor(mcpConfigManager, mcpServerManager, toolRegistry) {
+    this.mcpConfigManager = mcpConfigManager;
+    this.mcpServerManager = mcpServerManager;
+    this.toolRegistry = toolRegistry;
+  }
+  onShowMCPAddScreen;
+  onShowMCPRemoveScreen;
+  onShowMCPEnableScreen;
+  onShowMCPDisableScreen;
+  onShowMCPListScreen;
+  // Callback setters for UI screens
+  setOnShowMCPAddScreen(callback) {
+    this.onShowMCPAddScreen = callback;
+  }
+  setOnShowMCPRemoveScreen(callback) {
+    this.onShowMCPRemoveScreen = callback;
+  }
+  setOnShowMCPEnableScreen(callback) {
+    this.onShowMCPEnableScreen = callback;
+  }
+  setOnShowMCPDisableScreen(callback) {
+    this.onShowMCPDisableScreen = callback;
+  }
+  setOnShowMCPListScreen(callback) {
+    this.onShowMCPListScreen = callback;
+  }
+  async initializeMCP() {
+    try {
+      const servers = this.mcpConfigManager.getServers();
+      for (const serverConfig of servers) {
+        try {
+          await this.mcpServerManager.connect(serverConfig);
+          const tools = await this.mcpServerManager.listTools(serverConfig.name);
+          const wrappedTools = createMCPToolWrappers(serverConfig.name, tools, this.mcpServerManager);
+          for (const tool of wrappedTools) {
+            this.toolRegistry.register(tool);
+          }
+        } catch (error) {
+          logWarning(`MCP server ${serverConfig.name} failed to connect during init`);
+        }
+      }
+    } catch (error) {
+      logError("Failed to initialize MCP", error instanceof Error ? error : new Error(error?.message || "Unknown error"));
     }
-    setOnShowMCPRemoveScreen(callback) {
-        this.onShowMCPRemoveScreen = callback;
+  }
+  async handleCommand(args) {
+    const subcommand = args[0]?.toLowerCase();
+    if (!subcommand) {
+      return "## MCP (Model Context Protocol) Server Management\n\nAvailable subcommands:\n\n/mcp list       - List configured MCP servers and tools\n/mcp refresh    - Restart all MCP servers\n/mcp add        - Add a new MCP server\n/mcp remove     - Remove an MCP server\n/mcp enable     - Enable a disabled server\n/mcp disable    - Disable an enabled server\n\nType any subcommand for more details.";
     }
-    setOnShowMCPEnableScreen(callback) {
-        this.onShowMCPEnableScreen = callback;
+    switch (subcommand) {
+      case "list":
+        return this.handleList();
+      case "refresh":
+        return await this.handleRefresh();
+      case "add":
+        if (this.onShowMCPAddScreen) {
+          this.onShowMCPAddScreen();
+          return "";
+        }
+        return "\u274C MCP add screen not available. Please restart the CLI.";
+      case "remove":
+        return this.handleRemove();
+      case "enable":
+        return this.handleEnable();
+      case "disable":
+        return this.handleDisable();
+      default:
+        return `\u274C Unknown MCP subcommand: ${subcommand}
+Type /mcp for help.`;
     }
-    setOnShowMCPDisableScreen(callback) {
-        this.onShowMCPDisableScreen = callback;
+  }
+  handleList() {
+    const servers = this.mcpConfigManager.getAllServers();
+    if (servers.length === 0) {
+      return `## No MCP Servers Configured
+
+Configuration file: \`${this.mcpConfigManager.getConfigPath()}\`
+
+Use \`/mcp add\` to add your first MCP server.`;
     }
-    setOnShowMCPListScreen(callback) {
-        this.onShowMCPListScreen = callback;
+    const listData = servers.map((serverConfig) => {
+      const statuses = this.mcpServerManager.getStatus(serverConfig.name);
+      const status = statuses[0];
+      const tools = status?.tools || [];
+      return {
+        name: serverConfig.name,
+        enabled: serverConfig.enabled !== false,
+        status: serverConfig.enabled === false ? "disconnected" : status?.status || "disconnected",
+        tools: tools.map((t) => ({ name: t.name }))
+      };
+    });
+    if (this.onShowMCPListScreen) {
+      this.onShowMCPListScreen(listData);
+      return "";
     }
-    async initializeMCP() {
-        try {
-            // Auto-connect to configured servers
-            const servers = this.mcpConfigManager.getServers();
-            for (const serverConfig of servers) {
-                try {
-                    await this.mcpServerManager.connect(serverConfig);
-                    // Register MCP tools to tool registry
-                    const tools = await this.mcpServerManager.listTools(serverConfig.name);
-                    const wrappedTools = createMCPToolWrappers(serverConfig.name, tools, this.mcpServerManager);
-                    for (const tool of wrappedTools) {
-                        this.toolRegistry.register(tool);
-                    }
-                }
-                catch (error) {
-                    // Silent fail - user can check status with /mcp status
-                    logWarning(`MCP server ${serverConfig.name} failed to connect during init`);
-                }
-            }
-        }
-        catch (error) {
-            logError('Failed to initialize MCP', error instanceof Error ? error : new Error(error?.message || 'Unknown error'));
+    let output = "## MCP Servers and Tools\n\n";
+    for (const serverConfig of servers) {
+      const statuses = this.mcpServerManager.getStatus(serverConfig.name);
+      const status = statuses[0];
+      const tools = status?.tools || [];
+      const statusIcon = serverConfig.enabled === false ? "\u23F8\uFE0F" : status?.status === "connected" ? "\u2705" : status?.status === "error" ? "\u274C" : "\u26A0\uFE0F";
+      const enabledLabel = serverConfig.enabled === false ? " (disabled)" : "";
+      output += `### ${statusIcon} ${serverConfig.name}${enabledLabel}
+
+`;
+      output += `- **Command**: \`${serverConfig.command}\`
+`;
+      if (serverConfig.args && serverConfig.args.length > 0) {
+        output += `- **Args**: \`${serverConfig.args.join(" ")}\`
+`;
+      }
+      output += `- **Status**: ${serverConfig.enabled === false ? "disabled" : status?.status || "unknown"}
+`;
+      output += `- **Tools**: ${tools.length}
+
+`;
+      if (tools.length > 0) {
+        for (const tool of tools) {
+          output += `  - **${tool.name}**`;
+          if (tool.description) {
+            output += `: ${tool.description}`;
+          }
+          output += "\n";
         }
+        output += "\n";
+      }
     }
-    async handleCommand(args) {
-        const subcommand = args[0]?.toLowerCase();
-        if (!subcommand) {
-            return '## MCP (Model Context Protocol) Server Management\n\n' +
-                'Available subcommands:\n\n' +
-                '/mcp list       - List configured MCP servers and tools\n' +
-                '/mcp refresh    - Restart all MCP servers\n' +
-                '/mcp add        - Add a new MCP server\n' +
-                '/mcp remove     - Remove an MCP server\n' +
-                '/mcp enable     - Enable a disabled server\n' +
-                '/mcp disable    - Disable an enabled server\n\n' +
-                'Type any subcommand for more details.';
-        }
-        switch (subcommand) {
-            case 'list':
-                return this.handleList();
-            case 'refresh':
-                return await this.handleRefresh();
-            case 'add':
-                if (this.onShowMCPAddScreen) {
-                    this.onShowMCPAddScreen();
-                    return ''; // UI will handle display
-                }
-                return '❌ MCP add screen not available. Please restart the CLI.';
-            case 'remove':
-                return this.handleRemove();
-            case 'enable':
-                return this.handleEnable();
-            case 'disable':
-                return this.handleDisable();
-            default:
-                return `❌ Unknown MCP subcommand: ${subcommand}\nType /mcp for help.`;
-        }
+    return output;
+  }
+  handleRemove() {
+    const servers = this.mcpConfigManager.getAllServers();
+    if (servers.length === 0) {
+      return "\u274C No MCP servers to remove.\n\nUse `/mcp add` to add servers first.";
     }
-    handleList() {
-        const servers = this.mcpConfigManager.getAllServers();
-        if (servers.length === 0) {
-            return '## No MCP Servers Configured\n\n' +
-                `Configuration file: \`${this.mcpConfigManager.getConfigPath()}\`\n\n` +
-                'Use `/mcp add` to add your first MCP server.';
-        }
-        // Prepare data for the list screen
-        const listData = servers.map(serverConfig => {
-            const statuses = this.mcpServerManager.getStatus(serverConfig.name);
-            const status = statuses[0];
-            const tools = status?.tools || [];
-            return {
-                name: serverConfig.name,
-                enabled: serverConfig.enabled !== false,
-                status: serverConfig.enabled === false ? 'disconnected' : (status?.status || 'disconnected'),
-                tools: tools.map(t => ({ name: t.name }))
-            };
-        });
-        if (this.onShowMCPListScreen) {
-            this.onShowMCPListScreen(listData);
-            return ''; // UI will handle display
-        }
-        // Fallback to text output if screen not available
-        let output = '## MCP Servers and Tools\n\n';
-        for (const serverConfig of servers) {
-            const statuses = this.mcpServerManager.getStatus(serverConfig.name);
-            const status = statuses[0];
-            const tools = status?.tools || [];
-            const statusIcon = serverConfig.enabled === false ? '⏸️' :
-                status?.status === 'connected' ? '✅' :
-                    status?.status === 'error' ? '❌' : '⚠️';
-            const enabledLabel = serverConfig.enabled === false ? ' (disabled)' : '';
-            output += `### ${statusIcon} ${serverConfig.name}${enabledLabel}\n\n`;
-            output += `- **Command**: \`${serverConfig.command}\`\n`;
-            if (serverConfig.args && serverConfig.args.length > 0) {
-                output += `- **Args**: \`${serverConfig.args.join(' ')}\`\n`;
-            }
-            output += `- **Status**: ${serverConfig.enabled === false ? 'disabled' : (status?.status || 'unknown')}\n`;
-            output += `- **Tools**: ${tools.length}\n\n`;
-            if (tools.length > 0) {
-                for (const tool of tools) {
-                    output += `  - **${tool.name}**`;
-                    if (tool.description) {
-                        output += `: ${tool.description}`;
-                    }
-                    output += '\n';
-                }
-                output += '\n';
-            }
-        }
-        return output;
+    if (this.onShowMCPRemoveScreen) {
+      this.onShowMCPRemoveScreen(servers);
+      return "";
     }
-    handleRemove() {
-        const servers = this.mcpConfigManager.getAllServers();
-        if (servers.length === 0) {
-            return '❌ No MCP servers to remove.\n\nUse `/mcp add` to add servers first.';
-        }
-        if (this.onShowMCPRemoveScreen) {
-            this.onShowMCPRemoveScreen(servers);
-            return ''; // UI will handle display
-        }
-        return '❌ MCP remove screen not available. Please restart the CLI.';
+    return "\u274C MCP remove screen not available. Please restart the CLI.";
+  }
+  handleEnable() {
+    const disabledServers = this.mcpConfigManager.getAllServers().filter((s) => s.enabled === false);
+    if (disabledServers.length === 0) {
+      return "\u2705 All MCP servers are already enabled.\n\nUse `/mcp list` to see all servers.";
     }
-    handleEnable() {
-        const disabledServers = this.mcpConfigManager.getAllServers().filter(s => s.enabled === false);
-        if (disabledServers.length === 0) {
-            return '✅ All MCP servers are already enabled.\n\nUse `/mcp list` to see all servers.';
-        }
-        if (this.onShowMCPEnableScreen) {
-            this.onShowMCPEnableScreen(disabledServers);
-            return ''; // UI will handle display
-        }
-        return '❌ MCP enable screen not available. Please restart the CLI.';
+    if (this.onShowMCPEnableScreen) {
+      this.onShowMCPEnableScreen(disabledServers);
+      return "";
     }
-    handleDisable() {
-        const enabledServers = this.mcpConfigManager.getAllServers().filter(s => s.enabled !== false);
-        if (enabledServers.length === 0) {
-            return '⚠️ All MCP servers are already disabled.\n\nUse `/mcp list` to see all servers.';
-        }
-        if (this.onShowMCPDisableScreen) {
-            this.onShowMCPDisableScreen(enabledServers);
-            return ''; // UI will handle display
-        }
-        return '❌ MCP disable screen not available. Please restart the CLI.';
+    return "\u274C MCP enable screen not available. Please restart the CLI.";
+  }
+  handleDisable() {
+    const enabledServers = this.mcpConfigManager.getAllServers().filter((s) => s.enabled !== false);
+    if (enabledServers.length === 0) {
+      return "\u26A0\uFE0F All MCP servers are already disabled.\n\nUse `/mcp list` to see all servers.";
     }
-    async handleRefresh() {
+    if (this.onShowMCPDisableScreen) {
+      this.onShowMCPDisableScreen(enabledServers);
+      return "";
+    }
+    return "\u274C MCP disable screen not available. Please restart the CLI.";
+  }
+  async handleRefresh() {
+    try {
+      const config = this.mcpConfigManager.reload();
+      const servers = config.servers.filter((s) => s.enabled !== false);
+      if (servers.length === 0) {
+        return "\u26A0\uFE0F No enabled MCP servers in configuration.";
+      }
+      await this.mcpServerManager.refresh(servers);
+      for (const serverConfig of servers) {
         try {
-            // Reload config from disk (updates cache)
-            const config = this.mcpConfigManager.reload();
-            const servers = config.servers.filter(s => s.enabled !== false);
-            if (servers.length === 0) {
-                return '⚠️ No enabled MCP servers in configuration.';
-            }
-            // Refresh servers
-            await this.mcpServerManager.refresh(servers);
-            // Re-register tools
-            for (const serverConfig of servers) {
-                try {
-                    if (this.mcpServerManager.isConnected(serverConfig.name)) {
-                        const tools = await this.mcpServerManager.listTools(serverConfig.name);
-                        const wrappedTools = createMCPToolWrappers(serverConfig.name, tools, this.mcpServerManager);
-                        for (const tool of wrappedTools) {
-                            this.toolRegistry.register(tool);
-                        }
-                    }
-                }
-                catch (error) {
-                    // Continue with other servers
-                }
-            }
-            // Get all servers (including disabled ones) for comprehensive status
-            const allServers = this.mcpConfigManager.getAllServers();
-            const statuses = this.mcpServerManager.getStatus();
-            // Create a map for quick status lookup
-            const statusMap = new Map(statuses.map(s => [s.name, s]));
-            // Categorize servers
-            const connectedServers = [];
-            const disabledServers = [];
-            const errorServers = [];
-            for (const server of allServers) {
-                if (server.enabled === false) {
-                    disabledServers.push(server.name);
-                }
-                else {
-                    const status = statusMap.get(server.name);
-                    if (status?.status === 'connected') {
-                        connectedServers.push(server.name);
-                    }
-                    else if (status?.status === 'error') {
-                        errorServers.push(server.name);
-                    }
-                }
-            }
-            let output = '✓ MCP servers refreshed!\n\n';
-            // Connected & Enabled section
-            if (connectedServers.length > 0) {
-                output += `**Connected & Enabled** (${connectedServers.length}):\n`;
-                for (const name of connectedServers) {
-                    output += `  - ${name}\n`;
-                }
-                output += '\n';
+          if (this.mcpServerManager.isConnected(serverConfig.name)) {
+            const tools = await this.mcpServerManager.listTools(serverConfig.name);
+            const wrappedTools = createMCPToolWrappers(serverConfig.name, tools, this.mcpServerManager);
+            for (const tool of wrappedTools) {
+              this.toolRegistry.register(tool);
             }
-            // Disabled section
-            if (disabledServers.length > 0) {
-                output += `**Disabled** (${disabledServers.length}):\n`;
-                for (const name of disabledServers) {
-                    output += `  - ${name}\n`;
-                }
-                output += '\n';
-            }
-            // Errors section
-            if (errorServers.length > 0) {
-                output += `**Errors** (${errorServers.length}):\n`;
-                for (const name of errorServers) {
-                    output += `  - ${name}\n`;
-                }
-                output += '\n';
-            }
-            output += `Use \`/mcp list\` for details.`;
-            return output;
+          }
+        } catch (error) {
         }
-        catch (error) {
-            return `❌ Failed to refresh MCP servers: ${error.message}`;
+      }
+      const allServers = this.mcpConfigManager.getAllServers();
+      const statuses = this.mcpServerManager.getStatus();
+      const statusMap = new Map(statuses.map((s) => [s.name, s]));
+      const connectedServers = [];
+      const disabledServers = [];
+      const errorServers = [];
+      for (const server of allServers) {
+        if (server.enabled === false) {
+          disabledServers.push(server.name);
+        } else {
+          const status = statusMap.get(server.name);
+          if (status?.status === "connected") {
+            connectedServers.push(server.name);
+          } else if (status?.status === "error") {
+            errorServers.push(server.name);
+          }
         }
-    }
-    // Server management methods (called by CLI adapter after UI interaction)
-    addServer(config) {
-        try {
-            this.mcpConfigManager.addServer(config);
-            return { success: true };
+      }
+      let output = "\u2713 MCP servers refreshed!\n\n";
+      if (connectedServers.length > 0) {
+        output += `**Connected & Enabled** (${connectedServers.length}):
+`;
+        for (const name of connectedServers) {
+          output += `  - ${name}
+`;
         }
-        catch (error) {
-            return { success: false, error: error.message };
+        output += "\n";
+      }
+      if (disabledServers.length > 0) {
+        output += `**Disabled** (${disabledServers.length}):
+`;
+        for (const name of disabledServers) {
+          output += `  - ${name}
+`;
         }
+        output += "\n";
+      }
+      if (errorServers.length > 0) {
+        output += `**Errors** (${errorServers.length}):
+`;
+        for (const name of errorServers) {
+          output += `  - ${name}
+`;
+        }
+        output += "\n";
+      }
+      output += `Use \`/mcp list\` for details.`;
+      return output;
+    } catch (error) {
+      return `\u274C Failed to refresh MCP servers: ${error.message}`;
     }
-    removeServer(name) {
-        this.mcpConfigManager.removeServer(name);
-    }
-    enableServer(name) {
-        this.mcpConfigManager.enableServer(name);
-    }
-    disableServer(name) {
-        this.mcpConfigManager.disableServer(name);
-    }
-    validateServerConfig(jsonString) {
-        return this.mcpConfigManager.validateServerConfigJSON(jsonString);
+  }
+  /**
+   * Programmatically add an MCP server, connect to it, register its tools, and return status.
+   * Used by the add_mcp AI tool for automatic MCP provisioning.
+   */
+  async addAndConnect(config, timeoutMs = 6e4) {
+    const logs = [];
+    try {
+      logs.push(`[add_mcp] Saving config for server "${config.name}"...`);
+      this.mcpConfigManager.addServer(config);
+      logs.push(`[add_mcp] Config saved to ${this.mcpConfigManager.getConfigPath()}`);
+      logs.push(`[add_mcp] Connecting to MCP server "${config.name}" (command: ${config.command} ${(config.args || []).join(" ")})...`);
+      await this.mcpServerManager.connect(config, timeoutMs);
+      logs.push(`[add_mcp] Connected successfully.`);
+      const mcpTools = await this.mcpServerManager.listTools(config.name);
+      logs.push(`[add_mcp] Server "${config.name}" provides ${mcpTools.length} tool(s).`);
+      this.toolRegistry.unregisterByPrefix(`mcp_${config.name}_`);
+      const wrappedTools = createMCPToolWrappers(config.name, mcpTools, this.mcpServerManager);
+      for (const tool of wrappedTools) {
+        this.toolRegistry.register(tool);
+        logs.push(`[add_mcp] Registered tool: ${tool.schema.name}`);
+      }
+      const toolNames = mcpTools.map((t) => t.name);
+      logs.push(`[add_mcp] All tools registered. Server "${config.name}" is ready.`);
+      return {
+        success: true,
+        serverName: config.name,
+        tools: toolNames,
+        logs: logs.join("\n")
+      };
+    } catch (error) {
+      logs.push(`[add_mcp] ERROR: ${error.message}`);
+      try {
+        this.mcpConfigManager.removeServer(config.name);
+        logs.push(`[add_mcp] Rolled back config for "${config.name}" due to connection failure.`);
+      } catch (rollbackError) {
+        logs.push(`[add_mcp] Warning: Could not rollback config: ${rollbackError.message}`);
+      }
+      return {
+        success: false,
+        serverName: config.name,
+        tools: [],
+        error: error.message,
+        logs: logs.join("\n")
+      };
     }
-    getConfigManager() {
-        return this.mcpConfigManager;
+  }
+  // Server management methods (called by CLI adapter after UI interaction)
+  addServer(config) {
+    try {
+      this.mcpConfigManager.addServer(config);
+      return { success: true };
+    } catch (error) {
+      return { success: false, error: error.message };
     }
+  }
+  removeServer(name) {
+    this.mcpConfigManager.removeServer(name);
+  }
+  enableServer(name) {
+    this.mcpConfigManager.enableServer(name);
+  }
+  disableServer(name) {
+    this.mcpConfigManager.disableServer(name);
+  }
+  validateServerConfig(jsonString) {
+    return this.mcpConfigManager.validateServerConfigJSON(jsonString);
+  }
+  getConfigManager() {
+    return this.mcpConfigManager;
+  }
+  /**
+   * Get the names of all currently connected MCP servers.
+   */
+  getConnectedServerNames() {
+    const statuses = this.mcpServerManager.getStatus();
+    return statuses.filter((s) => s.status === "connected").map((s) => s.name);
+  }
 }
+export {
+  MCPCommandHandler
+};
 //# sourceMappingURL=mcp-command-handler.js.map