qat-cli 0.3.5 → 0.3.6

package/README.md

@@ -4,7 +4,7 @@
 
 **Quick Auto Testing — 面向 Vue 项目，集成 Vitest & Playwright，AI 驱动覆盖测试全流程**
 
-[![npm version](https://img.shields.io/badge/version-0.3.05-blue.svg)](https://www.npmjs.com/package/qat-cli)
+[![npm version](https://img.shields.io/badge/version-0.3.06-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

@@ -1156,6 +1156,8 @@ ${this.compressSourceCode(req.context, 3e3, importPathRewrites)}
     if (srcDirMatch) {
       variations.add(`@/${srcDirMatch[1]}`);
       variations.add(`@/${srcDirMatch[1].replace(/\.(ts|js|tsx|jsx)$/, "")}`);
+      variations.add(`#/${srcDirMatch[1]}`);
+      variations.add(`#/${srcDirMatch[1].replace(/\.(ts|js|tsx|jsx)$/, "")}`);
     }
     const pathParts = targetPath.replace(/^\.\//, "").split("/");
     const fileName = pathParts[pathParts.length - 1];
@@ -1890,7 +1892,7 @@ function extractImports(content) {
 function analyzeImportSignatures(imports, baseDir) {
   const signatures = [];
   for (const imp of imports) {
-    if (!imp.source.startsWith(".") && !imp.source.startsWith("@/") && !imp.source.startsWith("~")) {
+    if (!imp.source.startsWith(".") && !imp.source.startsWith("@/") && !imp.source.startsWith("#/") && !imp.source.startsWith("~")) {
       continue;
     }
     const resolvedPath = resolveImportPath(imp.source, baseDir);
@@ -1932,8 +1934,8 @@ function analyzeImportSignatures(imports, baseDir) {
 }
 function resolveImportPath(importSource, baseDir) {
   let resolved;
-  if (importSource.startsWith("@/") || importSource.startsWith("~")) {
-    const relativePath = importSource.replace(/^[@~]\//, "src/");
+  if (importSource.startsWith("@/") || importSource.startsWith("#/") || importSource.startsWith("~")) {
+    const relativePath = importSource.replace(/^[@#~]\//, "src/");
     resolved = path4.resolve(process.cwd(), relativePath);
   } else if (importSource.startsWith(".")) {
     resolved = path4.resolve(baseDir, importSource);
@@ -2914,59 +2916,101 @@ async function executeCreate(options) {
     ]);
     useAI = ai;
   }
+  const tasks = [];
+  for (const testType of types) {
+    for (const testTarget of targets) {
+      tasks.push({ testType, target: testTarget });
+    }
+  }
   const createdFiles = [];
   let skippedCount = 0;
   const reviewReportEntries = [];
-  for (const testType of types) {
-    for (const testTarget of targets) {
-      const spinner = ora3(`\u6B63\u5728\u751F\u6210 ${TEST_TYPE_LABELS[testType]} - ${path9.basename(testTarget)}...`).start();
-      try {
-        let content;
-        if (useAI && testTarget) {
-          const aiResult = await generateWithAI(testType, name, testTarget, config, (text) => {
-            spinner.text = `${text}`;
-          });
-          content = aiResult.code;
-          if (aiResult.reviewEntry) {
-            reviewReportEntries.push(aiResult.reviewEntry);
-          }
+  if (useAI && tasks.length > 0) {
+    const spinner = ora3(`\u5E76\u884C\u751F\u6210 ${tasks.length} \u4E2A\u6D4B\u8BD5\u7528\u4F8B...`).start();
+    let completed = 0;
+    let approvedCount = 0;
+    let failedCount = 0;
+    const results = await Promise.allSettled(
+      tasks.map(async (task) => {
+        const result = await generateWithAI(task.testType, name, task.target, config, (text) => {
+          completed++;
+          spinner.text = `\u5E76\u884C\u751F\u6210\u4E2D (${completed}/${tasks.length}) ${text}`;
+        });
+        return { ...task, ...result };
+      })
+    );
+    spinner.stop();
+    for (let i = 0; i < results.length; i++) {
+      const r = results[i];
+      const task = tasks[i];
+      if (r.status === "rejected") {
+        console.log(chalk5.red(`  \u2717 ${TEST_TYPE_LABELS[task.testType]} - ${path9.basename(task.target)}: ${r.reason instanceof Error ? r.reason.message : String(r.reason)}`));
+        failedCount++;
+        continue;
+      }
+      const { code, reviewEntry } = r.value;
+      const outputDir = getTestOutputDir(task.testType);
+      const filePath = path9.join(outputDir, `${name}.${task.testType === "e2e" ? "spec" : "test"}.ts`);
+      if (fs8.existsSync(filePath)) {
+        console.log(chalk5.yellow(`  \u26A0 \u6D4B\u8BD5\u6587\u4EF6\u5DF2\u5B58\u5728: ${path9.relative(process.cwd(), filePath)}`));
+        skippedCount++;
+        continue;
+      }
+      if (!fs8.existsSync(outputDir)) {
+        fs8.mkdirSync(outputDir, { recursive: true });
+      }
+      fs8.writeFileSync(filePath, code, "utf-8");
+      createdFiles.push({ type: task.testType, filePath });
+      if (reviewEntry) {
+        reviewReportEntries.push(reviewEntry);
+        if (reviewEntry.approved) {
+          approvedCount++;
         } else {
-          const analysis = testTarget ? analyzeFile(testTarget) : void 0;
-          content = renderTemplate(testType, {
-            name,
-            target: testTarget || `./${config.project.srcDir}`,
-            framework: projectInfo.framework,
-            vueVersion: projectInfo.vueVersion,
-            typescript: projectInfo.typescript,
-            uiLibrary: projectInfo.uiLibrary,
-            extraImports: projectInfo.componentTestSetup?.extraImports,
-            globalPlugins: projectInfo.componentTestSetup?.globalPlugins,
-            globalStubs: projectInfo.componentTestSetup?.globalStubs,
-            mountOptions: projectInfo.componentTestSetup?.mountOptions,
-            // 注入源码分析结果
-            exports: analysis?.exports.map((e) => ({
-              name: e.name,
-              kind: e.kind,
-              params: e.params,
-              isAsync: e.isAsync,
-              returnType: e.returnType
-            })),
-            props: analysis?.vueAnalysis?.props.map((p) => ({
-              name: p.name,
-              type: p.type,
-              required: p.required,
-              defaultValue: p.defaultValue
-            })),
-            emits: analysis?.vueAnalysis?.emits.map((e) => ({
-              name: e.name,
-              params: e.params
-            })),
-            methods: analysis?.vueAnalysis?.methods,
-            computed: analysis?.vueAnalysis?.computed
-          });
+          failedCount++;
         }
-        const outputDir = getTestOutputDir(testType);
-        const filePath = path9.join(outputDir, `${name}.${testType === "e2e" ? "spec" : "test"}.ts`);
+      }
+    }
+    console.log(chalk5.cyan(`
+  \u5E76\u884C\u751F\u6210\u5B8C\u6210: ${createdFiles.length} \u6210\u529F, ${skippedCount} \u8DF3\u8FC7, ${failedCount} \u5BA1\u8BA1\u672A\u901A\u8FC7`));
+  } else {
+    for (const task of tasks) {
+      const spinner = ora3(`\u6B63\u5728\u751F\u6210 ${TEST_TYPE_LABELS[task.testType]} - ${path9.basename(task.target)}...`).start();
+      try {
+        const analysis = task.target ? analyzeFile(task.target) : void 0;
+        const content = renderTemplate(task.testType, {
+          name,
+          target: task.target || `./${config.project.srcDir}`,
+          framework: projectInfo.framework,
+          vueVersion: projectInfo.vueVersion,
+          typescript: projectInfo.typescript,
+          uiLibrary: projectInfo.uiLibrary,
+          extraImports: projectInfo.componentTestSetup?.extraImports,
+          globalPlugins: projectInfo.componentTestSetup?.globalPlugins,
+          globalStubs: projectInfo.componentTestSetup?.globalStubs,
+          mountOptions: projectInfo.componentTestSetup?.mountOptions,
+          // 注入源码分析结果
+          exports: analysis?.exports.map((e) => ({
+            name: e.name,
+            kind: e.kind,
+            params: e.params,
+            isAsync: e.isAsync,
+            returnType: e.returnType
+          })),
+          props: analysis?.vueAnalysis?.props.map((p) => ({
+            name: p.name,
+            type: p.type,
+            required: p.required,
+            defaultValue: p.defaultValue
+          })),
+          emits: analysis?.vueAnalysis?.emits.map((e) => ({
+            name: e.name,
+            params: e.params
+          })),
+          methods: analysis?.vueAnalysis?.methods,
+          computed: analysis?.vueAnalysis?.computed
+        });
+        const outputDir = getTestOutputDir(task.testType);
+        const filePath = path9.join(outputDir, `${name}.${task.testType === "e2e" ? "spec" : "test"}.ts`);
         if (fs8.existsSync(filePath)) {
           spinner.warn(`\u6D4B\u8BD5\u6587\u4EF6\u5DF2\u5B58\u5728: ${path9.relative(process.cwd(), filePath)}`);
           skippedCount++;
@@ -2976,10 +3020,10 @@ async function executeCreate(options) {
           fs8.mkdirSync(outputDir, { recursive: true });
         }
         fs8.writeFileSync(filePath, content, "utf-8");
-        spinner.succeed(`${TEST_TYPE_LABELS[testType]} - ${path9.basename(testTarget)}`);
-        createdFiles.push({ type: testType, filePath });
+        spinner.succeed(`${TEST_TYPE_LABELS[task.testType]} - ${path9.basename(task.target)}`);
+        createdFiles.push({ type: task.testType, filePath });
       } catch (error) {
-        spinner.fail(`${TEST_TYPE_LABELS[testType]} - ${path9.basename(testTarget)} \u521B\u5EFA\u5931\u8D25`);
+        spinner.fail(`${TEST_TYPE_LABELS[task.testType]} - ${path9.basename(task.target)} \u521B\u5EFA\u5931\u8D25`);
         console.log(chalk5.gray(`    ${error instanceof Error ? error.message : String(error)}`));
       }
     }
@@ -2988,6 +3032,34 @@ async function executeCreate(options) {
   if (reviewReportEntries.length > 0) {
     printReviewReport(reviewReportEntries);
   }
+  const failedEntries = reviewReportEntries.filter((e) => !e.approved);
+  if (failedEntries.length > 0 && useAI) {
+    const { retry } = await inquirer2.prompt([
+      {
+        type: "confirm",
+        name: "retry",
+        message: `${failedEntries.length} \u4E2A\u6D4B\u8BD5\u7528\u4F8B\u5BA1\u8BA1\u672A\u901A\u8FC7\uFF0C\u662F\u5426\u91CD\u65B0\u751F\u6210\uFF1F`,
+        default: false
+      }
+    ]);
+    if (retry) {
+      console.log(chalk5.cyan("\n  \u91CD\u65B0\u751F\u6210\u672A\u901A\u8FC7\u7684\u6D4B\u8BD5\u7528\u4F8B...\n"));
+      for (const entry of failedEntries) {
+        const spinner = ora3(`\u91CD\u65B0\u751F\u6210 ${path9.basename(entry.target)}...`).start();
+        try {
+          const aiResult = await generateWithAI(entry.testType, name, entry.target, config, (text) => {
+            spinner.text = text;
+          });
+          const outputDir = getTestOutputDir(entry.testType);
+          const filePath = path9.join(outputDir, `${name}.${entry.testType === "e2e" ? "spec" : "test"}.ts`);
+          fs8.writeFileSync(filePath, aiResult.code, "utf-8");
+          spinner.succeed(`${path9.basename(entry.target)} \u5DF2\u91CD\u65B0\u751F\u6210${aiResult.reviewEntry?.approved ? chalk5.green(" (\u5BA1\u8BA1\u901A\u8FC7)") : chalk5.yellow(" (\u4ECD\u9700\u5BA1\u67E5)")}`);
+        } catch (error) {
+          spinner.fail(`${path9.basename(entry.target)} \u91CD\u65B0\u751F\u6210\u5931\u8D25`);
+        }
+      }
+    }
+  }
 }
 async function selectTargets(types, projectInfo, srcDir) {
   const allTargets = [];
@@ -5922,7 +5994,7 @@ async function executeChange(_options) {
 }
 
 // src/cli.ts
-var VERSION = "0.3.05";
+var VERSION = "0.3.06";
 function printLogo() {
   const logo = `
   ${chalk13.bold.cyan("   ___            _          _           _              _               _____                _     _                 ")}