@wundr.io/cli 1.0.11 → 1.0.13

package/src/commands/ai.ts

@@ -47,7 +47,7 @@ export class AICommands {
   constructor(
     private program: Command,
     private configManager: ConfigManager,
-    pluginManager: PluginManager,
+    pluginManager: PluginManager
   ) {
     this.pluginManager = pluginManager;
     this.aiService = new AIService(configManager);
@@ -79,7 +79,7 @@ export class AICommands {
       .option(
         '--focus <aspect>',
         'review focus (security, performance, style)',
-        'all',
+        'all'
       )
       .option('--severity <level>', 'minimum issue severity', 'info')
       .option('--suggest-fixes', 'suggest fixes for issues')
@@ -94,12 +94,12 @@ export class AICommands {
       .option(
         '--type <type>',
         'refactoring type (extract, rename, optimize)',
-        'optimize',
+        'optimize'
       )
       .option(
         '--scope <scope>',
         'refactoring scope (function, class, file)',
-        'function',
+        'function'
       )
       .option('--dry-run', 'show refactoring plan without applying')
       .action(async (target, options) => {
@@ -113,12 +113,12 @@ export class AICommands {
       .option(
         '--type <type>',
         'documentation type (api, readme, comments)',
-        'api',
+        'api'
       )
       .option(
         '--format <format>',
         'output format (markdown, html, json)',
-        'markdown',
+        'markdown'
       )
       .option('--include-examples', 'include code examples')
       .action(async (target, options) => {
@@ -132,12 +132,12 @@ export class AICommands {
       .option(
         '--framework <framework>',
         'testing framework (jest, mocha, vitest)',
-        'jest',
+        'jest'
       )
       .option(
         '--coverage <level>',
         'coverage level (unit, integration, e2e)',
-        'unit',
+        'unit'
       )
       .option('--mocks', 'generate mock objects')
       .action(async (target, options) => {
@@ -163,7 +163,7 @@ export class AICommands {
       .option(
         '--aspect <aspect>',
         'analysis aspect (complexity, maintainability, security)',
-        'all',
+        'all'
       )
       .option('--suggestions', 'include improvement suggestions')
       .action(async (target, options) => {
@@ -177,7 +177,7 @@ export class AICommands {
       .option(
         '--focus <focus>',
         'optimization focus (speed, memory, bundle)',
-        'speed',
+        'speed'
       )
       .option('--benchmarks', 'run before/after benchmarks')
       .action(async (target, options) => {
@@ -238,7 +238,7 @@ export class AICommands {
       if (!this.aiService.isReady()) {
         console.log(chalk.red('\n❌ AI service not configured'));
         console.log(
-          chalk.yellow('Run `wundr ai setup` to configure your API key first'),
+          chalk.yellow('Run `wundr ai setup` to configure your API key first')
         );
         return;
       }
@@ -269,7 +269,7 @@ export class AICommands {
         'WUNDR_AI_GENERATE_FAILED',
         'Failed to generate code',
         { type, options },
-        true,
+        true
       );
     }
   }
@@ -283,7 +283,7 @@ export class AICommands {
       if (!this.aiService.isReady()) {
         console.log(chalk.red('\n❌ AI service not configured'));
         console.log(
-          chalk.yellow('Run `wundr ai setup` to configure your API key first'),
+          chalk.yellow('Run `wundr ai setup` to configure your API key first')
         );
         return;
       }
@@ -323,7 +323,7 @@ export class AICommands {
         'WUNDR_AI_REVIEW_FAILED',
         'Failed to review code',
         { files, options },
-        true,
+        true
       );
     }
   }
@@ -337,7 +337,7 @@ export class AICommands {
       if (!this.aiService.isReady()) {
         console.log(chalk.red('\n❌ AI service not configured'));
         console.log(
-          chalk.yellow('Run `wundr ai setup` to configure your API key first'),
+          chalk.yellow('Run `wundr ai setup` to configure your API key first')
         );
         return;
       }
@@ -380,7 +380,7 @@ export class AICommands {
         'WUNDR_AI_REFACTOR_FAILED',
         'Failed to refactor code',
         { target, options },
-        true,
+        true
       );
     }
   }
@@ -394,13 +394,13 @@ export class AICommands {
       if (!this.aiService.isReady()) {
         console.log(chalk.red('\n❌ AI service not configured'));
         console.log(
-          chalk.yellow('Run `wundr ai setup` to configure your API key first'),
+          chalk.yellow('Run `wundr ai setup` to configure your API key first')
         );
         return;
       }
 
       logger.info(
-        `Generating ${options.type} documentation for ${chalk.cyan(target)}...`,
+        `Generating ${options.type} documentation for ${chalk.cyan(target)}...`
       );
 
       const code = await this.readFile(target);
@@ -415,7 +415,7 @@ export class AICommands {
       const outputPath = this.getDocsOutputPath(
         target,
         options.type,
-        options.format,
+        options.format
       );
       await this.saveGeneratedDocs(docs, outputPath);
 
@@ -425,7 +425,7 @@ export class AICommands {
         'WUNDR_AI_DOCS_FAILED',
         'Failed to generate documentation',
         { target, options },
-        true,
+        true
       );
     }
   }
@@ -439,7 +439,7 @@ export class AICommands {
       if (!this.aiService.isReady()) {
         console.log(chalk.red('\n❌ AI service not configured'));
         console.log(
-          chalk.yellow('Run `wundr ai setup` to configure your API key first'),
+          chalk.yellow('Run `wundr ai setup` to configure your API key first')
         );
         return;
       }
@@ -464,7 +464,7 @@ export class AICommands {
         'WUNDR_AI_TEST_FAILED',
         'Failed to generate tests',
         { target, options },
-        true,
+        true
       );
     }
   }
@@ -478,7 +478,7 @@ export class AICommands {
       if (!this.aiService.isReady()) {
         console.log(chalk.red('\n❌ AI service not configured'));
         console.log(
-          chalk.yellow('Run `wundr ai setup` to configure your API key first'),
+          chalk.yellow('Run `wundr ai setup` to configure your API key first')
         );
         return;
       }
@@ -495,7 +495,7 @@ export class AICommands {
         'WUNDR_AI_CHAT_FAILED',
         'Failed to start chat session',
         { options },
-        true,
+        true
       );
     }
   }
@@ -509,7 +509,7 @@ export class AICommands {
       if (!this.aiService.isReady()) {
         console.log(chalk.red('\n❌ AI service not configured'));
         console.log(
-          chalk.yellow('Run `wundr ai setup` to configure your API key first'),
+          chalk.yellow('Run `wundr ai setup` to configure your API key first')
         );
         return;
       }
@@ -530,7 +530,7 @@ export class AICommands {
         'WUNDR_AI_ANALYZE_FAILED',
         'Failed to analyze code',
         { target, options },
-        true,
+        true
       );
     }
   }
@@ -544,7 +544,7 @@ export class AICommands {
       if (!this.aiService.isReady()) {
         console.log(chalk.red('\n❌ AI service not configured'));
         console.log(
-          chalk.yellow('Run `wundr ai setup` to configure your API key first'),
+          chalk.yellow('Run `wundr ai setup` to configure your API key first')
         );
         return;
       }
@@ -588,7 +588,7 @@ export class AICommands {
         'WUNDR_AI_OPTIMIZE_FAILED',
         'Failed to optimize code',
         { target, options },
-        true,
+        true
       );
     }
   }
@@ -606,7 +606,7 @@ export class AICommands {
         'WUNDR_AI_CONFIG_SET_FAILED',
         'Failed to set AI configuration',
         { key, value },
-        true,
+        true
       );
     }
   }
@@ -628,7 +628,7 @@ export class AICommands {
         'WUNDR_AI_CONFIG_GET_FAILED',
         'Failed to get AI configuration',
         { key },
-        true,
+        true
       );
     }
   }
@@ -640,7 +640,7 @@ export class AICommands {
     try {
       console.log(chalk.blue('\n🤖 Wundr AI Setup'));
       console.log(
-        chalk.gray('Configure your AI assistant for enhanced CLI features\n'),
+        chalk.gray('Configure your AI assistant for enhanced CLI features\n')
       );
 
       let { provider, apiKey } = options;
@@ -697,7 +697,7 @@ export class AICommands {
           console.log(chalk.green('✅ API key is valid and working!'));
         } else {
           console.log(
-            chalk.red(`❌ API key validation failed: ${validation.error}`),
+            chalk.red(`❌ API key validation failed: ${validation.error}`)
           );
           return;
         }
@@ -714,7 +714,7 @@ export class AICommands {
         'WUNDR_AI_SETUP_FAILED',
         'Failed to setup AI configuration',
         { options },
-        true,
+        true
       );
     }
   }
@@ -731,24 +731,24 @@ export class AICommands {
       console.log(`Provider: ${chalk.cyan(status.provider)}`);
       console.log(`Model: ${chalk.cyan(status.model)}`);
       console.log(
-        `API Key: ${status.hasApiKey ? chalk.green('✅ Configured') : chalk.red('❌ Missing')}`,
+        `API Key: ${status.hasApiKey ? chalk.green('✅ Configured') : chalk.red('❌ Missing')}`
       );
       console.log(
-        `Status: ${status.ready ? chalk.green('✅ Ready') : chalk.yellow('⚠️  Not Ready')}`,
+        `Status: ${status.ready ? chalk.green('✅ Ready') : chalk.yellow('⚠️  Not Ready')}`
       );
 
       if (!status.hasApiKey) {
         console.log(chalk.yellow('\n⚠️  API key not configured'));
         console.log(
-          chalk.gray('Run `wundr ai setup` to configure your API key'),
+          chalk.gray('Run `wundr ai setup` to configure your API key')
         );
         console.log(
-          chalk.gray('Or set the CLAUDE_API_KEY environment variable'),
+          chalk.gray('Or set the CLAUDE_API_KEY environment variable')
         );
       } else if (!status.ready) {
         console.log(chalk.yellow('\n⚠️  AI service not ready'));
         console.log(
-          chalk.gray('Try running `wundr ai validate` to check connection'),
+          chalk.gray('Try running `wundr ai validate` to check connection')
         );
       }
     } catch (error) {
@@ -756,7 +756,7 @@ export class AICommands {
         'WUNDR_AI_STATUS_FAILED',
         'Failed to get AI status',
         {},
-        true,
+        true
       );
     }
   }
@@ -783,12 +783,12 @@ export class AICommands {
         if (validation.error?.includes('API key')) {
           console.log(chalk.yellow('\n💡 Try:'));
           console.log(
-            chalk.gray('1. Run `wundr ai setup` to configure your API key'),
+            chalk.gray('1. Run `wundr ai setup` to configure your API key')
           );
           console.log(
             chalk.gray(
-              '2. Check your API key is valid and has sufficient credits',
-            ),
+              '2. Check your API key is valid and has sufficient credits'
+            )
           );
           console.log(chalk.gray('3. Verify your internet connection'));
         }
@@ -798,7 +798,7 @@ export class AICommands {
         'WUNDR_AI_VALIDATE_FAILED',
         'Failed to validate AI connection',
         {},
-        true,
+        true
       );
     }
   }
@@ -809,29 +809,153 @@ export class AICommands {
   private async callAI(operation: string, params: any): Promise<any> {
     // Check if AI service is ready
     if (!this.aiService.isReady()) {
-      throw new Error('AI service not configured. Run `wundr ai setup` first.');
+      throw new Error(
+        'AI service not configured. Set ANTHROPIC_API_KEY or OPENAI_API_KEY, or run `wundr ai setup`.'
+      );
     }
 
     logger.debug(`Calling AI for operation: ${operation}`);
 
-    // For now, return mock responses - these would be replaced with actual AI calls
+    const sessionId = `ai-cmd-${operation}-${Date.now()}`;
+
     switch (operation) {
-      case 'generate':
-        return `// Generated ${params.type} code\nfunction ${params.type}() {\n  // Implementation here\n}`;
-      case 'review':
-        return { issues: [], suggestions: [], score: 95 };
-      case 'refactor':
-        return { description: 'Refactoring plan', changes: [] };
-      case 'docs':
-        return `# ${params.target} Documentation\n\nGenerated documentation...`;
-      case 'test':
-        return `// Generated tests\ndescribe('${params.target}', () => {\n  // Tests here\n});`;
-      case 'analyze':
-        return { complexity: 'low', maintainability: 'high', suggestions: [] };
-      case 'optimize':
-        return { description: 'Optimization plan', changes: [] };
+      case 'generate': {
+        const prompt = [
+          `Generate ${params.type} code.`,
+          params.prompt ? `Requirements: ${params.prompt}` : '',
+          params.template ? `Use template style: ${params.template}` : '',
+          params.context ? `Context:\n${params.context}` : '',
+          'Return only the code, no explanation.',
+        ]
+          .filter(Boolean)
+          .join('\n');
+        return this.aiService.sendMessage(sessionId, prompt);
+      }
+
+      case 'review': {
+        const prompt = [
+          `Review the following ${params.file} code for issues.`,
+          `Focus: ${params.focus || 'all'}. Minimum severity: ${params.severity || 'info'}.`,
+          'Return a JSON object with this exact shape: {"issues":[{"severity":"string","description":"string"}],"suggestions":["string"],"score":number}',
+          `Code:\n${params.code}`,
+        ].join('\n');
+        const raw = await this.aiService.sendMessage(sessionId, prompt);
+        try {
+          const jsonMatch = raw.match(/\{[\s\S]*\}/);
+          return jsonMatch
+            ? JSON.parse(jsonMatch[0])
+            : { issues: [], suggestions: [], score: 0 };
+        } catch {
+          return { issues: [], suggestions: [], score: 0 };
+        }
+      }
+
+      case 'refactor': {
+        const prompt = [
+          `Create a refactoring plan for the following code in ${params.target}.`,
+          `Refactoring type: ${params.type || 'optimize'}. Scope: ${params.scope || 'function'}.`,
+          'Return a JSON object with this exact shape: {"description":"string","changes":[{"description":"string","code":"string"}]}',
+          `Code:\n${params.code}`,
+        ].join('\n');
+        const raw = await this.aiService.sendMessage(sessionId, prompt);
+        try {
+          const jsonMatch = raw.match(/\{[\s\S]*\}/);
+          return jsonMatch
+            ? JSON.parse(jsonMatch[0])
+            : { description: '', changes: [] };
+        } catch {
+          return { description: '', changes: [] };
+        }
+      }
+
+      case 'docs': {
+        const prompt = [
+          `Generate ${params.type || 'api'} documentation in ${params.format || 'markdown'} format for ${params.target}.`,
+          params.includeExamples ? 'Include code examples.' : '',
+          `Code:\n${params.code}`,
+        ]
+          .filter(Boolean)
+          .join('\n');
+        return this.aiService.sendMessage(sessionId, prompt);
+      }
+
+      case 'test': {
+        const prompt = [
+          `Generate ${params.coverage || 'unit'} tests using ${params.framework || 'jest'} for ${params.target}.`,
+          params.mocks ? 'Include mock objects where appropriate.' : '',
+          'Return only the test code, no explanation.',
+          `Code:\n${params.code}`,
+        ]
+          .filter(Boolean)
+          .join('\n');
+        return this.aiService.sendMessage(sessionId, prompt);
+      }
+
+      case 'analyze': {
+        const prompt = [
+          `Analyze the following code from ${params.target}.`,
+          `Aspect: ${params.aspect || 'all'}.`,
+          params.suggestions ? 'Include improvement suggestions.' : '',
+          'Return a JSON object with this exact shape: {"complexity":"string","maintainability":"string","suggestions":["string"]}',
+          `Code:\n${params.code}`,
+        ]
+          .filter(Boolean)
+          .join('\n');
+        const raw = await this.aiService.sendMessage(sessionId, prompt);
+        try {
+          const jsonMatch = raw.match(/\{[\s\S]*\}/);
+          return jsonMatch
+            ? JSON.parse(jsonMatch[0])
+            : {
+                complexity: 'unknown',
+                maintainability: 'unknown',
+                suggestions: [],
+              };
+        } catch {
+          return {
+            complexity: 'unknown',
+            maintainability: 'unknown',
+            suggestions: [],
+          };
+        }
+      }
+
+      case 'optimize': {
+        const prompt = [
+          `Create a performance optimization plan for the following code in ${params.target}.`,
+          `Optimization focus: ${params.focus || 'speed'}.`,
+          'Return a JSON object with this exact shape: {"description":"string","changes":[{"description":"string"}],"optimizedCode":"string"}',
+          `Code:\n${params.code}`,
+        ].join('\n');
+        const raw = await this.aiService.sendMessage(sessionId, prompt);
+        try {
+          const jsonMatch = raw.match(/\{[\s\S]*\}/);
+          return jsonMatch
+            ? JSON.parse(jsonMatch[0])
+            : { description: '', changes: [], optimizedCode: params.code };
+        } catch {
+          return { description: '', changes: [], optimizedCode: params.code };
+        }
+      }
+
       case 'chat':
-        return 'This is a mock response. In a real implementation, this would use the AI service.';
+        return this.aiService.sendMessage(sessionId, params.message || '');
+
+      case 'suggest-fix': {
+        const prompt = [
+          `Suggest a fix for the following issue in ${params.file}:`,
+          `Issue (${params.severity}): ${params.issue}`,
+          'Return a JSON object with this shape: {"suggestion":"string"}',
+        ].join('\n');
+        const raw = await this.aiService.sendMessage(sessionId, prompt);
+        try {
+          const jsonMatch = raw.match(/\{[\s\S]*\}/);
+          return jsonMatch ? JSON.parse(jsonMatch[0]) : { suggestion: raw };
+        } catch {
+          return { suggestion: raw };
+        }
+      }
+
       default:
         throw new Error(`Unknown AI operation: ${operation}`);
     }
@@ -850,13 +974,28 @@ export class AICommands {
   }
 
   private async loadContext(contextPath: string): Promise<string> {
-    // Load project context from specified path
-    return `// Context from ${contextPath}`;
+    const fs = await import('fs/promises');
+    const path = await import('path');
+    const stat = await fs.stat(contextPath);
+    if (stat.isDirectory()) {
+      // Read all TypeScript/JavaScript files in the directory (non-recursive, shallow)
+      const entries = await fs.readdir(contextPath);
+      const parts: string[] = [];
+      for (const entry of entries) {
+        if (/\.(ts|tsx|js|jsx)$/.test(entry)) {
+          const filePath = path.join(contextPath, entry);
+          const content = await fs.readFile(filePath, 'utf-8');
+          parts.push(`// ${filePath}\n${content}`);
+        }
+      }
+      return parts.join('\n\n');
+    }
+    return fs.readFile(contextPath, 'utf-8');
   }
 
   private async saveGeneratedCode(
     code: string,
-    outputPath: string,
+    outputPath: string
   ): Promise<void> {
     const fs = await import('fs/promises');
     const path = await import('path');
@@ -871,13 +1010,34 @@ export class AICommands {
   }
 
   private async readFile(filePath: string): Promise<string> {
-    // Read file content
-    return `// Content of ${filePath}`;
+    const fs = await import('fs/promises');
+    return fs.readFile(filePath, 'utf-8');
   }
 
   private async getChangedFiles(): Promise<string[]> {
-    // Get list of changed files from git
-    return ['src/example.ts'];
+    const { execSync } = await import('child_process');
+    try {
+      const output = execSync('git diff --name-only HEAD', {
+        encoding: 'utf-8',
+        stdio: ['pipe', 'pipe', 'pipe'],
+      }).trim();
+      if (!output) {
+        return [];
+      }
+      return output.split('\n').filter(Boolean);
+    } catch {
+      // Fall back to staged changes if HEAD diff fails (e.g. no commits yet)
+      try {
+        const { execSync: exec } = await import('child_process');
+        const output = exec('git diff --name-only --cached', {
+          encoding: 'utf-8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+        }).trim();
+        return output ? output.split('\n').filter(Boolean) : [];
+      } catch {
+        return [];
+      }
+    }
   }
 
   private displayReviewResults(results: FileReviewResult[]): void {
@@ -885,7 +1045,7 @@ export class AICommands {
     results.forEach(result => {
       console.log(`\n${chalk.cyan(result.file)} (Score: ${result.score}/100)`);
       if (result.issues.length > 0) {
-        result.issues.forEach((issue) => {
+        result.issues.forEach(issue => {
           console.log(`  ${issue.severity}: ${issue.description}`);
         });
       }
@@ -904,13 +1064,18 @@ export class AICommands {
             issue: issue.description,
             severity: issue.severity,
           });
-          console.log(`  - ${issue.description}: ${fix.suggestion || 'No automatic fix available'}`);
+          console.log(
+            `  - ${issue.description}: ${fix.suggestion || 'No automatic fix available'}`
+          );
         }
       }
     }
   }
 
-  private async applyRefactoring(target: string, plan: RefactoringPlan): Promise<void> {
+  private async applyRefactoring(
+    target: string,
+    plan: RefactoringPlan
+  ): Promise<void> {
     const fs = await import('fs/promises');
 
     logger.debug(`Applying refactoring to ${target}`);
@@ -928,7 +1093,7 @@ export class AICommands {
   private getDocsOutputPath(
     target: string,
     type: string,
-    format: string,
+    format: string
   ): string {
     const ext = format === 'markdown' ? 'md' : format;
     return `docs/${target}.${type}.${ext}`;
@@ -936,7 +1101,7 @@ export class AICommands {
 
   private async saveGeneratedDocs(
     docs: string,
-    outputPath: string,
+    outputPath: string
   ): Promise<void> {
     const fs = await import('fs/promises');
     const path = await import('path');
@@ -1019,8 +1184,8 @@ export class AICommands {
         console.log(chalk.red(`Error: ${error.message}`));
         console.log(
           chalk.yellow(
-            '\nTip: Try `wundr ai validate` to check your connection',
-          ),
+            '\nTip: Try `wundr ai validate` to check your connection'
+          )
         );
       }
     }
@@ -1066,7 +1231,7 @@ export class AICommands {
 
   private async applyOptimization(
     target: string,
-    optimization: OptimizationPlan,
+    optimization: OptimizationPlan
   ): Promise<void> {
     const fs = await import('fs/promises');
 
@@ -1085,7 +1250,10 @@ export class AICommands {
     }
   }
 
-  private displayBenchmarkComparison(before: BenchmarkResult, after: BenchmarkResult): void {
+  private displayBenchmarkComparison(
+    before: BenchmarkResult,
+    after: BenchmarkResult
+  ): void {
     console.log(chalk.blue('\nBenchmark Comparison:'));
     const timeDiff = after.time - before.time;
     const memDiff = after.memory - before.memory;
@@ -1094,10 +1262,10 @@ export class AICommands {
     const memColor = memDiff < 0 ? chalk.green : chalk.yellow;
 
     console.log(
-      `Time: ${before.time}ms → ${after.time}ms (${timeColor(timeDiff > 0 ? '+' : '')}${timeColor(timeDiff + 'ms')})`,
+      `Time: ${before.time}ms → ${after.time}ms (${timeColor(timeDiff > 0 ? '+' : '')}${timeColor(timeDiff + 'ms')})`
     );
     console.log(
-      `Memory: ${before.memory}MB → ${after.memory}MB (${memColor(memDiff > 0 ? '+' : '')}${memColor(memDiff + 'MB')})`,
+      `Memory: ${before.memory}MB → ${after.memory}MB (${memColor(memDiff > 0 ? '+' : '')}${memColor(memDiff + 'MB')})`
     );
   }
 }