ai-unit-test-generator 1.4.5 → 2.0.1

package/lib/core/scorer.mjs

@@ -25,11 +25,136 @@ function loadJson(p) {
 function toFixedDown(num, digits = 2) { const m = Math.pow(10, digits); return Math.floor(num * m) / m }
 function clamp(n, min, max) { return Math.max(min, Math.min(max, n)) }
 
+// 匹配 AI 建议的模式（支持 glob 语法）
+function matchPattern(filePath, pattern) {
+  if (!pattern || !filePath) return false
+  
+  // 移除开头的 src/ 如果存在（标准化路径）
+  const normalizedPath = filePath.replace(/^src\//, '')
+  
+  // 简单 glob 匹配：支持 ** 和 *
+  const regexPattern = pattern
+    .replace(/\./g, '\\.')         // . 转义
+    .replace(/\*\*/g, '__DSTAR__') // ** 临时占位
+    .replace(/\*/g, '[^/]*')       // * 匹配非斜杠字符
+    .replace(/__DSTAR__/g, '.*')   // ** 匹配任意字符
+  
+  const regex = new RegExp(`^${regexPattern}$`)
+  return regex.test(normalizedPath) || regex.test(filePath)
+}
+
 function stripJsonComments(s) {
   return String(s)
     .replace(/\/\*[\s\S]*?\*\//g, '')
     .replace(/(^|\s)\/\/.*$/gm, '')
 }
+
+// 根据路径匹配层级（作为 fallback）
+function matchLayerByPath(filePath, cfg) {
+  const layers = cfg?.layers || {}
+  
+  // 按配置文件中定义的顺序遍历层级（foundation → business → state → ui）
+  const layerOrder = ['foundation', 'business', 'state', 'ui']
+  
+  for (const layerKey of layerOrder) {
+    const layerDef = layers[layerKey]
+    if (!layerDef) continue
+    
+    const patterns = layerDef.patterns || []
+    for (const pattern of patterns) {
+      // 简单的 glob 匹配：支持 ** 和 *
+      const regexPattern = pattern
+        .replace(/\./g, '\\.')         // . 转义
+        .replace(/\*\*/g, '__DSTAR__') // ** 临时占位
+        .replace(/\*/g, '[^/]*')       // * 匹配单层目录
+        .replace(/__DSTAR__/g, '.*')   // ** 匹配任意层级目录
+      const regex = new RegExp(`^${regexPattern}$`)
+      if (regex.test(filePath)) {
+        return layerKey
+      }
+    }
+  }
+  
+  // 如果都不匹配，尝试配置中的其他层级
+  for (const [layerKey, layerDef] of Object.entries(layers)) {
+    if (layerOrder.includes(layerKey)) continue // 已经检查过了
+    const patterns = layerDef.patterns || []
+    for (const pattern of patterns) {
+      const regexPattern = pattern
+        .replace(/\./g, '\\.')
+        .replace(/\*\*/g, '__DSTAR__')
+        .replace(/\*/g, '[^/]*')
+        .replace(/__DSTAR__/g, '.*')
+      const regex = new RegExp(`^${regexPattern}$`)
+      if (regex.test(filePath)) {
+        return layerKey
+      }
+    }
+  }
+  
+  return 'unknown'
+}
+
+// 综合判断层级：基于代码特征（roiHint）+ 路径约定
+function matchLayer(target, cfg) {
+  const { path, type, roiHint = {} } = target
+  
+  // ===== 第一优先级：明确的状态管理路径约定 =====
+  // atoms/stores 是明确的架构约定，优先识别
+  if (path.match(/\/(atoms|stores)\//)) {
+    return 'state'
+  }
+  
+  // ===== 第二优先级：基于代码特征（roiHint）=====
+  
+  // UI 层：needsUI = true（代码中包含 JSX 或 React Hooks）
+  if (roiHint.needsUI === true) {
+    // 排除 context（虽然有 JSX 但本质是状态管理）
+    if (path.match(/\/context\//)) {
+      return 'state'
+    }
+    return 'ui'
+  }
+  
+  // Foundation 层：纯函数 + 依赖可注入（真正的工具函数）
+  if (roiHint.isPure === true && roiHint.dependenciesInjectable === true) {
+    return 'foundation'
+  }
+  
+  // Foundation 层：纯函数（即使依赖不可注入，如某些 utils）
+  // 但排除状态管理相关的纯函数（如 createStore）
+  if (roiHint.isPure === true && !path.match(/\/(atoms|stores|context)\//)) {
+    return 'foundation'
+  }
+  
+  // Business 层：非纯函数 + 不需要 UI（业务逻辑、API 调用）
+  if (roiHint.isPure === false && roiHint.needsUI === false) {
+    // 排除状态管理
+    if (!path.match(/\/(atoms|stores|context)\//)) {
+      return 'business'
+    }
+  }
+  
+  // ===== 第三优先级：基于类型 =====
+  
+  // component 类型一定是 UI 层
+  if (type === 'component') {
+    return 'ui'
+  }
+  
+  // hook 类型：根据位置判断
+  if (type === 'hook') {
+    // 在 components/pages 下的 hook 是 UI 层
+    if (path.match(/\/(components|pages)\//)) {
+      return 'ui'
+    }
+    // 独立的 hook 更可能是业务逻辑
+    return 'business'
+  }
+  
+  // ===== 第四优先级：路径 fallback =====
+  return matchLayerByPath(path, cfg)
+}
 function loadConfig(pathFromArg) {
   const paths = [pathFromArg, 'ai-test.config.jsonc', 'ai-test.config.json']
   for (const p of paths) {
@@ -577,7 +702,62 @@ async function main() {
       coverageScore = mapCoverageScore(linesPct, cfg)
     }
 
-    let { score, priority, layer, layerName } = computeScore({ BC, CC: CCFinal, ER, ROI, testability, dependencyCount, coverageScore }, t, cfg)
+    // 根据代码特征和路径匹配层级（如果还没有匹配）
+    if (!t.layer || t.layer === 'unknown') {
+      t.layer = matchLayer(t, cfg)  // 传入整个 target，而不是路径
+    }
+
+    // ✅ AI 增强：应用 AI 分析建议（如果启用且已分析）
+    let enhancedBC = BC
+    let enhancedER = ER
+    let enhancedTestability = testability
+    
+    if (cfg?.aiEnhancement?.enabled && cfg?.aiEnhancement?.analyzed && cfg?.aiEnhancement?.suggestions) {
+      const suggestions = cfg.aiEnhancement.suggestions
+      
+      // 匹配 businessCriticalPaths（提升 BC）
+      if (suggestions.businessCriticalPaths && Array.isArray(suggestions.businessCriticalPaths)) {
+        for (const item of suggestions.businessCriticalPaths) {
+          if (matchPattern(path, item.pattern)) {
+            enhancedBC = Math.max(enhancedBC, item.suggestedBC)
+            break // 只应用第一个匹配
+          }
+        }
+      }
+      
+      // 匹配 highRiskModules（提升 ER）
+      if (suggestions.highRiskModules && Array.isArray(suggestions.highRiskModules)) {
+        for (const item of suggestions.highRiskModules) {
+          if (matchPattern(path, item.pattern)) {
+            enhancedER = Math.max(enhancedER, item.suggestedER)
+            break
+          }
+        }
+      }
+      
+      // 匹配 testabilityAdjustments（调整 testability）
+      if (suggestions.testabilityAdjustments && Array.isArray(suggestions.testabilityAdjustments)) {
+        for (const item of suggestions.testabilityAdjustments) {
+          if (matchPattern(path, item.pattern)) {
+            const adj = parseInt(item.adjustment)
+            if (!isNaN(adj)) {
+              enhancedTestability = clamp(enhancedTestability + adj, 0, 10)
+            }
+            break
+          }
+        }
+      }
+    }
+
+    let { score, priority, layer, layerName } = computeScore({ 
+      BC: enhancedBC, 
+      CC: CCFinal, 
+      ER: enhancedER, 
+      ROI, 
+      testability: enhancedTestability, 
+      dependencyCount, 
+      coverageScore 
+    }, t, cfg)
 
     // 覆盖率加权（可选）：对低覆盖文件小幅加分
     if (coverageSummary) {
@@ -605,11 +785,11 @@ async function main() {
       type, 
       layer: layer || 'N/A',
       layerName: layerName || 'N/A',
-      BC, 
+      BC: enhancedBC,  // 使用 AI 增强后的值
       CC: CCFinal, 
-      ER, 
+      ER: enhancedER,  // 使用 AI 增强后的值
       ROI, 
-      testability, 
+      testability: enhancedTestability,  // 使用 AI 增强后的值
       dependencyCount,
       coveragePct: coveragePct ?? 'N/A',
       coverageScore: coverageScore ?? 'N/A',

package/lib/utils/config-manager.mjs

@@ -0,0 +1,110 @@
+/**
+ * 配置文件管理工具
+ */
+
+import { existsSync, readFileSync, writeFileSync, copyFileSync } from 'node:fs'
+import { join } from 'node:path'
+import { fileURLToPath } from 'node:url'
+import { dirname } from 'node:path'
+
+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, '')     // 行注释
+}
+
+/**
+ * 自动探测配置文件
+ */
+export function detectConfig(providedPath) {
+  const detectOrder = [
+    providedPath,
+    'ai-test.config.jsonc',
+    'ai-test.config.json',
+    'ut_scoring_config.json' // 向后兼容
+  ].filter(Boolean)
+  
+  return detectOrder.find(p => existsSync(p)) || null
+}
+
+/**
+ * 确保配置文件存在（不存在则创建）
+ */
+export function ensureConfig(configPath = 'ai-test.config.jsonc') {
+  const detected = detectConfig(configPath)
+  
+  if (detected) {
+    return detected
+  }
+  
+  // 创建默认配置
+  console.log('⚙️  Config not found, creating default config...')
+  const templatePath = join(PKG_ROOT, 'templates', 'default.config.jsonc')
+  copyFileSync(templatePath, configPath)
+  console.log(`✅ Config created: ${configPath}\n`)
+  
+  return configPath
+}
+
+/**
+ * 读取配置文件（支持 JSONC）
+ */
+export function readConfig(configPath) {
+  if (!existsSync(configPath)) {
+    throw new Error(`Config file not found: ${configPath}`)
+  }
+  
+  const content = readFileSync(configPath, 'utf-8')
+  const cleaned = stripJsonComments(content)
+  
+  try {
+    return JSON.parse(cleaned)
+  } catch (err) {
+    throw new Error(`Failed to parse config file: ${configPath}\n${err.message}`)
+  }
+}
+
+/**
+ * 写入配置文件（保留 JSONC 格式）
+ */
+export function writeConfig(configPath, config) {
+  const content = JSON.stringify(config, null, 2)
+  writeFileSync(configPath, content, 'utf-8')
+}
+
+/**
+ * 检查配置是否已经 AI 分析过
+ */
+export function isAnalyzed(config) {
+  return config?.aiEnhancement?.analyzed === true
+}
+
+/**
+ * 验证配置结构
+ */
+export function validateConfig(config) {
+  const errors = []
+  
+  // 检查必需字段
+  if (!config.scoringMode) {
+    errors.push('Missing required field: scoringMode')
+  }
+  
+  if (config.scoringMode === 'layered' && !config.layers) {
+    errors.push('Layered mode requires "layers" field')
+  }
+  
+  if (config.scoringMode === 'legacy' && (!config.weights || !config.thresholds)) {
+    errors.push('Legacy mode requires "weights" and "thresholds" fields')
+  }
+  
+  return errors
+}
+

package/lib/utils/index.mjs

@@ -5,6 +5,8 @@
  */
 
 export { markDone } from './marker.mjs'
+export * from './config-manager.mjs'
+export * from './scan-manager.mjs'
 
 // 可以在未来添加更多工具函数
 // export { readJson, writeJson } from './file-io.mjs'

package/lib/utils/scan-manager.mjs

@@ -0,0 +1,121 @@
+/**
+ * 扫描结果管理工具
+ */
+
+import { existsSync, statSync, readdirSync } from 'node:fs'
+import { join } from 'node:path'
+
+/**
+ * 检查源代码是否有变化
+ */
+export function hasSourceCodeChanged(targetsPath, srcDir = 'src') {
+  if (!existsSync(targetsPath)) {
+    return true
+  }
+  
+  const targetsTime = statSync(targetsPath).mtimeMs
+  return checkDirectoryRecursive(srcDir, targetsTime)
+}
+
+/**
+ * 递归检查目录中的文件修改时间
+ */
+function checkDirectoryRecursive(dir, compareTime) {
+  if (!existsSync(dir)) {
+    return false
+  }
+  
+  try {
+    const entries = readdirSync(dir, { withFileTypes: true })
+    
+    for (const entry of entries) {
+      const fullPath = join(dir, entry.name)
+      
+      if (entry.isDirectory()) {
+        // 跳过不需要检查的目录
+        if (['node_modules', 'dist', '.git', 'reports', 'coverage'].includes(entry.name)) {
+          continue
+        }
+        
+        if (checkDirectoryRecursive(fullPath, compareTime)) {
+          return true
+        }
+      } else if (entry.isFile()) {
+        // 只检查代码文件
+        if (!/\.(ts|tsx|js|jsx)$/.test(entry.name)) {
+          continue
+        }
+        
+        const fileTime = statSync(fullPath).mtimeMs
+        if (fileTime > compareTime) {
+          console.log(`   Detected change: ${fullPath}`)
+          return true
+        }
+      }
+    }
+  } catch (err) {
+    // 忽略权限错误等
+    console.warn(`   Warning: Could not access ${dir}`)
+  }
+  
+  return false
+}
+
+/**
+ * 检查配置文件是否有变化
+ */
+export function hasConfigChanged(configPath, scoresPath) {
+  if (!existsSync(configPath) || !existsSync(scoresPath)) {
+    return false
+  }
+  
+  return statSync(configPath).mtimeMs > statSync(scoresPath).mtimeMs
+}
+
+/**
+ * 检查是否需要重新扫描
+ * @returns {boolean}
+ */
+export function needsRescan(options) {
+  const { force, targetsPath, scoresPath, configPath } = options
+  
+  // 1. 强制重扫
+  if (force) {
+    return true
+  }
+  
+  // 2. 目标文件不存在
+  if (!existsSync(targetsPath)) {
+    return true
+  }
+  
+  // 3. 打分结果不存在
+  if (scoresPath && !existsSync(scoresPath)) {
+    return true
+  }
+  
+  // 4. 源代码有变化
+  if (hasSourceCodeChanged(targetsPath)) {
+    return true
+  }
+  
+  // 5. 配置文件有变化
+  if (configPath && scoresPath && hasConfigChanged(configPath, scoresPath)) {
+    return true
+  }
+  
+  return false
+}
+
+/**
+ * 获取文件的年龄（小时）
+ */
+export function getFileAgeHours(filePath) {
+  if (!existsSync(filePath)) {
+    return Infinity
+  }
+  
+  const stats = statSync(filePath)
+  return (Date.now() - stats.mtimeMs) / (1000 * 60 * 60)
+}
+

package/lib/workflows/analyze.mjs

@@ -0,0 +1,157 @@
+/**
+ * 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 responsePath = 'reports/ai_analyze_response.txt'
+    
+    const child = spawn('cursor-agent', [
+      '--prompt-file', promptPath,
+      '--output', responsePath
+    ], {
+      stdio: 'inherit',
+      shell: true,
+      cwd: process.cwd()
+    })
+    
+    child.on('close', (code) => {
+      if (code !== 0) {
+        reject(new Error(`cursor-agent exited with code ${code}`))
+        return
+      }
+      
+      // 读取响应
+      try {
+        const { readFileSync } = require('node:fs')
+        const response = readFileSync(responsePath, 'utf-8')
+        resolve(response)
+      } catch (err) {
+        reject(new Error(`Failed to read AI response: ${err.message}`))
+      }
+    })
+    
+    child.on('error', (err) => {
+      reject(new Error(`Failed to spawn cursor-agent: ${err.message}`))
+    })
+  })
+}
+