ai-unit-test-generator 1.4.6 → 2.0.1

package/lib/ai/config-writer.mjs

@@ -0,0 +1,99 @@
+/**
+ * 安全的配置写入器
+ */
+
+import { readConfig, writeConfig } from '../utils/config-manager.mjs'
+
+const LOCKED_PATHS = [
+  'scoringMode',
+  'layers',
+  'weights',
+  'thresholds',
+  'ccMapping',
+  'dependencyCountMapping',
+  'coverageScoring',
+  'fallbacks',
+  'aiEnhancement.enabled'
+]
+
+const WRITABLE_PATHS = [
+  'aiEnhancement.analyzed',
+  'aiEnhancement.analyzedAt',
+  'aiEnhancement.suggestions'
+]
+
+/**
+ * 应用 AI 建议到配置文件
+ */
+export async function applyAISuggestions(configPath, suggestions) {
+  // 1. 读取现有配置
+  const config = readConfig(configPath)
+  
+  // 2. 验证权限
+  validateWritePermissions(suggestions)
+  
+  // 3. 深拷贝配置
+  const newConfig = JSON.parse(JSON.stringify(config))
+  
+  // 4. 初始化 aiEnhancement（如果不存在）
+  if (!newConfig.aiEnhancement) {
+    newConfig.aiEnhancement = {
+      enabled: true,
+      analyzed: false,
+      analyzedAt: null,
+      suggestions: {
+        businessCriticalPaths: [],
+        highRiskModules: [],
+        testabilityAdjustments: []
+      }
+    }
+  }
+  
+  // 5. 只写入允许的字段
+  newConfig.aiEnhancement.analyzed = true
+  newConfig.aiEnhancement.analyzedAt = new Date().toISOString()
+  newConfig.aiEnhancement.suggestions = suggestions
+  
+  // 6. 验证核心配置未被修改
+  validateCoreConfigIntact(config, newConfig)
+  
+  // 7. 写入文件
+  writeConfig(configPath, newConfig)
+  
+  return newConfig
+}
+
+/**
+ * 验证写入权限
+ */
+function validateWritePermissions(suggestions) {
+  const allowedKeys = ['businessCriticalPaths', 'highRiskModules', 'testabilityAdjustments']
+  
+  for (const key of Object.keys(suggestions)) {
+    if (!allowedKeys.includes(key)) {
+      throw new Error(`AI attempted to write forbidden field: ${key}`)
+    }
+  }
+}
+
+/**
+ * 验证核心配置完整性
+ */
+function validateCoreConfigIntact(oldConfig, newConfig) {
+  for (const path of LOCKED_PATHS) {
+    const oldValue = getNestedValue(oldConfig, path)
+    const newValue = getNestedValue(newConfig, path)
+    
+    if (JSON.stringify(oldValue) !== JSON.stringify(newValue)) {
+      throw new Error(`Core config was modified (locked path: ${path})`)
+    }
+  }
+}
+
+/**
+ * 获取嵌套对象的值
+ */
+function getNestedValue(obj, path) {
+  return path.split('.').reduce((curr, key) => curr?.[key], obj)
+}
+

package/lib/ai/context-builder.mjs

@@ -0,0 +1,52 @@
+/**
+ * 项目上下文构建工具
+ */
+
+import { existsSync, readFileSync } from 'node:fs'
+
+/**
+ * 构建项目上下文信息
+ */
+export async function buildProjectContext() {
+  const context = {
+    framework: 'Unknown',
+    criticalDeps: [],
+    devDeps: []
+  }
+  
+  // 读取 package.json
+  if (existsSync('package.json')) {
+    try {
+      const pkg = JSON.parse(readFileSync('package.json', 'utf-8'))
+      
+      context.name = pkg.name
+      context.framework = detectFramework(pkg.dependencies || {})
+      
+      // 识别关键依赖
+      const deps = Object.keys(pkg.dependencies || {})
+      const criticalKeywords = ['stripe', 'payment', 'auth', 'prisma', 'db', 'axios', 'fetch', 'jotai', 'zustand', 'redux']
+      
+      context.criticalDeps = deps.filter(dep => 
+        criticalKeywords.some(kw => dep.toLowerCase().includes(kw))
+      )
+      
+      context.devDeps = Object.keys(pkg.devDependencies || {})
+    } catch (err) {
+      console.warn('Warning: Could not read package.json')
+    }
+  }
+  
+  return context
+}
+
+/**
+ * 检测项目框架
+ */
+function detectFramework(deps) {
+  if (deps['next']) return 'Next.js'
+  if (deps['react']) return 'React'
+  if (deps['vue']) return 'Vue'
+  if (deps['@angular/core']) return 'Angular'
+  return 'Node.js'
+}
+

package/lib/ai/index.mjs

@@ -3,9 +3,20 @@
  * 
  * 提供 Prompt 构建、AI 调用和测试提取功能
  * 支持多种 LLM（目前实现：cursor-agent）
+ * 
+ * AI 分析功能：代码库分析、配置优化建议
  */
 
+// 测试生成相关
 export { buildBatchPrompt, runCLI as buildPrompt } from './prompt-builder.mjs'
 export { main as callAI } from './client.mjs'
 export { extractTests } from './extractor.mjs'
 
+// AI 分析相关
+export * from './sampler.mjs'
+export * from './context-builder.mjs'
+export * from './analyzer-prompt.mjs'
+export * from './validator.mjs'
+export * from './reviewer.mjs'
+export * from './config-writer.mjs'
+

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
+}
+