ai-unit-test-generator 1.4.0 → 1.4.2

package/bin/cli.js

@@ -24,13 +24,16 @@ program
 program
   .command('scan')
   .description('Scan code and generate priority scoring report')
-  .option('-c, --config <path>', 'Config file path', 'ut_scoring_config.json')
+  .option('-c, --config <path>', 'Config file path', 'ai-test.config.jsonc')
   .option('-o, --output <dir>', 'Output directory', 'reports')
   .option('--skip-git', 'Skip Git signals generation')
   .action(async (options) => {
-    const { config, output, skipGit } = options
+    let { config, output, skipGit } = options
 
-    // 自动初始化配置（如果不存在）
+    // 自动探测现有配置 & 初始化（不存在则创建）
+    const detectOrder = [config, 'ai-test.config.json', 'ai-test.config.jsonc']
+    const detected = detectOrder.find(p => existsSync(p))
+    if (detected) config = detected
     if (!existsSync(config)) {
       console.log('⚙️  Config not found, creating default config...')
       const templatePath = join(PKG_ROOT, 'templates', 'default.config.json')
@@ -43,6 +46,25 @@ program
       mkdirSync(output, { recursive: true })
     }
 
+    // 可选：在扫描前自动运行覆盖率（由配置控制）
+    try {
+      const cfgText = existsSync(config) ? readFileSync(config, 'utf-8') : '{}'
+      const cfg = JSON.parse(cfgText)
+      const covCfg = cfg?.coverage || { runBeforeScan: false }
+      if (covCfg.runBeforeScan) {
+        console.log('🧪 Running coverage before scan...')
+        await new Promise((resolve, reject) => {
+          const cmd = covCfg.command || 'npx jest --coverage --silent'
+          const child = spawn(cmd, { stdio: 'inherit', shell: true, cwd: process.cwd() })
+          child.on('close', code => code === 0 ? resolve(0) : reject(new Error(`coverage exited ${code}`)))
+          child.on('error', reject)
+        })
+      }
+    } catch (err) {
+      console.warn('⚠️  Coverage step failed or Jest not installed. Skipping coverage and continuing scan.')
+      console.warn('    - npm i -D jest@29 ts-jest@29 @types/jest@29 jest-environment-jsdom@29 --legacy-peer-deps')
+    }
+
     console.log('🚀 Starting code scan...\n')
 
     try {

package/lib/core/scanner.mjs

@@ -22,7 +22,12 @@ function parseArgs(argv) {
   return args
 }
 
-function loadJson(p) { try { return JSON.parse(readFileSync(p, 'utf8')) } catch { return null } }
+function stripJsonComments(s) {
+  return String(s)
+    .replace(/\/\*[\s\S]*?\*\//g, '') // block comments
+    .replace(/(^|\s)\/\/.*$/gm, '')     // line comments
+}
+function loadJson(p) { try { return JSON.parse(stripJsonComments(readFileSync(p, 'utf8'))) } catch { return null } }
 
 async function listFiles(excludeDirs = []) {
   const fg = (await req('fast-glob', 'fast-glob')).default

package/lib/core/scorer.mjs

@@ -25,7 +25,25 @@ 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)) }
 
-function loadConfig(pathFromArg) { return loadJson(pathFromArg) || loadJson('ut_scoring_config.json') || {} }
+function stripJsonComments(s) {
+  return String(s)
+    .replace(/\/\*[\s\S]*?\*\//g, '')
+    .replace(/(^|\s)\/\/.*$/gm, '')
+}
+function loadConfig(pathFromArg) {
+  const fs = require('fs')
+  const paths = [pathFromArg, 'ai-test.config.jsonc', 'ai-test.config.json']
+  for (const p of paths) {
+    if (!p) continue
+    try {
+      if (fs.existsSync(p)) {
+        const raw = fs.readFileSync(p, 'utf8')
+        return JSON.parse(stripJsonComments(raw))
+      }
+    } catch {}
+  }
+  return {}
+}
 function pickWeight(cfg, key, def) { return cfg?.weights?.[key] ?? def }
 function pickThreshold(cfg, key, def) { return cfg?.thresholds?.[key] ?? def }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-unit-test-generator",
-  "version": "1.4.0",
+  "version": "1.4.2",
   "description": "AI-powered unit test generator with smart priority scoring",
   "keywords": [
     "unit-test",
@@ -66,4 +66,3 @@
   },
   "homepage": "https://github.com/YuhengZhou/ai-unit-test-generator#readme"
 }
-

package/templates/default.config.json

@@ -98,6 +98,10 @@
       { "lte": 100, "score": 1 }
     ]
   },
+  "coverage": {
+    "runBeforeScan": true,
+    "command": "npx jest --coverage --silent"
+  },
   "weights": {
     "BC": 0.25,
     "CC": 0.15,