@chiway/contextweaver 1.1.0 → 1.5.0

package/dist/{chunk-7G5V7YT5.js → chunk-V3K4YVAR.js}

@@ -1,14 +1,15 @@
+import {
+  checkEnvOrRespond,
+  ensureIndexed
+} from "./chunk-VWBKZ6QL.js";
 import {
   generateProjectId
-} from "./chunk-6Z4JEEVJ.js";
+} from "./chunk-PPLFJGO3.js";
 import {
   logger
-} from "./chunk-AMQQK4P7.js";
+} from "./chunk-JVKVSTQ3.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(
@@ -21,80 +22,6 @@ var codebaseRetrievalSchema = z.object({
     "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-RFG4YWYI.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(
@@ -105,17 +32,16 @@ async function handleCodebaseRetrieval(args, onProgress) {
     },
     "MCP codebase-retrieval \u8C03\u7528\u5F00\u59CB"
   );
-  const { checkEmbeddingEnv, checkRerankerEnv } = await import("./config-BWZ6CU3W.js");
+  const { checkEmbeddingEnv, checkRerankerEnv } = await import("./config-LCOJHTCF.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);
+    return await checkEnvOrRespond(allMissingVars);
   }
   const projectId = generateProjectId(repo_path);
-  await ensureIndexed(repo_path, projectId, onProgress);
+  await ensureIndexed(repo_path, projectId, { onProgress });
   const query = [information_request, ...technical_terms || []].filter(Boolean).join(" ");
   logger.info(
     {
@@ -124,8 +50,9 @@ async function handleCodebaseRetrieval(args, onProgress) {
     },
     "MCP \u67E5\u8BE2\u6784\u5EFA"
   );
-  const { SearchService } = await import("./SearchService-MYPOCM3B.js");
-  const service = new SearchService(projectId, repo_path);
+  const { SearchService } = await import("./SearchService-WVD6THR3.js");
+  const { getSearchConfigOverrides } = await import("./loadConfig-XTVT2OWW.js");
+  const service = new SearchService(projectId, repo_path, getSearchConfigOverrides());
   await service.init();
   logger.debug("SearchService \u521D\u59CB\u5316\u5B8C\u6210");
   const contextPack = await service.buildContextPack(query);
@@ -244,43 +171,8 @@ function detectLanguage(filePath) {
   };
   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-7G5V7YT5.js.map

package/dist/chunk-VWBKZ6QL.js

@@ -0,0 +1,115 @@
+import {
+  getDefaultEnvFileContent
+} from "./chunk-TPM6YP43.js";
+import {
+  logger
+} from "./chunk-JVKVSTQ3.js";
+
+// src/mcp/tools/shared.ts
+import fs from "fs";
+import os from "os";
+import path from "path";
+var BASE_DIR = path.join(os.homedir(), ".contextweaver");
+var INDEX_LOCK_TIMEOUT_MS = 10 * 60 * 1e3;
+function isProjectIndexed(projectId) {
+  const dbPath = path.join(BASE_DIR, projectId, "index.db");
+  return fs.existsSync(dbPath);
+}
+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");
+  }
+  fs.writeFileSync(envFile, getDefaultEnvFileContent());
+  logger.info({ envFile }, "\u5DF2\u521B\u5EFA\u9ED8\u8BA4 .env \u914D\u7F6E\u6587\u4EF6");
+}
+async function ensureIndexed(repoPath, projectId, options = {}) {
+  const { onProgress, vectorIndex = true } = options;
+  const { withLock } = await import("./lock-HNKQ6X5B.js");
+  const { scan } = await import("./scanner-QDFZJLP7.js");
+  await withLock(
+    projectId,
+    "index",
+    async () => {
+      const wasIndexed = isProjectIndexed(projectId);
+      if (!wasIndexed) {
+        logger.info(
+          { repoPath, projectId: projectId.slice(0, 10), vectorIndex },
+          "\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), vectorIndex }, "\u6267\u884C\u589E\u91CF\u7D22\u5F15...");
+      }
+      const startTime = Date.now();
+      const stats = await scan(repoPath, { vectorIndex, 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
+  );
+}
+function formatTextResponse(text) {
+  return {
+    content: [
+      {
+        type: "text",
+        text
+      }
+    ]
+  };
+}
+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 formatTextResponse(text);
+}
+async function checkEnvOrRespond(missingVars) {
+  await ensureDefaultEnvFile();
+  return formatEnvMissingResponse(missingVars);
+}
+
+export {
+  ensureIndexed,
+  formatTextResponse,
+  checkEnvOrRespond
+};

package/dist/chunk-XFIM2T6S.js

@@ -0,0 +1,57 @@
+// src/search/ChunkContentLoader.ts
+var ChunkContentLoader = class _ChunkContentLoader {
+  constructor(db) {
+    this.db = db;
+  }
+  /**
+   * 生成 cache key
+   */
+  static key(slice) {
+    return `${slice.filePath}#${slice.start_index}#${slice.end_index}`;
+  }
+  /**
+   * 批量加载 chunk 正文
+   *
+   * @returns Map<key, code>，key 由 ChunkContentLoader.key 生成
+   */
+  loadMany(slices) {
+    const result = /* @__PURE__ */ new Map();
+    if (slices.length === 0) return result;
+    const byPath = /* @__PURE__ */ new Map();
+    for (const s of slices) {
+      let arr = byPath.get(s.filePath);
+      if (!arr) {
+        arr = [];
+        byPath.set(s.filePath, arr);
+      }
+      arr.push(s);
+    }
+    const stmt = this.db.prepare("SELECT content FROM files WHERE path = ?");
+    for (const [path, spans] of byPath) {
+      const row = stmt.get(path);
+      const content = row?.content ?? null;
+      for (const s of spans) {
+        const k = _ChunkContentLoader.key(s);
+        if (content === null) {
+          result.set(k, "");
+          continue;
+        }
+        const safeStart = Math.max(0, Math.min(s.start_index, content.length));
+        const safeEnd = Math.max(safeStart, Math.min(s.end_index, content.length));
+        result.set(k, content.slice(safeStart, safeEnd));
+      }
+    }
+    return result;
+  }
+  /**
+   * 加载单个 chunk 正文（便捷方法，不推荐在批量场景使用）
+   */
+  loadOne(slice) {
+    const map = this.loadMany([slice]);
+    return map.get(_ChunkContentLoader.key(slice)) ?? "";
+  }
+};
+
+export {
+  ChunkContentLoader
+};