ai-unit-test-generator 1.3.0

package/lib/testing/analyzer.mjs

@@ -0,0 +1,43 @@
+#!/usr/bin/env node
+/**
+ * 解析 reports/jest-report.json，提取失败原因并分类
+ * 输出一个简要的提示文本，可注入下一轮 Prompt
+ */
+
+import { readFileSync, existsSync } from 'fs'
+
+export function analyze(path = 'reports/jest-report.json') {
+  if (!existsSync(path)) return { summary: 'No jest-report.json', total: 0, failed: 0, hints: [] }
+  const json = JSON.parse(readFileSync(path, 'utf8'))
+  const testResults = json.testResults || []
+  const hints = []
+  let total = 0, failed = 0
+
+  for (const f of testResults) {
+    for (const a of (f.assertionResults || [])) {
+      total++
+      if (a.status === 'failed') failed++
+    }
+    for (const msg of (f.message ? [f.message] : [])) {
+      const m = String(msg)
+      if (m.includes('Cannot find module')) hints.push('修正导入路径或添加模块别名')
+      if (m.includes('TypeError')) hints.push('检查被测函数或 mock 是否返回预期类型')
+      if (m.includes('Timed out')) hints.push('为异步/计时器逻辑添加等待与 fakeTimers')
+      if (m.includes('not found') && m.includes('element')) hints.push('使用稳定的查询方式，如 getByRole/LabelText/TestId')
+    }
+  }
+
+  const uniqueHints = Array.from(new Set(hints))
+  const summary = `总断言: ${total}, 失败: ${failed}, 建议: ${uniqueHints.join('；')}`
+  return { summary, total, failed, hints: uniqueHints }
+}
+
+export function runCLI(argv = process.argv) {
+  const path = argv[2] || 'reports/jest-report.json'
+  const res = analyze(path)
+  console.log(JSON.stringify(res, null, 2))
+}
+
+if (import.meta.url === `file://${process.argv[1]}`) runCLI()
+
+

package/lib/testing/index.mjs

@@ -0,0 +1,10 @@
+/**
+ * Testing 模块：测试运行与失败分析
+ * 
+ * 提供测试执行和失败报告分析功能
+ * 支持多种测试框架（目前实现：Jest）
+ */
+
+export { main as runTests } from './runner.mjs'
+export { main as analyzeFailures } from './analyzer.mjs'
+

package/lib/testing/runner.mjs

@@ -0,0 +1,32 @@
+#!/usr/bin/env node
+/**
+ * 运行 Jest，产出 JSON 与覆盖率
+ */
+
+import { spawn } from 'child_process'
+
+function runJest(args = []) {
+  return new Promise((resolve, reject) => {
+    const child = spawn('npx', ['jest', '--json', '--outputFile=reports/jest-report.json', '--coverage', ...args], {
+      stdio: 'inherit',
+      cwd: process.cwd()
+    })
+    child.on('close', code => code === 0 ? resolve(0) : reject(new Error(`jest exited ${code}`)))
+    child.on('error', reject)
+  })
+}
+
+async function main(argv = process.argv) {
+  const extra = argv.slice(2)
+  try {
+    await runJest(extra)
+    process.exit(0)
+  } catch (err) {
+    console.error(err.message)
+    process.exit(1)
+  }
+}
+
+if (import.meta.url === `file://${process.argv[1]}`) main()
+
+

package/lib/utils/index.mjs

@@ -0,0 +1,11 @@
+/**
+ * Utils 模块：通用工具函数
+ * 
+ * 提供跨模块使用的工具函数
+ */
+
+export { markDone } from './marker.mjs'
+
+// 可以在未来添加更多工具函数
+// export { readJson, writeJson } from './file-io.mjs'
+

package/lib/utils/marker.mjs

@@ -0,0 +1,182 @@
+#!/usr/bin/env node
+/**
+ * 批量标记 ut_scores.md 中的完成状态
+ */
+
+import { readFileSync, writeFileSync, existsSync } from 'fs';
+
+/**
+ * 标记函数为完成状态
+ */
+export function markDone(functionNames, options = {}) {
+  const { 
+    reportPath = 'reports/ut_scores.md',
+    status = 'DONE',
+    dryRun = false 
+  } = options;
+  
+  if (!existsSync(reportPath)) {
+    throw new Error(`报告文件不存在: ${reportPath}`);
+  }
+  
+  let content = readFileSync(reportPath, 'utf8');
+  const originalContent = content;
+  
+  const results = {
+    success: [],
+    notFound: []
+  };
+  
+  functionNames.forEach(name => {
+    // 匹配 | TODO | [score] | [priority] | [name] | 格式
+    const escapedName = name.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+    const regex = new RegExp(
+      `\\| TODO (\\| [^|]+ \\| [^|]+ \\| ${escapedName} \\|)`,
+      'g'
+    );
+    
+    const newContent = content.replace(regex, `| ${status} $1`);
+    
+    if (newContent !== content) {
+      results.success.push(name);
+      content = newContent;
+    } else {
+      results.notFound.push(name);
+    }
+  });
+  
+  if (!dryRun && results.success.length > 0) {
+    writeFileSync(reportPath, content);
+  }
+  
+  // 统计进度
+  const lines = content.split('\n');
+  const todoCount = lines.filter(l => l.includes('| TODO |')).length;
+  const doneCount = lines.filter(l => l.includes('| DONE |')).length;
+  const skipCount = lines.filter(l => l.includes('| SKIP |')).length;
+  const total = todoCount + doneCount + skipCount;
+  
+  // 按优先级统计
+  const priorities = ['P0', 'P1', 'P2', 'P3'];
+  const priorityStats = {};
+  priorities.forEach(p => {
+    const pTotal = lines.filter(l => l.includes(`| ${p} |`)).length;
+    const pDone = lines.filter(l => l.includes('| DONE |') && l.includes(`| ${p} |`)).length;
+    const pTodo = lines.filter(l => l.includes('| TODO |') && l.includes(`| ${p} |`)).length;
+    if (pTotal > 0) {
+      priorityStats[p] = { total: pTotal, done: pDone, todo: pTodo };
+    }
+  });
+  
+  return {
+    ...results,
+    stats: {
+      total,
+      todo: todoCount,
+      done: doneCount,
+      skip: skipCount,
+      priorities: priorityStats
+    }
+  };
+}
+
+/**
+ * CLI 入口
+ */
+export function runCLI(argv = process.argv) {
+  const args = argv.slice(2);
+  
+  if (args.length === 0) {
+    console.error('❌ 缺少参数\n');
+    console.error('用法:');
+    console.error('  ut-score mark-done <function-names> [options]\n');
+    console.error('参数:');
+    console.error('  function-names    函数名列表（逗号或空格分隔）\n');
+    console.error('选项:');
+    console.error('  --report PATH     报告文件路径 (默认: reports/ut_scores.md)');
+    console.error('  --status STATUS   标记状态 (默认: DONE, 可选: SKIP)');
+    console.error('  --dry-run         仅显示将要修改的内容，不实际写入\n');
+    console.error('示例:');
+    console.error('  ut-score mark-done disableDragBack getMediumScale');
+    console.error('  ut-score mark-done "formatToDate,handleQuery"');
+    console.error('  ut-score mark-done func1 --status SKIP');
+    process.exit(1);
+  }
+  
+  // 解析参数
+  const options = { reportPath: 'reports/ut_scores.md', status: 'DONE' };
+  const names = [];
+  
+  for (let i = 0; i < args.length; i++) {
+    if (args[i] === '--report') {
+      options.reportPath = args[++i];
+    } else if (args[i] === '--status') {
+      options.status = args[++i];
+    } else if (args[i] === '--dry-run') {
+      options.dryRun = true;
+    } else {
+      names.push(args[i]);
+    }
+  }
+  
+  // 解析函数名（支持逗号分隔和空格分隔）
+  const functionNames = names.join(' ').split(/[,\s]+/).map(s => s.trim()).filter(Boolean);
+  
+  if (functionNames.length === 0) {
+    console.error('❌ 没有指定函数名');
+    process.exit(1);
+  }
+  
+  try {
+    console.log(`🔍 准备标记 ${functionNames.length} 个函数为 ${options.status}...\n`);
+    
+    const result = markDone(functionNames, options);
+    
+    if (result.success.length > 0) {
+      console.log(`✅ 成功标记 ${result.success.length} 个函数为 ${options.status}:`);
+      result.success.forEach(name => console.log(`   - ${name}`));
+      console.log('');
+    }
+    
+    if (result.notFound.length > 0) {
+      console.log(`⚠️  未找到 ${result.notFound.length} 个函数（可能已标记或不存在）:`);
+      result.notFound.forEach(name => console.log(`   - ${name}`));
+      console.log('');
+    }
+    
+    // 显示进度
+    const { stats } = result;
+    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
+    console.log('📊 当前进度:');
+    console.log(`   总计: ${stats.total}`);
+    console.log(`   已完成: ${stats.done} (${(stats.done / stats.total * 100).toFixed(1)}%)`);
+    console.log(`   待完成: ${stats.todo}`);
+    if (stats.skip > 0) {
+      console.log(`   已跳过: ${stats.skip}`);
+    }
+    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
+    
+    // 按优先级统计
+    if (Object.keys(stats.priorities).length > 0) {
+      console.log('');
+      console.log('📈 按优先级统计:');
+      Object.entries(stats.priorities).forEach(([p, data]) => {
+        console.log(`   ${p}: ${data.done}/${data.total} 完成，${data.todo} 待处理`);
+      });
+    }
+    
+    if (options.dryRun) {
+      console.log('');
+      console.log('ℹ️  Dry-run 模式，未实际修改文件');
+    }
+  } catch (err) {
+    console.error(`❌ 错误: ${err.message}`);
+    process.exit(1);
+  }
+}
+
+// 作为脚本直接运行时
+if (import.meta.url === `file://${process.argv[1]}`) {
+  runCLI();
+}
+

package/lib/workflows/all.mjs

@@ -0,0 +1,51 @@
+#!/usr/bin/env node
+/**
+ * 循环所有批次：每批 N 个，直到没有 TODO
+ */
+
+import { readFileSync, existsSync } from 'fs'
+import { spawn } from 'child_process'
+
+function sh(cmd, args = []) { return new Promise((resolve, reject) => {
+  const child = spawn(cmd, args, { stdio: 'inherit', cwd: process.cwd() })
+  child.on('close', code => code === 0 ? resolve(0) : reject(new Error(`${cmd} exited ${code}`)))
+  child.on('error', reject)
+})}
+
+function countTodos(path = 'reports/ut_scores.md') {
+  if (!existsSync(path)) return 0
+  const md = readFileSync(path, 'utf8')
+  return (md.match(/\|\s*TODO\s*\|/g) || []).length
+}
+
+async function main(argv = process.argv) {
+  const args = argv.slice(2)
+  const priority = args[0] || 'P0'
+  const batchSize = Number(args[1] || 10)
+  let skip = Number(args[2] || 0)
+
+  while (true) {
+    const remain = countTodos()
+    if (remain === 0) {
+      console.log('✅ All TODO items are done.')
+      break
+    }
+    console.log(`
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+Batch starting... priority=${priority}, size=${batchSize}, skip=${skip}
+Remaining TODO: ${remain}
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━`)
+
+    try {
+      await sh('node', ['ai-test-generator/lib/run-batch.mjs', priority, String(batchSize), String(skip)])
+    } catch (err) {
+      console.error('Batch failed:', err.message)
+    }
+
+    skip += batchSize
+  }
+}
+
+if (import.meta.url === `file://${process.argv[1]}`) main()
+
+

package/lib/workflows/batch.mjs

@@ -0,0 +1,187 @@
+#!/usr/bin/env node
+/**
+ * 单批次：生成 prompt → 调用 AI → 提取测试 → 运行 Jest → 自动标记状态
+ */
+
+import { spawn } from 'child_process'
+import { existsSync, readFileSync, writeFileSync } from 'fs'
+
+function sh(cmd, args = []) {
+  return new Promise((resolve, reject) => {
+    const child = spawn(cmd, args, { stdio: 'inherit', cwd: process.cwd() })
+    child.on('close', code => code === 0 ? resolve(0) : reject(new Error(`${cmd} exited ${code}`)))
+    child.on('error', reject)
+  })
+}
+
+function readCoverageSummary() {
+  const path = 'coverage/coverage-summary.json'
+  if (!existsSync(path)) return null
+  try { return JSON.parse(readFileSync(path, 'utf8')) } catch { return null }
+}
+
+function getCoveragePercent(summary) {
+  if (!summary || !summary.total) return 0
+  return summary.total.lines?.pct ?? 0
+}
+
+/**
+ * 从报告中读取 TODO 函数列表
+ */
+function readTodoFunctions(reportPath, priority, limit) {
+  if (!existsSync(reportPath)) {
+    throw new Error(`Report not found: ${reportPath}`)
+  }
+  
+  const content = readFileSync(reportPath, 'utf-8')
+  const lines = content.split('\n')
+  
+  const todoFunctions = []
+  for (const line of lines) {
+    if (line.includes('| TODO |') && line.includes(`| ${priority} |`)) {
+      // 解析表格行: | Name | Path | ... | Priority | Score | Status |
+      const parts = line.split('|').map(p => p.trim()).filter(Boolean)
+      if (parts.length >= 6) {
+        todoFunctions.push({
+          name: parts[0],
+          path: parts[1]
+        })
+      }
+    }
+  }
+  
+  return todoFunctions.slice(0, limit)
+}
+
+/**
+ * 标记函数状态为 DONE
+ */
+function markFunctionsDone(reportPath, functionNames) {
+  if (!existsSync(reportPath)) return
+  
+  let content = readFileSync(reportPath, 'utf-8')
+  
+  for (const name of functionNames) {
+    // 查找包含该函数名且状态为 TODO 的行，替换为 DONE
+    const lines = content.split('\n')
+    for (let i = 0; i < lines.length; i++) {
+      if (lines[i].includes(`| ${name} |`) && lines[i].includes('| TODO |')) {
+        lines[i] = lines[i].replace('| TODO |', '| DONE |')
+      }
+    }
+    content = lines.join('\n')
+  }
+  
+  writeFileSync(reportPath, content, 'utf-8')
+}
+
+async function main(argv = process.argv) {
+  const args = argv.slice(2)
+  const priority = args[0] || 'P0'
+  const limit = Number(args[1] || 10)
+  const skip = Number(args[2] || 0)
+  const reportPath = args[3] || 'reports/ut_scores.md'
+  const minCovDelta = priority === 'P0' ? 2 : (priority === 'P1' ? 1 : 0)
+
+  // 读取 TODO 函数列表（跳过 DONE）
+  console.log(`📋 Reading TODO functions from ${reportPath}...`)
+  const todoFunctions = readTodoFunctions(reportPath, priority, limit)
+  
+  if (todoFunctions.length === 0) {
+    console.log(`✅ No TODO functions found for ${priority}`)
+    return
+  }
+  
+  console.log(`📝 Found ${todoFunctions.length} TODO functions`)
+  
+  // 记录初始覆盖率
+  const beforeCov = getCoveragePercent(readCoverageSummary())
+
+  // 1) 生成 Prompt（只针对 TODO 函数，加入上一轮失败提示）
+  const promptArgs = [
+    'ai-test-generator/lib/ai/prompt-builder.mjs',
+    '--report', reportPath,
+    '-p', priority,
+    '-n', String(limit),
+    '--skip', String(skip),
+    '--only-todo' // 新增：只处理 TODO 状态
+  ]
+  try {
+    await sh('node', [...promptArgs, '--hints-file', 'reports/hints.txt'])
+  } catch {
+    await sh('node', promptArgs)
+  }
+
+  // 2) 调用 AI
+  console.log('\n🤖 Calling AI...')
+  await sh('node', ['ai-test-generator/lib/ai/client.mjs', '--prompt', 'prompt.txt', '--out', 'reports/ai_response.txt'])
+
+  // 3) 提取测试
+  console.log('\n📦 Extracting tests...')
+  await sh('node', ['ai-test-generator/lib/ai/extractor.mjs', 'reports/ai_response.txt', '--overwrite'])
+
+  // 4) 运行 Jest（按优先级自适应重跑）
+  console.log('\n🧪 Running tests...')
+  const reruns = priority === 'P0' ? 1 : 0
+  let testsPassed = false
+  
+  for (let i = 0; i < Math.max(1, reruns + 1); i++) {
+    try {
+      await sh('node', ['ai-test-generator/lib/testing/runner.mjs'])
+      testsPassed = true
+      break
+    } catch {
+      if (i === reruns) {
+        console.warn('⚠️  Tests failed after retries')
+      }
+    }
+  }
+
+  // 5) 校验覆盖率增量
+  if (minCovDelta > 0) {
+    const afterCov = getCoveragePercent(readCoverageSummary())
+    const delta = afterCov - beforeCov
+    if (delta < minCovDelta) {
+      console.warn(`⚠️  Coverage delta ${delta.toFixed(2)}% < required ${minCovDelta}% (before: ${beforeCov.toFixed(2)}%, after: ${afterCov.toFixed(2)}%)`)
+    } else {
+      console.log(`✅ Coverage improved: ${beforeCov.toFixed(2)}% → ${afterCov.toFixed(2)}% (+${delta.toFixed(2)}%)`)
+    }
+  }
+
+  // 6) 自动标记 DONE（如果测试通过）
+  if (testsPassed) {
+    console.log('\n✏️  Marking functions as DONE...')
+    const functionNames = todoFunctions.map(f => f.name)
+    markFunctionsDone(reportPath, functionNames)
+    console.log(`✅ Marked ${functionNames.length} functions as DONE`)
+  } else {
+    console.log('\n⚠️  Tests failed, keeping functions as TODO for retry')
+  }
+
+  // 7) 失败分析并落盘 hints（用于下次重试）
+  console.log('\n🔍 Analyzing failures...')
+  const { spawn: spawnLocal } = await import('child_process')
+  const { writeFileSync: writeFileSyncLocal } = await import('fs')
+  await new Promise((resolve) => {
+    const child = spawnLocal('node', ['ai-test-generator/lib/testing/analyzer.mjs'], { stdio: ['inherit','pipe','inherit'] })
+    const chunks = []
+    child.stdout.on('data', d => chunks.push(Buffer.from(d)))
+    child.on('close', () => {
+      try {
+        const obj = JSON.parse(Buffer.concat(chunks).toString('utf8'))
+        if (obj.hints?.length) {
+          writeFileSyncLocal('reports/hints.txt', `# 上一轮失败修复建议\n- ${obj.hints.join('\n- ')}`)
+          console.log(`💡 Saved ${obj.hints.length} hints for next run`)
+        }
+      } catch {}
+      resolve()
+    })
+  })
+
+  console.log('\n✅ Batch completed!')
+}
+
+main().catch(err => {
+  console.error('❌ Batch failed:', err.message)
+  process.exit(1)
+})

package/lib/workflows/index.mjs

@@ -0,0 +1,10 @@
+/**
+ * Workflows 模块：编排工作流
+ * 
+ * 提供批量测试生成和全自动化流程编排
+ * 组合 Core、AI 和 Testing 模块实现端到端工作流
+ */
+
+export { main as runBatch } from './batch.mjs'
+export { main as runAll } from './all.mjs'
+

package/package.json

@@ -0,0 +1,69 @@
+{
+  "name": "ai-unit-test-generator",
+  "version": "1.3.0",
+  "description": "AI-powered unit test generator with smart priority scoring",
+  "keywords": [
+    "unit-test",
+    "testing",
+    "priority",
+    "scoring",
+    "ai-test-generation",
+    "code-analysis",
+    "typescript",
+    "react"
+  ],
+  "author": "YuhengZhou",
+  "license": "MIT",
+  "type": "module",
+  "main": "./lib/index.mjs",
+  "bin": {
+    "ai-test": "./bin/cli.js"
+  },
+  "exports": {
+    ".": "./lib/index.mjs",
+    "./core": "./lib/core/index.mjs",
+    "./ai": "./lib/ai/index.mjs",
+    "./testing": "./lib/testing/index.mjs",
+    "./workflows": "./lib/workflows/index.mjs",
+    "./utils": "./lib/utils/index.mjs"
+  },
+  "files": [
+    "bin",
+    "lib",
+    "templates",
+    "README.md",
+    "CHANGELOG.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "prepublishOnly": "echo 'Ready to publish'",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "dependencies": {
+    "@eslint/js": "^9.17.0",
+    "commander": "^12.1.0",
+    "csv-stringify": "^6.5.2",
+    "escomplex": "^2.0.0-alpha",
+    "esprima": "^4.0.1",
+    "eslint": "^9.17.0",
+    "eslint-plugin-sonarjs": "^3.0.1",
+    "fast-glob": "^3.3.3",
+    "markdown-table": "^3.0.4",
+    "ts-morph": "^24.0.0",
+    "typescript": "^5.7.3",
+    "typescript-eslint": "^8.19.1"
+  },
+  "devDependencies": {},
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/YuhengZhou/ai-unit-test-generator.git"
+  },
+  "bugs": {
+    "url": "https://github.com/YuhengZhou/ai-unit-test-generator/issues"
+  },
+  "homepage": "https://github.com/YuhengZhou/ai-unit-test-generator#readme"
+}
+