@chiway/contextweaver 1.0.0

package/dist/chunk-EZG4H4MN.js

@@ -0,0 +1,286 @@
+import {
+  generateProjectId
+} from "./chunk-B6OWNBOD.js";
+import {
+  logger
+} from "./chunk-AMQQK4P7.js";
+
+// src/mcp/tools/codebaseRetrieval.ts
+import fs from "fs";
+import os from "os";
+import path from "path";
+import { z } from "zod";
+var codebaseRetrievalSchema = z.object({
+  repo_path: z.string().describe(
+    "The absolute file system path to the repository root. (e.g., '/Users/dev/my-project')"
+  ),
+  information_request: z.string().describe(
+    "The SEMANTIC GOAL. Describe the functionality, logic, or behavior you are looking for in full natural language sentences. Focus on 'how it works' rather than exact names. (e.g., 'Trace the execution flow of the login process')"
+  ),
+  technical_terms: z.array(z.string()).optional().describe(
+    "HARD FILTERS. Precise identifiers to narrow down results. Only use symbols KNOWN to exist to avoid false negatives."
+  )
+});
+var BASE_DIR = path.join(os.homedir(), ".contextweaver");
+var INDEX_LOCK_TIMEOUT_MS = 10 * 60 * 1e3;
+async function ensureDefaultEnvFile() {
+  const configDir = BASE_DIR;
+  const envFile = path.join(configDir, ".env");
+  if (fs.existsSync(envFile)) {
+    return;
+  }
+  if (!fs.existsSync(configDir)) {
+    fs.mkdirSync(configDir, { recursive: true });
+    logger.info({ configDir }, "\u521B\u5EFA\u914D\u7F6E\u76EE\u5F55");
+  }
+  const defaultEnvContent = `# ContextWeaver \u793A\u4F8B\u73AF\u5883\u53D8\u91CF\u914D\u7F6E\u6587\u4EF6
+
+# Embedding API \u914D\u7F6E\uFF08\u5FC5\u9700\uFF09
+EMBEDDINGS_API_KEY=your-api-key-here
+EMBEDDINGS_BASE_URL=https://api.siliconflow.cn/v1/embeddings
+EMBEDDINGS_MODEL=BAAI/bge-m3
+EMBEDDINGS_MAX_CONCURRENCY=10
+EMBEDDINGS_DIMENSIONS=1024
+
+# Reranker \u914D\u7F6E\uFF08\u5FC5\u9700\uFF09
+RERANK_API_KEY=your-api-key-here
+RERANK_BASE_URL=https://api.siliconflow.cn/v1/rerank
+RERANK_MODEL=BAAI/bge-reranker-v2-m3
+RERANK_TOP_N=20
+
+# \u7D22\u5F15\u5FFD\u7565\u6A21\u5F0F\uFF08\u53EF\u9009\uFF0C\u9017\u53F7\u5206\u9694\uFF0C\u9ED8\u8BA4\u5DF2\u5305\u542B\u5E38\u89C1\u5FFD\u7565\u9879\uFF09
+# IGNORE_PATTERNS=.venv,node_modules
+`;
+  fs.writeFileSync(envFile, defaultEnvContent);
+  logger.info({ envFile }, "\u5DF2\u521B\u5EFA\u9ED8\u8BA4 .env \u914D\u7F6E\u6587\u4EF6");
+}
+function isProjectIndexed(projectId) {
+  const dbPath = path.join(BASE_DIR, projectId, "index.db");
+  return fs.existsSync(dbPath);
+}
+async function ensureIndexed(repoPath, projectId, onProgress) {
+  const { withLock } = await import("./lock-DVY3KJSK.js");
+  const { scan } = await import("./scanner-SZ2BDYDS.js");
+  await withLock(
+    projectId,
+    "index",
+    async () => {
+      const wasIndexed = isProjectIndexed(projectId);
+      if (!wasIndexed) {
+        logger.info(
+          { repoPath, projectId: projectId.slice(0, 10) },
+          "\u4EE3\u7801\u5E93\u672A\u521D\u59CB\u5316\uFF0C\u5F00\u59CB\u9996\u6B21\u7D22\u5F15..."
+        );
+        onProgress?.(0, 100, "\u4EE3\u7801\u5E93\u672A\u7D22\u5F15\uFF0C\u5F00\u59CB\u9996\u6B21\u7D22\u5F15...");
+      } else {
+        logger.debug({ projectId: projectId.slice(0, 10) }, "\u6267\u884C\u589E\u91CF\u7D22\u5F15...");
+      }
+      const startTime = Date.now();
+      const stats = await scan(repoPath, { vectorIndex: true, onProgress });
+      const elapsed = Date.now() - startTime;
+      logger.info(
+        {
+          projectId: projectId.slice(0, 10),
+          isFirstTime: !wasIndexed,
+          totalFiles: stats.totalFiles,
+          added: stats.added,
+          modified: stats.modified,
+          deleted: stats.deleted,
+          vectorIndex: stats.vectorIndex,
+          elapsedMs: elapsed
+        },
+        "\u7D22\u5F15\u5B8C\u6210"
+      );
+    },
+    INDEX_LOCK_TIMEOUT_MS
+  );
+}
+async function handleCodebaseRetrieval(args, onProgress) {
+  const { repo_path, information_request, technical_terms } = args;
+  logger.info(
+    {
+      repo_path,
+      information_request,
+      technical_terms
+    },
+    "MCP codebase-retrieval \u8C03\u7528\u5F00\u59CB"
+  );
+  const { checkEmbeddingEnv, checkRerankerEnv } = await import("./config-BWZ6CU3W.js");
+  const embeddingCheck = checkEmbeddingEnv();
+  const rerankerCheck = checkRerankerEnv();
+  const allMissingVars = [...embeddingCheck.missingVars, ...rerankerCheck.missingVars];
+  if (allMissingVars.length > 0) {
+    logger.warn({ missingVars: allMissingVars }, "MCP \u73AF\u5883\u53D8\u91CF\u672A\u914D\u7F6E");
+    await ensureDefaultEnvFile();
+    return formatEnvMissingResponse(allMissingVars);
+  }
+  const projectId = generateProjectId(repo_path);
+  await ensureIndexed(repo_path, projectId, onProgress);
+  const query = [information_request, ...technical_terms || []].filter(Boolean).join(" ");
+  logger.info(
+    {
+      projectId: projectId.slice(0, 10),
+      query
+    },
+    "MCP \u67E5\u8BE2\u6784\u5EFA"
+  );
+  const { SearchService } = await import("./SearchService-533KL2HP.js");
+  const service = new SearchService(projectId, repo_path);
+  await service.init();
+  logger.debug("SearchService \u521D\u59CB\u5316\u5B8C\u6210");
+  const contextPack = await service.buildContextPack(query);
+  if (contextPack.seeds.length > 0) {
+    logger.info(
+      {
+        seeds: contextPack.seeds.map((s) => ({
+          file: s.filePath,
+          chunk: s.chunkIndex,
+          score: s.score.toFixed(4),
+          source: s.source
+        }))
+      },
+      "MCP \u641C\u7D22 seeds"
+    );
+  } else {
+    logger.warn("MCP \u641C\u7D22\u65E0 seeds \u547D\u4E2D");
+  }
+  if (contextPack.expanded.length > 0) {
+    logger.debug(
+      {
+        expandedCount: contextPack.expanded.length,
+        expanded: contextPack.expanded.slice(0, 5).map((e) => ({
+          file: e.filePath,
+          chunk: e.chunkIndex,
+          score: e.score.toFixed(4)
+        }))
+      },
+      "MCP \u6269\u5C55\u7ED3\u679C (\u524D5)"
+    );
+  }
+  logger.info(
+    {
+      seedCount: contextPack.seeds.length,
+      expandedCount: contextPack.expanded.length,
+      fileCount: contextPack.files.length,
+      totalSegments: contextPack.files.reduce((acc, f) => acc + f.segments.length, 0),
+      files: contextPack.files.map((f) => ({
+        path: f.filePath,
+        segments: f.segments.length,
+        lines: f.segments.map((s) => `L${s.startLine}-${s.endLine}`)
+      })),
+      timingMs: contextPack.debug?.timingMs
+    },
+    "MCP codebase-retrieval \u5B8C\u6210"
+  );
+  return formatMcpResponse(contextPack);
+}
+function formatMcpResponse(pack) {
+  const { files, seeds } = pack;
+  const fileBlocks = files.map((file) => {
+    const segments = file.segments.map((seg) => formatSegment(seg)).join("\n\n");
+    return segments;
+  }).join("\n\n---\n\n");
+  const summary = [
+    `Found ${seeds.length} relevant code blocks`,
+    `Files: ${files.length}`,
+    `Total segments: ${files.reduce((acc, f) => acc + f.segments.length, 0)}`
+  ].join(" | ");
+  const text = `${summary}
+
+${fileBlocks}`;
+  return {
+    content: [
+      {
+        type: "text",
+        text
+      }
+    ]
+  };
+}
+function formatSegment(seg) {
+  const lang = detectLanguage(seg.filePath);
+  const header = `## ${seg.filePath} (L${seg.startLine}-${seg.endLine})`;
+  const breadcrumb = seg.breadcrumb ? `> ${seg.breadcrumb}` : "";
+  const code = `\`\`\`${lang}
+${seg.text}
+\`\`\``;
+  return [header, breadcrumb, code].filter(Boolean).join("\n");
+}
+function detectLanguage(filePath) {
+  const ext = filePath.split(".").pop()?.toLowerCase() || "";
+  const langMap = {
+    ts: "typescript",
+    tsx: "typescript",
+    js: "javascript",
+    jsx: "javascript",
+    py: "python",
+    rs: "rust",
+    go: "go",
+    java: "java",
+    c: "c",
+    cpp: "cpp",
+    h: "c",
+    hpp: "cpp",
+    cs: "csharp",
+    rb: "ruby",
+    php: "php",
+    swift: "swift",
+    kt: "kotlin",
+    scala: "scala",
+    sql: "sql",
+    sh: "bash",
+    bash: "bash",
+    zsh: "bash",
+    json: "json",
+    yaml: "yaml",
+    yml: "yaml",
+    xml: "xml",
+    html: "html",
+    css: "css",
+    scss: "scss",
+    less: "less",
+    md: "markdown",
+    toml: "toml"
+  };
+  return langMap[ext] || ext || "plaintext";
+}
+function formatEnvMissingResponse(missingVars) {
+  const configPath = "~/.contextweaver/.env";
+  const text = `## \u26A0\uFE0F \u914D\u7F6E\u7F3A\u5931
+
+ContextWeaver \u9700\u8981\u914D\u7F6E Embedding API \u624D\u80FD\u5DE5\u4F5C\u3002
+
+### \u7F3A\u5931\u7684\u73AF\u5883\u53D8\u91CF
+${missingVars.map((v) => `- \`${v}\``).join("\n")}
+
+### \u914D\u7F6E\u6B65\u9AA4
+
+\u5DF2\u81EA\u52A8\u521B\u5EFA\u914D\u7F6E\u6587\u4EF6\uFF1A\`${configPath}\`
+
+\u8BF7\u7F16\u8F91\u8BE5\u6587\u4EF6\uFF0C\u586B\u5199\u4F60\u7684 API Key\uFF1A
+
+\`\`\`bash
+# Embedding API \u914D\u7F6E\uFF08\u5FC5\u9700\uFF09
+EMBEDDINGS_API_KEY=your-api-key-here  # \u2190 \u66FF\u6362\u4E3A\u4F60\u7684 API Key
+
+# Reranker \u914D\u7F6E\uFF08\u5FC5\u9700\uFF09
+RERANK_API_KEY=your-api-key-here      # \u2190 \u66FF\u6362\u4E3A\u4F60\u7684 API Key
+\`\`\`
+
+\u4FDD\u5B58\u6587\u4EF6\u540E\u91CD\u65B0\u8C03\u7528\u6B64\u5DE5\u5177\u5373\u53EF\u3002
+`;
+  return {
+    content: [
+      {
+        type: "text",
+        text
+      }
+    ]
+  };
+}
+
+export {
+  codebaseRetrievalSchema,
+  handleCodebaseRetrieval
+};
+//# sourceMappingURL=chunk-EZG4H4MN.js.map