skannr 0.1.2 → 0.1.4

package/src/agent-cli.ts

@@ -1,397 +1,397 @@
-#!/usr/bin/env node
-
-/**
- * Agent CLI - Interactive code analysis with Gemini AI
- * This demonstrates the agentic code analysis workflow
- */
-
-import { createCodeAnalysisAgent } from './agent';
-import { analyzeProject } from './index';
-import type { AnalysisResult } from './types';
-import * as readline from 'readline';
-import * as path from 'path';
-
-interface AgentCliArgs {
-  root: string;
-  question: string;
-  limit: number;
-  apiKey?: string;
-  model?: string;
-  interactive?: boolean;
-}
-
-interface InteractiveState {
-  result: AnalysisResult;
-  skeletonContext: string;
-}
-
-function buildSkeletonContext(result: AnalysisResult, includeScores: boolean): string {
-  return result.files
-    .map((file) => {
-      if (includeScores) {
-        return `=== ${file.path} (score: ${file.score.toFixed(2)}) ===\n${file.skeleton}`;
-      }
-      return `=== ${file.path} ===\n${file.skeleton}`;
-    })
-    .join('\n\n');
-}
-
-function printEvidence(result: AnalysisResult): void {
-  console.log('📚 Retrieval Evidence:');
-  for (const file of result.files.slice(0, 10)) {
-    console.log(`- ${file.path} (score: ${file.score.toFixed(2)})`);
-  }
-  console.log('');
-}
-
-function printInteractiveHelp(): void {
-  console.log('Commands:');
-  console.log('  /help                 Show this help');
-  console.log('  /files                List currently retrieved files');
-  console.log('  /symbols <query>      Search symbols in loaded mapping');
-  console.log('  /symbol <symbolId>    Show full implementation for a symbol');
-  console.log('  /deps <filePath>      Show imports and symbols for a file');
-  console.log('  /refresh              Re-analyze project and refresh context');
-  console.log('  /stats                Show mapping statistics');
-  console.log('  /exit                 Quit interactive mode');
-  console.log('');
-}
-
-/**
- * Parse command line arguments
- */
-function parseArgs(): AgentCliArgs | null {
-  const args = process.argv.slice(2);
-
-  let root = process.cwd();
-  let question = '';
-  let limit = 10;
-  let apiKey = process.env.GEMINI_API_KEY;
-  let model = 'gemini-2.0-flash-exp';
-  let interactive = false;
-
-  for (let i = 0; i < args.length; i++) {
-    const arg = args[i];
-
-    if (arg === '--root' && i + 1 < args.length) {
-      root = args[i + 1];
-      i++;
-    } else if (arg === '--question' && i + 1 < args.length) {
-      question = args[i + 1];
-      i++;
-    } else if (arg === '--limit' && i + 1 < args.length) {
-      limit = parseInt(args[i + 1], 10);
-      if (isNaN(limit) || limit < 1) {
-        console.error('  ✗ --limit must be a positive number');
-        return null;
-      }
-      i++;
-    } else if (arg === '--api-key' && i + 1 < args.length) {
-      apiKey = args[i + 1];
-      i++;
-    } else if (arg === '--model' && i + 1 < args.length) {
-      model = args[i + 1];
-      i++;
-    } else if (arg === '--interactive' || arg === '-i') {
-      interactive = true;
-    } else if (arg === '--help' || arg === '-h') {
-      printHelp();
-      return null;
-    } else {
-      console.error(`  ✗ Unknown argument: ${arg}`);
-      printHelp();
-      return null;
-    }
-  }
-
-  if (!apiKey) {
-    console.error('');
-    console.error('  ✗ No Gemini API key. Set GEMINI_API_KEY or pass --api-key <key>.');
-    console.error('');
-    return null;
-  }
-
-  if (!question && !interactive) {
-    console.error('');
-    console.error('  ✗ Provide --question "<text>" or use --interactive for chat mode.');
-    console.error('');
-    printHelp();
-    return null;
-  }
-
-  return { root, question, limit, apiKey, model, interactive };
-}
-
-/**
- * Print help message
- */
-function printHelp(): void {
-  console.log(`
-skannr-agent — AI-powered code analysis (Gemini)
-
-Usage:
-  skannr-agent --question "<text>" [options]
-  skannr-agent --interactive [options]
-
-Options:
-  --root <path>        Project root directory (default: current directory)
-  --question "<text>"  Question to analyze (required unless --interactive)
-  --limit <number>     Number of top files to analyze (default: 10)
-  --api-key <key>      Gemini API key (or set GEMINI_API_KEY env var)
-  --model <name>       Gemini model name (default: gemini-2.0-flash-exp)
-  --interactive, -i  Start interactive chat mode
-  --help, -h           Show this help message
-
-Examples:
-  skannr-agent --question "How does authentication work?"
-  skannr-agent --interactive --root ./my-project
-  skannr-agent --question "Find all API endpoints" --limit 15
-
-Environment variables:
-  GEMINI_API_KEY       Gemini API key (alternative to --api-key)
-
-Docs: https://skannr-ten.vercel.app
-  `);
-}
-
-/**
- * Run single question mode
- */
-async function runSingleQuestion(args: AgentCliArgs): Promise<void> {
-  console.log('🔍 Analyzing codebase...\n');
-
-  // Step 1: Generate skeletons with mapping
-  const result = await analyzeProject({
-    root: args.root,
-    question: args.question,
-    limit: args.limit,
-    generateMapping: true,
-  });
-
-  if (result.files.length === 0) {
-    console.error('');
-    console.error('  ✗ No files found to analyze.');
-    console.error('');
-    console.error('  Possible causes:');
-    console.error('    · --root is wrong or the tree has no matching source files');
-    console.error('    · Files are excluded by config or default excludes');
-    console.error('');
-    console.error('  Try: skannr-agent --question "..." --root .');
-    console.error('');
-    return;
-  }
-
-  console.log(`📊 Found ${result.files.length} relevant files\n`);
-
-  // Prepare skeleton context
-  const skeletonContext = buildSkeletonContext(result, true);
-
-  // Step 2: Create agent and load mapping
-  const agent = createCodeAnalysisAgent(args.apiKey!, args.model);
-  const mappingPath = path.join(result.root, 'code-analyzer.mapping.json');
-  agent.loadMapping(mappingPath);
-
-  const stats = agent.getStats();
-  console.log(`📝 Loaded mapping: ${stats.totalFiles} files, ${stats.totalSymbols} symbols\n`);
-  console.log(`🤖 Agent is analyzing...\n`);
-
-  // Step 3: Send to agent
-  const response = await agent.chat(args.question, skeletonContext);
-
-  console.log('📋 Agent Response:\n');
-  console.log(response);
-  console.log('\n');
-  printEvidence(result);
-}
-
-/**
- * Run interactive mode
- */
-async function runInteractive(args: AgentCliArgs): Promise<void> {
-  console.log('🚀 Starting interactive code analysis agent...\n');
-  console.log('First, let me analyze your codebase.\n');
-
-  const buildState = async (): Promise<InteractiveState> => {
-    const result = await analyzeProject({
-      root: args.root,
-      question: 'code structure and organization',
-      limit: args.limit,
-      generateMapping: true,
-    });
-
-    if (result.files.length === 0) {
-      return {
-        result,
-        skeletonContext: '',
-      };
-    }
-
-    return {
-      result,
-      skeletonContext: buildSkeletonContext(result, false),
-    };
-  };
-
-  // Step 1: Generate skeletons with mapping
-  let state = await buildState();
-
-  if (state.result.files.length === 0) {
-    console.log('No files found for analysis.');
-    return;
-  }
-
-  console.log(`📊 Analyzed ${state.result.files.length} files\n`);
-
-  // Step 2: Create agent and load mapping
-  const agent = createCodeAnalysisAgent(args.apiKey!, args.model);
-  const mappingPath = path.join(state.result.root, 'code-analyzer.mapping.json');
-  agent.loadMapping(mappingPath);
-
-  const stats = agent.getStats();
-  console.log(`📝 Mapping: ${stats.totalFiles} files, ${stats.totalSymbols} symbols\n`);
-  console.log('💬 Ready for questions! Type /help for commands.\n');
-
-  // Step 3: Interactive loop
-  const rl = readline.createInterface({
-    input: process.stdin,
-    output: process.stdout,
-  });
-
-  const askQuestion = () => {
-    rl.question('You: ', async (input) => {
-      const question = input.trim();
-
-      if (question.toLowerCase() === 'exit' || question.toLowerCase() === '/exit') {
-        console.log('\n👋 Goodbye!');
-        rl.close();
-        return;
-      }
-
-      if (question.toLowerCase() === 'stats' || question.toLowerCase() === '/stats') {
-        console.log('\n📊 Statistics:');
-        console.log(JSON.stringify(agent.getStats(), null, 2));
-        console.log('');
-        askQuestion();
-        return;
-      }
-
-      if (question.toLowerCase() === '/help') {
-        console.log('');
-        printInteractiveHelp();
-        askQuestion();
-        return;
-      }
-
-      if (question.toLowerCase() === '/files') {
-        console.log('');
-        printEvidence(state.result);
-        askQuestion();
-        return;
-      }
-
-      if (question.toLowerCase() === '/refresh') {
-        console.log('\n🔄 Refreshing project analysis...');
-        state = await buildState();
-        const refreshedMappingPath = path.join(state.result.root, 'code-analyzer.mapping.json');
-        agent.loadMapping(refreshedMappingPath);
-        console.log(`✅ Refreshed ${state.result.files.length} files\n`);
-        askQuestion();
-        return;
-      }
-
-      if (question.startsWith('/symbols ')) {
-        const query = question.slice('/symbols '.length).trim();
-        if (!query) {
-          console.log('\nUsage: /symbols <query>\n');
-          askQuestion();
-          return;
-        }
-        const symbols = agent.searchSymbols(query);
-        console.log('\n🔎 Symbol Search Result:');
-        console.log(JSON.stringify(symbols, null, 2));
-        console.log('');
-        askQuestion();
-        return;
-      }
-
-      if (question.startsWith('/symbol ')) {
-        const symbolId = question.slice('/symbol '.length).trim();
-        if (!symbolId) {
-          console.log('\nUsage: /symbol <symbolId>\n');
-          askQuestion();
-          return;
-        }
-        const symbol = agent.getSymbolDetails(symbolId);
-        console.log('\n🧩 Symbol Details:');
-        console.log(JSON.stringify(symbol, null, 2));
-        console.log('');
-        askQuestion();
-        return;
-      }
-
-      if (question.startsWith('/deps ')) {
-        const filePath = question.slice('/deps '.length).trim();
-        if (!filePath) {
-          console.log('\nUsage: /deps <filePath>\n');
-          askQuestion();
-          return;
-        }
-        const deps = agent.analyzeFileDependencies(filePath);
-        console.log('\n🕸️ File Dependencies:');
-        console.log(JSON.stringify(deps, null, 2));
-        console.log('');
-        askQuestion();
-        return;
-      }
-
-      if (!question) {
-        askQuestion();
-        return;
-      }
-
-      console.log('\n🤖 Agent: ');
-
-      try {
-        const response = await agent.chat(question, state.skeletonContext);
-        console.log(response);
-        console.log('');
-        printEvidence(state.result);
-      } catch (error) {
-        console.error('Error:', error instanceof Error ? error.message : error);
-      }
-
-      console.log('');
-      askQuestion();
-    });
-  };
-
-  askQuestion();
-}
-
-/**
- * Main CLI function
- */
-async function main(): Promise<void> {
-  const args = parseArgs();
-
-  if (!args) {
-    process.exit(1);
-  }
-
-  try {
-    if (args.interactive) {
-      await runInteractive(args);
-    } else {
-      await runSingleQuestion(args);
-    }
-  } catch (error) {
-    console.error('Error:', error instanceof Error ? error.message : error);
-    process.exit(1);
-  }
-}
-
-// Run CLI
-main().catch((error) => {
-  console.error('Unexpected error:', error);
-  process.exit(1);
-});
+#!/usr/bin/env node
+
+/**
+ * Agent CLI - Interactive code analysis with Gemini AI
+ * This demonstrates the agentic code analysis workflow
+ */
+
+import { createCodeAnalysisAgent } from './agent';
+import { analyzeProject } from './index';
+import type { AnalysisResult } from './types';
+import * as readline from 'readline';
+import * as path from 'path';
+
+interface AgentCliArgs {
+  root: string;
+  question: string;
+  limit: number;
+  apiKey?: string;
+  model?: string;
+  interactive?: boolean;
+}
+
+interface InteractiveState {
+  result: AnalysisResult;
+  skeletonContext: string;
+}
+
+function buildSkeletonContext(result: AnalysisResult, includeScores: boolean): string {
+  return result.files
+    .map((file) => {
+      if (includeScores) {
+        return `=== ${file.path} (score: ${file.score.toFixed(2)}) ===\n${file.skeleton}`;
+      }
+      return `=== ${file.path} ===\n${file.skeleton}`;
+    })
+    .join('\n\n');
+}
+
+function printEvidence(result: AnalysisResult): void {
+  console.log('📚 Retrieval Evidence:');
+  for (const file of result.files.slice(0, 10)) {
+    console.log(`- ${file.path} (score: ${file.score.toFixed(2)})`);
+  }
+  console.log('');
+}
+
+function printInteractiveHelp(): void {
+  console.log('Commands:');
+  console.log('  /help                 Show this help');
+  console.log('  /files                List currently retrieved files');
+  console.log('  /symbols <query>      Search symbols in loaded mapping');
+  console.log('  /symbol <symbolId>    Show full implementation for a symbol');
+  console.log('  /deps <filePath>      Show imports and symbols for a file');
+  console.log('  /refresh              Re-analyze project and refresh context');
+  console.log('  /stats                Show mapping statistics');
+  console.log('  /exit                 Quit interactive mode');
+  console.log('');
+}
+
+/**
+ * Parse command line arguments
+ */
+function parseArgs(): AgentCliArgs | null {
+  const args = process.argv.slice(2);
+
+  let root = process.cwd();
+  let question = '';
+  let limit = 10;
+  let apiKey = process.env.GEMINI_API_KEY;
+  let model = 'gemini-2.0-flash-exp';
+  let interactive = false;
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+
+    if (arg === '--root' && i + 1 < args.length) {
+      root = args[i + 1];
+      i++;
+    } else if (arg === '--question' && i + 1 < args.length) {
+      question = args[i + 1];
+      i++;
+    } else if (arg === '--limit' && i + 1 < args.length) {
+      limit = parseInt(args[i + 1], 10);
+      if (isNaN(limit) || limit < 1) {
+        console.error('  ✗ --limit must be a positive number');
+        return null;
+      }
+      i++;
+    } else if (arg === '--api-key' && i + 1 < args.length) {
+      apiKey = args[i + 1];
+      i++;
+    } else if (arg === '--model' && i + 1 < args.length) {
+      model = args[i + 1];
+      i++;
+    } else if (arg === '--interactive' || arg === '-i') {
+      interactive = true;
+    } else if (arg === '--help' || arg === '-h') {
+      printHelp();
+      return null;
+    } else {
+      console.error(`  ✗ Unknown argument: ${arg}`);
+      printHelp();
+      return null;
+    }
+  }
+
+  if (!apiKey) {
+    console.error('');
+    console.error('  ✗ No Gemini API key. Set GEMINI_API_KEY or pass --api-key <key>.');
+    console.error('');
+    return null;
+  }
+
+  if (!question && !interactive) {
+    console.error('');
+    console.error('  ✗ Provide --question "<text>" or use --interactive for chat mode.');
+    console.error('');
+    printHelp();
+    return null;
+  }
+
+  return { root, question, limit, apiKey, model, interactive };
+}
+
+/**
+ * Print help message
+ */
+function printHelp(): void {
+  console.log(`
+skannr-agent — AI-powered code analysis (Gemini)
+
+Usage:
+  skannr-agent --question "<text>" [options]
+  skannr-agent --interactive [options]
+
+Options:
+  --root <path>        Project root directory (default: current directory)
+  --question "<text>"  Question to analyze (required unless --interactive)
+  --limit <number>     Number of top files to analyze (default: 10)
+  --api-key <key>      Gemini API key (or set GEMINI_API_KEY env var)
+  --model <name>       Gemini model name (default: gemini-2.0-flash-exp)
+  --interactive, -i  Start interactive chat mode
+  --help, -h           Show this help message
+
+Examples:
+  skannr-agent --question "How does authentication work?"
+  skannr-agent --interactive --root ./my-project
+  skannr-agent --question "Find all API endpoints" --limit 15
+
+Environment variables:
+  GEMINI_API_KEY       Gemini API key (alternative to --api-key)
+
+Docs: https://skannr-ten.vercel.app
+  `);
+}
+
+/**
+ * Run single question mode
+ */
+async function runSingleQuestion(args: AgentCliArgs): Promise<void> {
+  console.log('🔍 Analyzing codebase...\n');
+
+  // Step 1: Generate skeletons with mapping
+  const result = await analyzeProject({
+    root: args.root,
+    question: args.question,
+    limit: args.limit,
+    generateMapping: true,
+  });
+
+  if (result.files.length === 0) {
+    console.error('');
+    console.error('  ✗ No files found to analyze.');
+    console.error('');
+    console.error('  Possible causes:');
+    console.error('    · --root is wrong or the tree has no matching source files');
+    console.error('    · Files are excluded by config or default excludes');
+    console.error('');
+    console.error('  Try: skannr-agent --question "..." --root .');
+    console.error('');
+    return;
+  }
+
+  console.log(`📊 Found ${result.files.length} relevant files\n`);
+
+  // Prepare skeleton context
+  const skeletonContext = buildSkeletonContext(result, true);
+
+  // Step 2: Create agent and load mapping
+  const agent = createCodeAnalysisAgent(args.apiKey!, args.model);
+  const mappingPath = path.join(result.root, 'code-analyzer.mapping.json');
+  agent.loadMapping(mappingPath);
+
+  const stats = agent.getStats();
+  console.log(`📝 Loaded mapping: ${stats.totalFiles} files, ${stats.totalSymbols} symbols\n`);
+  console.log(`🤖 Agent is analyzing...\n`);
+
+  // Step 3: Send to agent
+  const response = await agent.chat(args.question, skeletonContext);
+
+  console.log('📋 Agent Response:\n');
+  console.log(response);
+  console.log('\n');
+  printEvidence(result);
+}
+
+/**
+ * Run interactive mode
+ */
+async function runInteractive(args: AgentCliArgs): Promise<void> {
+  console.log('🚀 Starting interactive code analysis agent...\n');
+  console.log('First, let me analyze your codebase.\n');
+
+  const buildState = async (): Promise<InteractiveState> => {
+    const result = await analyzeProject({
+      root: args.root,
+      question: 'code structure and organization',
+      limit: args.limit,
+      generateMapping: true,
+    });
+
+    if (result.files.length === 0) {
+      return {
+        result,
+        skeletonContext: '',
+      };
+    }
+
+    return {
+      result,
+      skeletonContext: buildSkeletonContext(result, false),
+    };
+  };
+
+  // Step 1: Generate skeletons with mapping
+  let state = await buildState();
+
+  if (state.result.files.length === 0) {
+    console.log('No files found for analysis.');
+    return;
+  }
+
+  console.log(`📊 Analyzed ${state.result.files.length} files\n`);
+
+  // Step 2: Create agent and load mapping
+  const agent = createCodeAnalysisAgent(args.apiKey!, args.model);
+  const mappingPath = path.join(state.result.root, 'code-analyzer.mapping.json');
+  agent.loadMapping(mappingPath);
+
+  const stats = agent.getStats();
+  console.log(`📝 Mapping: ${stats.totalFiles} files, ${stats.totalSymbols} symbols\n`);
+  console.log('💬 Ready for questions! Type /help for commands.\n');
+
+  // Step 3: Interactive loop
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  const askQuestion = () => {
+    rl.question('You: ', async (input) => {
+      const question = input.trim();
+
+      if (question.toLowerCase() === 'exit' || question.toLowerCase() === '/exit') {
+        console.log('\n👋 Goodbye!');
+        rl.close();
+        return;
+      }
+
+      if (question.toLowerCase() === 'stats' || question.toLowerCase() === '/stats') {
+        console.log('\n📊 Statistics:');
+        console.log(JSON.stringify(agent.getStats(), null, 2));
+        console.log('');
+        askQuestion();
+        return;
+      }
+
+      if (question.toLowerCase() === '/help') {
+        console.log('');
+        printInteractiveHelp();
+        askQuestion();
+        return;
+      }
+
+      if (question.toLowerCase() === '/files') {
+        console.log('');
+        printEvidence(state.result);
+        askQuestion();
+        return;
+      }
+
+      if (question.toLowerCase() === '/refresh') {
+        console.log('\n🔄 Refreshing project analysis...');
+        state = await buildState();
+        const refreshedMappingPath = path.join(state.result.root, 'code-analyzer.mapping.json');
+        agent.loadMapping(refreshedMappingPath);
+        console.log(`✅ Refreshed ${state.result.files.length} files\n`);
+        askQuestion();
+        return;
+      }
+
+      if (question.startsWith('/symbols ')) {
+        const query = question.slice('/symbols '.length).trim();
+        if (!query) {
+          console.log('\nUsage: /symbols <query>\n');
+          askQuestion();
+          return;
+        }
+        const symbols = agent.searchSymbols(query);
+        console.log('\n🔎 Symbol Search Result:');
+        console.log(JSON.stringify(symbols, null, 2));
+        console.log('');
+        askQuestion();
+        return;
+      }
+
+      if (question.startsWith('/symbol ')) {
+        const symbolId = question.slice('/symbol '.length).trim();
+        if (!symbolId) {
+          console.log('\nUsage: /symbol <symbolId>\n');
+          askQuestion();
+          return;
+        }
+        const symbol = agent.getSymbolDetails(symbolId);
+        console.log('\n🧩 Symbol Details:');
+        console.log(JSON.stringify(symbol, null, 2));
+        console.log('');
+        askQuestion();
+        return;
+      }
+
+      if (question.startsWith('/deps ')) {
+        const filePath = question.slice('/deps '.length).trim();
+        if (!filePath) {
+          console.log('\nUsage: /deps <filePath>\n');
+          askQuestion();
+          return;
+        }
+        const deps = agent.analyzeFileDependencies(filePath);
+        console.log('\n🕸️ File Dependencies:');
+        console.log(JSON.stringify(deps, null, 2));
+        console.log('');
+        askQuestion();
+        return;
+      }
+
+      if (!question) {
+        askQuestion();
+        return;
+      }
+
+      console.log('\n🤖 Agent: ');
+
+      try {
+        const response = await agent.chat(question, state.skeletonContext);
+        console.log(response);
+        console.log('');
+        printEvidence(state.result);
+      } catch (error) {
+        console.error('Error:', error instanceof Error ? error.message : error);
+      }
+
+      console.log('');
+      askQuestion();
+    });
+  };
+
+  askQuestion();
+}
+
+/**
+ * Main CLI function
+ */
+async function main(): Promise<void> {
+  const args = parseArgs();
+
+  if (!args) {
+    process.exit(1);
+  }
+
+  try {
+    if (args.interactive) {
+      await runInteractive(args);
+    } else {
+      await runSingleQuestion(args);
+    }
+  } catch (error) {
+    console.error('Error:', error instanceof Error ? error.message : error);
+    process.exit(1);
+  }
+}
+
+// Run CLI
+main().catch((error) => {
+  console.error('Unexpected error:', error);
+  process.exit(1);
+});