@angri450/zvec-grep 0.2.2

package/dist/mcp/request-state.js

@@ -0,0 +1,210 @@
+import { createHash, randomBytes, randomUUID } from "node:crypto";
+import { chmod, mkdir, open, readFile, readdir, unlink, } from "node:fs/promises";
+import { dirname, join } from "node:path";
+import { createRequestStateCodec, } from "@modelcontextprotocol/server";
+import { defaultHome } from "../engine/utils/path.js";
+const REQUEST_STATE_KEY_BYTES = 32;
+const REQUEST_STATE_TTL_SECONDS = 10 * 60;
+const MAX_IN_MEMORY_CONSUMED_STATES = 4_096;
+export class InMemoryRemoteEmbeddingRequestStateReplayGuard {
+    consumed = new Map();
+    issue(state) {
+        return issueState(state);
+    }
+    async consume(state) {
+        const now = Date.now();
+        this.expire(now);
+        if (this.consumed.has(state.nonce) ||
+            this.consumed.size >= MAX_IN_MEMORY_CONSUMED_STATES) {
+            return false;
+        }
+        this.consumed.set(state.nonce, now);
+        return true;
+    }
+    expire(now) {
+        const expiresBefore = now - REQUEST_STATE_TTL_SECONDS * 1_000;
+        for (const [nonce, consumedAt] of this.consumed) {
+            if (consumedAt >= expiresBefore)
+                continue;
+            this.consumed.delete(nonce);
+        }
+    }
+}
+export class PersistentRemoteEmbeddingRequestStateReplayGuard {
+    directory;
+    consumed = new Map();
+    loaded = false;
+    saturated = false;
+    operationTail = Promise.resolve();
+    constructor(home) {
+        this.directory = join(home ?? defaultHome(), "mcp-request-state-consumed");
+    }
+    issue(state) {
+        return issueState(state);
+    }
+    async consume(state) {
+        const operation = this.operationTail.then(() => this.consumeLocked(state));
+        this.operationTail = operation.then(() => undefined, () => undefined);
+        return await operation;
+    }
+    async consumeLocked(state) {
+        if (!/^\d{13}-[0-9a-f-]{36}$/.test(state.nonce))
+            return false;
+        await mkdir(this.directory, { recursive: true, mode: 0o700 });
+        if (!this.loaded)
+            await this.loadTombstones();
+        await this.removeExpiredTombstones();
+        if (this.saturated) {
+            await this.loadTombstones();
+        }
+        if (this.saturated ||
+            this.consumed.has(state.nonce) ||
+            this.consumed.size >= MAX_IN_MEMORY_CONSUMED_STATES) {
+            return false;
+        }
+        try {
+            const file = await open(join(this.directory, state.nonce), "wx", 0o600);
+            await file.close();
+        }
+        catch (error) {
+            if (isExistingFile(error))
+                return false;
+            // Force reload on next call if filesystem state is uncertain.
+            this.loaded = false;
+            throw error;
+        }
+        this.consumed.set(state.nonce, Number(state.nonce.slice(0, 13)));
+        return true;
+    }
+    async loadTombstones() {
+        this.consumed.clear();
+        this.saturated = false;
+        const expiresBefore = Date.now() - REQUEST_STATE_TTL_SECONDS * 1_000;
+        for (const name of (await readdir(this.directory)).sort()) {
+            const consumedAt = Number(name.slice(0, 13));
+            if (!Number.isFinite(consumedAt))
+                continue;
+            if (consumedAt < expiresBefore) {
+                await unlink(join(this.directory, name)).catch(() => undefined);
+                continue;
+            }
+            if (this.consumed.size >= MAX_IN_MEMORY_CONSUMED_STATES) {
+                this.saturated = true;
+                break;
+            }
+            this.consumed.set(name, consumedAt);
+        }
+        this.loaded = true;
+    }
+    async removeExpiredTombstones() {
+        const expiresBefore = Date.now() - REQUEST_STATE_TTL_SECONDS * 1_000;
+        for (const [name, consumedAt] of this.consumed) {
+            if (consumedAt >= expiresBefore)
+                break;
+            this.consumed.delete(name);
+            await unlink(join(this.directory, name)).catch(() => undefined);
+        }
+    }
+}
+export function mcpRequestStateKeyPath(home) {
+    return join(home ?? defaultHome(), "mcp-request-state.key");
+}
+export async function loadOrCreateMcpRequestStateKey(home) {
+    const path = mcpRequestStateKeyPath(home);
+    try {
+        const existing = await readFile(path);
+        await chmod(path, 0o600);
+        return validateKey(existing);
+    }
+    catch (error) {
+        if (!isMissingFile(error))
+            throw error;
+    }
+    await mkdir(dirname(path), { recursive: true, mode: 0o700 });
+    const candidate = randomBytes(REQUEST_STATE_KEY_BYTES);
+    try {
+        const file = await open(path, "wx", 0o600);
+        try {
+            await file.writeFile(candidate);
+        }
+        finally {
+            await file.close();
+        }
+    }
+    catch (error) {
+        if (!isExistingFile(error))
+            throw error;
+    }
+    await chmod(path, 0o600);
+    return validateKey(await readFile(path));
+}
+export function createRemoteEmbeddingRequestStateCodec(key, ttlSeconds = REQUEST_STATE_TTL_SECONDS) {
+    return createRequestStateCodec({
+        key,
+        ttlSeconds,
+        bind: (ctx) => `${ctx.mcpReq.method}\0${ctx.http?.authInfo?.clientId ?? "anonymous"}`,
+    });
+}
+export function remoteEmbeddingRequestState(tool, args, plan) {
+    return {
+        version: 1,
+        method: "tools/call",
+        tool,
+        argumentsFingerprint: fingerprint(args),
+        targetFingerprint: plan.target.targetFingerprint,
+        disclosureFingerprint: fingerprint(plan.disclosure),
+    };
+}
+export function matchesRemoteEmbeddingRequestState(actual, expected) {
+    return (actual?.version === expected.version &&
+        actual.method === expected.method &&
+        actual.tool === expected.tool &&
+        actual.argumentsFingerprint === expected.argumentsFingerprint &&
+        actual.targetFingerprint === expected.targetFingerprint &&
+        actual.disclosureFingerprint === expected.disclosureFingerprint);
+}
+function issueState(state) {
+    return {
+        ...state,
+        nonce: `${Date.now()}-${randomUUID()}`,
+    };
+}
+export function requestPrincipal(token) {
+    return {
+        token: token ?? "",
+        clientId: token ? fingerprint(token) : "loopback-anonymous",
+        scopes: ["zvec-grep"],
+    };
+}
+function fingerprint(value) {
+    return createHash("sha256").update(stableJson(value)).digest("hex");
+}
+function stableJson(value) {
+    if (Array.isArray(value)) {
+        return `[${value.map(stableJson).join(",")}]`;
+    }
+    if (value && typeof value === "object") {
+        return `{${Object.entries(value)
+            .sort(([left], [right]) => left.localeCompare(right))
+            .map(([key, item]) => `${JSON.stringify(key)}:${stableJson(item)}`)
+            .join(",")}}`;
+    }
+    return JSON.stringify(value) ?? "undefined";
+}
+function validateKey(value) {
+    if (value.byteLength !== REQUEST_STATE_KEY_BYTES) {
+        throw new Error("MCP request-state key must be exactly 32 bytes.");
+    }
+    return value;
+}
+function isMissingFile(error) {
+    return (error instanceof Error &&
+        "code" in error &&
+        error.code === "ENOENT");
+}
+function isExistingFile(error) {
+    return (error instanceof Error &&
+        "code" in error &&
+        error.code === "EEXIST");
+}
+//# sourceMappingURL=request-state.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"request-state.js","sourceRoot":"","sources":["../../src/mcp/request-state.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAClE,OAAO,EACL,KAAK,EACL,KAAK,EACL,IAAI,EACJ,QAAQ,EACR,OAAO,EACP,MAAM,GACP,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EACL,uBAAuB,GAGxB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAGtD,MAAM,uBAAuB,GAAG,EAAE,CAAC;AACnC,MAAM,yBAAyB,GAAG,EAAE,GAAG,EAAE,CAAC;AAC1C,MAAM,6BAA6B,GAAG,KAAK,CAAC;AAmB5C,MAAM,OAAO,8CAA8C;IACxC,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEtD,KAAK,CACH,KAAiD;QAEjD,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,KAAkC;QAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjB,IACE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC;YAC9B,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,6BAA6B,EACnD,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACpC,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,GAAW;QACxB,MAAM,aAAa,GAAG,GAAG,GAAG,yBAAyB,GAAG,KAAK,CAAC;QAC9D,KAAK,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAChD,IAAI,UAAU,IAAI,aAAa;gBAAE,SAAS;YAC1C,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,gDAAgD;IAC1C,SAAS,CAAS;IAClB,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC9C,MAAM,GAAG,KAAK,CAAC;IACf,SAAS,GAAG,KAAK,CAAC;IAClB,aAAa,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IAEzD,YAAY,IAAa;QACvB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,IAAI,IAAI,WAAW,EAAE,EAAE,4BAA4B,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CACH,KAAiD;QAEjD,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,KAAkC;QAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3E,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC,IAAI,CACjC,GAAG,EAAE,CAAC,SAAS,EACf,GAAG,EAAE,CAAC,SAAS,CAChB,CAAC;QACF,OAAO,MAAM,SAAS,CAAC;IACzB,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,KAAkC;QAElC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAC9D,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9D,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC9C,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACrC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC9B,CAAC;QACD,IACE,IAAI,CAAC,SAAS;YACd,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC;YAC9B,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,6BAA6B,EACnD,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;YACxE,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,cAAc,CAAC,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAC;YACxC,8DAA8D;YAC9D,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;YACpB,MAAM,KAAK,CAAC;QACd,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QACjE,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,yBAAyB,GAAG,KAAK,CAAC;QACrE,KAAK,MAAM,IAAI,IAAI,CAAC,MAAM,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1D,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAAE,SAAS;YAC3C,IAAI,UAAU,GAAG,aAAa,EAAE,CAAC;gBAC/B,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBAChE,SAAS;YACX,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,6BAA6B,EAAE,CAAC;gBACxD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;gBACtB,MAAM;YACR,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,uBAAuB;QACnC,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,yBAAyB,GAAG,KAAK,CAAC;QACrE,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/C,IAAI,UAAU,IAAI,aAAa;gBAAE,MAAM;YACvC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CAAC,IAAa;IAClD,OAAO,IAAI,CAAC,IAAI,IAAI,WAAW,EAAE,EAAE,uBAAuB,CAAC,CAAC;AAC9D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,8BAA8B,CAClD,IAAa;IAEb,MAAM,IAAI,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;IAC1C,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC;QACtC,MAAM,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACzB,OAAO,WAAW,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;YAAE,MAAM,KAAK,CAAC;IACzC,CAAC;IAED,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7D,MAAM,SAAS,GAAG,WAAW,CAAC,uBAAuB,CAAC,CAAC;IACvD,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;gBAAS,CAAC;YACT,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC;YAAE,MAAM,KAAK,CAAC;IAC1C,CAAC;IACD,MAAM,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACzB,OAAO,WAAW,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,sCAAsC,CACpD,GAAe,EACf,UAAU,GAAG,yBAAyB;IAEtC,OAAO,uBAAuB,CAA8B;QAC1D,GAAG;QACH,UAAU;QACV,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE,CACZ,GAAG,GAAG,CAAC,MAAM,CAAC,MAAM,KAAK,GAAG,CAAC,IAAI,EAAE,QAAQ,EAAE,QAAQ,IAAI,WAAW,EAAE;KACzE,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,IAAY,EACZ,IAAa,EACb,IAAsC;IAEtC,OAAO;QACL,OAAO,EAAE,CAAC;QACV,MAAM,EAAE,YAAY;QACpB,IAAI;QACJ,oBAAoB,EAAE,WAAW,CAAC,IAAI,CAAC;QACvC,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB;QAChD,qBAAqB,EAAE,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC;KACpD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kCAAkC,CAChD,MAA+C,EAC/C,QAAoD;IAEpD,OAAO,CACL,MAAM,EAAE,OAAO,KAAK,QAAQ,CAAC,OAAO;QACpC,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,MAAM;QACjC,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI;QAC7B,MAAM,CAAC,oBAAoB,KAAK,QAAQ,CAAC,oBAAoB;QAC7D,MAAM,CAAC,iBAAiB,KAAK,QAAQ,CAAC,iBAAiB;QACvD,MAAM,CAAC,qBAAqB,KAAK,QAAQ,CAAC,qBAAqB,CAChE,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CACjB,KAAiD;IAEjD,OAAO;QACL,GAAG,KAAK;QACR,KAAK,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,UAAU,EAAE,EAAE;KACvC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAyB;IACxD,OAAO;QACL,KAAK,EAAE,KAAK,IAAI,EAAE;QAClB,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,oBAAoB;QAC3D,MAAM,EAAE,CAAC,WAAW,CAAC;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACtE,CAAC;AAED,SAAS,UAAU,CAAC,KAAc;IAChC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,IAAI,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IAChD,CAAC;IACD,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC;aACxD,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;aACpD,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;aAClE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IAClB,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,WAAW,CAAC;AAC9C,CAAC;AAED,SAAS,WAAW,CAAC,KAAiB;IACpC,IAAI,KAAK,CAAC,UAAU,KAAK,uBAAuB,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;IACrE,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,CACL,KAAK,YAAY,KAAK;QACtB,MAAM,IAAI,KAAK;QACd,KAA+B,CAAC,IAAI,KAAK,QAAQ,CACnD,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,CACL,KAAK,YAAY,KAAK;QACtB,MAAM,IAAI,KAAK;QACd,KAA+B,CAAC,IAAI,KAAK,QAAQ,CACnD,CAAC;AACJ,CAAC"}

package/dist/mcp/result-format.d.ts

@@ -0,0 +1,13 @@
+import type { ZvecGrepContextResult } from "../index.js";
+export type ToolResult = {
+    content: Array<{
+        type: "text";
+        text: string;
+    }>;
+    structuredContent?: Record<string, unknown>;
+};
+export declare function textToolResult(text: string): ToolResult;
+export declare function toolResult(text: string, structuredContent: Record<string, unknown>): ToolResult;
+export declare function contextToolResult(result: ZvecGrepContextResult): ToolResult;
+export declare function contextText(result: ZvecGrepContextResult): string;
+export declare function simplifyContextResult(result: ZvecGrepContextResult): Record<string, unknown>;

package/dist/mcp/result-format.js

@@ -0,0 +1,75 @@
+export function textToolResult(text) {
+    return {
+        content: [{ type: "text", text }],
+    };
+}
+export function toolResult(text, structuredContent) {
+    return {
+        content: [{ type: "text", text }],
+        structuredContent,
+    };
+}
+export function contextToolResult(result) {
+    return toolResult(contextText(result), {
+        result: simplifyContextResult(result),
+    });
+}
+export function contextText(result) {
+    const lines = [
+        `query: ${result.query}`,
+        `root: ${result.root}`,
+        `source: ${result.source}`,
+        `coverage: ${result.coverage}`,
+        `hits: ${result.items.length}`,
+    ];
+    for (const item of result.items) {
+        lines.push("");
+        lines.push(`${item.file.relativePath}:${rangeLabel(item.range)} rank=${item.rank} matchedBy=${item.matchedBy}`);
+        if (item.outline) {
+            lines.push(`outline: ${item.outline}`);
+        }
+        lines.push("source:");
+        lines.push(item.content);
+    }
+    return lines.join("\n");
+}
+export function simplifyContextResult(result) {
+    return {
+        query: result.query,
+        root: result.root,
+        source: result.source,
+        coverage: result.coverage,
+        workspaceIndex: result.workspaceIndex,
+        diagnostics: result.diagnostics,
+        items: result.items.map((item) => ({
+            kind: item.kind,
+            rank: item.rank,
+            file: item.file,
+            range: item.range,
+            excerptRange: item.excerptRange,
+            outline: item.outline,
+            content: item.content,
+            contentRole: item.contentRole,
+            status: item.status,
+            score: item.score,
+            matchedBy: item.matchedBy,
+            metadata: item.metadata,
+            entityId: item.entityId,
+            container: item.container,
+            trace: item.trace,
+        })),
+    };
+}
+function rangeLabel(range) {
+    if (range.kind === "text" &&
+        typeof range.startLine === "number" &&
+        typeof range.endLine === "number") {
+        return `${range.startLine}-${range.endLine}`;
+    }
+    if (typeof range.startOffset === "number" &&
+        typeof range.endOffset === "number") {
+        return `${range.startOffset}-${range.endOffset}`;
+    }
+    return range.kind;
+}
+//# sourceMappingURL=result-format.js.map

package/dist/mcp/result-format.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"result-format.js","sourceRoot":"","sources":["../../src/mcp/result-format.ts"],"names":[],"mappings":"AAOA,MAAM,UAAU,cAAc,CAAC,IAAY;IACzC,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;KAClC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,UAAU,CACxB,IAAY,EACZ,iBAA0C;IAE1C,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACjC,iBAAiB;KAClB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,MAA6B;IAC7D,OAAO,UAAU,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE;QACrC,MAAM,EAAE,qBAAqB,CAAC,MAAM,CAAC;KACtC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,MAA6B;IACvD,MAAM,KAAK,GAAG;QACZ,UAAU,MAAM,CAAC,KAAK,EAAE;QACxB,SAAS,MAAM,CAAC,IAAI,EAAE;QACtB,WAAW,MAAM,CAAC,MAAM,EAAE;QAC1B,aAAa,MAAM,CAAC,QAAQ,EAAE;QAC9B,SAAS,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE;KAC/B,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CACR,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,IAAI,cAAc,IAAI,CAAC,SAAS,EAAE,CACpG,CAAC;QACF,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,KAAK,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACzC,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,MAA6B;IAE7B,OAAO;QACL,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,cAAc,EAAE,MAAM,CAAC,cAAc;QACrC,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACjC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,KAMnB;IACC,IACE,KAAK,CAAC,IAAI,KAAK,MAAM;QACrB,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;QACnC,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,EACjC,CAAC;QACD,OAAO,GAAG,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;IAC/C,CAAC;IACD,IACE,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ;QACrC,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,EACnC,CAAC;QACD,OAAO,GAAG,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;IACnD,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC;AACpB,CAAC"}