upfynai-code 2.2.0 → 2.4.0

package/server/cli.js

@@ -6,8 +6,10 @@
  * Visual AI coding interface with Upfyn-Canvas for AI coding assistants
  *
  * Commands:
- *   (no args)     - Start the server (default)
+ *   (no args)     - Launch Claude Code with Upfyn branding (default)
  *   start         - Start the server
+ *   connect       - Connect to hosted server (relay bridge)
+ *   config        - View/set configuration (API key, server, etc.)
  *   status        - Show configuration and data locations
  *   help          - Show help information
  *   version       - Show version information
@@ -18,41 +20,32 @@ import path from 'path';
 import os from 'os';
 import { fileURLToPath } from 'url';
 import { dirname } from 'path';
+import { execSync, spawn } from 'child_process';
+import {
+    c,
+    showStyledHelp,
+    showStyledStatus,
+    showServerBanner,
+    showConnectStartup,
+    showConnectionBanner,
+    showLaunchScreen,
+    showConfig,
+    logRelayEvent,
+    createSpinner,
+    playRocketAnimation,
+    clearScreen,
+} from './cli-ui.js';
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = dirname(__filename);
 
-// ANSI color codes for terminal output
-const colors = {
-    reset: '\x1b[0m',
-    bright: '\x1b[1m',
-    dim: '\x1b[2m',
-
-    // Foreground colors
-    cyan: '\x1b[36m',
-    green: '\x1b[32m',
-    yellow: '\x1b[33m',
-    blue: '\x1b[34m',
-    magenta: '\x1b[35m',
-    white: '\x1b[37m',
-    gray: '\x1b[90m',
-};
-
-// Helper to colorize text
-const c = {
-    info: (text) => `${colors.cyan}${text}${colors.reset}`,
-    ok: (text) => `${colors.green}${text}${colors.reset}`,
-    warn: (text) => `${colors.yellow}${text}${colors.reset}`,
-    error: (text) => `${colors.yellow}${text}${colors.reset}`,
-    tip: (text) => `${colors.blue}${text}${colors.reset}`,
-    bright: (text) => `${colors.bright}${text}${colors.reset}`,
-    dim: (text) => `${colors.dim}${text}${colors.reset}`,
-};
-
 // Load package.json for version info
 const packageJsonPath = path.join(__dirname, '../package.json');
 const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'));
 
+const CONFIG_DIR = path.join(os.homedir(), '.upfynai');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+
 // Load environment variables from .env file if it exists
 function loadEnvFile() {
     try {
@@ -72,6 +65,26 @@ function loadEnvFile() {
     }
 }
 
+// Load saved config from ~/.upfynai/config.json
+function loadConfig() {
+    try {
+        if (fs.existsSync(CONFIG_FILE)) {
+            return JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf8'));
+        }
+    } catch (e) { /* ignore */ }
+    return {};
+}
+
+// Save config to ~/.upfynai/config.json
+function saveConfig(config) {
+    if (!fs.existsSync(CONFIG_DIR)) {
+        fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    }
+    // Don't save internal keys
+    const { _path, ...rest } = config;
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(rest, null, 2));
+}
+
 // Get the database path (same logic as db.js)
 function getDatabasePath() {
     loadEnvFile();
@@ -83,123 +96,54 @@ function getInstallDir() {
     return path.join(__dirname, '..');
 }
 
-// Show status command
-function showStatus() {
-    console.log(`\n${c.bright('Upfyn-Code - Status')}\n`);
-    console.log(c.dim('═'.repeat(60)));
-
-    // Version info
-    console.log(`\n${c.info('[INFO]')} Version: ${c.bright(packageJson.version)}`);
+// --- OS Detection + Claude Binary Discovery ---
+
+function findClaudeBinary() {
+    const isWindows = process.platform === 'win32';
+    const candidates = isWindows
+        ? ['claude.exe', 'claude.cmd', 'claude']
+        : ['claude'];
+
+    for (const cmd of candidates) {
+        try {
+            const whichCmd = isWindows ? 'where' : 'which';
+            const result = execSync(`${whichCmd} ${cmd}`, { stdio: 'pipe', encoding: 'utf8' }).trim();
+            if (result) return cmd;
+        } catch {
+            // not found, try next
+        }
+    }
+    return null;
+}
 
-    // Installation location
+// --- Show status command ---
+function showStatus() {
+    const config = loadConfig();
     const installDir = getInstallDir();
-    console.log(`\n${c.info('[INFO]')} Installation Directory:`);
-    console.log(`       ${c.dim(installDir)}`);
-
-    // Database location
     const dbPath = getDatabasePath();
     const dbExists = fs.existsSync(dbPath);
-    console.log(`\n${c.info('[INFO]')} Database Location:`);
-    console.log(`       ${c.dim(dbPath)}`);
-    console.log(`       Status: ${dbExists ? c.ok('[OK] Exists') : c.warn('[WARN] Not created yet (will be created on first run)')}`);
-
+    let dbSize = '';
     if (dbExists) {
         const stats = fs.statSync(dbPath);
-        console.log(`       Size: ${c.dim((stats.size / 1024).toFixed(2) + ' KB')}`);
-        console.log(`       Modified: ${c.dim(stats.mtime.toLocaleString())}`);
+        dbSize = (stats.size / 1024).toFixed(2) + ' KB';
     }
 
-    // Environment variables
-    console.log(`\n${c.info('[INFO]')} Configuration:`);
-    console.log(`       PORT: ${c.bright(process.env.PORT || '3001')} ${c.dim(process.env.PORT ? '' : '(default)')}`);
-    console.log(`       DATABASE_PATH: ${c.dim(process.env.DATABASE_PATH || '(using default location)')}`);
-    console.log(`       CLAUDE_CLI_PATH: ${c.dim(process.env.CLAUDE_CLI_PATH || 'claude (default)')}`);
-    console.log(`       CONTEXT_WINDOW: ${c.dim(process.env.CONTEXT_WINDOW || '160000 (default)')}`);
-
-    // Claude projects folder
-    const claudeProjectsPath = path.join(os.homedir(), '.claude', 'projects');
-    const projectsExists = fs.existsSync(claudeProjectsPath);
-    console.log(`\n${c.info('[INFO]')} Claude Projects Folder:`);
-    console.log(`       ${c.dim(claudeProjectsPath)}`);
-    console.log(`       Status: ${projectsExists ? c.ok('[OK] Exists') : c.warn('[WARN] Not found')}`);
-
-    // Config file location
-    const envFilePath = path.join(__dirname, '../.env');
-    const envExists = fs.existsSync(envFilePath);
-    console.log(`\n${c.info('[INFO]')} Configuration File:`);
-    console.log(`       ${c.dim(envFilePath)}`);
-    console.log(`       Status: ${envExists ? c.ok('[OK] Exists') : c.warn('[WARN] Not found (using defaults)')}`);
-
-    console.log('\n' + c.dim('═'.repeat(60)));
-    console.log(`\n${c.tip('[TIP]')} Hints:`);
-    console.log(`      ${c.dim('>')} Use ${c.bright('uc --port 8080')} to run on a custom port`);
-    console.log(`      ${c.dim('>')} Use ${c.bright('uc --database-path /path/to/db')} for custom database`);
-    console.log(`      ${c.dim('>')} Run ${c.bright('uc help')} for all options`);
-    console.log(`      ${c.dim('>')} Access the UI at http://localhost:${process.env.PORT || '3001'}\n`);
+    showStyledStatus({
+        version: packageJson.version,
+        installDir,
+        dbPath,
+        dbExists,
+        dbSize,
+        port: process.env.PORT || '3001',
+        portDefault: !process.env.PORT,
+        claudeCli: findClaudeBinary() || null,
+        apiKey: config.anthropicApiKey || null,
+    });
 }
 
 // Show help
 function showHelp() {
-    console.log(`
-╔═══════════════════════════════════════════════════════════════╗
-║         Upfyn-Code — by Thinqmesh Technologies          ║
-║      Visual AI Coding Interface with tldraw Canvas            ║
-╚═══════════════════════════════════════════════════════════════╝
-
-Usage:
-  upfynai-code [command] [options]
-  uc [command] [options]
-
-Commands:
-  start              Start the server (default)
-  connect            Connect local machine to hosted server (relay bridge)
-  status             Show configuration and data locations
-  install-commands   Install /upfynai-* slash commands to ~/.claude/commands/
-  uninstall-commands Remove /upfynai-* slash commands
-  update             Update to the latest version
-  help               Show this help information
-  version            Show version information
-
-Slash Commands (inside your AI CLI):
-  /upfynai           Start the web UI server
-  /upfynai-connect   Connect CLI session to web UI
-  /upfynai-disconnect Disconnect from web UI
-  /upfynai-status    Show connection status
-  /upfynai-doctor    Run diagnostics
-  /upfynai-export    Export session/canvas data
-  /upfynai-stop      Stop the web UI server
-  /upfynai-local     Full local installation setup
-  /upfynai-uninstall Remove Upfyn-Code
-
-Options:
-  -p, --port <port>           Set server port (default: 3001)
-  --database-path <path>      Set custom database location
-  -h, --help                  Show this help information
-  -v, --version               Show version information
-
-Options:
-  --server <url>     Server URL for connect (default: https://upfynai.thinqmesh.com)
-  --key <token>      Relay token for connect (get from Settings > Relay Tokens)
-
-Examples:
-  $ uc                              # Start with defaults
-  $ uc --port 8080                  # Start on port 8080
-  $ uc connect --key upfyn_xxx      # Bridge to hosted server
-  $ uc start --port 4000            # Explicit start command
-  $ uc status                       # Show configuration
-
-Environment Variables:
-  PORT                Set server port (default: 3001)
-  DATABASE_PATH       Set custom database location
-  CLAUDE_CLI_PATH     Set custom Claude CLI path
-  CONTEXT_WINDOW      Set context window size (default: 160000)
-
-Documentation:
-  ${packageJson.homepage || 'https://github.com/AnitChaudhry/UpfynAI-Code'}
-
-Report Issues:
-  ${packageJson.bugs?.url || 'https://github.com/AnitChaudhry/UpfynAI-Code/issues'}
-`);
+    showStyledHelp(packageJson.version);
 }
 
 // Show version
@@ -221,21 +165,22 @@ function isNewerVersion(v1, v2) {
 // Check for updates
 async function checkForUpdates(silent = false) {
     try {
-        const { execSync } = await import('child_process');
         const latestVersion = execSync('npm show upfynai-code version', { encoding: 'utf8' }).trim();
         const currentVersion = packageJson.version;
 
         if (isNewerVersion(latestVersion, currentVersion)) {
-            console.log(`\n${c.warn('[UPDATE]')} New version available: ${c.bright(latestVersion)} (current: ${currentVersion})`);
-            console.log(`         Run ${c.bright('uc update')} to update\n`);
+            if (!silent) {
+                console.log(`\n  ${c.yellow('!')} New version available: ${c.bBright(latestVersion)} ${c.dim(`(current: ${currentVersion})`)}`);
+                console.log(`    Run ${c.bright('uc update')} to update\n`);
+            }
             return { hasUpdate: true, latestVersion, currentVersion };
         } else if (!silent) {
-            console.log(`${c.ok('[OK]')} You are on the latest version (${currentVersion})`);
+            console.log(`  ${c.green('OK')} You are on the latest version (${currentVersion})`);
         }
         return { hasUpdate: false, latestVersion, currentVersion };
     } catch (e) {
         if (!silent) {
-            console.log(`${c.warn('[WARN]')} Could not check for updates`);
+            console.log(`  ${c.yellow('!')} Could not check for updates`);
         }
         return { hasUpdate: false, error: e.message };
     }
@@ -244,27 +189,28 @@ async function checkForUpdates(silent = false) {
 // Update the package
 async function updatePackage() {
     try {
-        const { execSync } = await import('child_process');
-        console.log(`${c.info('[INFO]')} Checking for updates...`);
+        const spinner = createSpinner('Checking for updates...');
+        spinner.start();
 
         const { hasUpdate, latestVersion, currentVersion } = await checkForUpdates(true);
-
         if (!hasUpdate) {
-            console.log(`${c.ok('[OK]')} Already on the latest version (${currentVersion})`);
+            spinner.stop(`Already on the latest version (${currentVersion})`);
             return;
         }
+        spinner.stop(`Update available: ${currentVersion} -> ${latestVersion}`);
 
-        console.log(`${c.info('[INFO]')} Updating from ${currentVersion} to ${latestVersion}...`);
-        execSync('npm update -g upfynai-code', { stdio: 'inherit' });
-        console.log(`${c.ok('[OK]')} Update complete! Restart uc to use the new version.`);
+        const installSpinner = createSpinner(`Updating to v${latestVersion}...`);
+        installSpinner.start();
+        execSync('npm update -g upfynai-code', { stdio: 'pipe' });
+        installSpinner.stop(`Updated to v${latestVersion}! Restart uc to use the new version.`);
     } catch (e) {
-        console.error(`${c.error('[ERROR]')} Update failed: ${e.message}`);
-        console.log(`${c.tip('[TIP]')} Try running manually: npm update -g upfynai-code`);
+        console.log(`  ${c.red('FAIL')} Update failed: ${e.message}`);
+        console.log(`  ${c.dim('Try running manually:')} ${c.bright('npm update -g upfynai-code')}`);
     }
 }
 
 // Install slash commands to ~/.claude/commands/
-async function installCommands() {
+async function installCommands(silent = false) {
     const commandsSource = path.join(__dirname, '..', 'commands');
     const commandsDest = path.join(os.homedir(), '.claude', 'commands');
 
@@ -273,8 +219,8 @@ async function installCommands() {
     }
 
     if (!fs.existsSync(commandsSource)) {
-        console.error(`${c.error('[ERROR]')} Commands directory not found`);
-        process.exit(1);
+        if (!silent) console.log(`  ${c.red('FAIL')} Commands directory not found`);
+        return 0;
     }
 
     const files = fs.readdirSync(commandsSource).filter(f => f.endsWith('.md'));
@@ -285,12 +231,17 @@ async function installCommands() {
             path.join(commandsSource, file),
             path.join(commandsDest, file)
         );
-        console.log(`${c.ok('[OK]')} Installed /${file.replace('.md', '')}`);
+        if (!silent) {
+            console.log(`  ${c.green('OK')} Installed ${c.violet('/' + file.replace('.md', ''))}`);
+        }
         count++;
     }
 
-    console.log(`\n${c.bright(`${count} slash commands installed!`)}`);
-    console.log(`${c.dim('Available in your AI CLI as /upfynai-*')}\n`);
+    if (!silent) {
+        console.log(`\n  ${c.bBright(`${count} slash commands installed!`)}`);
+        console.log(`  ${c.dim('Available in your AI CLI as /upfynai-*')}\n`);
+    }
+    return count;
 }
 
 // Remove slash commands from ~/.claude/commands/
@@ -301,16 +252,118 @@ async function uninstallCommands() {
         : [];
 
     if (files.length === 0) {
-        console.log(`${c.warn('[WARN]')} No Upfyn-Code commands found to remove`);
+        console.log(`  ${c.yellow('!')} No Upfyn-Code commands found to remove`);
         return;
     }
 
     for (const file of files) {
         fs.unlinkSync(path.join(commandsDest, file));
-        console.log(`${c.ok('[OK]')} Removed /${file.replace('.md', '')}`);
+        console.log(`  ${c.green('OK')} Removed ${c.violet('/' + file.replace('.md', ''))}`);
+    }
+
+    console.log(`\n  ${c.bBright(`${files.length} slash commands removed.`)}\n`);
+}
+
+// --- Config command ---
+function handleConfig(options) {
+    const config = loadConfig();
+
+    if (options.apiKey) {
+        // Set API key
+        config.anthropicApiKey = options.apiKey;
+        saveConfig(config);
+        console.log(`\n  ${c.green('OK')} ${c.white('Anthropic API key saved')}`);
+        console.log(`  ${c.dim('Key:')} sk-ant-...${options.apiKey.slice(-4)}`);
+        console.log(`  ${c.dim('Stored in:')} ${CONFIG_FILE}\n`);
+        return;
     }
 
-    console.log(`\n${c.bright(`${files.length} slash commands removed.`)}\n`);
+    if (options.clearApiKey) {
+        delete config.anthropicApiKey;
+        saveConfig(config);
+        console.log(`\n  ${c.green('OK')} ${c.white('Anthropic API key removed')}\n`);
+        return;
+    }
+
+    // Show current config
+    showConfig({ ...config, _path: CONFIG_FILE });
+}
+
+// --- Launch Interactive (default command) ---
+async function launchInteractive() {
+    // 1. Play rocket animation
+    await playRocketAnimation();
+    clearScreen();
+
+    // 2. Ensure slash commands are installed (silent)
+    await installCommands(true);
+
+    // 3. Find Claude Code binary
+    const claudeBin = findClaudeBinary();
+
+    // 4. Load config for relay + API key
+    const config = loadConfig();
+
+    // 5. Show launch screen
+    showLaunchScreen(packageJson.version, claudeBin, {
+        relayStatus: config.relayKey && config.server
+            ? `Auto-connecting to ${config.server}`
+            : null,
+        apiKey: config.anthropicApiKey || null,
+    });
+
+    // 6. If Claude Code not found, fall back to server mode
+    if (!claudeBin) {
+        console.log(`  ${c.dim('Falling back to server mode...')}\n`);
+        await startServer();
+        return;
+    }
+
+    // 7. Build environment for Claude Code
+    const childEnv = { ...process.env };
+    if (config.anthropicApiKey) {
+        childEnv.ANTHROPIC_API_KEY = config.anthropicApiKey;
+    }
+
+    // 8. Start background relay if config exists
+    if (config.relayKey && config.server) {
+        startBackgroundRelay(config);
+    }
+
+    // 9. Spawn Claude Code interactively
+    const child = spawn(claudeBin, [], {
+        stdio: 'inherit',
+        cwd: process.cwd(),
+        shell: true,
+        env: childEnv,
+    });
+
+    child.on('exit', (code) => {
+        process.exit(code || 0);
+    });
+
+    child.on('error', (err) => {
+        console.log(`\n  ${c.red('FAIL')} Failed to launch Claude Code: ${err.message}`);
+        console.log(`  ${c.dim('Make sure Claude Code is installed:')}`);
+        console.log(`    ${c.bright('npm install -g @anthropic-ai/claude-code')}\n`);
+        process.exit(1);
+    });
+}
+
+// --- Background relay connection ---
+function startBackgroundRelay(config) {
+    // Import and start relay in background (non-blocking)
+    import('./relay-client.js').then(({ connectToServerBackground }) => {
+        if (typeof connectToServerBackground === 'function') {
+            connectToServerBackground({
+                server: config.server,
+                key: config.relayKey,
+                silent: true,
+            });
+        }
+    }).catch(() => {
+        // Relay module not available or no background connect — that's ok
+    });
 }
 
 // Start the server
@@ -318,13 +371,17 @@ async function startServer() {
     // Check for updates silently on startup
     checkForUpdates(true);
 
+    // Show server banner
+    const port = process.env.PORT || '3001';
+    showServerBanner(port, packageJson.version);
+
     // Import and run the server
     await import('./index.js');
 }
 
 // Parse CLI arguments
 function parseArgs(args) {
-    const parsed = { command: 'start', options: {} };
+    const parsed = { command: null, options: {} };
 
     for (let i = 0; i < args.length; i++) {
         const arg = args[i];
@@ -345,15 +402,26 @@ function parseArgs(args) {
             parsed.options.key = args[++i];
         } else if (arg.startsWith('--key=')) {
             parsed.options.key = arg.split('=')[1];
+        } else if (arg === '--api-key') {
+            parsed.options.apiKey = args[++i];
+        } else if (arg.startsWith('--api-key=')) {
+            parsed.options.apiKey = arg.split('=')[1];
+        } else if (arg === '--clear-api-key') {
+            parsed.options.clearApiKey = true;
         } else if (arg === '--help' || arg === '-h') {
             parsed.command = 'help';
         } else if (arg === '--version' || arg === '-v') {
             parsed.command = 'version';
-        } else if (!arg.startsWith('-')) {
+        } else if (!arg.startsWith('-') && !parsed.command) {
             parsed.command = arg;
         }
     }
 
+    // Default command: launch interactive
+    if (!parsed.command) {
+        parsed.command = 'launch';
+    }
+
     return parsed;
 }
 
@@ -371,6 +439,9 @@ async function main() {
     }
 
     switch (command) {
+        case 'launch':
+            await launchInteractive();
+            break;
         case 'start':
             await startServer();
             break;
@@ -391,6 +462,9 @@ async function main() {
         case 'update':
             await updatePackage();
             break;
+        case 'config':
+            handleConfig(options);
+            break;
         case 'install-commands':
             await installCommands();
             break;
@@ -406,14 +480,14 @@ async function main() {
             break;
         }
         default:
-            console.error(`\n❌ Unknown command: ${command}`);
-            console.log('   Run "uc help" for usage information.\n');
+            console.error(`\n  ${c.red('FAIL')} Unknown command: ${command}`);
+            console.log(`    Run ${c.bright('"uc help"')} for usage information.\n`);
             process.exit(1);
     }
 }
 
 // Run the CLI
 main().catch(error => {
-    console.error('\n❌ Error:', error.message);
+    console.error(`\n  ${c.red('FAIL')} Error: ${error.message}`);
     process.exit(1);
 });