ai-unit-test-generator 1.3.0

package/lib/core/git-analyzer.mjs

@@ -0,0 +1,151 @@
+#!/usr/bin/env node
+
+import { execSync } from 'node:child_process'
+import { existsSync, readFileSync, writeFileSync } from 'node:fs'
+import { join } from 'node:path'
+
+function parseArgs(argv) {
+  const args = {}
+  for (let i = 2; i < argv.length; i++) {
+    const a = argv[i]
+    if (a.startsWith('--')) {
+      const [k, v] = a.includes('=') ? a.split('=') : [a, argv[i + 1]]
+      args[k.replace(/^--/, '')] = v === undefined || v.startsWith('--') ? true : v
+      if (v !== undefined && !v.startsWith('--') && !a.includes('=')) i++
+    }
+  }
+  return args
+}
+
+function run(cmd) {
+  const out = execSync(cmd, { stdio: ['ignore', 'pipe', 'pipe'] })
+    .toString()
+    .trim()
+  return out
+}
+
+function loadJson(p) { if (!p || !existsSync(p)) return null; try { return JSON.parse(readFileSync(p, 'utf8')) } catch { return null } }
+
+function assertGitRepo() {
+  try {
+    const inside = run('git rev-parse --is-inside-work-tree')
+    if (inside !== 'true') throw new Error('not a git work tree')
+  } catch (e) {
+    throw new Error('Not inside a git repository. Ensure git is initialized and files are tracked.')
+  }
+}
+
+function gitFiles(pattern = 'src') {
+  const out = run(`git ls-files "${pattern}"`)
+  const files = out ? out.split('\n').filter(Boolean) : []
+  if (!files.length) throw new Error(`No tracked files under '${pattern}'. Ensure files are added to git.`)
+  return files
+}
+
+function detectMultiPlatform(filePath, content) {
+  if (filePath.includes('.h5.') || filePath.includes('.crn.')) return true
+  if (!content) return false
+  return content.includes('xEnv') || content.includes('xUa')
+}
+
+function topCategory(path) {
+  const parts = path.replace(/\\/g, '/').split('/')
+  const idx = parts.indexOf('src')
+  return idx >= 0 && idx + 1 < parts.length ? parts[idx + 1] : parts[0]
+}
+
+// P1-1: 重命名为 inCategory（文件是否属于核心类别白名单）
+function computeInCategory(path, categories = []) {
+  const cat = topCategory(path)
+  return categories.includes(cat)
+}
+
+// P2-2: 批量获取 Git 数据优化
+function collectCommitsBatch(files) {
+  const byFile = {}
+  files.forEach(f => { byFile[f] = { c30: 0, c90: 0, c180: 0, authors: new Set() } })
+  
+  // 一次性获取所有 commits（180天内）
+  const log180 = run('git log --since="180 days ago" --pretty=format:"%H|%an|%ar" --name-only').split('\n\n')
+  const now = Date.now()
+  const day30 = 30 * 24 * 60 * 60 * 1000
+  const day90 = 90 * 24 * 60 * 60 * 1000
+  
+  for (const block of log180) {
+    if (!block.trim()) continue
+    const lines = block.split('\n')
+    const [hash, author, relTime] = (lines[0] || '').split('|')
+    if (!hash) continue
+    
+    const daysAgo = parseRelativeTime(relTime)
+    const isIn30 = daysAgo <= 30
+    const isIn90 = daysAgo <= 90
+    
+    for (let i = 1; i < lines.length; i++) {
+      const file = lines[i].trim()
+      if (!file || !byFile[file]) continue
+      if (isIn30) {
+        byFile[file].c30++
+        byFile[file].authors.add(author)
+      }
+      if (isIn90) byFile[file].c90++
+      byFile[file].c180++
+    }
+  }
+  
+  return byFile
+}
+
+function parseRelativeTime(rel) {
+  if (!rel) return 999
+  const match = rel.match(/(\d+)\s+(day|week|month|year)/)
+  if (!match) return 999
+  const [, num, unit] = match
+  const n = Number(num)
+  if (unit.startsWith('day')) return n
+  if (unit.startsWith('week')) return n * 7
+  if (unit.startsWith('month')) return n * 30
+  if (unit.startsWith('year')) return n * 365
+  return 999
+}
+
+function collectWithExec(files, config) {
+  const results = {}
+  const crossCats = config?.crossModuleCategories || []
+  
+  // P2-2: 批量获取所有 commits
+  const batchData = collectCommitsBatch(files)
+  
+  for (const file of files) {
+    const data = batchData[file] || { c30: 0, c90: 0, c180: 0, authors: new Set() }
+    const commits30d = data.c30
+    const commits90d = data.c90
+    const commits180d = data.c180
+    const authors30d = data.authors.size
+    
+    let content = ''
+    try { content = readFileSync(join(process.cwd(), file), 'utf8') } catch {}
+    const multiPlatform = detectMultiPlatform(file, content)
+    
+    // P1-1: 改名为 inCategory
+    const inCategory = computeInCategory(file, crossCats)
+    results[file] = { commits30d, commits90d, commits180d, authors30d, inCategory, multiPlatform }
+  }
+  return results
+}
+
+function main() {
+  const args = parseArgs(process.argv)
+  const outPath = args.out
+  const config = loadJson('ut_scoring_config.json') || {}
+
+  assertGitRepo()
+  const files = gitFiles('src')
+  const results = collectWithExec(files, config)
+
+  const json = JSON.stringify(results, null, 2)
+  if (outPath) writeFileSync(outPath, json)
+  else process.stdout.write(json)
+}
+
+main()

package/lib/core/index.mjs

@@ -0,0 +1,11 @@
+/**
+ * Core 模块：代码分析与评分引擎
+ * 
+ * 提供核心的代码扫描、Git 历史分析和优先级评分功能
+ * 该模块不依赖 AI，可独立使用
+ */
+
+export { main as scanCode } from './scanner.mjs'
+export { main as analyzeGit } from './git-analyzer.mjs'
+export { main as scoreTargets } from './scorer.mjs'
+

package/lib/core/scanner.mjs

@@ -0,0 +1,233 @@
+#!/usr/bin/env node
+
+import { readFileSync, writeFileSync } from 'node:fs'
+
+async function req(mod, installHint) {
+  try { return await import(mod) } catch {
+    const hint = installHint || mod
+    throw new Error(`${mod} not installed. Run: npm i -D ${hint}`)
+  }
+}
+
+function parseArgs(argv) {
+  const args = {}
+  for (let i = 2; i < argv.length; i++) {
+    const a = argv[i]
+    if (a.startsWith('--')) {
+      const [k, v] = a.includes('=') ? a.split('=') : [a, argv[i + 1]]
+      args[k.replace(/^--/, '')] = v === undefined || v.startsWith('--') ? true : v
+      if (v !== undefined && !v.startsWith('--') && !a.includes('=')) i++
+    }
+  }
+  return args
+}
+
+function loadJson(p) { try { return JSON.parse(readFileSync(p, 'utf8')) } catch { return null } }
+
+async function listFiles(excludeDirs = []) {
+  const fg = (await req('fast-glob', 'fast-glob')).default
+  
+  // 基础排除规则
+  const baseExcludes = ['!**/*.d.ts', '!**/node_modules/**']
+  
+  // 添加用户指定的排除目录
+  const userExcludes = excludeDirs.map(dir => {
+    // 标准化路径：移除前后斜杠，确保以 src/ 开头
+    const normalized = dir.replace(/^\/+|\/+$/g, '')
+    const prefixed = normalized.startsWith('src/') ? normalized : `src/${normalized}`
+    return `!${prefixed}/**`
+  })
+  
+  const patterns = ['src/**/*.{ts,tsx}', ...baseExcludes, ...userExcludes]
+  const files = await fg(patterns, { dot: false })
+  
+  if (!files?.length) throw new Error('No source files found under src (check exclude patterns)')
+  return files
+}
+
+function decideTypeByPathAndName(filePath, exportName) {
+  const p = filePath.toLowerCase()
+  if (p.includes('/hooks/') || /^use[A-Z]/.test(exportName)) return 'hook'
+  if (p.includes('/components/') || /^[A-Z]/.test(exportName)) return 'component'
+  return 'function'
+}
+
+function decideLayer(filePath, cfg) {
+  const layers = cfg?.layers
+  if (!layers) return 'unknown'
+  
+  // 标准化路径：移除反斜杠，去掉 src/ 前缀
+  let normalizedPath = filePath.replace(/\\/g, '/')
+  if (normalizedPath.startsWith('src/')) {
+    normalizedPath = normalizedPath.substring(4)
+  }
+  
+  // 按优先级匹配层级（从最具体到最通用）
+  for (const [layerKey, layerDef] of Object.entries(layers)) {
+    const patterns = layerDef.patterns || []
+    for (const pattern of patterns) {
+      // 简单的 glob 匹配：支持 ** 和 *
+      const regexPattern = pattern
+        .replace(/\*\*/g, '.*')
+        .replace(/\*/g, '[^/]*')
+      const regex = new RegExp(regexPattern)
+      if (regex.test(normalizedPath)) {
+        return layerKey
+      }
+    }
+  }
+  
+  return 'unknown'
+}
+
+function buildImpactHint(path) {
+  const lower = path.toLowerCase()
+  if (/(price|booking|soldout)/.test(lower)) return 'price/booking/soldout'
+  if (/(roomlist|filter)/.test(lower)) return 'roomlist/filter'
+  if (/login|order/.test(lower)) return 'login/order'
+  return 'display'
+}
+
+function buildRoiHint(path, content) {
+  const isPure = !/fetch\(|xRouter|xStorage|xUbt|xEnv|xUa/.test(content)
+  const needsUI = /<\w+/.test(content)
+  return {
+    isPure: isPure && !needsUI,
+    dependenciesInjectable: /\(.*:.*\)/.test(content) && !needsUI,
+    needsUI,
+    multiPlatformStrong: /xEnv|xUa|\.h5\.|\.crn\./.test(path)
+  }
+}
+
+function relativize(path) {
+  const cwd = process.cwd().replace(/\\/g, '/')
+  const norm = String(path).replace(/\\/g, '/')
+  return norm.startsWith(cwd) ? norm.slice(cwd.length + 1) : norm
+}
+
+function getLoc(text, start, end) {
+  const slice = text.slice(start, end)
+  return slice.split(/\r?\n/).length
+}
+
+async function extractTargets(files) {
+  const { Project, SyntaxKind } = await req('ts-morph', 'ts-morph')
+  const cfg = loadJson('ut_scoring_config.json') || {}
+  const internalInclude = cfg.internalInclude === true
+  const minLoc = cfg?.internalThresholds?.minLoc ?? 15
+
+  const project = new Project({ skipAddingFilesFromTsConfig: true })
+  files.forEach(f => project.addSourceFileAtPathIfExists(f))
+
+  const targets = []
+  
+  // 辅助函数：判断变量是否为可测试的函数/组件
+  function isTestableVariable(v) {
+    const init = v?.getInitializer()
+    if (!init) return false
+    const kind = init.getKind()
+    // 箭头函数、函数表达式、React组件（JSX）、HOC包装
+    if (kind === SyntaxKind.ArrowFunction || kind === SyntaxKind.FunctionExpression) return true
+    if (kind === SyntaxKind.CallExpression) {
+      const text = init.getText()
+      return /(React\.memo|forwardRef|memo|observer)\(/.test(text)
+    }
+    return false
+  }
+  
+  for (const sf of project.getSourceFiles()) {
+    const absPath = sf.getFilePath()
+    const relPath = relativize(absPath)
+    const content = sf.getFullText()
+
+    // 导出符号 - 仅包含函数和组件
+    const exported = Array.from(new Set(sf.getExportSymbols().map(s => s.getName()).filter(Boolean)))
+    const fileLoc = content.split('\n').length
+    for (const name of exported) {
+      // 检查是否为函数声明
+      const fn = sf.getFunction(name)
+      if (fn) {
+        const type = decideTypeByPathAndName(relPath, name)
+        const layer = decideLayer(relPath, cfg)
+        targets.push({
+          name,
+          path: relPath,
+          type,
+          layer,
+          internal: false,
+          loc: fileLoc,
+          impactHint: buildImpactHint(relPath),
+          roiHint: buildRoiHint(relPath, content)
+        })
+        continue
+      }
+      
+      // 检查是否为变量（可能是箭头函数或组件）
+          const v = sf.getVariableDeclaration(name)
+          if (v && isTestableVariable(v)) {
+            const type = decideTypeByPathAndName(relPath, name)
+            const layer = decideLayer(relPath, cfg)
+            targets.push({
+              name,
+              path: relPath,
+              type,
+              layer,
+              internal: false,
+              loc: fileLoc,
+              impactHint: buildImpactHint(relPath),
+              roiHint: buildRoiHint(relPath, content)
+            })
+          }
+      // 其他（interface/type/const/enum）直接跳过
+    }
+
+    // 内部顶层命名函数（非导出）
+    if (internalInclude) {
+      const fnDecls = sf.getFunctions().filter(fn => !fn.isExported() && !!fn.getName())
+      for (const fn of fnDecls) {
+        const name = fn.getName()
+        const start = fn.getStart()
+        const end = fn.getEnd()
+        const loc = getLoc(content, start, end)
+        if (loc < minLoc) continue
+            const type = decideTypeByPathAndName(relPath, name)
+            const layer = decideLayer(relPath, cfg)
+            targets.push({
+              name,
+              path: relPath,
+              type,
+              layer,
+              internal: true,
+              loc,
+              impactHint: buildImpactHint(relPath),
+              roiHint: buildRoiHint(relPath, content)
+            })
+      }
+    }
+  }
+  if (!targets.length) throw new Error('No targets (exported or internal) found. Check config thresholds and sources.')
+  return targets
+}
+
+async function main() {
+  const args = parseArgs(process.argv)
+  const cfg = loadJson('ut_scoring_config.json') || {}
+  
+  // 从配置文件和命令行参数获取排除目录
+  const configExcludes = cfg?.targetGeneration?.excludeDirs || []
+  const cliExcludes = args.exclude ? args.exclude.split(',').map(s => s.trim()) : []
+  const allExcludes = [...configExcludes, ...cliExcludes]
+  
+  if (allExcludes.length > 0) {
+    process.stdout.write(`Excluding directories: ${allExcludes.join(', ')}\n`)
+  }
+  
+  const files = await listFiles(allExcludes)
+  const targets = await extractTargets(files)
+  
+  const outPath = args.out || 'reports/targets.json'
+  writeFileSync(outPath, JSON.stringify(targets, null, 2))
+  process.stdout.write(`Generated ${targets.length} targets -> ${outPath}\n`)
+}
+
+main()