ai-unit-test-generator 1.4.5 → 2.0.1

package/lib/ai/reviewer.mjs

@@ -0,0 +1,283 @@
+/**
+ * 交互式 AI 建议审核器
+ * 支持分类审核、逐条选择、批量调整分数
+ */
+
+import readline from 'readline'
+
+/**
+ * 创建 readline 接口
+ */
+function createInterface() {
+  return readline.createInterface({
+    input: process.stdin,
+    output: process.stdout
+  })
+}
+
+/**
+ * 询问用户输入
+ */
+function ask(rl, question) {
+  return new Promise(resolve => rl.question(question, resolve))
+}
+
+/**
+ * 获取分类图标
+ */
+function getCategoryIcon(category) {
+  const icons = {
+    businessCriticalPaths: '🔴',
+    highRiskModules: '⚠️',
+    testabilityAdjustments: '✅'
+  }
+  return icons[category] || '📝'
+}
+
+/**
+ * 获取分类名称
+ */
+function getCategoryName(category) {
+  const names = {
+    businessCriticalPaths: 'Business Critical Paths',
+    highRiskModules: 'High Risk Modules',
+    testabilityAdjustments: 'Testability Adjustments'
+  }
+  return names[category] || category
+}
+
+/**
+ * 获取分类名称（小写）
+ */
+function getCategoryNameLower(category) {
+  const names = {
+    businessCriticalPaths: 'business critical paths',
+    highRiskModules: 'high risk modules',
+    testabilityAdjustments: 'testability adjustments'
+  }
+  return names[category] || category
+}
+
+/**
+ * 格式化单个建议
+ */
+function formatSuggestion(item, index, category) {
+  let output = `\n  [${index}] ${item.pattern}\n`
+  
+  if (category === 'businessCriticalPaths') {
+    output += `      BC: ${item.suggestedBC} | Confidence: ${(item.confidence * 100).toFixed(0)}%\n`
+  } else if (category === 'highRiskModules') {
+    output += `      ER: ${item.suggestedER} | Confidence: ${(item.confidence * 100).toFixed(0)}%\n`
+  } else if (category === 'testabilityAdjustments') {
+    output += `      Adjustment: ${item.adjustment} | Confidence: ${(item.confidence * 100).toFixed(0)}%\n`
+  }
+  
+  output += `      Reason: ${item.reason}\n`
+  output += `      Evidence:\n`
+  item.evidence.forEach(e => {
+    output += `      - ${e}\n`
+  })
+  
+  return output
+}
+
+/**
+ * 显示建议列表
+ */
+function displaySuggestions(category, items) {
+  const icon = getCategoryIcon(category)
+  const name = getCategoryName(category)
+  
+  console.log(`\n━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n`)
+  console.log(`${icon} ${name} (${items.length} suggestions):`)
+  
+  items.forEach((item, i) => {
+    console.log(formatSuggestion(item, i + 1, category))
+  })
+  
+  console.log(`\n━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n`)
+}
+
+/**
+ * 解析用户选择
+ */
+function parseSelection(input, maxCount) {
+  if (!input || input.trim() === '') {
+    return []
+  }
+  
+  return input.split(',')
+    .map(s => parseInt(s.trim()))
+    .filter(n => !isNaN(n) && n >= 1 && n <= maxCount)
+}
+
+/**
+ * 批量调整分数
+ */
+async function adjustScores(rl, items, category) {
+  console.log(`\nAdjust scores (press Enter to keep original):`)
+  
+  const scoreField = category === 'businessCriticalPaths' ? 'suggestedBC' : 'suggestedER'
+  const minScore = category === 'businessCriticalPaths' ? 8 : 7
+  const maxScore = 10
+  
+  const adjusted = []
+  
+  for (let i = 0; i < items.length; i++) {
+    const item = { ...items[i] }
+    const currentScore = item[scoreField]
+    
+    const input = await ask(rl, `  [${i + 1}] ${item.pattern} (${scoreField}: ${currentScore}): `)
+    
+    if (input && input.trim() !== '') {
+      const newScore = parseInt(input.trim())
+      if (!isNaN(newScore) && newScore >= minScore && newScore <= maxScore) {
+        item[scoreField] = newScore
+        console.log(`  ✅ Updated: ${currentScore} → ${newScore}`)
+      } else {
+        console.log(`  ⚠️  Invalid score (must be ${minScore}-${maxScore}), keeping original`)
+      }
+    }
+    
+    adjusted.push(item)
+  }
+  
+  return adjusted
+}
+
+/**
+ * 审核单个分类
+ */
+async function reviewCategory(rl, category, items) {
+  if (items.length === 0) {
+    return []
+  }
+  
+  // 显示建议
+  displaySuggestions(category, items)
+  
+  // 询问操作
+  const categoryName = getCategoryNameLower(category)
+  const action = await ask(rl,
+    `❓ Review ${categoryName}:\n` +
+    `   [a] Accept all (${items.length})\n` +
+    `   [r] Reject all\n` +
+    `   [s] Select individually\n` +
+    `   [n] Skip (keep empty)\n` +
+    `> `
+  )
+  
+  const actionLower = action.trim().toLowerCase()
+  
+  if (actionLower === 'a') {
+    // 接受全部
+    console.log(`✅ Accepted: ${items.length}/${items.length} ${categoryName}`)
+    
+    // 询问是否调整分数
+    if (category === 'businessCriticalPaths' || category === 'highRiskModules') {
+      const adjustInput = await ask(rl, `\nAdjust scores? (y/n) `)
+      if (adjustInput.trim().toLowerCase() === 'y') {
+        return await adjustScores(rl, items, category)
+      }
+    }
+    
+    return items
+  } else if (actionLower === 's') {
+    // 逐条选择
+    const selection = await ask(rl, `\nSelect which suggestions to accept (comma-separated, e.g. 1,3):\n> `)
+    const indices = parseSelection(selection, items.length)
+    
+    if (indices.length === 0) {
+      console.log(`❌ No suggestions selected`)
+      return []
+    }
+    
+    const selected = indices.map(i => items[i - 1])
+    console.log(`✅ Selected: ${selected.length}/${items.length} ${categoryName}`)
+    
+    // 询问是否调整分数
+    if (category === 'businessCriticalPaths' || category === 'highRiskModules') {
+      const adjustInput = await ask(rl, `\nAdjust scores? (y/n) `)
+      if (adjustInput.trim().toLowerCase() === 'y') {
+        return await adjustScores(rl, selected, category)
+      }
+    }
+    
+    return selected
+  } else if (actionLower === 'r') {
+    // 拒绝全部
+    console.log(`❌ Rejected: all ${categoryName}`)
+    return []
+  } else {
+    // 跳过
+    console.log(`⏭️  Skipped ${categoryName}`)
+    return []
+  }
+}
+
+/**
+ * 显示总结
+ */
+function displaySummary(result, validated) {
+  const totalSuggested = Object.values(validated).reduce((sum, arr) => sum + arr.length, 0)
+  const totalAccepted = Object.values(result).reduce((sum, arr) => sum + arr.length, 0)
+  
+  console.log(`\n━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n`)
+  console.log(`📊 Summary:`)
+  console.log(`   Business Critical Paths: ${result.businessCriticalPaths.length}/${validated.businessCriticalPaths?.length || 0} accepted`)
+  console.log(`   High Risk Modules: ${result.highRiskModules.length}/${validated.highRiskModules?.length || 0} accepted`)
+  console.log(`   Testability Adjustments: ${result.testabilityAdjustments.length}/${validated.testabilityAdjustments?.length || 0} accepted`)
+  console.log(`   Total: ${totalAccepted}/${totalSuggested} suggestions accepted`)
+  
+  if (totalAccepted === 0) {
+    console.log(`\n⚠️  No suggestions accepted. Config will not be modified.`)
+  }
+}
+
+/**
+ * 交互式审核
+ * @param {Object} validated - 已验证的建议
+ * @returns {Object|null} - 用户批准的建议，或 null（取消）
+ */
+export async function interactiveReview(validated) {
+  const rl = createInterface()
+  
+  const result = {
+    businessCriticalPaths: [],
+    highRiskModules: [],
+    testabilityAdjustments: []
+  }
+  
+  try {
+    // 审核每个分类
+    const categories = ['businessCriticalPaths', 'highRiskModules', 'testabilityAdjustments']
+    
+    for (const category of categories) {
+      const items = validated[category] || []
+      if (items.length === 0) continue
+      
+      result[category] = await reviewCategory(rl, category, items)
+    }
+    
+    // 显示总结
+    displaySummary(result, validated)
+    
+    const totalAccepted = Object.values(result).reduce((sum, arr) => sum + arr.length, 0)
+    
+    if (totalAccepted === 0) {
+      rl.close()
+      return null
+    }
+    
+    // 最终确认
+    const save = await ask(rl, `\n💾 Save these changes to ai-test.config.jsonc? (y/n)\n> `)
+    
+    rl.close()
+    
+    return save.trim().toLowerCase() === 'y' ? result : null
+  } catch (err) {
+    rl.close()
+    throw err
+  }
+}
+

package/lib/ai/sampler.mjs

@@ -0,0 +1,97 @@
+/**
+ * 代码文件采样工具
+ * 智能选择代表性代码文件供 AI 分析
+ */
+
+import { readFileSync, statSync } from 'node:fs'
+import fg from 'fast-glob'
+
+/**
+ * 智能采样代码文件
+ * @returns {Array} 采样结果
+ */
+export async function sampleCodeFiles() {
+  const samples = []
+  
+  // 策略 1: 按目录分层采样（确保覆盖各层）
+  const layers = {
+    services: await fg('src/services/**/*.{ts,tsx}'),
+    components: await fg('src/components/**/*.{ts,tsx}'),
+    utils: await fg('src/utils/**/*.{ts,tsx}'),
+    atoms: await fg('src/atoms/**/*.{ts,tsx}'),
+    stores: await fg('src/stores/**/*.{ts,tsx}'),
+    hooks: await fg('src/hooks/**/*.{ts,tsx}'),
+    context: await fg('src/context/**/*.{ts,tsx}')
+  }
+  
+  for (const [layer, files] of Object.entries(layers)) {
+    if (files.length === 0) continue
+    
+    // 每层选 2-3 个文件
+    const selected = files
+      .map(f => ({
+        path: f,
+        size: statSync(f).size,
+        name: f.split('/').pop()
+      }))
+      .sort((a, b) => b.size - a.size) // 按文件大小排序
+      .slice(0, 3)
+    
+    for (const file of selected) {
+      samples.push({
+        path: file.path,
+        layer,
+        reason: `${layer}_representative`,
+        preview: readFileSync(file.path, 'utf-8').slice(0, 1500) // 前1500字符
+      })
+    }
+  }
+  
+  // 策略 2: 关键词匹配（业务关键代码）
+  const keywords = ['payment', 'order', 'booking', 'checkout', 'price', 'cart', 'hotel', 'room']
+  const criticalFiles = await fg('src/**/*.{ts,tsx}')
+  
+  for (const keyword of keywords) {
+    const matched = criticalFiles.filter(f => f.toLowerCase().includes(keyword))
+    if (matched.length > 0) {
+      const file = matched[0] // 取第一个
+      if (!samples.find(s => s.path === file)) {
+        samples.push({
+          path: file,
+          layer: 'business',
+          reason: `critical_keyword_${keyword}`,
+          preview: readFileSync(file, 'utf-8').slice(0, 1500)
+        })
+      }
+    }
+  }
+  
+  // 去重
+  const uniqueSamples = samples.filter((s, i, arr) => 
+    arr.findIndex(x => x.path === s.path) === i
+  )
+  
+  return uniqueSamples.slice(0, 25) // 最多 25 个样本
+}
+
+/**
+ * 分析项目结构
+ */
+export async function analyzeProjectStructure() {
+  const allFiles = await fg('src/**/*.{ts,tsx,js,jsx}')
+  
+  let totalLines = 0
+  for (const file of allFiles) {
+    try {
+      const content = readFileSync(file, 'utf-8')
+      totalLines += content.split('\n').length
+    } catch {}
+  }
+  
+  return {
+    totalFiles: allFiles.length,
+    totalLines,
+    avgLinesPerFile: Math.round(totalLines / allFiles.length)
+  }
+}
+

package/lib/ai/validator.mjs

@@ -0,0 +1,101 @@
+/**
+ * AI 响应验证器
+ */
+
+const SCHEMA = {
+  businessCriticalPaths: {
+    minConfidence: 0.85,
+    maxCount: 10,
+    requiredFields: ['pattern', 'confidence', 'reason', 'suggestedBC', 'evidence'],
+    validators: {
+      pattern: (v) => /^[a-z0-9_/-]+\/?\*?\*?$/.test(v),
+      confidence: (v) => v >= 0.85 && v <= 1.0,
+      suggestedBC: (v) => [8, 9, 10].includes(v),
+      reason: (v) => v.length > 0 && v.length <= 200,
+      evidence: (v) => Array.isArray(v) && v.length >= 2 && v.length <= 3
+    }
+  },
+  highRiskModules: {
+    minConfidence: 0.75,
+    maxCount: 10,
+    requiredFields: ['pattern', 'confidence', 'reason', 'suggestedER', 'evidence'],
+    validators: {
+      pattern: (v) => /^[a-z0-9_/-]+\/?\*?\*?$/.test(v),
+      confidence: (v) => v >= 0.75 && v <= 1.0,
+      suggestedER: (v) => [7, 8, 9, 10].includes(v),
+      reason: (v) => v.length > 0 && v.length <= 200,
+      evidence: (v) => Array.isArray(v) && v.length >= 2 && v.length <= 3
+    }
+  },
+  testabilityAdjustments: {
+    minConfidence: 0.80,
+    maxCount: 10,
+    requiredFields: ['pattern', 'confidence', 'reason', 'adjustment', 'evidence'],
+    validators: {
+      pattern: (v) => /^[a-z0-9_/-]+\/?\*?\*?$/.test(v),
+      confidence: (v) => v >= 0.80 && v <= 1.0,
+      adjustment: (v) => ['-2', '-1', '+1', '+2'].includes(v),
+      reason: (v) => v.length > 0 && v.length <= 200,
+      evidence: (v) => Array.isArray(v) && v.length >= 2 && v.length <= 3
+    }
+  }
+}
+
+/**
+ * 验证单个建议
+ */
+function validateSuggestion(item, schema) {
+  // 检查必需字段
+  for (const field of schema.requiredFields) {
+    if (!(field in item)) {
+      return false
+    }
+  }
+  
+  // 检查置信度
+  if (item.confidence < schema.minConfidence) {
+    return false
+  }
+  
+  // 运行字段验证器
+  for (const [field, validator] of Object.entries(schema.validators)) {
+    if (field in item && !validator(item[field])) {
+      return false
+    }
+  }
+  
+  return true
+}
+
+/**
+ * 验证并清洗 AI 响应
+ */
+export function validateAndSanitize(parsed) {
+  const result = {
+    businessCriticalPaths: [],
+    highRiskModules: [],
+    testabilityAdjustments: []
+  }
+  
+  if (!parsed || typeof parsed !== 'object' || !parsed.suggestions) {
+    console.warn('⚠️  Invalid AI response structure')
+    return result
+  }
+  
+  const suggestions = parsed.suggestions
+  
+  for (const [key, items] of Object.entries(suggestions)) {
+    const schema = SCHEMA[key]
+    if (!schema || !Array.isArray(items)) continue
+    
+    const validated = items
+      .filter(item => validateSuggestion(item, schema))
+      .sort((a, b) => b.confidence - a.confidence)
+      .slice(0, schema.maxCount)
+    
+    result[key] = validated
+  }
+  
+  return result
+}
+

package/lib/core/scanner.mjs

@@ -140,6 +140,111 @@ async function extractTargets(files) {
     return false
   }
   
+  // ✅ 文件级缓存：避免重复扫描每个文件的导入
+  const fileImportsCache = new Map()
+  
+  function getCachedFileImports(sf) {
+    const filePath = sf.getFilePath()
+    if (fileImportsCache.has(filePath)) {
+      return fileImportsCache.get(filePath)
+    }
+    
+    const imports = sf.getImportDeclarations()
+    const criticalKeywords = ['stripe', 'payment', 'auth', 'axios', 'fetch', 'prisma', 'db', 'api', 'jotai', 'zustand']
+    const criticalImports = imports
+      .map(imp => imp.getModuleSpecifierValue())
+      .filter(mod => criticalKeywords.some(kw => mod.toLowerCase().includes(kw)))
+      .slice(0, 5) // 限制数量
+    
+    fileImportsCache.set(filePath, criticalImports)
+    return criticalImports
+  }
+  
+  // ✅ 新增：提取函数的 AI 分析元数据
+  function extractMetadata(node, sf) {
+    const metadata = {
+      criticalImports: [],
+      businessEntities: [],
+      hasDocumentation: false,
+      documentation: '',
+      errorHandling: 0,
+      externalCalls: 0,
+      paramCount: 0,
+      returnType: ''
+    }
+    
+    if (!node) return metadata
+    
+    try {
+      // 1. 提取关键导入（使用缓存）
+      metadata.criticalImports = getCachedFileImports(sf)
+      
+      // 2. 提取参数类型（识别业务实体）
+      if (node.getParameters) {
+        const params = node.getParameters()
+        metadata.paramCount = params.length
+        
+        // ✅ 从配置读取业务实体关键词（支持项目定制）
+        const entityKeywords = cfg?.aiEnhancement?.entityKeywords || 
+          ['Payment', 'Order', 'Booking', 'User', 'Hotel', 'Room', 'Cart', 'Price', 'Guest', 'Request', 'Response']
+        
+        metadata.businessEntities = params
+          .map(p => {
+            try {
+              return p.getType().getText()
+            } catch {
+              return ''
+            }
+          })
+          .filter(type => entityKeywords.some(kw => type.includes(kw)))
+          .slice(0, 3) // 限制数量
+      }
+      
+      // 3. 提取返回类型
+      if (node.getReturnType) {
+        try {
+          const returnType = node.getReturnType().getText()
+          // 简化类型（避免过长）
+          metadata.returnType = returnType.length > 100 ? returnType.slice(0, 100) + '...' : returnType
+        } catch {}
+      }
+      
+      // 4. 提取 JSDoc 注释
+      if (node.getJsDocs) {
+        const jsDocs = node.getJsDocs()
+        if (jsDocs.length > 0) {
+          metadata.hasDocumentation = true
+          metadata.documentation = jsDocs
+            .map(doc => {
+              const comment = doc.getComment()
+              return typeof comment === 'string' ? comment : ''
+            })
+            .filter(Boolean)
+            .join(' ')
+            .slice(0, 200) // 限制长度
+        }
+      }
+      
+      // 5. 统计异常处理（try-catch）
+      if (node.getDescendantsOfKind) {
+        metadata.errorHandling = node.getDescendantsOfKind(SyntaxKind.TryStatement).length
+      }
+      
+      // 6. 统计外部 API 调用
+      if (node.getDescendantsOfKind) {
+        const callExpressions = node.getDescendantsOfKind(SyntaxKind.CallExpression)
+        metadata.externalCalls = callExpressions.filter(call => {
+          const expr = call.getExpression().getText()
+          return /fetch|axios|\.get\(|\.post\(|\.put\(|\.delete\(/.test(expr)
+        }).length
+      }
+    } catch (err) {
+      // 提取失败不影响主流程
+    }
+    
+    return metadata
+  }
+  
   for (const sf of project.getSourceFiles()) {
     const absPath = sf.getFilePath()
     const relPath = relativize(absPath)
@@ -154,6 +259,7 @@ async function extractTargets(files) {
       if (fn) {
         const type = decideTypeByPathAndName(relPath, name)
         const layer = decideLayer(relPath, cfg)
+        const metadata = extractMetadata(fn, sf) // ✅ 提取元数据
         targets.push({
           name,
           path: relPath,
@@ -162,7 +268,8 @@ async function extractTargets(files) {
           internal: false,
           loc: fileLoc,
           impactHint: buildImpactHint(relPath),
-          roiHint: buildRoiHint(relPath, content)
+          roiHint: buildRoiHint(relPath, content),
+          metadata // ✅ 添加元数据
         })
         continue
       }
@@ -172,6 +279,8 @@ async function extractTargets(files) {
           if (v && isTestableVariable(v)) {
             const type = decideTypeByPathAndName(relPath, name)
             const layer = decideLayer(relPath, cfg)
+            const init = v.getInitializer()
+            const metadata = extractMetadata(init, sf) // ✅ 提取元数据
             targets.push({
               name,
               path: relPath,
@@ -180,7 +289,8 @@ async function extractTargets(files) {
               internal: false,
               loc: fileLoc,
               impactHint: buildImpactHint(relPath),
-              roiHint: buildRoiHint(relPath, content)
+              roiHint: buildRoiHint(relPath, content),
+              metadata // ✅ 添加元数据
             })
           }
       // 其他（interface/type/const/enum）直接跳过