@zvec/zvec-grep 0.1.5 → 0.1.6

package/dist/index-progress.js

@@ -0,0 +1,159 @@
+const MCP_INDEX_PROGRESS_PREFIX = "zvec-grep:index-progress:";
+export function formatIndexProgressLine(progress) {
+    if (progress.phase === "scanning") {
+        return progress.detail;
+    }
+    if (progress.phase === "indexing") {
+        const modelDownload = formatModelDownloadProgress(progress);
+        if (modelDownload) {
+            return modelDownload;
+        }
+        const indexed = progress.filesIndexed ?? 0;
+        const total = progress.filesTotal ?? 0;
+        const detail = progress.detail ? ` ${truncate(progress.detail, 100)}` : "";
+        const failed = formatFailedProgress(progress);
+        const embedding = formatEmbeddingProgress(progress);
+        return total > 0
+            ? `Indexing files: ${indexed}/${total}${detail}${failed}${embedding}`
+            : `Indexing files...${detail}${failed}${embedding}`;
+    }
+    return progress.detail ?? "Indexing complete";
+}
+export function formatModelDownloadProgress(progress) {
+    const embedding = progress.embedding;
+    if (!embedding?.model) {
+        return undefined;
+    }
+    if (embedding.stage === "preparing") {
+        return `Preparing ${embedding.model}`;
+    }
+    if (embedding.stage === "warning") {
+        return `zvec-grep warning: ${singleLineText(embedding.message ?? "Embedding model warning")}`;
+    }
+    if (embedding.stage === "ready") {
+        return `Model ready: ${embedding.model}`;
+    }
+    if (embedding.stage !== "downloading") {
+        return undefined;
+    }
+    const parts = [`Downloading ${embedding.model}`];
+    const downloadedBytes = embedding.downloadedBytes;
+    const totalBytes = embedding.totalBytes;
+    if (typeof downloadedBytes === "number" &&
+        typeof totalBytes === "number" &&
+        totalBytes > 0) {
+        const percent = Math.min(100, Math.max(0, Math.round((downloadedBytes / totalBytes) * 100)));
+        parts.push(`${percent}%`);
+        parts.push(`${formatByteCount(downloadedBytes)}/${formatByteCount(totalBytes)}`);
+    }
+    else if (typeof downloadedBytes === "number") {
+        parts.push(formatByteCount(downloadedBytes));
+    }
+    return parts.join(" · ");
+}
+export function indexProgressMessage(progress) {
+    const line = formatIndexProgressLine(progress);
+    return line
+        ? `${MCP_INDEX_PROGRESS_PREFIX}${JSON.stringify({ line, progress })}`
+        : undefined;
+}
+export function indexProgressFromMessage(message) {
+    if (!message?.startsWith(MCP_INDEX_PROGRESS_PREFIX))
+        return undefined;
+    const payload = message.slice(MCP_INDEX_PROGRESS_PREFIX.length);
+    if (payload.startsWith("{")) {
+        try {
+            const decoded = JSON.parse(payload);
+            if (typeof decoded.line === "string" &&
+                isIndexProgress(decoded.progress)) {
+                return {
+                    phase: decoded.progress.phase,
+                    line: decoded.line,
+                    progress: decoded.progress,
+                };
+            }
+        }
+        catch {
+            return undefined;
+        }
+    }
+    // Backward-compatible decoding for progress emitted by an older daemon.
+    const separator = payload.indexOf(":");
+    if (separator < 0)
+        return undefined;
+    const phase = payload.slice(0, separator);
+    if (phase !== "scanning" && phase !== "indexing" && phase !== "done") {
+        return undefined;
+    }
+    const line = payload.slice(separator + 1);
+    return { phase, line, progress: { phase, detail: line } };
+}
+function isIndexProgress(value) {
+    if (!value || typeof value !== "object")
+        return false;
+    const progress = value;
+    return (progress.phase === "scanning" ||
+        progress.phase === "indexing" ||
+        progress.phase === "done");
+}
+function formatFailedProgress(progress) {
+    const failed = progress.filesFailed ?? 0;
+    return failed > 0 ? ` [failed=${failed}]` : "";
+}
+function formatEmbeddingProgress(progress) {
+    const embedding = progress.embedding;
+    if (!embedding) {
+        return "";
+    }
+    const parts = [];
+    if (typeof embedding.concurrency === "number") {
+        parts.push(`concurrency=${embedding.concurrency}`);
+    }
+    if (typeof embedding.retryableFailures === "number" &&
+        embedding.retryableFailures > 0) {
+        parts.push(`retries=${embedding.retryableFailures}`);
+    }
+    return parts.length > 0 ? ` [embed ${parts.join(" ")}]` : "";
+}
+function truncate(value, maxLength) {
+    if (value.length <= maxLength) {
+        return value;
+    }
+    return `${value.slice(0, Math.max(0, maxLength - 1))}...`;
+}
+function formatByteCount(value) {
+    const bytes = Math.max(0, value);
+    const units = ["B", "KiB", "MiB", "GiB"];
+    let amount = bytes;
+    let unitIndex = 0;
+    while (amount >= 1024 && unitIndex < units.length - 1) {
+        amount /= 1024;
+        unitIndex++;
+    }
+    const digits = unitIndex === 0 || amount >= 10 ? 0 : 1;
+    return `${amount.toFixed(digits)} ${units[unitIndex]}`;
+}
+function singleLineText(value) {
+    let result = "";
+    for (let index = 0; index < value.length; index++) {
+        const code = value.charCodeAt(index);
+        if (code === 0x1b && value.charCodeAt(index + 1) === 0x5b) {
+            index += 2;
+            while (index < value.length) {
+                const ansiCode = value.charCodeAt(index);
+                if (ansiCode >= 0x40 && ansiCode <= 0x7e) {
+                    break;
+                }
+                index++;
+            }
+            continue;
+        }
+        if (code < 0x20 || (code >= 0x7f && code <= 0x9f)) {
+            result += " ";
+            continue;
+        }
+        result += value[index];
+    }
+    return result.replace(/\s+/g, " ").trim();
+}
+//# sourceMappingURL=index-progress.js.map

package/dist/index-progress.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index-progress.js","sourceRoot":"","sources":["../src/index-progress.ts"],"names":[],"mappings":"AAEA,MAAM,yBAAyB,GAAG,2BAA2B,CAAC;AAQ9D,MAAM,UAAU,uBAAuB,CACrC,QAAuB;IAEvB,IAAI,QAAQ,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;QAClC,OAAO,QAAQ,CAAC,MAAM,CAAC;IACzB,CAAC;IAED,IAAI,QAAQ,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;QAClC,MAAM,aAAa,GAAG,2BAA2B,CAAC,QAAQ,CAAC,CAAC;QAC5D,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,MAAM,OAAO,GAAG,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC;QAC3C,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,IAAI,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3E,MAAM,MAAM,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;QACpD,OAAO,KAAK,GAAG,CAAC;YACd,CAAC,CAAC,mBAAmB,OAAO,IAAI,KAAK,GAAG,MAAM,GAAG,MAAM,GAAG,SAAS,EAAE;YACrE,CAAC,CAAC,oBAAoB,MAAM,GAAG,MAAM,GAAG,SAAS,EAAE,CAAC;IACxD,CAAC;IAED,OAAO,QAAQ,CAAC,MAAM,IAAI,mBAAmB,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,QAAuB;IAEvB,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC;QACtB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;QACpC,OAAO,aAAa,SAAS,CAAC,KAAK,EAAE,CAAC;IACxC,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAClC,OAAO,sBAAsB,cAAc,CACzC,SAAS,CAAC,OAAO,IAAI,yBAAyB,CAC/C,EAAE,CAAC;IACN,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,KAAK,OAAO,EAAE,CAAC;QAChC,OAAO,gBAAgB,SAAS,CAAC,KAAK,EAAE,CAAC;IAC3C,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,KAAK,aAAa,EAAE,CAAC;QACtC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,eAAe,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC;IACjD,MAAM,eAAe,GAAG,SAAS,CAAC,eAAe,CAAC;IAClD,MAAM,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC;IACxC,IACE,OAAO,eAAe,KAAK,QAAQ;QACnC,OAAO,UAAU,KAAK,QAAQ;QAC9B,UAAU,GAAG,CAAC,EACd,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CACtB,GAAG,EACH,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,eAAe,GAAG,UAAU,CAAC,GAAG,GAAG,CAAC,CAAC,CAC9D,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC;QAC1B,KAAK,CAAC,IAAI,CACR,GAAG,eAAe,CAAC,eAAe,CAAC,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CACrE,CAAC;IACJ,CAAC;SAAM,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;QAC/C,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,QAAuB;IAEvB,MAAM,IAAI,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IAC/C,OAAO,IAAI;QACT,CAAC,CAAC,GAAG,yBAAyB,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE;QACrE,CAAC,CAAC,SAAS,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,OAA2B;IAE3B,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,yBAAyB,CAAC;QAAE,OAAO,SAAS,CAAC;IACtE,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC;IAChE,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAGjC,CAAC;YACF,IACE,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ;gBAChC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,EACjC,CAAC;gBACD,OAAO;oBACL,KAAK,EAAE,OAAO,CAAC,QAAQ,CAAC,KAAK;oBAC7B,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ;iBAC3B,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAED,wEAAwE;IACxE,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACvC,IAAI,SAAS,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IACpC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IAC1C,IAAI,KAAK,KAAK,UAAU,IAAI,KAAK,KAAK,UAAU,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;QACrE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;IAC1C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;AAC5D,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,QAAQ,GAAG,KAAgC,CAAC;IAClD,OAAO,CACL,QAAQ,CAAC,KAAK,KAAK,UAAU;QAC7B,QAAQ,CAAC,KAAK,KAAK,UAAU;QAC7B,QAAQ,CAAC,KAAK,KAAK,MAAM,CAC1B,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,QAAuB;IACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,WAAW,IAAI,CAAC,CAAC;IACzC,OAAO,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;AACjD,CAAC;AAED,SAAS,uBAAuB,CAAC,QAAuB;IACtD,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,OAAO,SAAS,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;QAC9C,KAAK,CAAC,IAAI,CAAC,eAAe,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC;IACrD,CAAC;IAED,IACE,OAAO,SAAS,CAAC,iBAAiB,KAAK,QAAQ;QAC/C,SAAS,CAAC,iBAAiB,GAAG,CAAC,EAC/B,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,WAAW,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;AAC/D,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa,EAAE,SAAiB;IAChD,IAAI,KAAK,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC;AAC5D,CAAC;AAED,SAAS,eAAe,CAAC,KAAa;IACpC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IACzC,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,OAAO,MAAM,IAAI,IAAI,IAAI,SAAS,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtD,MAAM,IAAI,IAAI,CAAC;QACf,SAAS,EAAE,CAAC;IACd,CAAC;IACD,MAAM,MAAM,GAAG,SAAS,KAAK,CAAC,IAAI,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACvD,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,cAAc,CAAC,KAAa;IACnC,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,IAAI,KAAK,IAAI,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YAC1D,KAAK,IAAI,CAAC,CAAC;YACX,OAAO,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBACzC,IAAI,QAAQ,IAAI,IAAI,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;oBACzC,MAAM;gBACR,CAAC;gBACD,KAAK,EAAE,CAAC;YACV,CAAC;YACD,SAAS;QACX,CAAC;QACD,IAAI,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,IAAI,GAAG,CAAC;YACd,SAAS;QACX,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AAC5C,CAAC"}

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 export { createZvecGrep } from "./engine/service/index.js";
-export type { CreateZvecGrepOptions, ZvecGrep, ZvecGrepCollectionIndexOptions, ZvecGrepCollections, ZvecGrepContent, ZvecGrepContextCollection, ZvecGrepContextContainer, ZvecGrepContextCoverage, ZvecGrepContextDiagnostics, ZvecGrepContextFile, ZvecGrepContextItem, ZvecGrepContextItemKind, ZvecGrepContextOptions, ZvecGrepContextResult, ZvecGrepContextRoute, ZvecGrepContextSource, ZvecGrepStructureEnrichmentDiagnostics, ZvecGrepIndexDiagnostics, ZvecGrepInfoOptions, ZvecGrepInfoResult, ZvecGrepIndexOptions, ZvecGrepRgDiagnostics, ZvecGrepSearchOptions, } from "./engine/service/index.js";
-export { EMBEDDING_MODEL_CATALOG, MODEL_CATALOG, createEmbeddingModel, createEmbeddingModelFromCatalog, createEmbeddingModelFromReference, EmbeddingModel, getEmbeddingModelCatalogEntry, getEmbeddingModelCatalogEntryByRef, listEmbeddingModels, } from "./engine/models/index.js";
+export type { CreateZvecGrepOptions, ZvecGrep, ZvecGrepContent, ZvecGrepContextWorkspaceIndex, ZvecGrepContextContainer, ZvecGrepContextCoverage, ZvecGrepContextDiagnostics, ZvecGrepContextFile, ZvecGrepContextItem, ZvecGrepContextItemKind, ZvecGrepContextOptions, ZvecGrepContextResult, ZvecGrepContextRoute, ZvecGrepContextSource, ZvecGrepStructureEnrichmentDiagnostics, ZvecGrepIndexDiagnostics, ZvecGrepInfoOptions, ZvecGrepInfoResult, ZvecGrepIndexOptions, ZvecGrepRgDiagnostics, ZvecGrepSearchOptions, } from "./engine/service/index.js";
+export { createEmbeddingModel, EmbeddingPurpose, } from "./engine/models/index.js";
 export { isImageContent, isTextContent } from "./engine/types.js";
-export type { CodeEntityMetadata, CodeEntityModifier, CodeSymbolType, CollectionEmbeddingSchema, CollectionIndexPolicy, CollectionIndexStatus, CollectionInfo, Content, ContentKind, EntityMetadata, FileFormat, FileIndexStatus, FileInfo, FileKind, ImageContent, ImageFormat, IndexOptions, IndexProgress, IndexProgressPhase, IndexResult, MarkdownEntityMetadata, RootPath, SearchFinalTrace, SearchHitTrace, SearchRecallTrace, SearchStageTrace, TextContent, TimingEntry, } from "./engine/types.js";
-export type { EmbeddingCatalogEntry, LlamaCppEmbeddingCatalogEntry, LlamaGpuMode, LocalEmbeddingFormat, ModelProviderOptions, } from "./engine/models/index.js";
+export type { CodeEntityMetadata, CodeEntityModifier, CodeSymbolType, WorkspaceIndexEmbeddingSchema, WorkspaceIndexPolicy, WorkspaceIndexStatus, WorkspaceIndexInfo, Content, ContentKind, EntityMetadata, FileFormat, FileIndexStatus, FileInfo, FileKind, ImageContent, ImageFormat, IndexOptions, IndexProgress, IndexProgressPhase, IndexResult, MarkdownEntityMetadata, RootPath, SearchFinalTrace, SearchHitTrace, SearchRecallTrace, SearchStageTrace, TextContent, TimingEntry, } from "./engine/types.js";
+export type { CreateEmbeddingModelOptions, EmbeddingModel, EmbeddingModelProgress, EmbeddingModelInfo, EmbeddingOptions, EmbeddingResult, } from "./engine/models/index.js";

package/dist/index.js

@@ -1,4 +1,4 @@
 export { createZvecGrep } from "./engine/service/index.js";
-export { EMBEDDING_MODEL_CATALOG, MODEL_CATALOG, createEmbeddingModel, createEmbeddingModelFromCatalog, createEmbeddingModelFromReference, EmbeddingModel, getEmbeddingModelCatalogEntry, getEmbeddingModelCatalogEntryByRef, listEmbeddingModels, } from "./engine/models/index.js";
+export { createEmbeddingModel, EmbeddingPurpose, } from "./engine/models/index.js";
 export { isImageContent, isTextContent } from "./engine/types.js";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AA0B3D,OAAO,EACL,uBAAuB,EACvB,aAAa,EACb,oBAAoB,EACpB,+BAA+B,EAC/B,iCAAiC,EACjC,cAAc,EACd,6BAA6B,EAC7B,kCAAkC,EAClC,mBAAmB,GACpB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAwB3D,OAAO,EACL,oBAAoB,EACpB,gBAAgB,GACjB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/mcp/http-transport.d.ts

@@ -0,0 +1,27 @@
+import type { IncomingMessage, ServerResponse } from "node:http";
+import type { ZvecGrepDaemonBackend, ZvecGrepMcpServerOptions } from "./tools.js";
+export type McpHttpEndpointOptions = {
+    legacySessionIdleTtlMs?: number;
+    maxLegacySessions?: number;
+    modernOnly?: boolean;
+};
+export declare class McpHttpEndpoint {
+    private readonly backend;
+    private readonly version;
+    private readonly sessions;
+    private readonly initializing;
+    private readonly serverOptions;
+    private readonly modern;
+    private readonly modernNodeHandler;
+    private readonly legacySessionIdleTtlMs;
+    private readonly maxLegacySessions;
+    private readonly modernOnly;
+    constructor(backend: ZvecGrepDaemonBackend, version: string, serverOptions?: Readonly<ZvecGrepMcpServerOptions>, endpointOptions?: Readonly<McpHttpEndpointOptions>);
+    handlePost(request: IncomingMessage, response: ServerResponse, body: unknown): Promise<void>;
+    handleSessionRequest(request: IncomingMessage, response: ServerResponse): Promise<void>;
+    close(): Promise<void>;
+    private handleLegacyPost;
+    private expireIdleSessions;
+    private withActiveSession;
+    private forget;
+}

package/dist/mcp/http-transport.js

@@ -0,0 +1,182 @@
+import { randomUUID } from "node:crypto";
+import { createMcpHandler, isInitializeRequest, isLegacyRequest, } from "@modelcontextprotocol/server";
+import { NodeStreamableHTTPServerTransport, toNodeHandler, toWebRequest, } from "@modelcontextprotocol/node";
+import { createZvecGrepMcpServer } from "./tools.js";
+import { InMemoryRemoteEmbeddingRequestStateReplayGuard } from "./request-state.js";
+const LEGACY_SESSION_IDLE_TTL_MS = 30 * 60 * 1_000;
+const MAX_LEGACY_SESSIONS = 256;
+export class McpHttpEndpoint {
+    backend;
+    version;
+    sessions = new Map();
+    initializing = new Set();
+    serverOptions;
+    modern;
+    modernNodeHandler;
+    legacySessionIdleTtlMs;
+    maxLegacySessions;
+    modernOnly;
+    constructor(backend, version, serverOptions = {}, endpointOptions = {}) {
+        this.backend = backend;
+        this.version = version;
+        this.serverOptions = Object.freeze({
+            ...serverOptions,
+            requestStateReplayGuard: serverOptions.requestStateReplayGuard ??
+                new InMemoryRemoteEmbeddingRequestStateReplayGuard(),
+        });
+        this.legacySessionIdleTtlMs =
+            endpointOptions.legacySessionIdleTtlMs ?? LEGACY_SESSION_IDLE_TTL_MS;
+        this.maxLegacySessions =
+            endpointOptions.maxLegacySessions ?? MAX_LEGACY_SESSIONS;
+        this.modernOnly = endpointOptions.modernOnly ?? false;
+        if (!Number.isFinite(this.legacySessionIdleTtlMs) ||
+            this.legacySessionIdleTtlMs <= 0 ||
+            !Number.isInteger(this.maxLegacySessions) ||
+            this.maxLegacySessions <= 0) {
+            throw new RangeError("Legacy MCP session limits must be positive.");
+        }
+        this.modern = createMcpHandler(() => createZvecGrepMcpServer(this.backend, this.version, this.serverOptions), { legacy: "reject" });
+        this.modernNodeHandler = toNodeHandler(this.modern);
+    }
+    async handlePost(request, response, body) {
+        if (this.modernOnly) {
+            await this.modernNodeHandler(request, response, body);
+            return;
+        }
+        const webRequest = await toWebRequest(request, body);
+        if (!(await isLegacyRequest(webRequest, body))) {
+            await this.modernNodeHandler(request, response, body);
+            return;
+        }
+        await this.handleLegacyPost(request, response, body);
+    }
+    async handleSessionRequest(request, response) {
+        if (this.modernOnly) {
+            await this.modernNodeHandler(request, response);
+            return;
+        }
+        this.expireIdleSessions();
+        const sessionId = requestSessionId(request);
+        if (!sessionId) {
+            response.statusCode = request.method === "GET" ? 405 : 400;
+            response.end();
+            return;
+        }
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            writeMcpError(response, 404, "Unknown or expired MCP session.");
+            return;
+        }
+        await this.withActiveSession(session, () => session.transport.handleRequest(request, response));
+    }
+    async close() {
+        const sessions = new Set([
+            ...this.sessions.values(),
+            ...this.initializing.values(),
+        ]);
+        this.sessions.clear();
+        this.initializing.clear();
+        await Promise.all([
+            this.modern.close(),
+            ...[...sessions].map((session) => session.server.close().catch(() => undefined)),
+        ]);
+    }
+    async handleLegacyPost(request, response, body) {
+        this.expireIdleSessions();
+        const sessionId = requestSessionId(request);
+        if (sessionId) {
+            const session = this.sessions.get(sessionId);
+            if (!session) {
+                writeMcpError(response, 404, "Unknown or expired MCP session.");
+                return;
+            }
+            await this.withActiveSession(session, () => session.transport.handleRequest(request, response, body));
+            return;
+        }
+        if (!isInitializeRequest(body)) {
+            writeMcpError(response, 400, "An MCP initialize request is required before other requests.");
+            return;
+        }
+        if (this.sessions.size + this.initializing.size >= this.maxLegacySessions) {
+            writeMcpError(response, 503, "MCP legacy session limit reached.");
+            return;
+        }
+        const holder = {};
+        const transport = new NodeStreamableHTTPServerTransport({
+            sessionIdGenerator: () => randomUUID(),
+            onsessioninitialized: (id) => {
+                const session = holder.session;
+                if (!session)
+                    return;
+                session.id = id;
+                session.lastAccessedAt = Date.now();
+                this.sessions.set(id, session);
+            },
+        });
+        const server = createZvecGrepMcpServer(this.backend, this.version, this.serverOptions);
+        const session = {
+            activeRequests: 0,
+            server,
+            transport,
+            lastAccessedAt: Date.now(),
+        };
+        holder.session = session;
+        this.initializing.add(session);
+        transport.onclose = () => this.forget(session);
+        try {
+            await server.connect(transport);
+            await transport.handleRequest(request, response, body);
+        }
+        catch (error) {
+            if (!session.id) {
+                await server.close().catch(() => undefined);
+            }
+            throw error;
+        }
+        finally {
+            this.initializing.delete(session);
+        }
+    }
+    expireIdleSessions(now = Date.now()) {
+        for (const session of this.sessions.values()) {
+            if (session.activeRequests > 0 ||
+                now - session.lastAccessedAt <= this.legacySessionIdleTtlMs) {
+                continue;
+            }
+            this.forget(session);
+            void session.server.close().catch(() => undefined);
+        }
+    }
+    async withActiveSession(session, operation) {
+        session.activeRequests += 1;
+        session.lastAccessedAt = Date.now();
+        try {
+            await operation();
+        }
+        finally {
+            session.activeRequests -= 1;
+            session.lastAccessedAt = Date.now();
+        }
+    }
+    forget(session) {
+        if (session.id && this.sessions.get(session.id) === session) {
+            this.sessions.delete(session.id);
+        }
+        this.initializing.delete(session);
+    }
+}
+function requestSessionId(request) {
+    const value = request.headers["mcp-session-id"];
+    return typeof value === "string" && value.length > 0 ? value : undefined;
+}
+function writeMcpError(response, status, message) {
+    response.statusCode = status;
+    response.setHeader("Content-Type", "application/json");
+    response.setHeader("Cache-Control", "no-store");
+    response.end(JSON.stringify({
+        jsonrpc: "2.0",
+        error: { code: -32000, message },
+        id: null,
+    }));
+}
+//# sourceMappingURL=http-transport.js.map

package/dist/mcp/http-transport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-transport.js","sourceRoot":"","sources":["../../src/mcp/http-transport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EACL,gBAAgB,EAChB,mBAAmB,EACnB,eAAe,GAChB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,iCAAiC,EACjC,aAAa,EACb,YAAY,GACb,MAAM,4BAA4B,CAAC;AAKpC,OAAO,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AACrD,OAAO,EAAE,8CAA8C,EAAE,MAAM,oBAAoB,CAAC;AAEpF,MAAM,0BAA0B,GAAG,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC;AACnD,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAgBhC,MAAM,OAAO,eAAe;IAWP;IACA;IAXF,QAAQ,GAAG,IAAI,GAAG,EAAsB,CAAC;IACzC,YAAY,GAAG,IAAI,GAAG,EAAc,CAAC;IACrC,aAAa,CAAqC;IAClD,MAAM,CAAC;IACP,iBAAiB,CAAC;IAClB,sBAAsB,CAAS;IAC/B,iBAAiB,CAAS;IAC1B,UAAU,CAAU;IAErC,YACmB,OAA8B,EAC9B,OAAe,EAChC,gBAAoD,EAAE,EACtD,kBAAoD,EAAE;QAHrC,YAAO,GAAP,OAAO,CAAuB;QAC9B,YAAO,GAAP,OAAO,CAAQ;QAIhC,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC;YACjC,GAAG,aAAa;YAChB,uBAAuB,EACrB,aAAa,CAAC,uBAAuB;gBACrC,IAAI,8CAA8C,EAAE;SACvD,CAAC,CAAC;QACH,IAAI,CAAC,sBAAsB;YACzB,eAAe,CAAC,sBAAsB,IAAI,0BAA0B,CAAC;QACvE,IAAI,CAAC,iBAAiB;YACpB,eAAe,CAAC,iBAAiB,IAAI,mBAAmB,CAAC;QAC3D,IAAI,CAAC,UAAU,GAAG,eAAe,CAAC,UAAU,IAAI,KAAK,CAAC;QACtD,IACE,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,sBAAsB,CAAC;YAC7C,IAAI,CAAC,sBAAsB,IAAI,CAAC;YAChC,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAC;YACzC,IAAI,CAAC,iBAAiB,IAAI,CAAC,EAC3B,CAAC;YACD,MAAM,IAAI,UAAU,CAAC,6CAA6C,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAC5B,GAAG,EAAE,CACH,uBAAuB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,EACzE,EAAE,MAAM,EAAE,QAAQ,EAAE,CACrB,CAAC;QACF,IAAI,CAAC,iBAAiB,GAAG,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAAwB,EACxB,QAAwB,EACxB,IAAa;QAEb,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;YACtD,OAAO;QACT,CAAC;QACD,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,CAAC,MAAM,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC;YAC/C,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;YACtD,OAAO;QACT,CAAC;QACD,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,OAAwB,EACxB,QAAwB;QAExB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,MAAM,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,QAAQ,CAAC,UAAU,GAAG,OAAO,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;YAC3D,QAAQ,CAAC,GAAG,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,aAAa,CAAC,QAAQ,EAAE,GAAG,EAAE,iCAAiC,CAAC,CAAC;YAChE,OAAO;QACT,CAAC;QACD,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,GAAG,EAAE,CACzC,OAAO,CAAC,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,CACnD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC;YACvB,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE;YACzB,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;SAC9B,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1B,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE;YACnB,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAC/B,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAC9C;SACF,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,OAAwB,EACxB,QAAwB,EACxB,IAAa;QAEb,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,MAAM,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,aAAa,CAAC,QAAQ,EAAE,GAAG,EAAE,iCAAiC,CAAC,CAAC;gBAChE,OAAO;YACT,CAAC;YACD,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,GAAG,EAAE,CACzC,OAAO,CAAC,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,CACzD,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/B,aAAa,CACX,QAAQ,EACR,GAAG,EACH,8DAA8D,CAC/D,CAAC;YACF,OAAO;QACT,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC1E,aAAa,CAAC,QAAQ,EAAE,GAAG,EAAE,mCAAmC,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAA6B,EAAE,CAAC;QAC5C,MAAM,SAAS,GAAG,IAAI,iCAAiC,CAAC;YACtD,kBAAkB,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE;YACtC,oBAAoB,EAAE,CAAC,EAAE,EAAE,EAAE;gBAC3B,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;gBAC/B,IAAI,CAAC,OAAO;oBAAE,OAAO;gBACrB,OAAO,CAAC,EAAE,GAAG,EAAE,CAAC;gBAChB,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACpC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACjC,CAAC;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,uBAAuB,CACpC,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,aAAa,CACnB,CAAC;QACF,MAAM,OAAO,GAAe;YAC1B,cAAc,EAAE,CAAC;YACjB,MAAM;YACN,SAAS;YACT,cAAc,EAAE,IAAI,CAAC,GAAG,EAAE;SAC3B,CAAC;QACF,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC/B,SAAS,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE/C,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAChC,MAAM,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;gBAChB,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC9C,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;QACzC,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IACE,OAAO,CAAC,cAAc,GAAG,CAAC;gBAC1B,GAAG,GAAG,OAAO,CAAC,cAAc,IAAI,IAAI,CAAC,sBAAsB,EAC3D,CAAC;gBACD,SAAS;YACX,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACrB,KAAK,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAC7B,OAAmB,EACnB,SAA8B;QAE9B,OAAO,CAAC,cAAc,IAAI,CAAC,CAAC;QAC5B,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,SAAS,EAAE,CAAC;QACpB,CAAC;gBAAS,CAAC;YACT,OAAO,CAAC,cAAc,IAAI,CAAC,CAAC;YAC5B,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtC,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,OAAmB;QAChC,IAAI,OAAO,CAAC,EAAE,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,OAAO,EAAE,CAAC;YAC5D,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,OAAwB;IAChD,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAChD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAC3E,CAAC;AAED,SAAS,aAAa,CACpB,QAAwB,EACxB,MAAc,EACd,OAAe;IAEf,QAAQ,CAAC,UAAU,GAAG,MAAM,CAAC;IAC7B,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;IACvD,QAAQ,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;IAChD,QAAQ,CAAC,GAAG,CACV,IAAI,CAAC,SAAS,CAAC;QACb,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE;QAChC,EAAE,EAAE,IAAI;KACT,CAAC,CACH,CAAC;AACJ,CAAC"}

package/dist/mcp/input-normalization.d.ts

@@ -0,0 +1,38 @@
+import type { CodeSymbolType, ZvecGrepContextOptions } from "../index.js";
+import type { StringListInput, TimeInput, ZvecGrepRgInput, ZvecGrepSearchInput } from "./schemas.js";
+export type NormalizedSearchInput = {
+    root: string;
+    apiKey?: string;
+    device?: "auto" | "cpu" | "metal" | "vulkan" | "cuda";
+    queries?: string[];
+    routes: Array<{
+        mode: "fts" | "vector";
+        query: string;
+    }>;
+    fuse?: boolean;
+    limit?: number;
+    freshness: "eventual" | "wait_for_fresh";
+    autoUpdate: boolean;
+    preferSymbol?: boolean;
+    symbolTypes?: CodeSymbolType[];
+    globs?: string[];
+    insensitiveGlobs?: string[];
+    fileTypes?: string[];
+    excludedFileTypes?: string[];
+    hidden?: boolean;
+    noIgnore?: boolean;
+    ignoreFiles?: string[];
+    maxDepth?: number;
+    maxFileSizeBytes?: number;
+    follow?: boolean;
+    embeddingConcurrency?: number;
+    modifiedAfter?: number;
+    modifiedBefore?: number;
+    trace?: boolean;
+};
+export declare function normalizeSearchInput(input: ZvecGrepSearchInput): NormalizedSearchInput;
+export declare function contextOptionsFromRgInput(input: ZvecGrepRgInput): ZvecGrepContextOptions;
+export declare function normalizeQueryList(value: StringListInput): string[];
+export declare function normalizePlainStringList(value: StringListInput): string[] | undefined;
+export declare function normalizePathFilters(value: StringListInput): string[];
+export declare function normalizeModifiedTime(value: TimeInput, option: string): number | undefined;

package/dist/mcp/input-normalization.js

@@ -0,0 +1,95 @@
+import { parseModifiedTime, splitPathFilters } from "../cli/args.js";
+import { parseManagedRgCommand } from "../cli/managed-rg.js";
+export function normalizeSearchInput(input) {
+    const common = normalizeSearchFields(input);
+    return {
+        root: input.root,
+        apiKey: input.apiKey,
+        device: input.device,
+        ...common,
+        freshness: input.freshness,
+        autoUpdate: input.autoUpdate,
+    };
+}
+export function contextOptionsFromRgInput(input) {
+    const { queries, options } = parseManagedRgCommand(input.root, input.command);
+    return {
+        queries: queries.length > 0 ? queries : undefined,
+        rg: true,
+        rgOptions: options.rgOptions,
+        rgPaths: options.rgPaths,
+        root: input.root,
+        limit: options.limit,
+        globs: options.globs,
+        insensitiveGlobs: options.insensitiveGlobs,
+        fileTypes: options.fileTypes,
+        excludedFileTypes: options.excludedFileTypes,
+        hidden: options.hidden,
+        noIgnore: options.noIgnore,
+        ignoreFiles: options.ignoreFiles,
+        maxDepth: options.maxDepth,
+        maxFileSizeBytes: options.maxFileSizeBytes,
+    };
+}
+function normalizeSearchFields(input) {
+    const queries = [
+        ...normalizeQueryList(input.query),
+        ...normalizeQueryList(input.queries),
+    ];
+    const fts = normalizeQueryList(input.fts);
+    const vector = normalizeQueryList(input.vector);
+    if (queries.length === 0 && fts.length === 0 && vector.length === 0) {
+        throw new Error("zvec_grep_search requires query, queries, fts, or vector.");
+    }
+    return {
+        queries: queries.length > 0 ? queries : undefined,
+        routes: [
+            ...fts.map((query) => ({ mode: "fts", query })),
+            ...vector.map((query) => ({ mode: "vector", query })),
+        ],
+        fuse: input.fuse,
+        limit: input.limit,
+        trace: input.trace,
+        preferSymbol: input.preferSymbol,
+        symbolTypes: input.symbolTypes.length > 0 ? input.symbolTypes : undefined,
+        globs: normalizePlainStringList(input.globs),
+        insensitiveGlobs: normalizePlainStringList(input.insensitiveGlobs),
+        fileTypes: normalizePlainStringList(input.fileTypes),
+        excludedFileTypes: normalizePlainStringList(input.excludedFileTypes),
+        hidden: input.hidden,
+        noIgnore: input.noIgnore,
+        ignoreFiles: normalizePlainStringList(input.ignoreFiles),
+        maxDepth: input.maxDepth,
+        maxFileSizeBytes: input.maxFileSizeBytes,
+        follow: input.follow,
+        embeddingConcurrency: input.embeddingConcurrency,
+        modifiedAfter: normalizeModifiedTime(input.modifiedAfter, "modifiedAfter"),
+        modifiedBefore: normalizeModifiedTime(input.modifiedBefore, "modifiedBefore"),
+    };
+}
+export function normalizeQueryList(value) {
+    return normalizePlainStringList(value) ?? [];
+}
+export function normalizePlainStringList(value) {
+    const items = value === undefined ? [] : Array.isArray(value) ? value : [value];
+    const normalized = items
+        .map((item) => item.trim())
+        .filter((item) => item.length > 0);
+    return normalized.length > 0 ? normalized : undefined;
+}
+export function normalizePathFilters(value) {
+    if (value === undefined) {
+        return [];
+    }
+    if (Array.isArray(value)) {
+        return value.map((item) => item.trim()).filter((item) => item.length > 0);
+    }
+    return splitPathFilters(value);
+}
+export function normalizeModifiedTime(value, option) {
+    if (value === undefined) {
+        return undefined;
+    }
+    return typeof value === "number" ? value : parseModifiedTime(value, option);
+}
+//# sourceMappingURL=input-normalization.js.map

package/dist/mcp/input-normalization.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"input-normalization.js","sourceRoot":"","sources":["../../src/mcp/input-normalization.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AACrE,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAoC7D,MAAM,UAAU,oBAAoB,CAClC,KAA0B;IAE1B,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;IAC5C,OAAO;QACL,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,GAAG,MAAM;QACT,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;KAC7B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,KAAsB;IAEtB,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,qBAAqB,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAC9E,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QACjD,EAAE,EAAE,IAAI;QACR,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;QAC1C,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,iBAAiB,EAAE,OAAO,CAAC,iBAAiB;QAC5C,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;KAC3C,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAC5B,KAwBC;IAED,MAAM,OAAO,GAAG;QACd,GAAG,kBAAkB,CAAC,KAAK,CAAC,KAAK,CAAC;QAClC,GAAG,kBAAkB,CAAC,KAAK,CAAC,OAAO,CAAC;KACrC,CAAC;IACF,MAAM,GAAG,GAAG,kBAAkB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAChD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACpE,MAAM,IAAI,KAAK,CACb,2DAA2D,CAC5D,CAAC;IACJ,CAAC;IAED,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QACjD,MAAM,EAAE;YACN,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAc,EAAE,KAAK,EAAE,CAAC,CAAC;YACxD,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;SAC/D;QACD,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,YAAY,EAAE,KAAK,CAAC,YAAY;QAChC,WAAW,EAAE,KAAK,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;QACzE,KAAK,EAAE,wBAAwB,CAAC,KAAK,CAAC,KAAK,CAAC;QAC5C,gBAAgB,EAAE,wBAAwB,CAAC,KAAK,CAAC,gBAAgB,CAAC;QAClE,SAAS,EAAE,wBAAwB,CAAC,KAAK,CAAC,SAAS,CAAC;QACpD,iBAAiB,EAAE,wBAAwB,CAAC,KAAK,CAAC,iBAAiB,CAAC;QACpE,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,wBAAwB,CAAC,KAAK,CAAC,WAAW,CAAC;QACxD,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;QACxC,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,oBAAoB,EAAE,KAAK,CAAC,oBAAoB;QAChD,aAAa,EAAE,qBAAqB,CAAC,KAAK,CAAC,aAAa,EAAE,eAAe,CAAC;QAC1E,cAAc,EAAE,qBAAqB,CACnC,KAAK,CAAC,cAAc,EACpB,gBAAgB,CACjB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,KAAsB;IACvD,OAAO,wBAAwB,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;AAC/C,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,KAAsB;IAEtB,MAAM,KAAK,GACT,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IACpE,MAAM,UAAU,GAAG,KAAK;SACrB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACrC,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,KAAsB;IACzD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC5E,CAAC;IACD,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,KAAgB,EAChB,MAAc;IAEd,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;AAC9E,CAAC"}

package/dist/mcp/progress-heartbeat.d.ts

@@ -0,0 +1,17 @@
+import type { ProgressNotification } from "@modelcontextprotocol/server";
+export declare const REMOTE_AUTHORIZATION_HEARTBEAT_MS = 15000;
+export declare const LONG_RUNNING_MCP_TIMEOUT_MS = 2147483647;
+type ProgressHeartbeatContext = {
+    mcpReq: {
+        _meta?: {
+            progressToken?: string | number;
+        };
+        signal: AbortSignal;
+        notify: (notification: ProgressNotification) => Promise<void>;
+    };
+};
+export declare function withProgressHeartbeat<T>(context: ProgressHeartbeatContext, operation: () => Promise<T>, options: {
+    intervalMs?: number;
+    message: string;
+}): Promise<T>;
+export {};

package/dist/mcp/progress-heartbeat.js

@@ -0,0 +1,46 @@
+export const REMOTE_AUTHORIZATION_HEARTBEAT_MS = 15_000;
+// The MCP SDK does not currently expose a disabled-timeout mode. Keep a
+// practical upper bound as a safety net while progress heartbeats make the
+// authorization wait open-ended.
+export const LONG_RUNNING_MCP_TIMEOUT_MS = 2_147_483_647;
+export async function withProgressHeartbeat(context, operation, options) {
+    const progressToken = context.mcpReq._meta?.progressToken;
+    if (progressToken === undefined) {
+        return await operation();
+    }
+    const intervalMs = options.intervalMs ?? REMOTE_AUTHORIZATION_HEARTBEAT_MS;
+    let progress = Date.now();
+    let inFlight;
+    const sendHeartbeat = async () => {
+        if (context.mcpReq.signal.aborted || inFlight)
+            return;
+        progress += 1;
+        inFlight = context.mcpReq
+            .notify({
+            method: "notifications/progress",
+            params: {
+                progressToken,
+                progress,
+                message: options.message,
+            },
+        })
+            .catch(() => undefined)
+            .finally(() => {
+            inFlight = undefined;
+        });
+        await inFlight;
+    };
+    await sendHeartbeat();
+    const timer = setInterval(() => {
+        void sendHeartbeat();
+    }, intervalMs);
+    timer.unref();
+    try {
+        return await operation();
+    }
+    finally {
+        clearInterval(timer);
+        await inFlight?.catch(() => undefined);
+    }
+}
+//# sourceMappingURL=progress-heartbeat.js.map

package/dist/mcp/progress-heartbeat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"progress-heartbeat.js","sourceRoot":"","sources":["../../src/mcp/progress-heartbeat.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,iCAAiC,GAAG,MAAM,CAAC;AAExD,wEAAwE;AACxE,2EAA2E;AAC3E,iCAAiC;AACjC,MAAM,CAAC,MAAM,2BAA2B,GAAG,aAAa,CAAC;AAUzD,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,OAAiC,EACjC,SAA2B,EAC3B,OAGC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,aAAa,CAAC;IAC1D,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QAChC,OAAO,MAAM,SAAS,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,iCAAiC,CAAC;IAC3E,IAAI,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC1B,IAAI,QAAmC,CAAC;IAExC,MAAM,aAAa,GAAG,KAAK,IAAmB,EAAE;QAC9C,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,IAAI,QAAQ;YAAE,OAAO;QACtD,QAAQ,IAAI,CAAC,CAAC;QACd,QAAQ,GAAG,OAAO,CAAC,MAAM;aACtB,MAAM,CAAC;YACN,MAAM,EAAE,wBAAwB;YAChC,MAAM,EAAE;gBACN,aAAa;gBACb,QAAQ;gBACR,OAAO,EAAE,OAAO,CAAC,OAAO;aACzB;SACF,CAAC;aACD,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACtB,OAAO,CAAC,GAAG,EAAE;YACZ,QAAQ,GAAG,SAAS,CAAC;QACvB,CAAC,CAAC,CAAC;QACL,MAAM,QAAQ,CAAC;IACjB,CAAC,CAAC;IAEF,MAAM,aAAa,EAAE,CAAC;IACtB,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,KAAK,aAAa,EAAE,CAAC;IACvB,CAAC,EAAE,UAAU,CAAC,CAAC;IACf,KAAK,CAAC,KAAK,EAAE,CAAC;IAEd,IAAI,CAAC;QACH,OAAO,MAAM,SAAS,EAAE,CAAC;IAC3B,CAAC;YAAS,CAAC;QACT,aAAa,CAAC,KAAK,CAAC,CAAC;QACrB,MAAM,QAAQ,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;AACH,CAAC"}

package/dist/mcp/request-metadata.d.ts

@@ -0,0 +1,2 @@
+export declare const EMBEDDING_ENVIRONMENT_META_KEY = "io.zvec-grep/embedding-environment";
+export declare function embeddingEnvironmentFromRequestMeta(meta: unknown): string | undefined;

package/dist/mcp/request-metadata.js

@@ -0,0 +1,11 @@
+export const EMBEDDING_ENVIRONMENT_META_KEY = "io.zvec-grep/embedding-environment";
+export function embeddingEnvironmentFromRequestMeta(meta) {
+    if (typeof meta !== "object" || meta === null)
+        return undefined;
+    const value = meta[EMBEDDING_ENVIRONMENT_META_KEY];
+    if (typeof value !== "string")
+        return undefined;
+    const normalized = value.trim();
+    return normalized.length > 0 ? normalized : undefined;
+}
+//# sourceMappingURL=request-metadata.js.map