@wundr.io/cli 1.0.1 → 1.0.4

package/src/commands/ai.ts

@@ -1,24 +1,55 @@
-import { Command } from 'commander';
-import inquirer from 'inquirer';
 import chalk from 'chalk';
-import { ConfigManager } from '../utils/config-manager';
-import { PluginManager } from '../plugins/plugin-manager';
+import inquirer from 'inquirer';
+
 import { AIService } from '../ai/ai-service';
-import { logger } from '../utils/logger';
 import { errorHandler } from '../utils/error-handler';
-import { ChatSession } from '../types';
+import { logger } from '../utils/logger';
+
+import type { PluginManager } from '../plugins/plugin-manager';
+import type { ChatSession } from '../types';
+import type { ConfigManager } from '../utils/config-manager';
+import type { Command } from 'commander';
+
+/** Review result for a single file */
+interface FileReviewResult {
+  file: string;
+  issues: Array<{ severity: string; description: string }>;
+  suggestions: string[];
+  score: number;
+}
+
+/** Refactoring plan from AI analysis */
+interface RefactoringPlan {
+  description: string;
+  changes: Array<{ description: string; code?: string }>;
+}
+
+/** Optimization plan from AI analysis */
+interface OptimizationPlan {
+  description: string;
+  changes: Array<{ description: string; code?: string }>;
+  optimizedCode: string;
+}
+
+/** Benchmark result metrics */
+interface BenchmarkResult {
+  time: number;
+  memory: number;
+}
 
 /**
  * AI commands for AI-powered development features
  */
 export class AICommands {
   private aiService: AIService;
+  private pluginManager: PluginManager;
 
   constructor(
     private program: Command,
     private configManager: ConfigManager,
-    private _pluginManager: PluginManager
+    pluginManager: PluginManager,
   ) {
+    this.pluginManager = pluginManager;
     this.aiService = new AIService(configManager);
     this.registerCommands();
   }
@@ -48,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')
@@ -63,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) => {
@@ -82,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) => {
@@ -101,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) => {
@@ -132,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) => {
@@ -146,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) => {
@@ -207,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;
       }
@@ -238,7 +269,7 @@ export class AICommands {
         'WUNDR_AI_GENERATE_FAILED',
         'Failed to generate code',
         { type, options },
-        true
+        true,
       );
     }
   }
@@ -252,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;
       }
@@ -261,12 +292,7 @@ export class AICommands {
 
       const filesToReview =
         files.length > 0 ? files : await this.getChangedFiles();
-      const reviewResults: Array<{
-        file: string;
-        issues: any;
-        suggestions: any;
-        score: any;
-      }> = [];
+      const reviewResults: FileReviewResult[] = [];
 
       for (const file of filesToReview) {
         logger.debug(`Reviewing ${file}...`);
@@ -297,7 +323,7 @@ export class AICommands {
         'WUNDR_AI_REVIEW_FAILED',
         'Failed to review code',
         { files, options },
-        true
+        true,
       );
     }
   }
@@ -311,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;
       }
@@ -354,7 +380,7 @@ export class AICommands {
         'WUNDR_AI_REFACTOR_FAILED',
         'Failed to refactor code',
         { target, options },
-        true
+        true,
       );
     }
   }
@@ -368,17 +394,17 @@ 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);
-      const _docs = await this.callAI('docs', {
+      const docs = await this.callAI('docs', {
         code,
         target,
         type: options.type,
@@ -389,9 +415,9 @@ export class AICommands {
       const outputPath = this.getDocsOutputPath(
         target,
         options.type,
-        options.format
+        options.format,
       );
-      await this.saveGeneratedDocs(_docs, outputPath);
+      await this.saveGeneratedDocs(docs, outputPath);
 
       logger.success(`Documentation generated: ${outputPath}`);
     } catch (error) {
@@ -399,7 +425,7 @@ export class AICommands {
         'WUNDR_AI_DOCS_FAILED',
         'Failed to generate documentation',
         { target, options },
-        true
+        true,
       );
     }
   }
@@ -413,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;
       }
@@ -438,7 +464,7 @@ export class AICommands {
         'WUNDR_AI_TEST_FAILED',
         'Failed to generate tests',
         { target, options },
-        true
+        true,
       );
     }
   }
@@ -452,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;
       }
@@ -469,7 +495,7 @@ export class AICommands {
         'WUNDR_AI_CHAT_FAILED',
         'Failed to start chat session',
         { options },
-        true
+        true,
       );
     }
   }
@@ -483,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;
       }
@@ -504,7 +530,7 @@ export class AICommands {
         'WUNDR_AI_ANALYZE_FAILED',
         'Failed to analyze code',
         { target, options },
-        true
+        true,
       );
     }
   }
@@ -518,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;
       }
@@ -526,13 +552,13 @@ export class AICommands {
       logger.info(`Optimizing ${chalk.cyan(target)}...`);
 
       const code = await this.readFile(target);
-      let beforeBenchmark = null;
+      let beforeBenchmark: BenchmarkResult | undefined;
 
       if (options.benchmarks) {
         beforeBenchmark = await this.runBenchmark(target);
       }
 
-      const _optimization = await this.callAI('optimize', {
+      const optimization: OptimizationPlan = await this.callAI('optimize', {
         code,
         target,
         focus: options.focus,
@@ -548,9 +574,9 @@ export class AICommands {
       ]);
 
       if (apply) {
-        await this.applyOptimization(target, _optimization);
+        await this.applyOptimization(target, optimization);
 
-        if (options.benchmarks) {
+        if (options.benchmarks && beforeBenchmark) {
           const afterBenchmark = await this.runBenchmark(target);
           this.displayBenchmarkComparison(beforeBenchmark, afterBenchmark);
         }
@@ -562,7 +588,7 @@ export class AICommands {
         'WUNDR_AI_OPTIMIZE_FAILED',
         'Failed to optimize code',
         { target, options },
-        true
+        true,
       );
     }
   }
@@ -580,7 +606,7 @@ export class AICommands {
         'WUNDR_AI_CONFIG_SET_FAILED',
         'Failed to set AI configuration',
         { key, value },
-        true
+        true,
       );
     }
   }
@@ -602,7 +628,7 @@ export class AICommands {
         'WUNDR_AI_CONFIG_GET_FAILED',
         'Failed to get AI configuration',
         { key },
-        true
+        true,
       );
     }
   }
@@ -614,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;
@@ -671,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;
         }
@@ -688,7 +714,7 @@ export class AICommands {
         'WUNDR_AI_SETUP_FAILED',
         'Failed to setup AI configuration',
         { options },
-        true
+        true,
       );
     }
   }
@@ -705,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) {
@@ -730,7 +756,7 @@ export class AICommands {
         'WUNDR_AI_STATUS_FAILED',
         'Failed to get AI status',
         {},
-        true
+        true,
       );
     }
   }
@@ -757,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'));
         }
@@ -772,7 +798,7 @@ export class AICommands {
         'WUNDR_AI_VALIDATE_FAILED',
         'Failed to validate AI connection',
         {},
-        true
+        true,
       );
     }
   }
@@ -829,11 +855,19 @@ export class AICommands {
   }
 
   private async saveGeneratedCode(
-    _code: string,
-    outputPath: string
+    code: string,
+    outputPath: string,
   ): Promise<void> {
-    // Save generated code to file
-    logger.debug(`Saving generated code to ${outputPath}`);
+    const fs = await import('fs/promises');
+    const path = await import('path');
+
+    // Ensure directory exists
+    const dir = path.dirname(outputPath);
+    await fs.mkdir(dir, { recursive: true });
+
+    // Write the generated code to file
+    await fs.writeFile(outputPath, code, 'utf-8');
+    logger.debug(`Saved generated code to ${outputPath}`);
   }
 
   private async readFile(filePath: string): Promise<string> {
@@ -846,47 +880,84 @@ export class AICommands {
     return ['src/example.ts'];
   }
 
-  private displayReviewResults(results: any[]): void {
+  private displayReviewResults(results: FileReviewResult[]): void {
     console.log(chalk.blue('\nCode Review Results:'));
     results.forEach(result => {
       console.log(`\n${chalk.cyan(result.file)} (Score: ${result.score}/100)`);
       if (result.issues.length > 0) {
-        result.issues.forEach((issue: any) => {
+        result.issues.forEach((issue) => {
           console.log(`  ${issue.severity}: ${issue.description}`);
         });
       }
     });
   }
 
-  private async suggestFixes(_results: any[]): Promise<void> {
-    // Implementation for suggesting fixes
+  private async suggestFixes(results: FileReviewResult[]): Promise<void> {
     logger.info('Generating fix suggestions...');
+
+    for (const result of results) {
+      if (result.issues.length > 0) {
+        console.log(chalk.yellow(`\nSuggested fixes for ${result.file}:`));
+        for (const issue of result.issues) {
+          const fix = await this.callAI('suggest-fix', {
+            file: result.file,
+            issue: issue.description,
+            severity: issue.severity,
+          });
+          console.log(`  - ${issue.description}: ${fix.suggestion || 'No automatic fix available'}`);
+        }
+      }
+    }
   }
 
-  private async applyRefactoring(target: string, _plan: any): Promise<void> {
-    // Apply refactoring changes
+  private async applyRefactoring(target: string, plan: RefactoringPlan): Promise<void> {
+    const fs = await import('fs/promises');
+
     logger.debug(`Applying refactoring to ${target}`);
+    logger.info(`Refactoring: ${plan.description}`);
+
+    // Apply each change from the refactoring plan
+    for (const change of plan.changes) {
+      logger.debug(`  Applying: ${change.description}`);
+      if (change.code) {
+        await fs.writeFile(target, change.code, 'utf-8');
+      }
+    }
   }
 
   private getDocsOutputPath(
     target: string,
     type: string,
-    format: string
+    format: string,
   ): string {
     const ext = format === 'markdown' ? 'md' : format;
     return `docs/${target}.${type}.${ext}`;
   }
 
   private async saveGeneratedDocs(
-    _docs: string,
-    outputPath: string
+    docs: string,
+    outputPath: string,
   ): Promise<void> {
-    // Save generated documentation
-    logger.debug(`Saving documentation to ${outputPath}`);
+    const fs = await import('fs/promises');
+    const path = await import('path');
+
+    // Ensure docs directory exists
+    const dir = path.dirname(outputPath);
+    await fs.mkdir(dir, { recursive: true });
+
+    // Write the generated documentation to file
+    await fs.writeFile(outputPath, docs, 'utf-8');
+    logger.debug(`Saved documentation to ${outputPath}`);
   }
 
-  private getTestOutputPath(target: string, _framework: string): string {
-    return `${target}.test.js`;
+  private getTestOutputPath(target: string, framework: string): string {
+    const path = require('path');
+    const baseName = path.basename(target, path.extname(target));
+    const dirName = path.dirname(target);
+
+    // Use framework-specific test file extensions
+    const extension = framework === 'vitest' ? '.test.ts' : '.test.js';
+    return path.join(dirName, '__tests__', `${baseName}${extension}`);
   }
 
   private async loadChatSession(sessionId: string): Promise<ChatSession> {
@@ -948,8 +1019,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',
+          ),
         );
       }
     }
@@ -970,26 +1041,63 @@ export class AICommands {
     }
   }
 
-  private async runBenchmark(_target: string): Promise<any> {
-    // Run performance benchmark
-    return { time: 100, memory: 50 };
+  private async runBenchmark(target: string): Promise<BenchmarkResult> {
+    const { performance } = await import('perf_hooks');
+
+    logger.debug(`Running benchmark for ${target}`);
+
+    const startTime = performance.now();
+    const startMemory = process.memoryUsage().heapUsed / 1024 / 1024;
+
+    // Simulate loading and analyzing the file for benchmarking
+    const code = await this.readFile(target);
+    // Simple analysis to measure time
+    const lines = code.split('\n').length;
+    logger.debug(`Analyzed ${lines} lines`);
+
+    const endTime = performance.now();
+    const endMemory = process.memoryUsage().heapUsed / 1024 / 1024;
+
+    return {
+      time: Math.round(endTime - startTime),
+      memory: Math.round(endMemory - startMemory),
+    };
   }
 
   private async applyOptimization(
     target: string,
-    _optimization: any
+    optimization: OptimizationPlan,
   ): Promise<void> {
-    // Apply optimization changes
+    const fs = await import('fs/promises');
+
     logger.debug(`Applying optimization to ${target}`);
+    logger.info(`Optimization: ${optimization.description}`);
+
+    // Log each change being applied
+    for (const change of optimization.changes) {
+      logger.debug(`  Applying: ${change.description}`);
+    }
+
+    // Write the optimized code to the file
+    if (optimization.optimizedCode) {
+      await fs.writeFile(target, optimization.optimizedCode, 'utf-8');
+      logger.success(`Optimization applied to ${target}`);
+    }
   }
 
-  private displayBenchmarkComparison(before: any, after: any): 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;
+
+    const timeColor = timeDiff < 0 ? chalk.green : chalk.yellow;
+    const memColor = memDiff < 0 ? chalk.green : chalk.yellow;
+
     console.log(
-      `Time: ${before.time}ms → ${after.time}ms (${after.time - before.time}ms)`
+      `Time: ${before.time}ms → ${after.time}ms (${timeColor(timeDiff > 0 ? '+' : '')}${timeColor(timeDiff + 'ms')})`,
     );
     console.log(
-      `Memory: ${before.memory}MB → ${after.memory}MB (${after.memory - before.memory}MB)`
+      `Memory: ${before.memory}MB → ${after.memory}MB (${memColor(memDiff > 0 ? '+' : '')}${memColor(memDiff + 'MB')})`,
     );
   }
 }