ai-unit-test-generator 1.4.6 → 2.0.2

package/lib/workflows/analyze.mjs

@@ -0,0 +1,174 @@
+/**
+ * Analyze 工作流：AI 分析代码库并生成配置建议
+ */
+
+import { spawn } from 'node:child_process'
+import { writeFileSync } from 'node:fs'
+import { detectConfig } from '../utils/config-manager.mjs'
+import { sampleCodeFiles, analyzeProjectStructure } from '../ai/sampler.mjs'
+import { buildProjectContext } from '../ai/context-builder.mjs'
+import { buildAnalysisPrompt } from '../ai/analyzer-prompt.mjs'
+import { validateAndSanitize } from '../ai/validator.mjs'
+import { interactiveReview } from '../ai/reviewer.mjs'
+import { applyAISuggestions } from '../ai/config-writer.mjs'
+
+/**
+ * AI 分析工作流
+ */
+export async function analyze(options) {
+  const { config, output } = options
+  
+  // 1. 检查配置是否存在
+  console.log('🔍 Step 1: Checking configuration...')
+  const configPath = detectConfig(config)
+  
+  if (!configPath) {
+    console.error('❌ Config not found. Run `ai-test init` first.')
+    process.exit(1)
+  }
+  
+  console.log(`   Using config: ${configPath}\n`)
+  
+  // 2. 分析项目结构
+  console.log('📊 Step 2: Analyzing project structure...')
+  const stats = await analyzeProjectStructure()
+  console.log(`   Total files: ${stats.totalFiles}`)
+  console.log(`   Total lines: ${stats.totalLines}`)
+  console.log(`   Avg lines/file: ${stats.avgLinesPerFile}\n`)
+  
+  // 3. 智能采样代码
+  console.log('🎯 Step 3: Sampling representative code...')
+  const samples = await sampleCodeFiles()
+  console.log(`   Selected ${samples.length} files across layers\n`)
+  
+  // 4. 构建项目上下文
+  console.log('📦 Step 4: Reading project context...')
+  const projectCtx = await buildProjectContext()
+  console.log(`   Framework: ${projectCtx.framework}`)
+  console.log(`   Critical deps: ${projectCtx.criticalDeps.length > 0 ? projectCtx.criticalDeps.join(', ') : 'None detected'}\n`)
+  
+  // 5. 构建 AI Prompt
+  console.log('✍️  Step 5: Building AI analysis prompt...')
+  const prompt = buildAnalysisPrompt(samples, stats, projectCtx)
+  
+  // 保存 prompt 到临时文件
+  const promptPath = 'prompt_analyze.txt'
+  writeFileSync(promptPath, prompt, 'utf-8')
+  console.log(`   Prompt saved to: ${promptPath}\n`)
+  
+  // 6. 调用 Cursor Agent
+  console.log('🤖 Step 6: Calling Cursor Agent...')
+  console.log('   Cursor will analyze the full codebase using its index...')
+  console.log('   This may take 1-2 minutes...\n')
+  
+  const responseText = await callCursorAgent(promptPath)
+  
+  if (!responseText) {
+    console.error('❌ AI analysis failed or returned empty response')
+    process.exit(1)
+  }
+  
+  // 7. 解析并验证响应
+  console.log('✅ Step 7: Validating AI suggestions...')
+  
+  let parsed
+  try {
+    // 尝试提取 JSON（AI 可能返回 markdown 包装）
+    const jsonMatch = responseText.match(/```json\s*([\s\S]*?)\s*```/) || 
+                      responseText.match(/```\s*([\s\S]*?)\s*```/)
+    
+    const jsonText = jsonMatch ? jsonMatch[1] : responseText
+    parsed = JSON.parse(jsonText)
+  } catch (err) {
+    console.error('❌ Failed to parse AI response as JSON')
+    console.error('   Response preview:', responseText.slice(0, 500))
+    process.exit(1)
+  }
+  
+  const validated = validateAndSanitize(parsed)
+  
+  const totalSuggestions = Object.values(validated).reduce((sum, arr) => sum + arr.length, 0)
+  console.log(`   Validated ${totalSuggestions} suggestions\n`)
+  
+  if (totalSuggestions === 0) {
+    console.log('⚠️  No valid suggestions from AI. Please check the response.')
+    process.exit(0)
+  }
+  
+  // 8. 交互式审核
+  console.log('📝 Step 8: Interactive review...\n')
+  const approved = await interactiveReview(validated)
+  
+  if (approved === null) {
+    console.log('\n❌ No changes saved.')
+    process.exit(0)
+  }
+  
+  // 9. 写入配置
+  console.log('\n💾 Step 9: Updating configuration...')
+  
+  try {
+    await applyAISuggestions(configPath, approved)
+    console.log('✅ Config updated!')
+    console.log('\n💡 Next: Run `ai-test scan` to recalculate scores with AI enhancements.')
+  } catch (err) {
+    console.error(`❌ Failed to update config: ${err.message}`)
+    process.exit(1)
+  }
+}
+
+/**
+ * 调用 Cursor Agent
+ */
+async function callCursorAgent(promptPath) {
+  return new Promise((resolve, reject) => {
+    const { readFileSync } = require('node:fs')
+    
+    // 读取 prompt
+    let prompt
+    try {
+      prompt = readFileSync(promptPath, 'utf-8')
+    } catch (err) {
+      reject(new Error(`Failed to read prompt file: ${err.message}`))
+      return
+    }
+    
+    // 调用 cursor-agent chat（通过 stdin 传递 prompt）
+    const child = spawn('cursor-agent', ['chat'], {
+      stdio: ['pipe', 'pipe', 'inherit'],
+      shell: true,
+      cwd: process.cwd()
+    })
+    
+    const chunks = []
+    child.stdout.on('data', (d) => chunks.push(Buffer.from(d)))
+    
+    // 写入 prompt 到 stdin
+    child.stdin.write(prompt)
+    child.stdin.end()
+    
+    // 超时处理（10 分钟）
+    const timeout = setTimeout(() => {
+      child.kill('SIGKILL')
+      reject(new Error('cursor-agent timeout after 600s'))
+    }, 600000)
+    
+    child.on('close', (code) => {
+      clearTimeout(timeout)
+      
+      if (code !== 0) {
+        reject(new Error(`cursor-agent exited with code ${code}`))
+        return
+      }
+      
+      const response = Buffer.concat(chunks).toString('utf-8')
+      resolve(response)
+    })
+    
+    child.on('error', (err) => {
+      clearTimeout(timeout)
+      reject(new Error(`Failed to spawn cursor-agent: ${err.message}`))
+    })
+  })
+}
+

package/lib/workflows/generate.mjs

@@ -0,0 +1,98 @@
+/**
+ * Generate 工作流：生成单元测试
+ */
+
+import { spawn } from 'node:child_process'
+import { existsSync, readFileSync } from 'node:fs'
+import { join, dirname } from 'node:path'
+import { fileURLToPath } from 'node:url'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const PKG_ROOT = join(__dirname, '../..')
+
+/**
+ * 生成单批测试
+ */
+async function generateBatch(priority, count, skip, report) {
+  const batchScript = join(PKG_ROOT, 'lib/workflows/batch.mjs')
+  
+  return new Promise((resolve, reject) => {
+    const child = spawn('node', [batchScript, priority, String(count), String(skip), report], {
+      stdio: 'inherit',
+      cwd: process.cwd()
+    })
+    
+    child.on('close', (code) => {
+      if (code === 0) {
+        resolve({ generated: count, passed: count })
+      } else {
+        reject(new Error(`Batch generation failed with code ${code}`))
+      }
+    })
+    
+    child.on('error', reject)
+  })
+}
+
+/**
+ * Generate 工作流
+ */
+export async function generate(options) {
+  const { count, priority, all, report } = options
+  
+  // 1. 检查报告是否存在
+  if (!existsSync(report)) {
+    console.error(`❌ Report not found: ${report}`)
+    console.log(`   Run: ai-test scan`)
+    process.exit(1)
+  }
+  
+  if (all) {
+    // 2. 持续生成直到所有 TODO 完成
+    console.log(`🚀 Generating all ${priority} TODO functions...\n`)
+    
+    let batchNum = 1
+    let totalGenerated = 0
+    let totalPassed = 0
+    
+    while (true) {
+      // 检查还有多少 TODO
+      const content = readFileSync(report, 'utf-8')
+      const lines = content.split('\n')
+      const todoLines = lines.filter(line => 
+        line.includes('| TODO |') && line.includes(`| ${priority} |`)
+      )
+      
+      if (todoLines.length === 0) {
+        console.log(`\n✅ All ${priority} functions completed!`)
+        console.log(`   Total generated: ${totalGenerated}`)
+        console.log(`   Total passed: ${totalPassed}`)
+        break
+      }
+      
+      console.log(`\n━━━━ Batch ${batchNum} (${todoLines.length} TODO remaining) ━━━━`)
+      
+      try {
+        const result = await generateBatch(priority, Math.min(count, todoLines.length), 0, report)
+        totalGenerated += result.generated
+        totalPassed += result.passed
+        batchNum++
+      } catch (err) {
+        console.error(`❌ Batch ${batchNum} failed:`, err.message)
+        break
+      }
+    }
+  } else {
+    // 3. 生成指定数量
+    console.log(`🚀 Generating ${count} ${priority} functions...\n`)
+    
+    try {
+      await generateBatch(priority, count, 0, report)
+    } catch (err) {
+      console.error('❌ Generation failed:', err.message)
+      process.exit(1)
+    }
+  }
+}
+

package/lib/workflows/index.mjs

@@ -5,6 +5,13 @@
  * 组合 Core、AI 和 Testing 模块实现端到端工作流
  */
 
+// 主要工作流
+export * from './init.mjs'
+export * from './analyze.mjs'
+export * from './scan.mjs'
+export * from './generate.mjs'
+
+// 批处理工作流（内部使用）
 export { main as runBatch } from './batch.mjs'
 export { main as runAll } from './all.mjs'
 

package/lib/workflows/init.mjs

@@ -0,0 +1,45 @@
+/**
+ * Init 工作流：初始化配置文件
+ */
+
+import { existsSync, copyFileSync } from 'node:fs'
+import { join, dirname } from 'node:path'
+import { fileURLToPath } from 'node:url'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const PKG_ROOT = join(__dirname, '../..')
+
+/**
+ * 初始化配置文件
+ */
+export async function init(options) {
+  const { config } = options
+  const configPath = config || 'ai-test.config.jsonc'
+  
+  // 检查配置文件是否已存在
+  if (existsSync(configPath)) {
+    console.log(`⚠️  Config already exists: ${configPath}`)
+    console.log('   Delete it first or use a different path with -c option')
+    process.exit(1)
+  }
+  
+  // 复制模板
+  console.log('⚙️  Creating default config...')
+  const templatePath = join(PKG_ROOT, 'templates', 'default.config.jsonc')
+  
+  try {
+    copyFileSync(templatePath, configPath)
+    console.log(`✅ Config created: ${configPath}`)
+    
+    // 显示下一步建议
+    console.log('\n💡 Next steps:')
+    console.log('   1. Review and customize the config (optional)')
+    console.log('   2. Run `ai-test analyze` to let AI analyze your codebase')
+    console.log('   3. Or run `ai-test scan` to start scoring immediately')
+  } catch (err) {
+    console.error(`❌ Failed to create config: ${err.message}`)
+    process.exit(1)
+  }
+}
+

package/lib/workflows/scan.mjs

@@ -0,0 +1,144 @@
+/**
+ * Scan 工作流：扫描代码 + 打分 + 生成报告
+ */
+
+import { spawn } from 'node:child_process'
+import { existsSync, mkdirSync, readFileSync } from 'node:fs'
+import { join, dirname } from 'node:path'
+import { fileURLToPath } from 'node:url'
+import { detectConfig, readConfig } from '../utils/config-manager.mjs'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const PKG_ROOT = join(__dirname, '../..')
+
+/**
+ * 移除 JSON 注释
+ */
+function stripJsonComments(str) {
+  return String(str)
+    .replace(/\/\*[\s\S]*?\*\//g, '')
+    .replace(/(^|\s)\/\/.*$/gm, '')
+}
+
+/**
+ * 运行脚本
+ */
+function runScript(scriptPath, args) {
+  return new Promise((resolve, reject) => {
+    const fullPath = join(PKG_ROOT, 'lib', scriptPath)
+    const child = spawn('node', [fullPath, ...args], {
+      stdio: 'inherit',
+      cwd: process.cwd()
+    })
+
+    child.on('close', (code) => {
+      if (code === 0) {
+        resolve()
+      } else {
+        reject(new Error(`Script ${scriptPath} exited with code ${code}`))
+      }
+    })
+
+    child.on('error', reject)
+  })
+}
+
+/**
+ * Scan 工作流
+ */
+export async function scan(options) {
+  let { config, output, skipGit } = options
+  
+  // 1. 检查配置
+  console.log('🔍 Step 1: Checking configuration...')
+  const configPath = detectConfig(config)
+  
+  if (!configPath) {
+    console.error('❌ Config not found. Run `ai-test init` first.')
+    process.exit(1)
+  }
+  
+  config = configPath
+  console.log(`   Using config: ${config}\n`)
+  
+  // 2. 创建输出目录
+  if (!existsSync(output)) {
+    mkdirSync(output, { recursive: true })
+  }
+  
+  // 3. 可选：运行覆盖率
+  try {
+    const cfgText = existsSync(config) ? readFileSync(config, 'utf-8') : '{}'
+    const cfg = JSON.parse(stripJsonComments(cfgText))
+    const covCfg = cfg?.coverage || { runBeforeScan: false }
+    
+    if (covCfg.runBeforeScan) {
+      console.log('🧪 Running coverage before scan...')
+      await new Promise((resolve) => {
+        const cmd = covCfg.command || 'npx jest --coverage --silent'
+        const child = spawn(cmd, { stdio: 'inherit', shell: true, cwd: process.cwd() })
+        child.on('close', () => resolve())
+        child.on('error', () => resolve())
+      })
+      console.log('✅ Coverage completed.\n')
+    }
+  } catch (err) {
+    console.warn('⚠️  Coverage step failed. Continuing scan.')
+  }
+  
+  console.log('🚀 Starting code scan...\n')
+  
+  try {
+    // 4. 扫描 AST + 复杂度
+    console.log('📋 Step 2: Scanning targets...')
+    await runScript('core/scanner.mjs', [
+      '--config', config,
+      '--out', join(output, 'targets.json')
+    ])
+    
+    // 5. Git 信号（可选）
+    if (!skipGit) {
+      console.log('\n📊 Step 3: Analyzing Git history...')
+      await runScript('core/git-analyzer.mjs', [
+        '--targets', join(output, 'targets.json'),
+        '--out', join(output, 'git_signals.json')
+      ])
+    }
+    
+    // 6. 打分
+    console.log('\n🎯 Step 4: Scoring targets...')
+    const scoreArgs = [
+      '--targets', join(output, 'targets.json'),
+      '--config', config,
+      '--out-md', join(output, 'ut_scores.md'),
+      '--out-csv', join(output, 'ut_scores.csv')
+    ]
+    
+    if (!skipGit && existsSync(join(output, 'git_signals.json'))) {
+      scoreArgs.push('--git', join(output, 'git_signals.json'))
+    }
+    
+    await runScript('core/scorer.mjs', scoreArgs)
+    
+    // 7. 统计结果
+    const reportPath = join(output, 'ut_scores.md')
+    if (existsSync(reportPath)) {
+      const content = readFileSync(reportPath, 'utf-8')
+      const todoCount = (content.match(/\| TODO \|/g) || []).length
+      const doneCount = (content.match(/\| DONE \|/g) || []).length
+      
+      console.log('\n✅ Scan completed!')
+      console.log(`\n📊 Status:`)
+      console.log(`   TODO: ${todoCount}`)
+      console.log(`   DONE: ${doneCount}`)
+      console.log(`   Total: ${todoCount + doneCount}`)
+      console.log(`\n📄 Report: ${reportPath}`)
+      console.log(`\n💡 Next: ai-test generate`)
+    }
+  } catch (err) {
+    console.error('❌ Scan failed:', err.message)
+    process.exit(1)
+  }
+}
+

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-unit-test-generator",
-  "version": "1.4.6",
+  "version": "2.0.2",
   "description": "AI-powered unit test generator with smart priority scoring",
   "keywords": [
     "unit-test",

package/templates/default.config.jsonc

@@ -310,5 +310,63 @@
     "CC": 6,           // 复杂度默认值
     "ERLikelihood": 3, // 错误可能性默认值
     "Testability": 6   // 可测试性默认值
+  },
+  
+  // ========== AI 增强配置（AI 唯一可写区域）==========
+  
+  "aiEnhancement": {
+    "enabled": true,        // 是否启用 AI 增强
+    "analyzed": false,      // 是否已完成 AI 分析
+    "analyzedAt": null,     // 分析时间戳
+    
+    // ✅ 业务实体关键词（用于 AST 扫描识别业务对象）
+    "entityKeywords": [
+      "Payment", "Order", "Booking", "User", "Hotel", "Room", 
+      "Cart", "Price", "Guest", "Request", "Response"
+    ],
+    
+    // AI 建议（由 ai-test analyze 填充）
+    "suggestions": {
+      "businessCriticalPaths": [
+        // 示例：
+        // {
+        //   "pattern": "services/payment/**",
+        //   "confidence": 0.95,
+        //   "reason": "Handles Stripe payment processing",
+        //   "suggestedBC": 10,
+        //   "evidence": [
+        //     "Contains processPayment with Stripe API",
+        //     "Referenced by checkout flow",
+        //     "Handles money transactions"
+        //   ]
+        // }
+      ],
+      "highRiskModules": [
+        // 示例：
+        // {
+        //   "pattern": "utils/date/**",
+        //   "confidence": 0.88,
+        //   "reason": "Complex timezone calculations",
+        //   "suggestedER": 8,
+        //   "evidence": [
+        //     "Multiple timezone conversions",
+        //     "Handles DST edge cases"
+        //   ]
+        // }
+      ],
+      "testabilityAdjustments": [
+        // 示例：
+        // {
+        //   "pattern": "utils/**",
+        //   "confidence": 0.92,
+        //   "reason": "Pure functions, easy to test",
+        //   "adjustment": "+1",
+        //   "evidence": [
+        //     "No side effects",
+        //     "No external dependencies"
+        //   ]
+        // }
+      ]
+    }
   }
 }