@angri450/zvec-grep 0.2.2

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, ZvecGrepCliSearchInput, 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 | ZvecGrepCliSearchInput): 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,104 @@
+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: "apiKey" in input ? input.apiKey : undefined,
+        device: "device" in input ? input.device : undefined,
+        ...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);
+    const orderedRoutes = (input.routes ?? []).flatMap((route) => {
+        const query = route.query.trim();
+        return query.length > 0 ? [{ mode: route.mode, query }] : [];
+    });
+    if (queries.length === 0 &&
+        fts.length === 0 &&
+        vector.length === 0 &&
+        orderedRoutes.length === 0) {
+        throw new Error("zvec_grep_search requires query, queries, fts, or vector.");
+    }
+    return {
+        queries: queries.length > 0 ? queries : undefined,
+        routes: orderedRoutes.length > 0
+            ? orderedRoutes
+            : [
+                ...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;AAqC7D,MAAM,UAAU,oBAAoB,CAClC,KAAmD;IAEnD,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;IAC5C,OAAO;QACL,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,MAAM,EAAE,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;QACpD,MAAM,EAAE,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;QACpD,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,KAyBiD;IAEjD,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,MAAM,aAAa,GAAG,CAAC,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;QAC3D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACjC,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/D,CAAC,CAAC,CAAC;IACH,IACE,OAAO,CAAC,MAAM,KAAK,CAAC;QACpB,GAAG,CAAC,MAAM,KAAK,CAAC;QAChB,MAAM,CAAC,MAAM,KAAK,CAAC;QACnB,aAAa,CAAC,MAAM,KAAK,CAAC,EAC1B,CAAC;QACD,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,EACJ,aAAa,CAAC,MAAM,GAAG,CAAC;YACtB,CAAC,CAAC,aAAa;YACf,CAAC,CAAC;gBACE,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAc,EAAE,KAAK,EAAE,CAAC,CAAC;gBACxD,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;aAC/D;QACP,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

package/dist/mcp/request-metadata.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-metadata.js","sourceRoot":"","sources":["../../src/mcp/request-metadata.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,8BAA8B,GACzC,oCAAoC,CAAC;AAEvC,MAAM,UAAU,mCAAmC,CACjD,IAAa;IAEb,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI;QAAE,OAAO,SAAS,CAAC;IAChE,MAAM,KAAK,GAAI,IAAgC,CAC7C,8BAA8B,CAC/B,CAAC;IACF,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAChD,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAChC,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;AACxD,CAAC"}

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

@@ -0,0 +1,40 @@
+import { type AuthInfo, type RequestStateCodec } from "@modelcontextprotocol/server";
+import type { RemoteEmbeddingAuthorizationPlan } from "../authorization/types.js";
+export type RemoteEmbeddingRequestState = {
+    version: 1;
+    nonce: string;
+    method: "tools/call";
+    tool: string;
+    argumentsFingerprint: string;
+    targetFingerprint: string;
+    disclosureFingerprint: string;
+};
+export interface RemoteEmbeddingRequestStateReplayGuard {
+    issue(state: Omit<RemoteEmbeddingRequestState, "nonce">): RemoteEmbeddingRequestState;
+    consume(state: RemoteEmbeddingRequestState): Promise<boolean>;
+}
+export declare class InMemoryRemoteEmbeddingRequestStateReplayGuard implements RemoteEmbeddingRequestStateReplayGuard {
+    private readonly consumed;
+    issue(state: Omit<RemoteEmbeddingRequestState, "nonce">): RemoteEmbeddingRequestState;
+    consume(state: RemoteEmbeddingRequestState): Promise<boolean>;
+    private expire;
+}
+export declare class PersistentRemoteEmbeddingRequestStateReplayGuard implements RemoteEmbeddingRequestStateReplayGuard {
+    private readonly directory;
+    private readonly consumed;
+    private loaded;
+    private saturated;
+    private operationTail;
+    constructor(home?: string);
+    issue(state: Omit<RemoteEmbeddingRequestState, "nonce">): RemoteEmbeddingRequestState;
+    consume(state: RemoteEmbeddingRequestState): Promise<boolean>;
+    private consumeLocked;
+    private loadTombstones;
+    private removeExpiredTombstones;
+}
+export declare function mcpRequestStateKeyPath(home?: string): string;
+export declare function loadOrCreateMcpRequestStateKey(home?: string): Promise<Uint8Array>;
+export declare function createRemoteEmbeddingRequestStateCodec(key: Uint8Array, ttlSeconds?: number): RequestStateCodec<RemoteEmbeddingRequestState>;
+export declare function remoteEmbeddingRequestState(tool: string, args: unknown, plan: RemoteEmbeddingAuthorizationPlan): Omit<RemoteEmbeddingRequestState, "nonce">;
+export declare function matchesRemoteEmbeddingRequestState(actual: RemoteEmbeddingRequestState | undefined, expected: Omit<RemoteEmbeddingRequestState, "nonce">): boolean;
+export declare function requestPrincipal(token: string | undefined): AuthInfo;