@lanonasis/cli 3.6.5 → 3.6.7

package/dist/commands/mcp.js

@@ -1,17 +1,21 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.mcpCommands = mcpCommands;
-const chalk_1 = __importDefault(require("chalk"));
-const ora_1 = __importDefault(require("ora"));
-const table_1 = require("table");
-const mcp_client_js_1 = require("../utils/mcp-client.js");
-const enhanced_client_js_1 = require("../mcp/client/enhanced-client.js");
-const config_js_1 = require("../utils/config.js");
-const ws_1 = __importDefault(require("ws"));
-function mcpCommands(program) {
+import chalk from 'chalk';
+import ora from 'ora';
+import { table } from 'table';
+import { getMCPClient } from '../utils/mcp-client.js';
+import { EnhancedMCPClient } from '../mcp/client/enhanced-client.js';
+import { CLIConfig } from '../utils/config.js';
+import WebSocket from 'ws';
+/**
+ * Register MCP-related CLI commands (mcp and mcp-server) on a Commander program.
+ *
+ * Adds commands and subcommands for MCP server initialization, connection management,
+ * status reporting, tool listing and invocation, memory create/search operations,
+ * preference configuration, and diagnostic routines, wiring each command to its
+ * corresponding action handlers.
+ *
+ * @param program - Commander program instance to extend with MCP commands
+ */
+export function mcpCommands(program) {
     const mcp = program
         .command('mcp')
         .description('MCP (Model Context Protocol) server operations');
@@ -22,35 +26,35 @@ function mcpCommands(program) {
     mcpServer.command('init')
         .description('Initialize MCP server configuration')
         .action(async () => {
-        console.log(chalk_1.default.cyan('🚀 Initializing MCP Server Configuration'));
+        console.log(chalk.cyan('🚀 Initializing MCP Server Configuration'));
         console.log('');
-        const config = new config_js_1.CLIConfig();
+        const config = new CLIConfig();
         const isAuthenticated = !!config.get('token');
         if (isAuthenticated) {
-            console.log(chalk_1.default.green('✓ Authenticated - Using remote MCP mode'));
+            console.log(chalk.green('✓ Authenticated - Using remote MCP mode'));
             console.log('  Your memory operations will use mcp.lanonasis.com');
             console.log('  with real-time SSE updates enabled');
         }
         else {
-            console.log(chalk_1.default.yellow('⚠️  Not authenticated - Using local MCP mode'));
+            console.log(chalk.yellow('⚠️  Not authenticated - Using local MCP mode'));
             console.log('  Run "lanonasis auth login" to enable remote mode');
         }
         console.log('');
-        console.log(chalk_1.default.cyan('Available MCP Commands:'));
+        console.log(chalk.cyan('Available MCP Commands:'));
         console.log('  lanonasis mcp connect       # Auto-connect to best mode');
         console.log('  lanonasis mcp connect -r    # Force remote mode');
         console.log('  lanonasis mcp connect -l    # Force local mode');
         console.log('  lanonasis mcp status        # Check connection status');
         console.log('  lanonasis mcp tools         # List available tools');
         console.log('');
-        console.log(chalk_1.default.cyan('Memory operations are MCP-powered by default!'));
+        console.log(chalk.cyan('Memory operations are MCP-powered by default!'));
         // Auto-connect to MCP
-        const spinner = (0, ora_1.default)('Auto-connecting to MCP...').start();
+        const spinner = ora('Auto-connecting to MCP...').start();
         try {
-            const client = (0, mcp_client_js_1.getMCPClient)();
+            const client = getMCPClient();
             const connected = await client.connect({ useRemote: isAuthenticated });
             if (connected) {
-                spinner.succeed(chalk_1.default.green(`Connected to ${isAuthenticated ? 'remote' : 'local'} MCP server`));
+                spinner.succeed(chalk.green(`Connected to ${isAuthenticated ? 'remote' : 'local'} MCP server`));
                 process.exit(0);
             }
             else {
@@ -73,8 +77,8 @@ function mcpCommands(program) {
         .option('-u, --url <url>', 'Remote/WebSocket server URL')
         .option('--local-args <args>', 'Extra args for local server (e.g., "--stdio --port 3001")')
         .action(async (options) => {
-        const spinner = (0, ora_1.default)('Connecting to MCP server...').start();
-        const config = new config_js_1.CLIConfig();
+        const spinner = ora('Connecting to MCP server...').start();
+        const config = new CLIConfig();
         try {
             let connectionMode;
             // Determine connection mode - WebSocket takes precedence over remote and local
@@ -88,8 +92,9 @@ function mcpCommands(program) {
                 connectionMode = 'local';
             }
             else {
-                // Default to remote if authenticated, otherwise local
-                connectionMode = config.get('token') ? 'remote' : 'local';
+                // Default to websocket (production mode) for all users
+                // Local mode should only be used explicitly for development
+                connectionMode = 'websocket';
             }
             // Save preferences
             config.set('mcpConnectionMode', connectionMode);
@@ -106,7 +111,7 @@ function mcpCommands(program) {
             }
             let connected = false;
             // Use Enhanced MCP Client for better connection handling
-            const enhancedClient = new enhanced_client_js_1.EnhancedMCPClient();
+            const enhancedClient = new EnhancedMCPClient();
             if (options.url) {
                 // Connect to specific URL (WebSocket or remote)
                 const serverConfig = {
@@ -117,14 +122,14 @@ function mcpCommands(program) {
                 };
                 connected = await enhancedClient.connectSingle(serverConfig);
                 if (connected) {
-                    spinner.succeed(chalk_1.default.green(`Connected to MCP server at ${options.url}`));
+                    spinner.succeed(chalk.green(`Connected to MCP server at ${options.url}`));
                     process.exit(0);
                     return;
                 }
             }
             else {
                 // Fall back to old client for local connections
-                const client = (0, mcp_client_js_1.getMCPClient)();
+                const client = getMCPClient();
                 const localArgs = typeof options.localArgs === 'string' && options.localArgs.trim().length > 0
                     ? options.localArgs.split(' ').map((s) => s.trim()).filter(Boolean)
                     : undefined;
@@ -136,15 +141,15 @@ function mcpCommands(program) {
                 });
             }
             if (connected) {
-                spinner.succeed(chalk_1.default.green(`Connected to MCP server in ${connectionMode} mode`));
+                spinner.succeed(chalk.green(`Connected to MCP server in ${connectionMode} mode`));
                 process.exit(0);
                 if (connectionMode === 'remote') {
-                    console.log(chalk_1.default.cyan('ℹ️  Using remote MCP via mcp.lanonasis.com'));
-                    console.log(chalk_1.default.cyan('📡 SSE endpoint active for real-time updates'));
+                    console.log(chalk.cyan('ℹ️  Using remote MCP via mcp.lanonasis.com'));
+                    console.log(chalk.cyan('📡 SSE endpoint active for real-time updates'));
                 }
                 else if (connectionMode === 'websocket') {
-                    console.log(chalk_1.default.cyan('ℹ️  Using enterprise WebSocket MCP server'));
-                    console.log(chalk_1.default.cyan('📡 WebSocket connection active with auto-reconnect'));
+                    console.log(chalk.cyan('ℹ️  Using enterprise WebSocket MCP server'));
+                    console.log(chalk.cyan('📡 WebSocket connection active with auto-reconnect'));
                 }
             }
             else {
@@ -161,47 +166,47 @@ function mcpCommands(program) {
     mcp.command('disconnect')
         .description('Disconnect from MCP server')
         .action(async () => {
-        const client = (0, mcp_client_js_1.getMCPClient)();
+        const client = getMCPClient();
         await client.disconnect();
-        console.log(chalk_1.default.green('✓ Disconnected from MCP server'));
+        console.log(chalk.green('✓ Disconnected from MCP server'));
     });
     // Status command
     mcp.command('status')
         .description('Show MCP connection status')
         .action(async () => {
-        const client = (0, mcp_client_js_1.getMCPClient)();
+        const client = getMCPClient();
         // Reload config from disk to get latest preference
         await client.init();
         const status = client.getConnectionStatus();
-        console.log(chalk_1.default.cyan('\n📊 MCP Connection Status'));
-        console.log(chalk_1.default.cyan('========================'));
-        console.log(`Status: ${status.connected ? chalk_1.default.green('Connected') : chalk_1.default.red('Disconnected')}`);
+        console.log(chalk.cyan('\n📊 MCP Connection Status'));
+        console.log(chalk.cyan('========================'));
+        console.log(`Status: ${status.connected ? chalk.green('Connected') : chalk.red('Disconnected')}`);
         // Display mode with proper labels
         let modeDisplay;
         switch (status.mode) {
             case 'websocket':
-                modeDisplay = chalk_1.default.blue('WebSocket');
+                modeDisplay = chalk.blue('WebSocket');
                 break;
             case 'remote':
-                modeDisplay = chalk_1.default.blue('Remote (HTTP/SSE)');
+                modeDisplay = chalk.blue('Remote (HTTP/SSE)');
                 break;
             case 'local':
-                modeDisplay = chalk_1.default.yellow('Local (stdio)');
+                modeDisplay = chalk.yellow('Local (stdio)');
                 break;
             default:
-                modeDisplay = chalk_1.default.gray(status.mode);
+                modeDisplay = chalk.gray(status.mode);
         }
         console.log(`Mode: ${modeDisplay}`);
         console.log(`Server: ${status.server}`);
         if (status.connected) {
             if (status.mode === 'remote') {
-                console.log(`\n${chalk_1.default.cyan('Features:')}`);
+                console.log(`\n${chalk.cyan('Features:')}`);
                 console.log('• Real-time updates via SSE');
                 console.log('• Authenticated API access');
                 console.log('• MCP-compatible tool interface');
             }
             else if (status.mode === 'websocket') {
-                console.log(`\n${chalk_1.default.cyan('Features:')}`);
+                console.log(`\n${chalk.cyan('Features:')}`);
                 console.log('• Bi-directional real-time communication');
                 console.log('• Authenticated WebSocket connection');
                 console.log('• Production-ready MCP server');
@@ -212,29 +217,29 @@ function mcpCommands(program) {
     mcp.command('tools')
         .description('List available MCP tools')
         .action(async () => {
-        const spinner = (0, ora_1.default)('Fetching available tools...').start();
+        const spinner = ora('Fetching available tools...').start();
         try {
-            const client = (0, mcp_client_js_1.getMCPClient)();
+            const client = getMCPClient();
             if (!client.isConnectedToServer()) {
                 spinner.info('Not connected. Attempting auto-connect...');
-                const config = new config_js_1.CLIConfig();
+                const config = new CLIConfig();
                 const useRemote = !!config.get('token');
                 await client.connect({ useRemote });
             }
             const tools = await client.listTools();
             spinner.succeed('Tools fetched successfully');
-            console.log(chalk_1.default.cyan('\n🔧 Available MCP Tools'));
-            console.log(chalk_1.default.cyan('====================='));
+            console.log(chalk.cyan('\n🔧 Available MCP Tools'));
+            console.log(chalk.cyan('====================='));
             const tableData = [
-                [chalk_1.default.bold('Tool Name'), chalk_1.default.bold('Description')]
+                [chalk.bold('Tool Name'), chalk.bold('Description')]
             ];
             tools.forEach(tool => {
                 tableData.push([
-                    chalk_1.default.green(tool.name),
+                    chalk.green(tool.name),
                     tool.description
                 ]);
             });
-            console.log((0, table_1.table)(tableData, {
+            console.log(table(tableData, {
                 border: {
                     topBody: '─',
                     topJoin: '┬',
@@ -265,12 +270,12 @@ function mcpCommands(program) {
         .argument('<tool>', 'Tool name to call')
         .option('-a, --args <json>', 'Tool arguments as JSON')
         .action(async (toolName, options) => {
-        const spinner = (0, ora_1.default)(`Calling tool: ${toolName}...`).start();
+        const spinner = ora(`Calling tool: ${toolName}...`).start();
         try {
-            const client = (0, mcp_client_js_1.getMCPClient)();
+            const client = getMCPClient();
             if (!client.isConnectedToServer()) {
                 spinner.info('Not connected. Attempting auto-connect...');
-                const config = new config_js_1.CLIConfig();
+                const config = new CLIConfig();
                 const useRemote = !!config.get('token');
                 await client.connect({ useRemote });
             }
@@ -286,7 +291,7 @@ function mcpCommands(program) {
             }
             const result = await client.callTool(toolName, args);
             spinner.succeed(`Tool ${toolName} executed successfully`);
-            console.log(chalk_1.default.cyan('\n📤 Tool Result:'));
+            console.log(chalk.cyan('\n📤 Tool Result:'));
             console.log(JSON.stringify(result, null, 2));
         }
         catch (error) {
@@ -304,12 +309,12 @@ function mcpCommands(program) {
         .option('-T, --type <type>', 'Memory type', 'context')
         .option('--tags <tags>', 'Comma-separated tags')
         .action(async (options) => {
-        const spinner = (0, ora_1.default)('Creating memory via MCP...').start();
+        const spinner = ora('Creating memory via MCP...').start();
         try {
-            const client = (0, mcp_client_js_1.getMCPClient)();
+            const client = getMCPClient();
             if (!client.isConnectedToServer()) {
                 spinner.info('Not connected. Attempting auto-connect...');
-                const config = new config_js_1.CLIConfig();
+                const config = new CLIConfig();
                 const useRemote = !!config.get('token');
                 await client.connect({ useRemote });
             }
@@ -320,8 +325,8 @@ function mcpCommands(program) {
                 tags: options.tags ? options.tags.split(',').map((t) => t.trim()) : []
             });
             spinner.succeed('Memory created successfully');
-            console.log(chalk_1.default.green('\n✓ Memory created'));
-            console.log(`ID: ${chalk_1.default.cyan(result.id)}`);
+            console.log(chalk.green('\n✓ Memory created'));
+            console.log(`ID: ${chalk.cyan(result.id)}`);
             console.log(`Title: ${result.title}`);
             console.log(`Type: ${result.memory_type}`);
         }
@@ -336,12 +341,12 @@ function mcpCommands(program) {
         .option('-l, --limit <number>', 'Maximum results', '10')
         .option('-t, --threshold <number>', 'Similarity threshold (0-1)', '0.7')
         .action(async (query, options) => {
-        const spinner = (0, ora_1.default)('Searching memories via MCP...').start();
+        const spinner = ora('Searching memories via MCP...').start();
         try {
-            const client = (0, mcp_client_js_1.getMCPClient)();
+            const client = getMCPClient();
             if (!client.isConnectedToServer()) {
                 spinner.info('Not connected. Attempting auto-connect...');
-                const config = new config_js_1.CLIConfig();
+                const config = new CLIConfig();
                 const useRemote = !!config.get('token');
                 await client.connect({ useRemote });
             }
@@ -352,15 +357,15 @@ function mcpCommands(program) {
             });
             spinner.succeed(`Found ${results.length} memories`);
             if (results.length === 0) {
-                console.log(chalk_1.default.yellow('\nNo memories found matching your query'));
+                console.log(chalk.yellow('\nNo memories found matching your query'));
                 return;
             }
-            console.log(chalk_1.default.cyan('\n🔍 Search Results:'));
+            console.log(chalk.cyan('\n🔍 Search Results:'));
             results.forEach((memory, index) => {
-                console.log(`\n${chalk_1.default.bold(`${index + 1}. ${memory.title}`)}`);
-                console.log(`   ID: ${chalk_1.default.gray(memory.id)}`);
-                console.log(`   Type: ${chalk_1.default.blue(memory.memory_type)}`);
-                console.log(`   Score: ${chalk_1.default.green((memory.relevance_score * 100).toFixed(1) + '%')}`);
+                console.log(`\n${chalk.bold(`${index + 1}. ${memory.title}`)}`);
+                console.log(`   ID: ${chalk.gray(memory.id)}`);
+                console.log(`   Type: ${chalk.blue(memory.memory_type)}`);
+                console.log(`   Score: ${chalk.green((memory.relevance_score * 100).toFixed(1) + '%')}`);
                 console.log(`   Content: ${memory.content.substring(0, 100)}...`);
             });
         }
@@ -372,30 +377,36 @@ function mcpCommands(program) {
     // Configure MCP preferences
     mcp.command('config')
         .description('Configure MCP preferences')
-        .option('--prefer-remote', 'Prefer remote MCP server when available')
-        .option('--prefer-local', 'Prefer local MCP server')
+        .option('--prefer-websocket', 'Prefer WebSocket MCP connection (recommended for production)')
+        .option('--prefer-remote', 'Prefer remote MCP server (REST/SSE mode)')
+        .option('--prefer-local', 'Prefer local MCP server (development only)')
         .option('--auto', 'Auto-detect best connection mode')
         .action(async (options) => {
-        const config = new config_js_1.CLIConfig();
-        if (options.preferRemote) {
+        const config = new CLIConfig();
+        if (options.preferWebsocket) {
+            await config.setAndSave('mcpPreference', 'websocket');
+            console.log(chalk.green('✓ Set MCP preference to WebSocket (production mode)'));
+        }
+        else if (options.preferRemote) {
             await config.setAndSave('mcpPreference', 'remote');
-            console.log(chalk_1.default.green('✓ Set MCP preference to remote'));
+            console.log(chalk.green('✓ Set MCP preference to remote (REST/SSE mode)'));
         }
         else if (options.preferLocal) {
             await config.setAndSave('mcpPreference', 'local');
-            console.log(chalk_1.default.green('✓ Set MCP preference to local'));
+            console.log(chalk.green('✓ Set MCP preference to local (development only)'));
         }
         else if (options.auto) {
             await config.setAndSave('mcpPreference', 'auto');
-            console.log(chalk_1.default.green('✓ Set MCP preference to auto-detect'));
+            console.log(chalk.green('✓ Set MCP preference to auto-detect'));
         }
         else {
             const current = config.get('mcpPreference') || 'auto';
-            console.log(`Current MCP preference: ${chalk_1.default.cyan(current)}`);
+            console.log(`Current MCP preference: ${chalk.cyan(current)}`);
             console.log('\nOptions:');
-            console.log('  --prefer-remote : Use remote MCP server (mcp.lanonasis.com)');
-            console.log('  --prefer-local  : Use local MCP server');
-            console.log('  --auto          : Auto-detect based on authentication');
+            console.log('  --prefer-websocket : Use WebSocket mode (recommended for production)');
+            console.log('  --prefer-remote    : Use remote REST/SSE mode (alternative)');
+            console.log('  --prefer-local     : Use local stdio mode (development only)');
+            console.log('  --auto             : Auto-detect based on configuration (default)');
         }
     });
     // Diagnose MCP connection issues
@@ -403,10 +414,10 @@ function mcpCommands(program) {
         .description('Diagnose MCP connection issues')
         .option('-v, --verbose', 'show detailed diagnostic information')
         .action(async (options) => {
-        const config = new config_js_1.CLIConfig();
+        const config = new CLIConfig();
         await config.init();
-        console.log(chalk_1.default.blue.bold('🔍 MCP Connection Diagnostic'));
-        console.log(chalk_1.default.cyan('━'.repeat(50)));
+        console.log(chalk.blue.bold('🔍 MCP Connection Diagnostic'));
+        console.log(chalk.cyan('━'.repeat(50)));
         console.log();
         const diagnostics = {
             authenticationValid: false,
@@ -422,64 +433,64 @@ function mcpCommands(program) {
             healthCheckPassing: false
         };
         // Step 1: Check authentication status
-        console.log(chalk_1.default.cyan('1. Authentication Status'));
+        console.log(chalk.cyan('1. Authentication Status'));
         const token = config.getToken();
         const vendorKey = config.getVendorKey();
         if (!token && !vendorKey) {
-            console.log(chalk_1.default.red('   ✖ No authentication credentials found'));
-            console.log(chalk_1.default.gray('   → Run: lanonasis auth login'));
-            console.log(chalk_1.default.gray('   → MCP requires authentication for remote access'));
+            console.log(chalk.red('   ✖ No authentication credentials found'));
+            console.log(chalk.gray('   → Run: lanonasis auth login'));
+            console.log(chalk.gray('   → MCP requires authentication for remote access'));
         }
         else {
             try {
                 const isValid = await config.validateStoredCredentials();
                 diagnostics.authenticationValid = isValid;
                 if (isValid) {
-                    console.log(chalk_1.default.green('   ✓ Authentication credentials are valid'));
+                    console.log(chalk.green('   ✓ Authentication credentials are valid'));
                 }
                 else {
-                    console.log(chalk_1.default.red('   ✖ Authentication credentials are invalid'));
-                    console.log(chalk_1.default.gray('   → Run: lanonasis auth login'));
+                    console.log(chalk.red('   ✖ Authentication credentials are invalid'));
+                    console.log(chalk.gray('   → Run: lanonasis auth login'));
                 }
             }
             catch (error) {
-                console.log(chalk_1.default.yellow('   ⚠ Could not validate authentication'));
-                console.log(chalk_1.default.gray(`     ${error instanceof Error ? error.message : 'Unknown error'}`));
+                console.log(chalk.yellow('   ⚠ Could not validate authentication'));
+                console.log(chalk.gray(`     ${error instanceof Error ? error.message : 'Unknown error'}`));
             }
         }
         // Step 2: Test endpoint availability
-        console.log(chalk_1.default.cyan('\n2. Endpoint Availability'));
-        const spinner1 = (0, ora_1.default)('Testing MCP endpoints...').start();
+        console.log(chalk.cyan('\n2. Endpoint Availability'));
+        const spinner1 = ora('Testing MCP endpoints...').start();
         try {
             await config.discoverServices(options.verbose);
             const services = config.get('discoveredServices');
             if (services) {
                 spinner1.succeed('MCP endpoints discovered');
                 diagnostics.endpointsReachable = true;
-                console.log(chalk_1.default.green('   ✓ Service discovery successful'));
+                console.log(chalk.green('   ✓ Service discovery successful'));
                 if (options.verbose) {
                     const svc = services;
-                    console.log(chalk_1.default.gray(`     HTTP: ${svc.mcp_base}`));
-                    console.log(chalk_1.default.gray(`     WebSocket: ${svc.mcp_ws_base}`));
-                    console.log(chalk_1.default.gray(`     SSE: ${svc.mcp_sse_base}`));
+                    console.log(chalk.gray(`     HTTP: ${svc.mcp_base}`));
+                    console.log(chalk.gray(`     WebSocket: ${svc.mcp_ws_base}`));
+                    console.log(chalk.gray(`     SSE: ${svc.mcp_sse_base}`));
                 }
             }
             else {
                 spinner1.warn('Using fallback endpoints');
-                console.log(chalk_1.default.yellow('   ⚠ Service discovery failed, using fallbacks'));
+                console.log(chalk.yellow('   ⚠ Service discovery failed, using fallbacks'));
                 diagnostics.endpointsReachable = true; // Fallbacks still work
             }
         }
         catch (error) {
             spinner1.fail('Endpoint discovery failed');
-            console.log(chalk_1.default.red('   ✖ Cannot discover MCP endpoints'));
-            console.log(chalk_1.default.gray(`     ${error instanceof Error ? error.message : 'Unknown error'}`));
+            console.log(chalk.red('   ✖ Cannot discover MCP endpoints'));
+            console.log(chalk.gray(`     ${error instanceof Error ? error.message : 'Unknown error'}`));
         }
         // Step 3: Test transport protocols
-        console.log(chalk_1.default.cyan('\n3. Transport Protocol Tests'));
+        console.log(chalk.cyan('\n3. Transport Protocol Tests'));
         // Test HTTP/REST endpoint
         if (diagnostics.authenticationValid) {
-            const httpSpinner = (0, ora_1.default)('Testing HTTP transport...').start();
+            const httpSpinner = ora('Testing HTTP transport...').start();
             try {
                 const startTime = Date.now();
                 const axios = (await import('axios')).default;
@@ -495,22 +506,22 @@ function mcpCommands(program) {
                 diagnostics.connectionLatency.http = latency;
                 diagnostics.transportTests.http = true;
                 httpSpinner.succeed(`HTTP transport working (${latency}ms)`);
-                console.log(chalk_1.default.green(`   ✓ HTTP/REST endpoint reachable`));
+                console.log(chalk.green(`   ✓ HTTP/REST endpoint reachable`));
             }
             catch (error) {
                 httpSpinner.fail('HTTP transport failed');
-                console.log(chalk_1.default.red('   ✖ HTTP/REST endpoint failed'));
+                console.log(chalk.red('   ✖ HTTP/REST endpoint failed'));
                 if (options.verbose) {
-                    console.log(chalk_1.default.gray(`     Error: ${error instanceof Error ? error.message : String(error)}`));
+                    console.log(chalk.gray(`     Error: ${error instanceof Error ? error.message : String(error)}`));
                 }
             }
             // Test WebSocket endpoint
-            const wsSpinner = (0, ora_1.default)('Testing WebSocket transport...').start();
+            const wsSpinner = ora('Testing WebSocket transport...').start();
             try {
                 const startTime = Date.now();
                 const wsUrl = config.getMCPServerUrl();
                 // Create a test WebSocket connection
-                const ws = new ws_1.default(wsUrl, [], {
+                const ws = new WebSocket(wsUrl, [], {
                     headers: {
                         'Authorization': `Bearer ${token}`,
                         'X-API-Key': String(token || vendorKey)
@@ -535,17 +546,17 @@ function mcpCommands(program) {
                     });
                 });
                 wsSpinner.succeed(`WebSocket transport working (${diagnostics.connectionLatency.websocket}ms)`);
-                console.log(chalk_1.default.green('   ✓ WebSocket endpoint reachable'));
+                console.log(chalk.green('   ✓ WebSocket endpoint reachable'));
             }
             catch (error) {
                 wsSpinner.fail('WebSocket transport failed');
-                console.log(chalk_1.default.red('   ✖ WebSocket endpoint failed'));
+                console.log(chalk.red('   ✖ WebSocket endpoint failed'));
                 if (options.verbose) {
-                    console.log(chalk_1.default.gray(`     Error: ${error instanceof Error ? error.message : String(error)}`));
+                    console.log(chalk.gray(`     Error: ${error instanceof Error ? error.message : String(error)}`));
                 }
             }
             // Test SSE endpoint
-            const sseSpinner = (0, ora_1.default)('Testing SSE transport...').start();
+            const sseSpinner = ora('Testing SSE transport...').start();
             try {
                 const startTime = Date.now();
                 const sseUrl = config.getMCPSSEUrl();
@@ -562,86 +573,86 @@ function mcpCommands(program) {
                 diagnostics.connectionLatency.sse = latency;
                 diagnostics.transportTests.sse = true;
                 sseSpinner.succeed(`SSE transport working (${latency}ms)`);
-                console.log(chalk_1.default.green('   ✓ SSE endpoint reachable'));
+                console.log(chalk.green('   ✓ SSE endpoint reachable'));
             }
             catch (error) {
                 sseSpinner.fail('SSE transport failed');
-                console.log(chalk_1.default.red('   ✖ SSE endpoint failed'));
+                console.log(chalk.red('   ✖ SSE endpoint failed'));
                 if (options.verbose) {
-                    console.log(chalk_1.default.gray(`     Error: ${error instanceof Error ? error.message : String(error)}`));
+                    console.log(chalk.gray(`     Error: ${error instanceof Error ? error.message : String(error)}`));
                 }
             }
         }
         else {
-            console.log(chalk_1.default.gray('   - Skipped transport tests (authentication required)'));
+            console.log(chalk.gray('   - Skipped transport tests (authentication required)'));
         }
         // Step 4: Test current MCP connection
-        console.log(chalk_1.default.cyan('\n4. Current MCP Connection'));
-        const client = (0, mcp_client_js_1.getMCPClient)();
+        console.log(chalk.cyan('\n4. Current MCP Connection'));
+        const client = getMCPClient();
         diagnostics.currentConnection = client.getConnectionStatus();
         if (diagnostics.currentConnection.connected) {
-            console.log(chalk_1.default.green('   ✓ MCP client is connected'));
-            console.log(chalk_1.default.gray(`     Mode: ${diagnostics.currentConnection.mode}`));
-            console.log(chalk_1.default.gray(`     Server: ${diagnostics.currentConnection.server}`));
+            console.log(chalk.green('   ✓ MCP client is connected'));
+            console.log(chalk.gray(`     Mode: ${diagnostics.currentConnection.mode}`));
+            console.log(chalk.gray(`     Server: ${diagnostics.currentConnection.server}`));
             if (diagnostics.currentConnection.connectionUptime) {
                 const uptimeSeconds = Math.floor(diagnostics.currentConnection.connectionUptime / 1000);
-                console.log(chalk_1.default.gray(`     Uptime: ${uptimeSeconds}s`));
+                console.log(chalk.gray(`     Uptime: ${uptimeSeconds}s`));
             }
             if (diagnostics.currentConnection.lastHealthCheck) {
                 const healthCheckAge = Date.now() - diagnostics.currentConnection.lastHealthCheck.getTime();
-                console.log(chalk_1.default.gray(`     Last health check: ${Math.floor(healthCheckAge / 1000)}s ago`));
+                console.log(chalk.gray(`     Last health check: ${Math.floor(healthCheckAge / 1000)}s ago`));
             }
         }
         else {
-            console.log(chalk_1.default.red('   ✖ MCP client is not connected'));
-            console.log(chalk_1.default.gray('   → Try: lanonasis mcp connect'));
+            console.log(chalk.red('   ✖ MCP client is not connected'));
+            console.log(chalk.gray('   → Try: lanonasis mcp connect'));
         }
         // Step 5: Test tool availability
-        console.log(chalk_1.default.cyan('\n5. Tool Availability'));
+        console.log(chalk.cyan('\n5. Tool Availability'));
         if (diagnostics.currentConnection.connected) {
-            const toolSpinner = (0, ora_1.default)('Testing MCP tools...').start();
+            const toolSpinner = ora('Testing MCP tools...').start();
             try {
                 const tools = await client.listTools();
                 diagnostics.toolsAvailable = tools.length > 0;
                 toolSpinner.succeed(`Found ${tools.length} available tools`);
-                console.log(chalk_1.default.green(`   ✓ ${tools.length} MCP tools available`));
+                console.log(chalk.green(`   ✓ ${tools.length} MCP tools available`));
                 if (options.verbose && tools.length > 0) {
-                    console.log(chalk_1.default.gray('     Available tools:'));
+                    console.log(chalk.gray('     Available tools:'));
                     tools.slice(0, 5).forEach(tool => {
-                        console.log(chalk_1.default.gray(`       • ${tool.name}`));
+                        console.log(chalk.gray(`       • ${tool.name}`));
                     });
                     if (tools.length > 5) {
-                        console.log(chalk_1.default.gray(`       ... and ${tools.length - 5} more`));
+                        console.log(chalk.gray(`       ... and ${tools.length - 5} more`));
                     }
                 }
             }
             catch (error) {
                 toolSpinner.fail('Tool listing failed');
-                console.log(chalk_1.default.red('   ✖ Cannot list MCP tools'));
+                console.log(chalk.red('   ✖ Cannot list MCP tools'));
                 if (options.verbose) {
-                    console.log(chalk_1.default.gray(`     Error: ${error instanceof Error ? error.message : 'Unknown error'}`));
+                    console.log(chalk.gray(`     Error: ${error instanceof Error ? error.message : 'Unknown error'}`));
                 }
             }
         }
         else {
-            console.log(chalk_1.default.gray('   - Skipped (not connected to MCP server)'));
+            console.log(chalk.gray('   - Skipped (not connected to MCP server)'));
         }
         // Step 6: Connection quality measurement
-        console.log(chalk_1.default.cyan('\n6. Connection Quality'));
+        console.log(chalk.cyan('\n6. Connection Quality'));
         if (Object.keys(diagnostics.connectionLatency).length > 0) {
-            console.log(chalk_1.default.green('   ✓ Latency measurements:'));
+            console.log(chalk.green('   ✓ Latency measurements:'));
             Object.entries(diagnostics.connectionLatency).forEach(([transport, latency]) => {
                 const quality = latency < 100 ? 'Excellent' : latency < 300 ? 'Good' : latency < 1000 ? 'Fair' : 'Poor';
-                const color = latency < 100 ? chalk_1.default.green : latency < 300 ? chalk_1.default.yellow : chalk_1.default.red;
+                const color = latency < 100 ? chalk.green : latency < 300 ? chalk.yellow : chalk.red;
                 console.log(color(`     ${transport.toUpperCase()}: ${latency}ms (${quality})`));
             });
         }
         else {
-            console.log(chalk_1.default.gray('   - No latency measurements available'));
+            console.log(chalk.gray('   - No latency measurements available'));
         }
         // Summary and recommendations
-        console.log(chalk_1.default.blue.bold('\n📋 MCP Diagnostic Summary'));
-        console.log(chalk_1.default.cyan('━'.repeat(50)));
+        console.log(chalk.blue.bold('\n📋 MCP Diagnostic Summary'));
+        console.log(chalk.cyan('━'.repeat(50)));
         const issues = [];
         const recommendations = [];
         if (!diagnostics.authenticationValid) {
@@ -671,30 +682,30 @@ function mcpCommands(program) {
         }
         // Show results
         if (issues.length === 0) {
-            console.log(chalk_1.default.green('✅ All MCP connection checks passed!'));
-            console.log(chalk_1.default.cyan('   Your MCP connection is working correctly.'));
+            console.log(chalk.green('✅ All MCP connection checks passed!'));
+            console.log(chalk.cyan('   Your MCP connection is working correctly.'));
             if (Object.keys(diagnostics.connectionLatency).length > 0) {
                 const avgLatency = Object.values(diagnostics.connectionLatency).reduce((a, b) => a + b, 0) / Object.values(diagnostics.connectionLatency).length;
-                console.log(chalk_1.default.cyan(`   Average latency: ${Math.round(avgLatency)}ms`));
+                console.log(chalk.cyan(`   Average latency: ${Math.round(avgLatency)}ms`));
             }
         }
         else {
-            console.log(chalk_1.default.red(`❌ Found ${issues.length} issue(s):`));
+            console.log(chalk.red(`❌ Found ${issues.length} issue(s):`));
             issues.forEach(issue => {
-                console.log(chalk_1.default.red(`   • ${issue}`));
+                console.log(chalk.red(`   • ${issue}`));
             });
-            console.log(chalk_1.default.yellow('\n💡 Recommended actions:'));
+            console.log(chalk.yellow('\n💡 Recommended actions:'));
             recommendations.forEach(rec => {
-                console.log(chalk_1.default.cyan(`   • ${rec}`));
+                console.log(chalk.cyan(`   • ${rec}`));
             });
         }
         // Additional troubleshooting info
         if (issues.length > 0) {
-            console.log(chalk_1.default.gray('\n🔧 Additional troubleshooting:'));
-            console.log(chalk_1.default.gray('   • Try different connection modes: --mode websocket|remote|local'));
-            console.log(chalk_1.default.gray('   • Check firewall settings for ports 80, 443, and WebSocket'));
-            console.log(chalk_1.default.gray('   • Verify your network allows outbound HTTPS connections'));
-            console.log(chalk_1.default.gray('   • Contact support if issues persist'));
+            console.log(chalk.gray('\n🔧 Additional troubleshooting:'));
+            console.log(chalk.gray('   • Try different connection modes: --mode websocket|remote|local'));
+            console.log(chalk.gray('   • Check firewall settings for ports 80, 443, and WebSocket'));
+            console.log(chalk.gray('   • Verify your network allows outbound HTTPS connections'));
+            console.log(chalk.gray('   • Contact support if issues persist'));
         }
     });
 }