codebakers 2.2.2 → 2.3.4

package/src/index.ts

@@ -5,6 +5,7 @@ import * as p from '@clack/prompts';
 import chalk from 'chalk';
 import boxen from 'boxen';
 import gradient from 'gradient-string';
+import * as path from 'path';
 import { Config } from './utils/config.js';
 import { checkForUpdates } from './utils/updates.js';
 import { setupCommand } from './commands/setup.js';
@@ -27,9 +28,9 @@ import { prdMakerCommand } from './commands/prd-maker.js';
 import { buildCommand } from './commands/build.js';
 import { integrateCommand, INTEGRATIONS } from './commands/integrate.js';
 import { websiteCommand } from './commands/website.js';
-import { parseNaturalLanguage, clarifyCommand, clarifyDeployTarget } from './utils/nlp.js';
+import { parseNaturalLanguage } from './utils/nlp.js';
 
-const VERSION = '2.2.2';
+const VERSION = '2.3.4';
 
 // ASCII art logo
 const logo = `
@@ -41,115 +42,195 @@ const logo = `
    ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝╚═════╝ ╚═╝  ╚═╝╚═╝  ╚═╝╚══════╝╚═╝  ╚═╝╚══════╝
 `;
 
-// Main interactive menu
+// ============================================================================
+// MAIN MENU
+// ============================================================================
+
 async function showMainMenu(): Promise<void> {
   const config = new Config();
-  const isSetup = config.isConfigured();
 
   // Show logo
   console.log(gradient.pastel.multiline(logo));
   console.log(chalk.dim(`  v${VERSION} — AI dev team that follows the rules\n`));
 
-  // If not setup, run setup wizard
-  if (!isSetup) {
+  // Check setup status
+  const hasAnthropic = !!config.getCredentials('anthropic')?.apiKey;
+
+  // STEP 1: First time? Run setup
+  if (!hasAnthropic) {
     console.log(boxen(
-      chalk.yellow('Welcome to CodeBakers! Let\'s get you set up.'),
+      chalk.yellow('Welcome to CodeBakers!\n\n') +
+      chalk.white('Let\'s connect your Anthropic API key so the AI can work.\n') +
+      chalk.dim('(Takes about 1 minute)'),
       { padding: 1, borderColor: 'yellow', borderStyle: 'round' }
     ));
+    
     await setupCommand();
+    
+    // After setup, show next steps
+    showPostSetupInstructions();
     return;
   }
 
-  // Check if in a project directory
+  // STEP 2: Detect context
   const inProject = config.isInProject();
+  const projectConfig = config.getProjectConfig();
+  const cwd = process.cwd();
+  const folderName = path.basename(cwd);
+
+  // Show context
+  if (inProject && projectConfig) {
+    const framework = projectConfig.framework || 'detected';
+    console.log(chalk.cyan(`  📁 Working in: ${chalk.bold(folderName)}`));
+    console.log(chalk.dim(`     ${framework} project\n`));
+    
+    // SHOW PROJECT MENU
+    await showProjectMenu(config);
+  } else {
+    console.log(chalk.cyan(`  📁 Current folder: ${chalk.bold(cwd)}`));
+    console.log(chalk.dim(`     Not a project folder\n`));
+    
+    // SHOW START MENU
+    await showStartMenu(config);
+  }
+}
 
-  // Build menu options based on context
-  const menuOptions = [];
-  
-  // Always show these creation options at top
-  menuOptions.push({ value: 'new', label: '🆕 Create new project' });
-  menuOptions.push({ value: 'website', label: '🌐 Website Builder', hint: 'Describe → AI builds it' });
-  menuOptions.push({ value: 'build', label: '🏗️  Parallel Build', hint: '3 agents from PRD' });
-  menuOptions.push({ value: 'prd', label: '📄 Build from PRD', hint: 'sequential build' });
-  menuOptions.push({ value: 'prd-maker', label: '✏️  Create PRD', hint: 'interview → generate PRD' });
-  menuOptions.push({ value: 'advisors', label: '🌟 Dream Team Advisors', hint: 'consult with experts' });
-  
-  // If in a project, show project-specific options
-  if (inProject) {
-    menuOptions.push({ value: 'separator1', label: '───── Current Project ─────' });
-    menuOptions.push({ value: 'code', label: '💬 Code with AI', hint: 'build features, fix bugs' });
-    menuOptions.push({ value: 'check', label: '🔍 Check code quality', hint: 'run pattern enforcement' });
-    menuOptions.push({ value: 'deploy', label: '🚀 Deploy', hint: 'deploy to Vercel' });
-    menuOptions.push({ value: 'fix', label: '🔧 Fix errors', hint: 'auto-fix with AI' });
-    menuOptions.push({ value: 'generate', label: '⚡ Generate', hint: 'scaffold components, pages' });
-    menuOptions.push({ value: 'migrate', label: '🗄️  Database migrations', hint: 'push, generate, status' });
+// ============================================================================
+// START MENU (Not in a project)
+// ============================================================================
+
+async function showStartMenu(config: Config): Promise<void> {
+  while (true) {
+    const action = await p.select({
+      message: 'What would you like to do?',
+      options: [
+        { value: 'website', label: '🌐 Build a website', hint: 'Describe it → AI builds it' },
+        { value: 'new', label: '🆕 Create new project', hint: 'Start fresh with Next.js, React, etc.' },
+        { value: 'prd-maker', label: '✏️  Create a PRD', hint: 'Plan your project first' },
+        { value: 'build', label: '🏗️  Build from PRD', hint: 'Have a PRD file? Build it now' },
+        { value: 'advisors', label: '🌟 Consult advisors', hint: 'Get advice from AI experts' },
+        { value: 'divider1', label: chalk.dim('─────────────────────────────') },
+        { value: 'integrate', label: '🔌 Add integration', hint: 'Stripe, Supabase, Twilio, etc.' },
+        { value: 'settings', label: '⚙️  Settings', hint: 'API keys & preferences' },
+        { value: 'help', label: '❓ Help', hint: 'Learn how to use CodeBakers' },
+        { value: 'divider2', label: chalk.dim('─────────────────────────────') },
+        { value: 'exit', label: '🚪 Return to terminal', hint: 'Exit CodeBakers' },
+      ],
+    });
+
+    // User pressed Ctrl+C or selected exit
+    if (p.isCancel(action) || action === 'exit') {
+      console.log('');
+      console.log(chalk.cyan('  ─────────────────────────────────────────────────'));
+      console.log(chalk.white('  You\'re back in the terminal.'));
+      console.log('');
+      console.log(chalk.dim('  To start CodeBakers again, type:'));
+      console.log(chalk.green('    codebakers'));
+      console.log('');
+      console.log(chalk.dim('  Quick commands you can run directly:'));
+      console.log(chalk.dim('    codebakers website') + chalk.gray(' - Build a website'));
+      console.log(chalk.dim('    codebakers code') + chalk.gray('    - Code with AI'));
+      console.log(chalk.dim('    codebakers help') + chalk.gray('    - See all commands'));
+      console.log(chalk.cyan('  ─────────────────────────────────────────────────'));
+      console.log('');
+      process.exit(0);
+    }
+
+    // Handle dividers - just continue the loop
+    if (action === 'divider1' || action === 'divider2') {
+      continue;
+    }
+
+    // Handle the selected action
+    await handleAction(action as string, config);
+    
+    // After action completes, loop back to show menu again
   }
-  
-  // Always show these
-  menuOptions.push({ value: 'separator2', label: '───── Tools & Settings ─────' });
-  menuOptions.push({ value: 'integrate', label: '🔌 One-Click Integrations', hint: '50+ services' });
-  menuOptions.push({ value: 'gateway', label: '📱 Channel gateway', hint: 'WhatsApp, Telegram, etc.' });
-  menuOptions.push({ value: 'connect', label: '🔗 Connect service', hint: 'API keys' });
-  menuOptions.push({ value: 'settings', label: '⚙️  Settings' });
-  menuOptions.push({ value: 'help', label: '❓ Help' });
-
-  // Main menu
-  const action = await p.select({
-    message: 'What do you want to do?',
-    options: menuOptions,
-  });
-
-  if (p.isCancel(action)) {
-    p.cancel('Goodbye!');
-    process.exit(0);
+}
+
+// ============================================================================
+// PROJECT MENU (In a project)
+// ============================================================================
+
+async function showProjectMenu(config: Config): Promise<void> {
+  while (true) {
+    const action = await p.select({
+      message: 'What would you like to do?',
+      options: [
+        { value: 'code', label: '💬 Code with AI', hint: 'Ask AI to build features or fix bugs' },
+        { value: 'deploy', label: '🚀 Deploy', hint: 'Push your project live' },
+        { value: 'check', label: '🔍 Check code', hint: 'Review quality & patterns' },
+        { value: 'fix', label: '🔧 Fix errors', hint: 'AI automatically fixes issues' },
+        { value: 'integrate', label: '🔌 Add integration', hint: 'Stripe, Supabase, Twilio, etc.' },
+        { value: 'generate', label: '⚡ Generate', hint: 'Create components, pages, APIs' },
+        { value: 'divider1', label: chalk.dim('─────────────────────────────') },
+        { value: 'new', label: '🆕 Start new project', hint: 'Create a different project' },
+        { value: 'settings', label: '⚙️  Settings', hint: 'API keys & preferences' },
+        { value: 'help', label: '❓ Help', hint: 'Learn how to use CodeBakers' },
+        { value: 'divider2', label: chalk.dim('─────────────────────────────') },
+        { value: 'exit', label: '🚪 Return to terminal', hint: 'Exit CodeBakers' },
+      ],
+    });
+
+    // User pressed Ctrl+C or selected exit
+    if (p.isCancel(action) || action === 'exit') {
+      console.log('');
+      console.log(chalk.cyan('  ─────────────────────────────────────────────────'));
+      console.log(chalk.white('  You\'re back in the terminal.'));
+      console.log('');
+      console.log(chalk.dim('  To start CodeBakers again, type:'));
+      console.log(chalk.green('    codebakers'));
+      console.log('');
+      console.log(chalk.dim('  Quick commands you can run directly:'));
+      console.log(chalk.dim('    codebakers code') + chalk.gray('    - Code with AI'));
+      console.log(chalk.dim('    codebakers deploy') + chalk.gray('  - Deploy your project'));
+      console.log(chalk.dim('    codebakers help') + chalk.gray('    - See all commands'));
+      console.log(chalk.cyan('  ─────────────────────────────────────────────────'));
+      console.log('');
+      process.exit(0);
+    }
+
+    // Handle dividers - just continue the loop
+    if (action === 'divider1' || action === 'divider2') {
+      continue;
+    }
+
+    // Handle the selected action
+    await handleAction(action as string, config);
+    
+    // After action completes, loop back to show menu again
   }
+}
+
+// ============================================================================
+// ACTION HANDLER
+// ============================================================================
 
-  // Handle actions
+async function handleAction(action: string, config: Config): Promise<void> {
   switch (action) {
+    case 'website':
+      await websiteCommand();
+      break;
+    case 'new':
+      await initCommand();
+      break;
     case 'code':
       await codeCommand();
       break;
-    case 'integrate':
-      await integrateCommand();
-      break;
     case 'check':
       await checkCommand();
       break;
     case 'deploy':
       await deployCommand();
       break;
-    case 'migrate':
-      await migrateCommand();
-      break;
     case 'fix':
       await fixCommand();
       break;
     case 'generate':
       await generateCommand();
       break;
-    case 'status':
-      await statusCommand();
-      break;
-    case 'connect':
-      await connectCommand();
-      break;
-    case 'gateway':
-      await gatewayCommand();
-      break;
-    case 'learn':
-      await learnCommand();
-      break;
-    case 'security':
-      await securityCommand();
-      break;
-    case 'design':
-      await designCommand();
-      break;
-    case 'new':
-      await initCommand();
-      break;
-    case 'website':
-      await websiteCommand();
+    case 'integrate':
+      await integrateCommand();
       break;
     case 'build':
       await buildCommand();
@@ -163,6 +244,12 @@ async function showMainMenu(): Promise<void> {
     case 'advisors':
       await advisorsCommand();
       break;
+    case 'gateway':
+      await gatewayCommand();
+      break;
+    case 'migrate':
+      await migrateCommand();
+      break;
     case 'settings':
       await setupCommand();
       break;
@@ -170,45 +257,89 @@ async function showMainMenu(): Promise<void> {
       showHelp();
       break;
     default:
-      // Separators, do nothing
-      await showMainMenu();
+      console.log(chalk.yellow('Coming soon!'));
   }
 }
 
+// ============================================================================
+// POST-SETUP INSTRUCTIONS
+// ============================================================================
+
+function showPostSetupInstructions(): void {
+  console.log(boxen(
+    chalk.green.bold('✓ Setup complete!\n\n') +
+    chalk.white('What\'s next?\n\n') +
+    chalk.cyan('1. ') + 'Navigate to where you want to build:\n' +
+    chalk.dim('   cd C:\\dev\\my-project\n\n') +
+    chalk.cyan('2. ') + 'Run CodeBakers:\n' +
+    chalk.dim('   codebakers\n\n') +
+    chalk.white('Or build a website right now:\n') +
+    chalk.dim('   codebakers website'),
+    { padding: 1, borderColor: 'green', borderStyle: 'round' }
+  ));
+}
+
+// ============================================================================
+// POST-BUILD INSTRUCTIONS (exported for other commands to use)
+// ============================================================================
+
+export function showPostBuildInstructions(projectName: string, projectPath?: string): void {
+  const displayPath = projectPath || projectName;
+  
+  console.log(boxen(
+    chalk.green.bold(`✓ ${projectName} created!\n\n`) +
+    chalk.white('Next steps:\n\n') +
+    chalk.cyan('1. ') + 'Go to your project:\n' +
+    chalk.dim(`   cd ${displayPath}\n\n`) +
+    chalk.cyan('2. ') + 'Install dependencies:\n' +
+    chalk.dim('   npm install\n\n') +
+    chalk.cyan('3. ') + 'Start the dev server:\n' +
+    chalk.dim('   npm run dev\n\n') +
+    chalk.cyan('4. ') + 'Open in browser:\n' +
+    chalk.dim('   http://localhost:3000\n\n') +
+    chalk.white('Ready to deploy?\n') +
+    chalk.dim('   codebakers deploy'),
+    { padding: 1, borderColor: 'green', borderStyle: 'round' }
+  ));
+}
+
+// ============================================================================
+// HELP
+// ============================================================================
+
 function showHelp(): void {
   console.log(boxen(`
 ${chalk.bold('CodeBakers CLI v' + VERSION)} — AI dev team that follows the rules
 
 ${chalk.bold.cyan('Getting Started:')}
-  ${chalk.cyan('codebakers setup')}        Connect your accounts (GitHub, Vercel, etc.)
-  ${chalk.cyan('codebakers init')}         Create a new project from scratch
-  ${chalk.cyan('codebakers website')}      Build a website by describing it
-  ${chalk.cyan('codebakers build')}        Build from PRD with parallel AI agents
+  ${chalk.cyan('codebakers')}              Interactive menu
+  ${chalk.cyan('codebakers setup')}        Connect API keys
+  ${chalk.cyan('codebakers website')}      Build website by describing it
+  ${chalk.cyan('codebakers init')}         Create new project
 
-${chalk.bold.cyan('Daily Development:')}
+${chalk.bold.cyan('In a Project:')}
   ${chalk.cyan('codebakers code')}         Chat with AI to build features
-  ${chalk.cyan('codebakers check')}        Check code quality & patterns
-  ${chalk.cyan('codebakers fix')}          Auto-fix errors with AI
   ${chalk.cyan('codebakers deploy')}       Deploy to Vercel
-
-${chalk.bold.cyan('Integrations:')}
-  ${chalk.cyan('codebakers integrate')}    50+ one-click integrations (Stripe, Supabase, etc.)
-  ${chalk.cyan('codebakers gateway')}      Connect WhatsApp, Telegram, Discord, etc.
+  ${chalk.cyan('codebakers check')}        Check code quality
+  ${chalk.cyan('codebakers fix')}          Auto-fix errors
 
 ${chalk.bold.cyan('Planning:')}
-  ${chalk.cyan('codebakers prd-maker')}    Create PRD through interview (voice supported)
-  ${chalk.cyan('codebakers advisors')}     Consult AI experts (CEO, CTO, Designer, etc.)
+  ${chalk.cyan('codebakers prd-maker')}    Create PRD through interview
+  ${chalk.cyan('codebakers build')}        Build from PRD (parallel agents)
+  ${chalk.cyan('codebakers advisors')}     Consult AI experts
 
-${chalk.bold.cyan('Tips:')}
-  • Type naturally: ${chalk.dim('codebakers "add a contact form"')}
-  • Voice input: Type ${chalk.yellow('"v"')} at any prompt
-  • Clipboard: Type ${chalk.yellow('"clip"')} to paste from clipboard
+${chalk.bold.cyan('Integrations:')}
+  ${chalk.cyan('codebakers integrate')}    50+ one-click integrations
+  ${chalk.cyan('codebakers gateway')}      WhatsApp, Telegram, Discord
 
 ${chalk.bold('Docs:')} ${chalk.dim('https://codebakers.dev/docs')}
 `, { padding: 1, borderColor: 'cyan', borderStyle: 'round' }));
 }
 
-// CLI setup with Commander
+// ============================================================================
+// CLI COMMANDS (Commander)
+// ============================================================================
+
 const program = new Command();
 
 program
@@ -219,225 +350,191 @@ program
 
 program
   .command('setup')
-  .description('Configure CodeBakers (first-time setup)')
+  .description('Configure CodeBakers (API keys)')
   .action(setupCommand);
 
 program
   .command('init')
   .description('Create a new project')
-  .option('-n, --name <name>', 'Project name')
-  .option('-t, --template <template>', 'Template (nextjs, voice-agent, saas)')
-  .option('--no-git', 'Skip GitHub repo creation')
-  .option('--no-vercel', 'Skip Vercel deployment')
-  .option('--no-supabase', 'Skip Supabase setup')
+  .option('-n, --name <n>', 'Project name')
+  .option('-t, --template <t>', 'Template')
   .action(initCommand);
 
 program
   .command('code [prompt]')
   .description('Start AI coding session')
-  .option('-w, --watch', 'Watch mode - continuous assistance')
+  .option('-w, --watch', 'Watch mode')
   .action(codeCommand);
 
 program
   .command('check')
-  .description('Run CodeBakers pattern enforcement')
-  .option('-f, --fix', 'Auto-fix violations')
-  .option('--watch', 'Watch mode')
+  .description('Run code quality checks')
   .action(checkCommand);
 
 program
   .command('deploy')
   .description('Deploy to production')
-  .option('-p, --preview', 'Deploy to preview only')
-  .option('--no-check', 'Skip pattern check')
+  .option('-p, --preview', 'Preview deployment')
+  .option('--no-check', 'Skip quality check')
   .action(deployCommand);
 
 program
   .command('fix')
-  .description('Auto-fix build and deployment errors')
-  .option('--auto', 'Fix without asking')
+  .description('Auto-fix errors with AI')
   .action(fixCommand);
 
 program
-  .command('generate <type>')
-  .alias('g')
-  .description('Generate components, pages, API routes')
-  .option('-n, --name <name>', 'Name of the generated item')
+  .command('generate')
+  .alias('gen')
+  .description('Generate components, pages, etc.')
   .action(generateCommand);
 
 program
-  .command('connect [service]')
-  .description('Connect external services (Stripe, VAPI, etc.)')
+  .command('status')
+  .description('View project status')
+  .action(statusCommand);
+
+program
+  .command('connect')
+  .description('Connect external services')
   .action(connectCommand);
 
 program
   .command('gateway')
-  .description('Manage messaging channel gateway')
-  .option('--start', 'Start the gateway')
-  .option('--stop', 'Stop the gateway')
-  .option('--status', 'Show gateway status')
+  .description('Manage messaging channels (WhatsApp, Telegram, etc.)')
+  .option('--start', 'Start gateway')
+  .option('--stop', 'Stop gateway')
+  .option('--status', 'Show status')
   .action(gatewayCommand);
 
-program
-  .command('status')
-  .description('View project status and health')
-  .option('-a, --all', 'Show all projects')
-  .action(statusCommand);
-
 program
   .command('security')
   .description('Run security audit')
-  .option('--fix', 'Auto-fix security issues')
   .action(securityCommand);
 
 program
   .command('learn')
-  .description('View and manage learning settings')
-  .option('--forget <item>', 'Forget a learned preference')
-  .option('--reset', 'Reset all learning')
-  .option('--export', 'Export learned patterns')
+  .description('View/manage what CodeBakers has learned')
   .action(learnCommand);
 
 program
-  .command('design [action]')
-  .description('Manage design system (profile, palette, check)')
+  .command('design')
+  .description('Manage design system')
   .action(designCommand);
 
 program
   .command('prd [file]')
-  .description('Build entire project from a PRD document')
+  .description('Build project from PRD')
   .action(prdCommand);
 
 program
   .command('advisors')
   .alias('dream-team')
-  .description('Consult with the CodeBakers Dream Team advisory board')
+  .description('Consult AI expert advisors')
   .action(advisorsCommand);
 
 program
   .command('migrate')
   .alias('db')
-  .description('Database migrations (push, generate, status)')
-  .option('--push', 'Push migrations to database')
-  .option('--generate', 'Generate new migration')
-  .option('--status', 'Check migration status')
+  .description('Database migrations')
+  .option('--push', 'Push migrations')
+  .option('--generate', 'Generate migration')
+  .option('--status', 'Check status')
   .action(migrateCommand);
 
 program
   .command('prd-maker')
   .alias('create-prd')
-  .description('Create a PRD through guided interview (supports voice input)')
+  .description('Create PRD through guided interview')
   .action(prdMakerCommand);
 
 program
   .command('build [prd-file]')
   .alias('swarm')
-  .description('Parallel build with 3 AI agents (self-healing)')
-  .option('--sequential', 'Disable parallel execution')
+  .description('Parallel build with AI agents')
+  .option('--sequential', 'Disable parallel')
   .action(buildCommand);
 
 program
   .command('integrate [integration]')
   .alias('add')
-  .description('One-click integrations (50+ services with browser auth)')
+  .description('One-click integrations (50+ services)')
   .action(integrateCommand);
 
 program
   .command('website')
   .alias('site')
-  .description('Build a website by describing it in plain English')
+  .description('Build website by describing it')
   .action(websiteCommand);
 
-// Natural language processing for free-form input
+program
+  .command('help')
+  .description('Show help')
+  .action(showHelp);
+
+// ============================================================================
+// NATURAL LANGUAGE HANDLER
+// ============================================================================
+
 async function handleNaturalLanguage(input: string): Promise<void> {
   const config = new Config();
   
-  console.log(chalk.dim('\n  Parsing your request...\n'));
+  console.log(chalk.dim('\n  Understanding your request...\n'));
   
   const parsed = await parseNaturalLanguage(input, config);
   
   if (!parsed) {
-    // Fall back to code command
+    // Fall back to code command with the input as prompt
     await codeCommand(input);
     return;
   }
 
-  // Clarify if needed
-  const clarified = await clarifyCommand(parsed);
-  
-  if (clarified.command === 'cancel') {
-    return;
-  }
-
-  // Execute the command
-  switch (clarified.command) {
-    case 'init':
-      await initCommand();
+  // Handle recognized commands
+  switch (parsed.command) {
+    case 'deploy':
+      await deployCommand();
       break;
     case 'build':
-      await buildCommand(clarified.args[0]);
+      await buildCommand();
       break;
     case 'code':
-      await codeCommand(clarified.args.join(' '));
+      await codeCommand(input);
+      break;
+    case 'init':
+      await initCommand();
+      break;
+    case 'website':
+      await websiteCommand();
       break;
     case 'check':
       await checkCommand();
       break;
-    case 'deploy':
-      const target = await clarifyDeployTarget();
-      if (target) {
-        await deployCommand({ preview: target === 'preview' });
-      }
-      break;
     case 'fix':
       await fixCommand();
       break;
-    case 'migrate':
-      await migrateCommand();
-      break;
-    case 'design':
-      await designCommand();
-      break;
-    case 'advisors':
-      await advisorsCommand();
-      break;
-    case 'prd-maker':
-      await prdMakerCommand();
-      break;
-    case 'security':
-      await securityCommand();
-      break;
-    case 'status':
-      await statusCommand();
-      break;
-    case 'setup':
-      await setupCommand();
-      break;
-    case 'help':
-      showHelp();
-      break;
     default:
-      // Unknown command - pass to code agent
+      // Fall back to code command
       await codeCommand(input);
   }
 }
 
-// Parse args or show menu
+// ============================================================================
+// ENTRY POINT
+// ============================================================================
+
 const args = process.argv.slice(2);
 
 if (args.length === 0) {
-  // Check for updates in background
+  // No args - show interactive menu
   checkForUpdates().catch(() => {});
-  
-  // Show interactive menu
   showMainMenu().catch(console.error);
 } else if (args.length > 0 && args[0] && !args[0].startsWith('-')) {
   // Check if it's a known command
   const knownCommands = [
-    'setup', 'init', 'code', 'check', 'deploy', 'fix', 'generate',
+    'setup', 'init', 'code', 'check', 'deploy', 'fix', 'generate', 'gen',
     'connect', 'status', 'gateway', 'security', 'learn', 'design',
-    'prd', 'advisors', 'migrate', 'prd-maker', 'build', 'swarm', 
-    'integrate', 'add', 'website', 'site', 'help'
+    'prd', 'advisors', 'dream-team', 'migrate', 'db', 'prd-maker', 'create-prd',
+    'build', 'swarm', 'integrate', 'add', 'website', 'site', 'help'
   ];
   
   const firstArg = args[0] as string;