qat-cli 0.2.4 → 0.2.5

package/README.md

@@ -4,7 +4,7 @@
 
 **Quick Auto Testing — 面向 Vue 项目，集成 Vitest & Playwright，AI 驱动覆盖测试全流程**
 
-[![npm version](https://img.shields.io/badge/version-0.2.4-blue.svg)](https://www.npmjs.com/package/qat-cli)
+[![npm version](https://img.shields.io/badge/version-0.2.5-blue.svg)](https://www.npmjs.com/package/qat-cli)
 [![Node.js](https://img.shields.io/badge/node-%3E%3D18.0.0-green.svg)](https://nodejs.org/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 [![TypeScript](https://img.shields.io/badge/TypeScript-5.6-blue.svg)](https://www.typescriptlang.org/)

package/dist/cli.js

@@ -2470,11 +2470,22 @@ async function autoGenerateTests(config, projectInfo, aiConfig, useAI) {
   for (const utilPath of utilities.slice(0, 30)) {
     allTargets.push({ filePath: utilPath, testType: inferTestType(utilPath) });
   }
-  const genSpinner = ora2(`\u6B63\u5728\u751F\u6210\u6D4B\u8BD5\u7528\u4F8B (AI \u8F85\u52A9 + \u5BA1\u8BA1) \u2014 \u53D1\u73B0 ${allTargets.length} \u4E2A\u6587\u4EF6...`).start();
+  const selectedTargets = await selectTargetFiles(allTargets);
+  if (selectedTargets.length === 0) {
+    console.log(chalk3.gray("\n  \u672A\u9009\u62E9\u4EFB\u4F55\u6587\u4EF6\uFF0C\u8DF3\u8FC7\u6D4B\u8BD5\u7528\u4F8B\u751F\u6210\u3002"));
+    return [];
+  }
+  const total = selectedTargets.length;
+  const genSpinner = ora2(`\u6B63\u5728\u751F\u6210\u6D4B\u8BD5\u7528\u4F8B [0/${total}] ...`).start();
   const generatedFiles = [];
   const typeCount = {};
   const reviewReport = [];
-  for (const { filePath, testType } of allTargets) {
+  let current = 0;
+  let failed = 0;
+  for (const { filePath, testType } of selectedTargets) {
+    current++;
+    const fileLabel = path7.basename(filePath);
+    genSpinner.text = `\u6B63\u5728\u751F\u6210\u6D4B\u8BD5\u7528\u4F8B [${current}/${total}] ${chalk3.cyan(fileLabel)} ...`;
     try {
       const result = await generateTestForTarget(
         testType,
@@ -2492,20 +2503,105 @@ async function autoGenerateTests(config, projectInfo, aiConfig, useAI) {
         }
       }
     } catch {
+      failed++;
     }
   }
   if (generatedFiles.length > 0) {
     const summary = Object.entries(typeCount).map(([type, count]) => `${count} ${type}`).join(", ");
     const approvedCount = reviewReport.filter((r) => r.approved).length;
-    genSpinner.succeed(`\u5DF2\u751F\u6210 ${generatedFiles.length} \u4E2A\u6D4B\u8BD5\u7528\u4F8B (${summary}) \u2014 AI \u751F\u6210+\u5BA1\u8BA1 ${approvedCount}/${reviewReport.length} \u901A\u8FC7`);
+    let msg = `\u5DF2\u751F\u6210 ${generatedFiles.length}/${total} \u4E2A\u6D4B\u8BD5\u7528\u4F8B (${summary}) \u2014 AI \u5BA1\u8BA1 ${approvedCount}/${reviewReport.length} \u901A\u8FC7`;
+    if (failed > 0) msg += chalk3.yellow(` ${failed} \u4E2A\u5931\u8D25`);
+    genSpinner.succeed(msg);
   } else {
-    genSpinner.warn("\u672A\u751F\u6210\u6D4B\u8BD5\u7528\u4F8B");
+    genSpinner.warn(`\u672A\u751F\u6210\u6D4B\u8BD5\u7528\u4F8B (${failed} \u4E2A\u5931\u8D25)`);
   }
   if (reviewReport.length > 0) {
     printReviewReport(reviewReport);
   }
   return generatedFiles;
 }
+async function selectTargetFiles(allTargets) {
+  const testTypeLabels = {
+    unit: chalk3.blue("[unit]"),
+    component: chalk3.magenta("[comp]"),
+    e2e: chalk3.green("[e2e]"),
+    api: chalk3.yellow("[api]"),
+    visual: chalk3.cyan("[visual]"),
+    performance: chalk3.gray("[perf]")
+  };
+  const choices = allTargets.map(({ filePath, testType }) => ({
+    name: `${testTypeLabels[testType]} ${filePath}`,
+    value: filePath,
+    short: filePath
+  }));
+  choices.push({
+    name: chalk3.gray("\u270E \u624B\u52A8\u8F93\u5165\u6587\u4EF6/\u76EE\u5F55\u8DEF\u5F84"),
+    value: "__manual__",
+    short: "\u624B\u52A8\u8F93\u5165"
+  });
+  const { selected } = await inquirer.prompt([
+    {
+      type: "checkbox",
+      name: "selected",
+      message: "\u9009\u62E9\u8981\u751F\u6210\u6D4B\u8BD5\u7528\u4F8B\u7684\u6587\u4EF6 (\u7A7A\u683C\u9009\u62E9/\u53D6\u6D88\uFF0C\u56DE\u8F66\u786E\u8BA4):",
+      choices,
+      pageSize: 15
+    }
+  ]);
+  const manualPaths = [];
+  if (selected.includes("__manual__")) {
+    const { manualInput } = await inquirer.prompt([
+      {
+        type: "input",
+        name: "manualInput",
+        message: "\u8F93\u5165\u6587\u4EF6\u6216\u76EE\u5F55\u8DEF\u5F84\uFF08\u591A\u4E2A\u7528\u9017\u53F7\u5206\u9694\uFF09:",
+        default: "",
+        filter: (input) => input.split(",").map((s) => s.trim()).filter(Boolean)
+      }
+    ]);
+    for (const p of manualInput) {
+      const resolved = path7.resolve(process.cwd(), p);
+      if (fs7.existsSync(resolved)) {
+        const stat = fs7.statSync(resolved);
+        if (stat.isDirectory()) {
+          const dirFiles = walkDirForTestableFiles(resolved);
+          manualPaths.push(...dirFiles);
+        } else if (stat.isFile()) {
+          manualPaths.push(p.replace(/\\/g, "/"));
+        }
+      } else {
+        console.log(chalk3.yellow(`    \u8DEF\u5F84\u4E0D\u5B58\u5728\uFF0C\u5DF2\u8DF3\u8FC7: ${p}`));
+      }
+    }
+  }
+  const selectedPaths = selected.filter((s) => s !== "__manual__");
+  const selectedSet = new Set(selectedPaths);
+  const result = allTargets.filter((t) => selectedSet.has(t.filePath));
+  const existingPaths = new Set(result.map((t) => t.filePath));
+  for (const mp of manualPaths) {
+    if (!existingPaths.has(mp)) {
+      result.push({ filePath: mp, testType: inferTestType(mp) });
+    }
+  }
+  return result;
+}
+function walkDirForTestableFiles(dir) {
+  const files = [];
+  try {
+    const entries = fs7.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.name === "node_modules" || entry.name === "dist" || entry.name.startsWith(".")) continue;
+      const fullPath = path7.join(dir, entry.name);
+      if (entry.isDirectory()) {
+        files.push(...walkDirForTestableFiles(fullPath));
+      } else if (entry.isFile() && /\.(vue|ts|js)$/.test(entry.name)) {
+        files.push(path7.relative(process.cwd(), fullPath).replace(/\\/g, "/"));
+      }
+    }
+  } catch {
+  }
+  return files;
+}
 async function generateTestForTarget(testType, targetPath, config, projectInfo, aiConfig, useAI) {
   const basename = path7.basename(targetPath, path7.extname(targetPath));
   const name = basename.replace(/[^a-zA-Z0-9]/g, "-").replace(/-+/g, "-").replace(/^-|-$/g, "") || `${testType}-test`;
@@ -5367,7 +5463,7 @@ async function executeChange(_options) {
 }
 
 // src/cli.ts
-var VERSION = "0.2.4";
+var VERSION = "0.2.5";
 function printLogo() {
   const logo = `
   ${chalk12.bold.cyan("   ___            _          _           _              _               _____                _     _                 ")}