ai-unit-test-generator 1.4.6 → 2.0.1

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）直接跳过

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