qat-cli 0.2.4 → 0.2.6

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.6-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

@@ -2342,14 +2342,41 @@ async function executeInit(options) {
     }
   }
   const config = buildProjectConfig(projectInfo);
-  const fileSpinner = ora2("\u6B63\u5728\u751F\u6210\u914D\u7F6E\u6587\u4EF6...").start();
   let configPath;
-  try {
-    configPath = await writeConfigFile(process.cwd(), config, options.force);
-    fileSpinner.succeed("\u914D\u7F6E\u6587\u4EF6\u5DF2\u751F\u6210");
-  } catch (error) {
-    fileSpinner.fail("\u914D\u7F6E\u6587\u4EF6\u751F\u6210\u5931\u8D25");
-    throw error;
+  const existingConfigPath = path7.join(process.cwd(), "qat.config.js");
+  const existingTsPath = path7.join(process.cwd(), "qat.config.ts");
+  const configExists = fs7.existsSync(existingConfigPath) || fs7.existsSync(existingTsPath);
+  if (configExists && !options.force) {
+    const { overwrite } = await inquirer.prompt([
+      {
+        type: "confirm",
+        name: "overwrite",
+        message: "\u914D\u7F6E\u6587\u4EF6 qat.config.js \u5DF2\u5B58\u5728\uFF0C\u662F\u5426\u8986\u76D6\uFF1F",
+        default: true
+      }
+    ]);
+    if (!overwrite) {
+      console.log(chalk3.gray("  \u4FDD\u7559\u73B0\u6709\u914D\u7F6E\u6587\u4EF6\uFF0C\u7EE7\u7EED\u540E\u7EED\u6B65\u9AA4..."));
+      configPath = existingConfigPath;
+    } else {
+      const fileSpinner = ora2("\u6B63\u5728\u8986\u76D6\u914D\u7F6E\u6587\u4EF6...").start();
+      try {
+        configPath = await writeConfigFile(process.cwd(), config, true);
+        fileSpinner.succeed("\u914D\u7F6E\u6587\u4EF6\u5DF2\u8986\u76D6");
+      } catch (error) {
+        fileSpinner.fail("\u914D\u7F6E\u6587\u4EF6\u8986\u76D6\u5931\u8D25");
+        throw error;
+      }
+    }
+  } else {
+    const fileSpinner = ora2("\u6B63\u5728\u751F\u6210\u914D\u7F6E\u6587\u4EF6...").start();
+    try {
+      configPath = await writeConfigFile(process.cwd(), config, options.force);
+      fileSpinner.succeed("\u914D\u7F6E\u6587\u4EF6\u5DF2\u751F\u6210");
+    } catch (error) {
+      fileSpinner.fail("\u914D\u7F6E\u6587\u4EF6\u751F\u6210\u5931\u8D25");
+      throw error;
+    }
   }
   const dirSpinner = ora2("\u6B63\u5728\u521B\u5EFA\u6D4B\u8BD5\u76EE\u5F55...").start();
   const createdDirs = createTestDirectories(config);
@@ -2470,11 +2497,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 +2530,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 +5490,7 @@ async function executeChange(_options) {
 }
 
 // src/cli.ts
-var VERSION = "0.2.4";
+var VERSION = "0.2.6";
 function printLogo() {
   const logo = `
   ${chalk12.bold.cyan("   ___            _          _           _              _               _____                _     _                 ")}