qat-cli 0.3.4 → 0.3.5

package/dist/index.cjs

@@ -2861,9 +2861,15 @@ ${errorDetails}`;
   /**
    * 压缩源码：保留签名和关键逻辑，剔除注释、空行、样式块
    * 目标：在保留准确性的前提下减少 token 消耗
+   * @param code 源码内容
+   * @param maxLength 最大长度
+   * @param importPathRewrites import 路径重写映射（原路径→正确路径）
    */
-  compressSourceCode(code, maxLength = 3e3) {
+  compressSourceCode(code, maxLength = 3e3, importPathRewrites) {
     let compressed = code;
+    if (importPathRewrites && importPathRewrites.size > 0) {
+      compressed = this.rewriteImportPaths(compressed, importPathRewrites);
+    }
     compressed = compressed.replace(/<style[^>]*>[\s\S]*?<\/style>/gi, "");
     compressed = compressed.replace(/<template[^>]*>([\s\S]*?)<\/template>/gi, (_match, content) => {
       return `<template>${content.replace(/\s*(?:class|style)\s*=\s*["'][^"']*["']/gi, "")}</template>`;
@@ -2901,7 +2907,8 @@ ${errorDetails}`;
   }
   buildGenerateTestUserPrompt(req) {
     const importPath = this.computeTestImportPath(req.type, req.target);
-    let prompt = `\u4E3A${req.target}\u751F\u6210${req.type}\u6D4B\u8BD5\u3002import\u8DEF\u5F84:${importPath}
+    let prompt = `\u4E3A${req.target}\u751F\u6210${req.type}\u6D4B\u8BD5\u3002
+\u3010\u5F3A\u5236\u3011import\u88AB\u6D4B\u6A21\u5757\u5FC5\u987B\u7528: ${importPath}
 `;
     if (req.analysis) {
       const parts = [];
@@ -2923,13 +2930,19 @@ ${errorDetails}`;
       if (req.analysis.computed?.length) {
         parts.push(`Computed:${req.analysis.computed.join(",")}`);
       }
+      if (req.analysis.importSignatures?.length) {
+        parts.push(`\u4F9D\u8D56\u7B7E\u540D:${req.analysis.importSignatures.map(
+          (imp) => `${imp.source}{${Object.entries(imp.signatures).map(([k, v]) => `${k}:${v}`).join(",")}}`
+        ).join(";")}`);
+      }
       prompt += parts.join("\n") + "\n";
     }
     if (req.context) {
+      const importPathRewrites = this.buildImportPathRewrites(req.type, req.target);
       prompt += `
 \u6E90\u7801:
 \`\`\`
-${this.compressSourceCode(req.context)}
+${this.compressSourceCode(req.context, 3e3, importPathRewrites)}
 \`\`\`
 `;
     }
@@ -2964,6 +2977,56 @@ ${this.compressSourceCode(req.context)}
     }
     return `${prefix}${cleanPath}`;
   }
+  /**
+   * 构建 import 路径重写映射
+   * 将源码中可能引用自身的各种写法，映射到测试文件中正确的导入路径
+   */
+  buildImportPathRewrites(testType, targetPath) {
+    const rewrites = /* @__PURE__ */ new Map();
+    if (!targetPath) return rewrites;
+    const correctImportPath = this.computeTestImportPath(testType, targetPath);
+    const variations = /* @__PURE__ */ new Set();
+    variations.add(targetPath);
+    variations.add(targetPath.replace(/^\.\//, ""));
+    const withoutExt = targetPath.replace(/\.(ts|js|tsx|jsx)$/, "");
+    variations.add(withoutExt);
+    const srcDirMatch = targetPath.match(/^(?:\.\/)?(src\/.+)$/);
+    if (srcDirMatch) {
+      variations.add(`@/${srcDirMatch[1]}`);
+      variations.add(`@/${srcDirMatch[1].replace(/\.(ts|js|tsx|jsx)$/, "")}`);
+    }
+    const pathParts = targetPath.replace(/^\.\//, "").split("/");
+    const fileName = pathParts[pathParts.length - 1];
+    const fileNameNoExt = fileName.replace(/\.(ts|js|tsx|jsx|vue)$/, "");
+    variations.add(`./${fileName}`);
+    variations.add(`./${fileNameNoExt}`);
+    for (let i = 1; i < pathParts.length; i++) {
+      const relPath = "../".repeat(i) + pathParts.slice(i).join("/");
+      variations.add(relPath);
+      variations.add(relPath.replace(/\.(ts|js|tsx|jsx)$/, ""));
+    }
+    for (const variant of variations) {
+      if (variant && variant !== correctImportPath) {
+        rewrites.set(variant, correctImportPath);
+      }
+    }
+    return rewrites;
+  }
+  /**
+   * 重写源码中的 import 路径
+   * 将 from 'oldPath' / from "oldPath" 替换为正确的路径
+   */
+  rewriteImportPaths(code, rewrites) {
+    let result = code;
+    for (const [oldPath, newPath] of rewrites) {
+      const escaped = oldPath.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+      const singleQuoteRegex = new RegExp(`(from\\s+')${escaped}(')`, "g");
+      const doubleQuoteRegex = new RegExp(`(from\\s+")${escaped}(")`, "g");
+      result = result.replace(singleQuoteRegex, `$1${newPath}$2`);
+      result = result.replace(doubleQuoteRegex, `$1${newPath}$2`);
+    }
+    return result;
+  }
   parseGenerateTestResponse(content) {
     const codeBlockMatch = content.match(/```(?:typescript|ts|javascript|js)?\s*\n([\s\S]*?)```/);
     const code = codeBlockMatch ? codeBlockMatch[1].trim() : content.replace(/^(?:```[\s\S]*?\n)?/, "").replace(/\n?```$/, "").trim();
@@ -2986,12 +3049,14 @@ SUGGESTIONS:- \u5EFA\u8BAE(\u6BCF\u884C\u4E00\u4E2A)`;
   }
   buildReviewTestUserPrompt(req) {
     const importPath = this.computeTestImportPath(req.testType, req.target);
-    let prompt = `\u5BA1\u67E5\u6D4B\u8BD5\u4EE3\u7801\u3002\u88AB\u6D4B:${req.target} \u7C7B\u578B:${req.testType} import\u8DEF\u5F84:${importPath}
+    let prompt = `\u5BA1\u67E5\u6D4B\u8BD5\u4EE3\u7801\u3002\u88AB\u6D4B:${req.target} \u7C7B\u578B:${req.testType}
+\u3010\u5F3A\u5236\u3011import\u88AB\u6D4B\u6A21\u5757\u5FC5\u987B\u7528: ${importPath}
 `;
+    const importPathRewrites = this.buildImportPathRewrites(req.testType, req.target);
     prompt += `
 \u6E90\u7801:
 \`\`\`
-${this.compressSourceCode(req.sourceCode, 2e3)}
+${this.compressSourceCode(req.sourceCode, 2e3, importPathRewrites)}
 \`\`\`
 `;
     prompt += `
@@ -3010,6 +3075,11 @@ ${req.testCode}
       if (req.analysis.emits?.length) {
         parts.push(`Emits:${req.analysis.emits.map((e) => `${e.name}(${e.params?.join(",") || ""})`).join(",")}`);
       }
+      if (req.analysis.importSignatures?.length) {
+        parts.push(`\u4F9D\u8D56:${req.analysis.importSignatures.map(
+          (imp) => `${imp.source}{${Object.entries(imp.signatures).map(([k, v]) => `${k}:${v}`).join(",")}}`
+        ).join(";")}`);
+      }
       if (parts.length > 0) {
         prompt += `
 \u5206\u6790:${parts.join("|")}`;
@@ -3621,15 +3691,18 @@ var import_node_path10 = __toESM(require("path"), 1);
 function analyzeFile(filePath) {
   const absolutePath = import_node_path10.default.resolve(process.cwd(), filePath);
   if (!import_node_fs9.default.existsSync(absolutePath)) {
-    return { filePath, exports: [], apiCalls: [] };
+    return { filePath, exports: [], imports: [], importSignatures: [], apiCalls: [] };
   }
   const content = import_node_fs9.default.readFileSync(absolutePath, "utf-8");
   const ext = import_node_path10.default.extname(filePath);
   const result = {
     filePath,
     exports: extractExports(content, ext),
+    imports: extractImports(content),
+    importSignatures: [],
     apiCalls: extractAPICalls(content, filePath)
   };
+  result.importSignatures = analyzeImportSignatures(result.imports, import_node_path10.default.dirname(absolutePath));
   if (ext === ".vue") {
     result.vueAnalysis = analyzeVueComponent(content, import_node_path10.default.basename(filePath, ".vue"));
     const scriptMatch = content.match(/<script[^>]*>([\s\S]*?)<\/script>/);
@@ -4052,6 +4125,90 @@ function generateMockRoutesFromAPICalls(apiCalls) {
   }
   return routes;
 }
+function extractImports(content) {
+  const imports = [];
+  const namedImportRegex = /import\s*\{([^}]+)\}\s*from\s*['"]([^'"]+)['"]/g;
+  let match;
+  while ((match = namedImportRegex.exec(content)) !== null) {
+    const names = match[1].split(",").map((n) => {
+      const parts = n.trim().split(/\s+as\s+/);
+      return parts[0].trim();
+    }).filter(Boolean);
+    imports.push({ names, source: match[2], isDefault: false, isNamespace: false });
+  }
+  const defaultImportRegex = /import\s+(\w+)\s+from\s*['"]([^'"]+)['"]/g;
+  while ((match = defaultImportRegex.exec(content)) !== null) {
+    const fullLine = content.slice(Math.max(0, match.index - 5), match.index + match[0].length);
+    if (fullLine.includes("{")) continue;
+    imports.push({ names: [match[1]], source: match[2], isDefault: true, isNamespace: false });
+  }
+  const namespaceImportRegex = /import\s*\*\s*as\s+(\w+)\s*from\s*['"]([^'"]+)['"]/g;
+  while ((match = namespaceImportRegex.exec(content)) !== null) {
+    imports.push({ names: [match[1]], source: match[2], isDefault: false, isNamespace: true });
+  }
+  return imports;
+}
+function analyzeImportSignatures(imports, baseDir) {
+  const signatures = [];
+  for (const imp of imports) {
+    if (!imp.source.startsWith(".") && !imp.source.startsWith("@/") && !imp.source.startsWith("~")) {
+      continue;
+    }
+    const resolvedPath = resolveImportPath2(imp.source, baseDir);
+    if (!resolvedPath || !import_node_fs9.default.existsSync(resolvedPath)) continue;
+    try {
+      const content = import_node_fs9.default.readFileSync(resolvedPath, "utf-8");
+      const ext = import_node_path10.default.extname(resolvedPath);
+      const exports2 = extractExports(content, ext);
+      const sigMap = {};
+      for (const name of imp.names) {
+        const exp = exports2.find((e) => e.name === name);
+        if (exp) {
+          const params = exp.params.length > 0 ? `(${exp.params.join(", ")})` : "";
+          const ret = exp.returnType ? `: ${exp.returnType}` : "";
+          const async = exp.isAsync ? "async " : "";
+          sigMap[name] = `${async}${exp.kind}${params}${ret}`;
+        } else if (imp.isDefault) {
+          const defaultExp = exports2.find((e) => e.kind === "default");
+          if (defaultExp) {
+            sigMap[name] = `default[${defaultExp.name || "anonymous"}]`;
+          }
+        } else if (imp.isNamespace) {
+          sigMap[name] = `{${exports2.map((e) => e.name).join(", ")}}`;
+        }
+      }
+      if (ext === ".vue") {
+        const vueAnalysis = analyzeVueComponent(content, import_node_path10.default.basename(resolvedPath, ".vue"));
+        if (vueAnalysis.props.length > 0) {
+          sigMap["__props__"] = vueAnalysis.props.map((p) => `${p.name}:${p.type}${p.required ? "!" : "?"}`).join(",");
+        }
+      }
+      if (Object.keys(sigMap).length > 0) {
+        signatures.push({ source: imp.source, signatures: sigMap });
+      }
+    } catch {
+    }
+  }
+  return signatures;
+}
+function resolveImportPath2(importSource, baseDir) {
+  let resolved;
+  if (importSource.startsWith("@/") || importSource.startsWith("~")) {
+    const relativePath = importSource.replace(/^[@~]\//, "src/");
+    resolved = import_node_path10.default.resolve(process.cwd(), relativePath);
+  } else if (importSource.startsWith(".")) {
+    resolved = import_node_path10.default.resolve(baseDir, importSource);
+  } else {
+    return null;
+  }
+  const extensions = [".ts", ".tsx", ".js", ".jsx", ".vue", "/index.ts", "/index.js"];
+  if (import_node_fs9.default.existsSync(resolved)) return resolved;
+  for (const ext of extensions) {
+    const withExt = resolved + ext;
+    if (import_node_fs9.default.existsSync(withExt)) return withExt;
+  }
+  return null;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AI_PRESET_PROVIDERS,