@zvec/zvec-grep 0.1.4 → 0.1.6

package/dist/mcp/tools.js

@@ -0,0 +1,448 @@
+import { randomBytes } from "node:crypto";
+import { acceptedContent, inputRequired, inputResponse, McpServer, ProtocolError, } from "@modelcontextprotocol/server";
+import { z } from "zod";
+import { formatAgentContextResult } from "../cli/format/context.js";
+import { formatRemoteEmbeddingAuthorizationPrompt, remoteEmbeddingDisclosureData, } from "../authorization/prompt.js";
+import { normalizeSearchInput, } from "./input-normalization.js";
+import { zvecGrepIndexInputSchema, zvecGrepIndexOutputSchema, zvecGrepIndexDropInputSchema, zvecGrepIndexDropOutputSchema, zvecGrepIndexStatusInputSchema, zvecGrepIndexStatusOutputSchema, zvecGrepRgInputSchema, zvecGrepSearchInputSchema, zvecGrepServerStatusInputSchema, zvecGrepServerStatusOutputSchema, } from "./schemas.js";
+import { embeddingEnvironmentFromRequestMeta } from "./request-metadata.js";
+import { textToolResult, toolResult } from "./result-format.js";
+import { indexProgressMessage } from "../index-progress.js";
+import { DEFAULT_MCP_TOOLSET } from "./toolset.js";
+import { createRemoteEmbeddingRequestStateCodec, InMemoryRemoteEmbeddingRequestStateReplayGuard, matchesRemoteEmbeddingRequestState, remoteEmbeddingRequestState, } from "./request-state.js";
+import { runWithTraceContext, traceContextFromMcpMeta, } from "../observability/trace-context.js";
+const ZVEC_GREP_AGENT_SEARCH_MCP_INSTRUCTIONS = [
+    "Choose the initial search tool by the scope of the requested answer, not merely by whether the question contains an identifier.",
+    "Use native Grep or rg first when the answer can be obtained by locating a specific definition, reference, filename, configuration key, error message, literal, source fragment, or regex.",
+    "Use zvec_grep_search first when the answer requires architecture, lifecycle, system design, conceptual discovery, data or control flow, comparison across implementations, design rationale, or performance analysis.",
+    "When exact symbols are present but the answer spans multiple components, files, stages, or implementations, treat it as a mixed task: use zvec_grep_search with the concept and known symbols, then use native Grep or rg for focused follow-up.",
+    "Before using broad file reads or a sub-agent for conceptual repository discovery, make at least one appropriate zvec_grep_search call.",
+    "Do not launch a sub-agent solely to locate code. Stop discovery when the evidence covers all components requested by the question.",
+];
+const ZVEC_GREP_FULL_SEARCH_MCP_INSTRUCTIONS = [
+    "Choose the initial search tool by the scope of the requested answer, not merely by whether the question contains an identifier.",
+    "Use zvec_grep_rg first only when the answer can be obtained by locating a specific definition, reference, filename, configuration key, error message, literal, source fragment, or regex.",
+    "Use zvec_grep_search first when the answer requires architecture, lifecycle, system design, conceptual discovery, data or control flow, comparison across implementations, design rationale, or performance analysis.",
+    "When exact symbols are present but the answer spans multiple components, files, stages, or implementations, treat it as a mixed task: search using the concept and known symbols, then use zvec_grep_rg for focused follow-up.",
+    "Before using native Grep, Glob, shell rg, broad file reads, or a sub-agent for repository discovery, make at least one appropriate zvec-grep call.",
+    "Do not launch a sub-agent solely to locate code. Stop discovery when the evidence covers all components requested by the question.",
+];
+export const ZVEC_GREP_AGENT_MCP_INSTRUCTIONS = [
+    ...ZVEC_GREP_AGENT_SEARCH_MCP_INSTRUCTIONS,
+    "Every repository operation requires an absolute root path visible to the daemon.",
+    "Read freshness and indexing directly from zvec_grep_search responses without a status preflight.",
+    "Use possibly_stale search results immediately when they are sufficient; do not perform extra diagnostics merely because a background update is active.",
+    "When an index is missing and literal or regex search can answer the task, use native Grep or rg.",
+].join(" ");
+export const ZVEC_GREP_FULL_MCP_INSTRUCTIONS = [
+    ...ZVEC_GREP_FULL_SEARCH_MCP_INSTRUCTIONS,
+    "Every repository operation requires an absolute root path visible to the daemon.",
+    "Use the zvec_grep_* tools directly for repository search, status, indexing, deletion, and exhaustive lexical search.",
+    "Use freshness and indexing from zvec_grep_search without a status preflight; call zvec_grep_index_status only for a missing index, failed or cancelled indexing, diagnostics, or explicit progress monitoring.",
+    "Use possibly_stale search results immediately when they are sufficient; do not call status merely because a background update is active.",
+    "Call zvec_grep_index only when persistent indexing or index deletion is explicitly requested. Never silently create, rebuild, or drop an index.",
+    "For a new index, use a user-selected embedding or omit it only when a server default model is known; never guess a model.",
+    "zvec_grep_index wait defaults to false; poll zvec_grep_index_status for background progress and set wait to true only when completion is required before continuing.",
+    "Use zvec_grep_index with drop: true, or zvec_grep_index_drop, only when index deletion is explicitly requested.",
+    "Call zvec_grep_server_status only for daemon diagnostics, not before ordinary searches.",
+].join(" ");
+export const ZVEC_GREP_MCP_INSTRUCTIONS = ZVEC_GREP_AGENT_MCP_INSTRUCTIONS;
+export function createZvecGrepMcpServer(backend, version, options = {}) {
+    const toolset = options.toolset ?? DEFAULT_MCP_TOOLSET;
+    const requestStateCodec = options.requestStateCodec ??
+        createRemoteEmbeddingRequestStateCodec(randomBytes(32));
+    const requestStateReplayGuard = options.requestStateReplayGuard ??
+        new InMemoryRemoteEmbeddingRequestStateReplayGuard();
+    const server = new McpServer({ name: "zvec-grep", version }, {
+        instructions: toolset === "full"
+            ? ZVEC_GREP_FULL_MCP_INSTRUCTIONS
+            : ZVEC_GREP_AGENT_MCP_INSTRUCTIONS,
+        cacheHints: {
+            "server/discover": { ttlMs: 60 * 60 * 1_000, cacheScope: "private" },
+            "tools/list": { ttlMs: 60 * 60 * 1_000, cacheScope: "private" },
+        },
+        requestState: { verify: requestStateCodec.verify },
+    });
+    registerZvecGrepTools(server, backend, {
+        ...options,
+        requestStateCodec,
+        requestStateReplayGuard,
+        toolset,
+    });
+    return server;
+}
+export function registerZvecGrepTools(server, backend, options = {}) {
+    const toolset = options.toolset ?? DEFAULT_MCP_TOOLSET;
+    const full = toolset === "full";
+    if (full) {
+        server.registerTool("zvec_grep_index", {
+            title: "Ensure or drop zvec-grep index",
+            description: "Activate an absolute repository root to create, incrementally update, rebuild, or explicitly drop its index. Do not call this tool to create, rebuild, or drop an index unless the user requested persistent indexing or index deletion.",
+            inputSchema: zvecGrepIndexInputSchema,
+            outputSchema: zvecGrepIndexOutputSchema,
+            annotations: {
+                readOnlyHint: false,
+                destructiveHint: true,
+                idempotentHint: true,
+                openWorldHint: true,
+            },
+        }, async (input, ctx) => await runWithTraceContext(traceContextFromMcpMeta(ctx.mcpReq._meta), async () => {
+            const request = {
+                ...input,
+                embeddingEnvironment: options.acceptEmbeddingEnvironmentMeta
+                    ? embeddingEnvironmentFromRequestMeta(ctx.mcpReq._meta)
+                    : undefined,
+            };
+            const plan = await backend.planIndexAuthorization?.(request);
+            const resolution = plan
+                ? await resolveRemoteEmbeddingAuthorization(backend, plan, undefined, ctx, "zvec_grep_index", request, options)
+                : { kind: "ready" };
+            if (resolution.kind === "input_required")
+                return resolution.result;
+            const progress = createMcpIndexProgressReporter(ctx);
+            let result;
+            try {
+                result = await backend.index(request, {
+                    authorization: resolution.authorization,
+                    onProgress: progress.report,
+                });
+            }
+            finally {
+                await progress.flush();
+            }
+            const structuredContent = {
+                root: result.root,
+                job_id: result.jobId,
+                state: result.state,
+                reused: result.reused,
+                action: result.action,
+                dropped: result.dropped,
+                error: result.error,
+            };
+            return toolResult([
+                `root: ${result.root}`,
+                `job_id: ${result.jobId}`,
+                `state: ${result.state}`,
+                `reused: ${result.reused}`,
+                ...(result.action ? [`action: ${result.action}`] : []),
+                ...(result.dropped !== undefined
+                    ? [`dropped: ${result.dropped}`]
+                    : []),
+                ...(result.error
+                    ? [
+                        `error_code: ${result.error.code}`,
+                        `error_message: ${result.error.message}`,
+                    ]
+                    : []),
+            ].join("\n"), structuredContent);
+        }));
+    }
+    server.registerTool("zvec_grep_search", {
+        title: "Search with zvec-grep",
+        description: full
+            ? "Search an existing repository index when the answer requires architecture, lifecycle, system design, conceptual discovery, data or control flow, comparison across implementations, design rationale, or performance analysis. Use search for mixed tasks whose question names exact symbols but whose answer spans multiple components, files, stages, or implementations; use managed ripgrep for focused follow-up. Read freshness and indexing from the response; use zvec_grep_index_status only for missing indexes, failed or cancelled indexing, diagnostics, or explicit progress monitoring."
+            : "Search an existing repository index when the answer requires architecture, lifecycle, system design, conceptual discovery, data or control flow, comparison across implementations, design rationale, or performance analysis. Use search for mixed tasks whose question names exact symbols but whose answer spans multiple components, files, stages, or implementations; use native Grep or rg for focused follow-up. Read freshness and indexing directly from the response without a status preflight. When an index is unavailable, use the returned diagnostics and native Grep or rg for exact fallback.",
+        inputSchema: zvecGrepSearchInputSchema,
+        annotations: {
+            readOnlyHint: false,
+            destructiveHint: false,
+            idempotentHint: false,
+            openWorldHint: false,
+        },
+    }, async (input, ctx) => await runWithTraceContext(traceContextFromMcpMeta(ctx.mcpReq._meta), async () => {
+        const normalized = normalizeSearchInput(input);
+        const plan = await backend.planSearchAuthorization?.(normalized);
+        const resolution = plan
+            ? await resolveRemoteEmbeddingAuthorization(backend, plan, "local_search", ctx, "zvec_grep_search", normalized, options)
+            : { kind: "ready" };
+        if (resolution.kind === "input_required")
+            return resolution.result;
+        const effectiveSearch = resolution.alternative === "local_search"
+            ? ftsFallbackSearch(normalized)
+            : normalized;
+        const response = await backend.search(effectiveSearch, {
+            authorization: resolution.authorization,
+        });
+        const statusLines = [
+            `freshness: ${response.freshness}`,
+            ...(response.indexing
+                ? [`indexing: ${formatSearchIndexing(response.indexing)}`]
+                : []),
+        ];
+        // Mirror the compact rg output: return agent-formatted text only and drop
+        // the verbose structuredContent (per-item outline + full source), which
+        // otherwise dominates the agent's context. `short` keeps a bounded source
+        // snippet per hit so relevance is judgeable without extra file reads.
+        return textToolResult(`${statusLines.join("\n")}\n${formatAgentContextResult(response.result, {
+            preview: "short",
+        })}`);
+    }));
+    if (full) {
+        server.registerTool("zvec_grep_index_drop", {
+            title: "Drop zvec-grep Workspace index",
+            description: "Delete the persisted index for an absolute Workspace root and release its daemon runtime.",
+            inputSchema: zvecGrepIndexDropInputSchema,
+            outputSchema: zvecGrepIndexDropOutputSchema,
+            annotations: {
+                readOnlyHint: false,
+                destructiveHint: true,
+                idempotentHint: true,
+                openWorldHint: false,
+            },
+        }, async (input) => {
+            const result = await backend.dropIndex(input);
+            const structuredContent = {
+                root: result.root,
+                removed: result.removed,
+            };
+            return toolResult(result.removed
+                ? `Dropped Workspace index for ${result.root}`
+                : `No Workspace index found for ${result.root}`, structuredContent);
+        });
+    }
+    if (full) {
+        server.registerTool("zvec_grep_rg", {
+            title: "Search with managed ripgrep",
+            description: "Run exhaustive, AST-enriched managed ripgrep locally without requiring an index. Use it first only when the answer can be obtained by locating a specific definition, reference, filename, configuration key, error message, literal, source fragment, or regex. Pass the rg command you would otherwise run. Results are exhaustive by default; append `| head -N` only when you intentionally want a bounded result set. Scope broad matches with command paths, `-g`/`--glob`, or `-t`/`--type` filters.",
+            inputSchema: zvecGrepRgInputSchema,
+            annotations: {
+                readOnlyHint: true,
+                destructiveHint: false,
+                idempotentHint: true,
+                openWorldHint: true,
+            },
+        }, async (input) => {
+            const response = await backend.rg(input);
+            return textToolResult(`${formatAgentContextResult(response.result, {})}${rgCoverageHint(response.result)}`);
+        });
+    }
+    if (full) {
+        server.registerTool("zvec_grep_index_status", {
+            title: "Inspect zvec-grep index status",
+            description: "Read persisted index status and, when active, daemon runtime and job status for an absolute root. Use only after a missing-index response, indexing failure or cancellation, explicit progress monitoring, or daemon diagnostics.",
+            inputSchema: zvecGrepIndexStatusInputSchema,
+            outputSchema: zvecGrepIndexStatusOutputSchema,
+            annotations: {
+                readOnlyHint: true,
+                destructiveHint: false,
+                idempotentHint: true,
+                openWorldHint: false,
+            },
+        }, async (input) => {
+            const result = await backend.indexStatus(input);
+            const structuredContent = formatIndexStatus(result);
+            return toolResult(JSON.stringify(structuredContent, null, 2), structuredContent);
+        });
+        server.registerTool("zvec_grep_server_status", {
+            title: "Inspect zvec-grep server status",
+            description: "Read daemon version, queue, runtime and model-pool summary without exposing repository paths.",
+            inputSchema: zvecGrepServerStatusInputSchema,
+            outputSchema: zvecGrepServerStatusOutputSchema,
+            annotations: {
+                readOnlyHint: true,
+                destructiveHint: false,
+                idempotentHint: true,
+                openWorldHint: false,
+            },
+        }, async () => {
+            const result = await backend.serverStatus();
+            const structuredContent = {
+                version: result.version,
+                uptime_ms: result.uptimeMs,
+                shutting_down: result.shuttingDown,
+                active_runtimes: result.activeRuntimes,
+                queued_jobs: result.queuedJobs,
+                running_jobs: result.runningJobs,
+                models: {
+                    loaded: result.models.loaded,
+                    active_leases: result.models.activeLeases,
+                },
+            };
+            return toolResult(JSON.stringify(structuredContent, null, 2), structuredContent);
+        });
+    }
+}
+function rgCoverageHint(result) {
+    return result.coverage === "rg_truncated"
+        ? "\n\nMore matches were omitted by the explicit output bound. Remove or increase the trailing `head` bound to see them."
+        : "";
+}
+function createMcpIndexProgressReporter(extra) {
+    const progressToken = extra.mcpReq._meta?.progressToken;
+    let progressValue = Date.now();
+    let pending = Promise.resolve();
+    let modelPreparationActive = false;
+    return {
+        report(progress) {
+            const modelStage = progress.embedding?.stage;
+            if (modelStage === "preparing" || modelStage === "downloading") {
+                modelPreparationActive = true;
+            }
+            else if (modelStage === "ready" || progress.phase === "done") {
+                modelPreparationActive = false;
+            }
+            else if (modelPreparationActive &&
+                modelStage === undefined &&
+                progress.phase === "indexing") {
+                return;
+            }
+            const message = indexProgressMessage(progress);
+            if (progressToken === undefined ||
+                !message ||
+                extra.mcpReq.signal.aborted) {
+                return;
+            }
+            progressValue = Math.max(progressValue + 1, Date.now());
+            pending = pending
+                .then(() => extra.mcpReq.notify({
+                method: "notifications/progress",
+                params: {
+                    progressToken,
+                    progress: progressValue,
+                    message,
+                },
+            }))
+                .catch(() => undefined);
+        },
+        async flush() {
+            await pending;
+        },
+    };
+}
+const REMOTE_EMBEDDING_SCOPE_DESCRIPTION = "Allow this operation once or remember permission for this workspace.";
+const remoteEmbeddingDecisionSchema = z.object({
+    decision: z.enum([
+        "allow_once",
+        "allow_workspace",
+        "use_local_search",
+        "cancel",
+    ]),
+});
+async function resolveRemoteEmbeddingAuthorization(backend, plan, alternative, ctx, tool, args, options = {}) {
+    const existing = await backend.existingRemoteEmbeddingPermit?.(plan);
+    if (existing)
+        return { kind: "ready", authorization: existing };
+    if (!backend.grantRemoteEmbedding) {
+        throw new Error("Remote Embedding authorization is required.");
+    }
+    const expectedState = remoteEmbeddingRequestState(tool, args, plan);
+    const echoedState = ctx.mcpReq.requestState();
+    if (echoedState) {
+        if (!matchesRemoteEmbeddingRequestState(echoedState, expectedState)) {
+            throw new ProtocolError(-32602, "Invalid or expired requestState");
+        }
+        const response = inputResponse(ctx.mcpReq.inputResponses, "remote_embedding_authorization");
+        const content = acceptedContent(ctx.mcpReq.inputResponses, "remote_embedding_authorization", remoteEmbeddingDecisionSchema);
+        if (response.kind !== "elicit" ||
+            response.action !== "accept" ||
+            !content) {
+            throw new Error("Remote Embedding authorization was declined. No remote data was sent.");
+        }
+        if (!(await options.requestStateReplayGuard?.consume(echoedState))) {
+            throw new ProtocolError(-32602, "Invalid or expired requestState");
+        }
+        if (content.decision === "use_local_search" && alternative) {
+            return { kind: "ready", alternative: "local_search" };
+        }
+        if (content.decision !== "allow_once" &&
+            content.decision !== "allow_workspace") {
+            throw new Error("Remote Embedding authorization was declined. No remote data was sent.");
+        }
+        const scope = content.decision === "allow_workspace" ? "workspace" : "once";
+        return {
+            kind: "ready",
+            authorization: await backend.grantRemoteEmbedding(plan, scope),
+        };
+    }
+    const requestStateCodec = options.requestStateCodec;
+    if (!requestStateCodec) {
+        throw new Error("MCP request-state signing is not configured.");
+    }
+    return {
+        kind: "input_required",
+        result: inputRequired({
+            requestState: await requestStateCodec.mint(options.requestStateReplayGuard.issue(expectedState), ctx),
+            inputRequests: {
+                remote_embedding_authorization: inputRequired.elicit({
+                    mode: "form",
+                    message: formatRemoteEmbeddingAuthorizationPrompt({
+                        workspaceRoots: plan.target.workspaceRoots,
+                        provider: plan.target.provider,
+                        model: plan.target.model,
+                        endpoint: plan.target.endpoint,
+                        data: remoteEmbeddingDisclosureData(plan.disclosure),
+                    }),
+                    requestedSchema: {
+                        type: "object",
+                        properties: {
+                            decision: {
+                                type: "string",
+                                title: "Remote Embedding permission",
+                                description: REMOTE_EMBEDDING_SCOPE_DESCRIPTION,
+                                oneOf: [
+                                    { const: "allow_once", title: "Allow once" },
+                                    {
+                                        const: "allow_workspace",
+                                        title: "Allow for this workspace",
+                                    },
+                                    ...(alternative === "local_search"
+                                        ? [
+                                            {
+                                                const: "use_local_search",
+                                                title: "Use FTS only",
+                                            },
+                                        ]
+                                        : []),
+                                    { const: "cancel", title: "Cancel" },
+                                ],
+                                default: "cancel",
+                            },
+                        },
+                        required: ["decision"],
+                    },
+                }),
+            },
+        }),
+    };
+}
+function ftsFallbackSearch(input) {
+    const queries = [
+        ...(input.queries ?? []),
+        ...input.routes.map((route) => route.query),
+    ];
+    return {
+        ...input,
+        queries: undefined,
+        routes: queries.map((query) => ({ mode: "fts", query })),
+        autoUpdate: false,
+        freshness: "eventual",
+    };
+}
+function formatSearchIndexing(indexing) {
+    if (indexing.completed === undefined || indexing.total === undefined) {
+        return indexing.state;
+    }
+    return `${indexing.state} (${indexing.completed}/${indexing.total})`;
+}
+function formatIndexStatus(result) {
+    return {
+        root: result.root,
+        indexed: result.indexed,
+        index_policy: result.indexPolicy,
+        source: result.source,
+        persistent: result.persistent,
+        runtime: result.runtime
+            ? {
+                watcher_active: result.runtime.watcherActive,
+                dirty_revision: result.runtime.dirtyRevision,
+                indexed_revision: result.runtime.indexedRevision,
+                active_job_id: result.runtime.activeJobId,
+                job_state: result.runtime.jobState,
+                progress: result.runtime.progress,
+                completion: result.runtime.completion,
+                error: result.runtime.error,
+            }
+            : undefined,
+    };
+}
+//# sourceMappingURL=tools.js.map

package/dist/mcp/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/mcp/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EACL,eAAe,EACf,aAAa,EACb,aAAa,EACb,SAAS,EACT,aAAa,GAId,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AACpE,OAAO,EACL,wCAAwC,EACxC,6BAA6B,GAC9B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,oBAAoB,GAErB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,wBAAwB,EACxB,yBAAyB,EACzB,4BAA4B,EAC5B,6BAA6B,EAC7B,8BAA8B,EAC9B,+BAA+B,EAC/B,qBAAqB,EACrB,yBAAyB,EACzB,+BAA+B,EAC/B,gCAAgC,GAMjC,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,mCAAmC,EAAE,MAAM,uBAAuB,CAAC;AAC5E,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAMhE,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAmB,MAAM,cAAc,CAAC;AACpE,OAAO,EACL,sCAAsC,EACtC,8CAA8C,EAC9C,kCAAkC,EAClC,2BAA2B,GAG5B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,mCAAmC,CAAC;AAwJ3C,MAAM,uCAAuC,GAAG;IAC9C,iIAAiI;IACjI,2LAA2L;IAC3L,uNAAuN;IACvN,kPAAkP;IAClP,wIAAwI;IACxI,oIAAoI;CACrI,CAAC;AAEF,MAAM,sCAAsC,GAAG;IAC7C,iIAAiI;IACjI,2LAA2L;IAC3L,uNAAuN;IACvN,gOAAgO;IAChO,oJAAoJ;IACpJ,oIAAoI;CACrI,CAAC;AAEF,MAAM,CAAC,MAAM,gCAAgC,GAAG;IAC9C,GAAG,uCAAuC;IAC1C,kFAAkF;IAClF,kGAAkG;IAClG,wJAAwJ;IACxJ,kGAAkG;CACnG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAEZ,MAAM,CAAC,MAAM,+BAA+B,GAAG;IAC7C,GAAG,sCAAsC;IACzC,kFAAkF;IAClF,sHAAsH;IACtH,gNAAgN;IAChN,0IAA0I;IAC1I,iJAAiJ;IACjJ,2HAA2H;IAC3H,sKAAsK;IACtK,iHAAiH;IACjH,yFAAyF;CAC1F,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAEZ,MAAM,CAAC,MAAM,0BAA0B,GAAG,gCAAgC,CAAC;AAS3E,MAAM,UAAU,uBAAuB,CACrC,OAA8B,EAC9B,OAAe,EACf,UAAoC,EAAE;IAEtC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,mBAAmB,CAAC;IACvD,MAAM,iBAAiB,GACrB,OAAO,CAAC,iBAAiB;QACzB,sCAAsC,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1D,MAAM,uBAAuB,GAC3B,OAAO,CAAC,uBAAuB;QAC/B,IAAI,8CAA8C,EAAE,CAAC;IACvD,MAAM,MAAM,GAAG,IAAI,SAAS,CAC1B,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAC9B;QACE,YAAY,EACV,OAAO,KAAK,MAAM;YAChB,CAAC,CAAC,+BAA+B;YACjC,CAAC,CAAC,gCAAgC;QACtC,UAAU,EAAE;YACV,iBAAiB,EAAE,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE;YACpE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE;SAChE;QACD,YAAY,EAAE,EAAE,MAAM,EAAE,iBAAiB,CAAC,MAAM,EAAE;KACnD,CACF,CAAC;IACF,qBAAqB,CAAC,MAAM,EAAE,OAAO,EAAE;QACrC,GAAG,OAAO;QACV,iBAAiB;QACjB,uBAAuB;QACvB,OAAO;KACR,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,MAAiB,EACjB,OAA8B,EAC9B,UAAoC,EAAE;IAEtC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,mBAAmB,CAAC;IACvD,MAAM,IAAI,GAAG,OAAO,KAAK,MAAM,CAAC;IAEhC,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,CAAC,YAAY,CACjB,iBAAiB,EACjB;YACE,KAAK,EAAE,gCAAgC;YACvC,WAAW,EACT,0OAA0O;YAC5O,WAAW,EAAE,wBAAwB;YACrC,YAAY,EAAE,yBAAyB;YACvC,WAAW,EAAE;gBACX,YAAY,EAAE,KAAK;gBACnB,eAAe,EAAE,IAAI;gBACrB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,IAAI;aACpB;SACF,EACD,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CACnB,MAAM,mBAAmB,CACvB,uBAAuB,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,EACzC,KAAK,IAAI,EAAE;YACT,MAAM,OAAO,GAAyB;gBACpC,GAAG,KAAK;gBACR,oBAAoB,EAAE,OAAO,CAAC,8BAA8B;oBAC1D,CAAC,CAAC,mCAAmC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC;oBACvD,CAAC,CAAC,SAAS;aACd,CAAC;YACF,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,sBAAsB,EAAE,CAAC,OAAO,CAAC,CAAC;YAC7D,MAAM,UAAU,GAAG,IAAI;gBACrB,CAAC,CAAC,MAAM,mCAAmC,CACvC,OAAO,EACP,IAAI,EACJ,SAAS,EACT,GAAG,EACH,iBAAiB,EACjB,OAAO,EACP,OAAO,CACR;gBACH,CAAC,CAAC,EAAE,IAAI,EAAE,OAAgB,EAAE,CAAC;YAC/B,IAAI,UAAU,CAAC,IAAI,KAAK,gBAAgB;gBAAE,OAAO,UAAU,CAAC,MAAM,CAAC;YACnE,MAAM,QAAQ,GAAG,8BAA8B,CAAC,GAAG,CAAC,CAAC;YACrD,IAAI,MAA2B,CAAC;YAChC,IAAI,CAAC;gBACH,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE;oBACpC,aAAa,EAAE,UAAU,CAAC,aAAa;oBACvC,UAAU,EAAE,QAAQ,CAAC,MAAM;iBAC5B,CAAC,CAAC;YACL,CAAC;oBAAS,CAAC;gBACT,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;YACD,MAAM,iBAAiB,GAAG;gBACxB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,MAAM,EAAE,MAAM,CAAC,KAAK;gBACpB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,MAAM,CAAC,KAAK;aACpB,CAAC;YACF,OAAO,UAAU,CACf;gBACE,SAAS,MAAM,CAAC,IAAI,EAAE;gBACtB,WAAW,MAAM,CAAC,KAAK,EAAE;gBACzB,UAAU,MAAM,CAAC,KAAK,EAAE;gBACxB,WAAW,MAAM,CAAC,MAAM,EAAE;gBAC1B,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACtD,GAAG,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS;oBAC9B,CAAC,CAAC,CAAC,YAAY,MAAM,CAAC,OAAO,EAAE,CAAC;oBAChC,CAAC,CAAC,EAAE,CAAC;gBACP,GAAG,CAAC,MAAM,CAAC,KAAK;oBACd,CAAC,CAAC;wBACE,eAAe,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE;wBAClC,kBAAkB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE;qBACzC;oBACH,CAAC,CAAC,EAAE,CAAC;aACR,CAAC,IAAI,CAAC,IAAI,CAAC,EACZ,iBAAiB,CAClB,CAAC;QACJ,CAAC,CACF,CACJ,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,YAAY,CACjB,kBAAkB,EAClB;QACE,KAAK,EAAE,uBAAuB;QAC9B,WAAW,EAAE,IAAI;YACf,CAAC,CAAC,wkBAAwkB;YAC1kB,CAAC,CAAC,klBAAklB;QACtlB,WAAW,EAAE,yBAAyB;QACtC,WAAW,EAAE;YACX,YAAY,EAAE,KAAK;YACnB,eAAe,EAAE,KAAK;YACtB,cAAc,EAAE,KAAK;YACrB,aAAa,EAAE,KAAK;SACrB;KACF,EACD,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CACnB,MAAM,mBAAmB,CACvB,uBAAuB,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,EACzC,KAAK,IAAI,EAAE;QACT,MAAM,UAAU,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,uBAAuB,EAAE,CAAC,UAAU,CAAC,CAAC;QACjE,MAAM,UAAU,GAAG,IAAI;YACrB,CAAC,CAAC,MAAM,mCAAmC,CACvC,OAAO,EACP,IAAI,EACJ,cAAc,EACd,GAAG,EACH,kBAAkB,EAClB,UAAU,EACV,OAAO,CACR;YACH,CAAC,CAAC,EAAE,IAAI,EAAE,OAAgB,EAAE,CAAC;QAC/B,IAAI,UAAU,CAAC,IAAI,KAAK,gBAAgB;YAAE,OAAO,UAAU,CAAC,MAAM,CAAC;QACnE,MAAM,eAAe,GACnB,UAAU,CAAC,WAAW,KAAK,cAAc;YACvC,CAAC,CAAC,iBAAiB,CAAC,UAAU,CAAC;YAC/B,CAAC,CAAC,UAAU,CAAC;QACjB,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,eAAe,EAAE;YACrD,aAAa,EAAE,UAAU,CAAC,aAAa;SACxC,CAAC,CAAC;QACH,MAAM,WAAW,GAAG;YAClB,cAAc,QAAQ,CAAC,SAAS,EAAE;YAClC,GAAG,CAAC,QAAQ,CAAC,QAAQ;gBACnB,CAAC,CAAC,CAAC,aAAa,oBAAoB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC1D,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QACF,0EAA0E;QAC1E,wEAAwE;QACxE,0EAA0E;QAC1E,sEAAsE;QACtE,OAAO,cAAc,CACnB,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,wBAAwB,CACpD,QAAQ,CAAC,MAAM,EACf;YACE,OAAO,EAAE,OAAO;SACjB,CACF,EAAE,CACJ,CAAC;IACJ,CAAC,CACF,CACJ,CAAC;IAEF,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,CAAC,YAAY,CACjB,sBAAsB,EACtB;YACE,KAAK,EAAE,gCAAgC;YACvC,WAAW,EACT,2FAA2F;YAC7F,WAAW,EAAE,4BAA4B;YACzC,YAAY,EAAE,6BAA6B;YAC3C,WAAW,EAAE;gBACX,YAAY,EAAE,KAAK;gBACnB,eAAe,EAAE,IAAI;gBACrB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,KAAK;aACrB;SACF,EACD,KAAK,EAAE,KAAK,EAAE,EAAE;YACd,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,iBAAiB,GAAG;gBACxB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC;YACF,OAAO,UAAU,CACf,MAAM,CAAC,OAAO;gBACZ,CAAC,CAAC,+BAA+B,MAAM,CAAC,IAAI,EAAE;gBAC9C,CAAC,CAAC,gCAAgC,MAAM,CAAC,IAAI,EAAE,EACjD,iBAAiB,CAClB,CAAC;QACJ,CAAC,CACF,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,CAAC,YAAY,CACjB,cAAc,EACd;YACE,KAAK,EAAE,6BAA6B;YACpC,WAAW,EACT,6eAA6e;YAC/e,WAAW,EAAE,qBAAqB;YAClC,WAAW,EAAE;gBACX,YAAY,EAAE,IAAI;gBAClB,eAAe,EAAE,KAAK;gBACtB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,IAAI;aACpB;SACF,EACD,KAAK,EAAE,KAAK,EAAE,EAAE;YACd,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;YACzC,OAAO,cAAc,CACnB,GAAG,wBAAwB,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,cAAc,CAC/D,QAAQ,CAAC,MAAM,CAChB,EAAE,CACJ,CAAC;QACJ,CAAC,CACF,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,CAAC,YAAY,CACjB,wBAAwB,EACxB;YACE,KAAK,EAAE,gCAAgC;YACvC,WAAW,EACT,mOAAmO;YACrO,WAAW,EAAE,8BAA8B;YAC3C,YAAY,EAAE,+BAA+B;YAC7C,WAAW,EAAE;gBACX,YAAY,EAAE,IAAI;gBAClB,eAAe,EAAE,KAAK;gBACtB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,KAAK;aACrB;SACF,EACD,KAAK,EAAE,KAAK,EAAE,EAAE;YACd,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACpD,OAAO,UAAU,CACf,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC,EAC1C,iBAAiB,CAClB,CAAC;QACJ,CAAC,CACF,CAAC;QAEF,MAAM,CAAC,YAAY,CACjB,yBAAyB,EACzB;YACE,KAAK,EAAE,iCAAiC;YACxC,WAAW,EACT,+FAA+F;YACjG,WAAW,EAAE,+BAA+B;YAC5C,YAAY,EAAE,gCAAgC;YAC9C,WAAW,EAAE;gBACX,YAAY,EAAE,IAAI;gBAClB,eAAe,EAAE,KAAK;gBACtB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,KAAK;aACrB;SACF,EACD,KAAK,IAAI,EAAE;YACT,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,EAAE,CAAC;YAC5C,MAAM,iBAAiB,GAAG;gBACxB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,QAAQ;gBAC1B,aAAa,EAAE,MAAM,CAAC,YAAY;gBAClC,eAAe,EAAE,MAAM,CAAC,cAAc;gBACtC,WAAW,EAAE,MAAM,CAAC,UAAU;gBAC9B,YAAY,EAAE,MAAM,CAAC,WAAW;gBAChC,MAAM,EAAE;oBACN,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;oBAC5B,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,YAAY;iBAC1C;aACF,CAAC;YACF,OAAO,UAAU,CACf,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC,EAC1C,iBAAiB,CAClB,CAAC;QACJ,CAAC,CACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,MAA6B;IACnD,OAAO,MAAM,CAAC,QAAQ,KAAK,cAAc;QACvC,CAAC,CAAC,uHAAuH;QACzH,CAAC,CAAC,EAAE,CAAC;AACT,CAAC;AAED,SAAS,8BAA8B,CAAC,KAAoB;IAI1D,MAAM,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,aAAa,CAAC;IACxD,IAAI,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC/B,IAAI,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAChC,IAAI,sBAAsB,GAAG,KAAK,CAAC;IACnC,OAAO;QACL,MAAM,CAAC,QAAQ;YACb,MAAM,UAAU,GAAG,QAAQ,CAAC,SAAS,EAAE,KAAK,CAAC;YAC7C,IAAI,UAAU,KAAK,WAAW,IAAI,UAAU,KAAK,aAAa,EAAE,CAAC;gBAC/D,sBAAsB,GAAG,IAAI,CAAC;YAChC,CAAC;iBAAM,IAAI,UAAU,KAAK,OAAO,IAAI,QAAQ,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;gBAC/D,sBAAsB,GAAG,KAAK,CAAC;YACjC,CAAC;iBAAM,IACL,sBAAsB;gBACtB,UAAU,KAAK,SAAS;gBACxB,QAAQ,CAAC,KAAK,KAAK,UAAU,EAC7B,CAAC;gBACD,OAAO;YACT,CAAC;YACD,MAAM,OAAO,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAC/C,IACE,aAAa,KAAK,SAAS;gBAC3B,CAAC,OAAO;gBACR,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAC3B,CAAC;gBACD,OAAO;YACT,CAAC;YACD,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,GAAG,CAAC,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YACxD,OAAO,GAAG,OAAO;iBACd,IAAI,CAAC,GAAG,EAAE,CACT,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC;gBAClB,MAAM,EAAE,wBAAwB;gBAChC,MAAM,EAAE;oBACN,aAAa;oBACb,QAAQ,EAAE,aAAa;oBACvB,OAAO;iBACR;aACF,CAAC,CACH;iBACA,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5B,CAAC;QACD,KAAK,CAAC,KAAK;YACT,MAAM,OAAO,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,kCAAkC,GACtC,sEAAsE,CAAC;AAEzE,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7C,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC;QACf,YAAY;QACZ,iBAAiB;QACjB,kBAAkB;QAClB,QAAQ;KACT,CAAC;CACH,CAAC,CAAC;AAUH,KAAK,UAAU,mCAAmC,CAChD,OAA8B,EAC9B,IAAsC,EACtC,WAAuC,EACvC,GAAkB,EAClB,IAAY,EACZ,IAAa,EACb,UAAoC,EAAE;IAEtC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,6BAA6B,EAAE,CAAC,IAAI,CAAC,CAAC;IACrE,IAAI,QAAQ;QAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,CAAC;IAChE,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,aAAa,GAAG,2BAA2B,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACpE,MAAM,WAAW,GAAG,GAAG,CAAC,MAAM,CAAC,YAAY,EAA+B,CAAC;IAC3E,IAAI,WAAW,EAAE,CAAC;QAChB,IAAI,CAAC,kCAAkC,CAAC,WAAW,EAAE,aAAa,CAAC,EAAE,CAAC;YACpE,MAAM,IAAI,aAAa,CAAC,CAAC,KAAK,EAAE,iCAAiC,CAAC,CAAC;QACrE,CAAC;QACD,MAAM,QAAQ,GAAG,aAAa,CAC5B,GAAG,CAAC,MAAM,CAAC,cAAc,EACzB,gCAAgC,CACjC,CAAC;QACF,MAAM,OAAO,GAAG,eAAe,CAC7B,GAAG,CAAC,MAAM,CAAC,cAAc,EACzB,gCAAgC,EAChC,6BAA6B,CAC9B,CAAC;QACF,IACE,QAAQ,CAAC,IAAI,KAAK,QAAQ;YAC1B,QAAQ,CAAC,MAAM,KAAK,QAAQ;YAC5B,CAAC,OAAO,EACR,CAAC;YACD,MAAM,IAAI,KAAK,CACb,uEAAuE,CACxE,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,OAAO,CAAC,uBAAuB,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC;YACnE,MAAM,IAAI,aAAa,CAAC,CAAC,KAAK,EAAE,iCAAiC,CAAC,CAAC;QACrE,CAAC;QACD,IAAI,OAAO,CAAC,QAAQ,KAAK,kBAAkB,IAAI,WAAW,EAAE,CAAC;YAC3D,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,CAAC;QACxD,CAAC;QACD,IACE,OAAO,CAAC,QAAQ,KAAK,YAAY;YACjC,OAAO,CAAC,QAAQ,KAAK,iBAAiB,EACtC,CAAC;YACD,MAAM,IAAI,KAAK,CACb,uEAAuE,CACxE,CAAC;QACJ,CAAC;QACD,MAAM,KAAK,GACT,OAAO,CAAC,QAAQ,KAAK,iBAAiB,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC;QAChE,OAAO;YACL,IAAI,EAAE,OAAO;YACb,aAAa,EAAE,MAAM,OAAO,CAAC,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC;SAC/D,CAAC;IACJ,CAAC;IAED,MAAM,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IACpD,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;IAClE,CAAC;IACD,OAAO;QACL,IAAI,EAAE,gBAAgB;QACtB,MAAM,EAAE,aAAa,CAAC;YACpB,YAAY,EAAE,MAAM,iBAAiB,CAAC,IAAI,CACxC,OAAO,CAAC,uBAAwB,CAAC,KAAK,CAAC,aAAa,CAAC,EACrD,GAAG,CACJ;YACD,aAAa,EAAE;gBACb,8BAA8B,EAAE,aAAa,CAAC,MAAM,CAAC;oBACnD,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,wCAAwC,CAAC;wBAChD,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;wBAC1C,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;wBAC9B,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;wBACxB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;wBAC9B,IAAI,EAAE,6BAA6B,CAAC,IAAI,CAAC,UAAU,CAAC;qBACrD,CAAC;oBACF,eAAe,EAAE;wBACf,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,QAAQ,EAAE;gCACR,IAAI,EAAE,QAAQ;gCACd,KAAK,EAAE,6BAA6B;gCACpC,WAAW,EAAE,kCAAkC;gCAC/C,KAAK,EAAE;oCACL,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE;oCAC5C;wCACE,KAAK,EAAE,iBAAiB;wCACxB,KAAK,EAAE,0BAA0B;qCAClC;oCACD,GAAG,CAAC,WAAW,KAAK,cAAc;wCAChC,CAAC,CAAC;4CACE;gDACE,KAAK,EAAE,kBAAkB;gDACzB,KAAK,EAAE,cAAc;6CACtB;yCACF;wCACH,CAAC,CAAC,EAAE,CAAC;oCACP,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;iCACrC;gCACD,OAAO,EAAE,QAAQ;6BAClB;yBACF;wBACD,QAAQ,EAAE,CAAC,UAAU,CAAC;qBACvB;iBACF,CAAC;aACH;SACF,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CACxB,KAA4B;IAE5B,MAAM,OAAO,GAAG;QACd,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;QACxB,GAAG,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC;KAC5C,CAAC;IACF,OAAO;QACL,GAAG,KAAK;QACR,OAAO,EAAE,SAAS;QAClB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACjE,UAAU,EAAE,KAAK;QACjB,SAAS,EAAE,UAAU;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAC3B,QAAuD;IAEvD,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QACrE,OAAO,QAAQ,CAAC,KAAK,CAAC;IACxB,CAAC;IACD,OAAO,GAAG,QAAQ,CAAC,KAAK,KAAK,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,KAAK,GAAG,CAAC;AACvE,CAAC;AAED,SAAS,iBAAiB,CACxB,MAAiC;IAEjC,OAAO;QACL,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,YAAY,EAAE,MAAM,CAAC,WAAW;QAChC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;YACrB,CAAC,CAAC;gBACE,cAAc,EAAE,MAAM,CAAC,OAAO,CAAC,aAAa;gBAC5C,cAAc,EAAE,MAAM,CAAC,OAAO,CAAC,aAAa;gBAC5C,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,eAAe;gBAChD,aAAa,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW;gBACzC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ;gBAClC,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ;gBACjC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU;gBACrC,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK;aAC5B;YACH,CAAC,CAAC,SAAS;KACd,CAAC;AACJ,CAAC"}

package/dist/mcp/toolset.d.ts

@@ -0,0 +1,5 @@
+export declare const MCP_TOOLSET_ENV = "ZVEC_GREP_MCP_TOOLSET";
+export declare const DEFAULT_MCP_TOOLSET = "agent";
+export type McpToolset = "agent" | "full";
+export declare function parseMcpToolset(value: string): McpToolset;
+export declare function resolveMcpToolset(explicitValue?: string, environmentValue?: string): McpToolset;

package/dist/mcp/toolset.js

@@ -0,0 +1,12 @@
+export const MCP_TOOLSET_ENV = "ZVEC_GREP_MCP_TOOLSET";
+export const DEFAULT_MCP_TOOLSET = "agent";
+export function parseMcpToolset(value) {
+    if (value === "agent" || value === "full")
+        return value;
+    throw new Error(`Unsupported MCP toolset "${value}". Expected "agent" or "full".`);
+}
+export function resolveMcpToolset(explicitValue, environmentValue) {
+    const value = explicitValue ?? environmentValue;
+    return value === undefined ? DEFAULT_MCP_TOOLSET : parseMcpToolset(value);
+}
+//# sourceMappingURL=toolset.js.map

package/dist/mcp/toolset.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolset.js","sourceRoot":"","sources":["../../src/mcp/toolset.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,eAAe,GAAG,uBAAuB,CAAC;AAEvD,MAAM,CAAC,MAAM,mBAAmB,GAAG,OAAO,CAAC;AAI3C,MAAM,UAAU,eAAe,CAAC,KAAa;IAC3C,IAAI,KAAK,KAAK,OAAO,IAAI,KAAK,KAAK,MAAM;QAAE,OAAO,KAAK,CAAC;IACxD,MAAM,IAAI,KAAK,CACb,4BAA4B,KAAK,gCAAgC,CAClE,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,aAAsB,EACtB,gBAAyB;IAEzB,MAAM,KAAK,GAAG,aAAa,IAAI,gBAAgB,CAAC;IAChD,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;AAC5E,CAAC"}

package/dist/observability/trace-context.d.ts

@@ -0,0 +1,11 @@
+export type TraceContext = {
+    traceparent: string;
+    traceId: string;
+    tracestate?: string;
+    baggage?: string;
+};
+export declare function currentTraceContext(): TraceContext | undefined;
+export declare function runWithTraceContext<T>(context: TraceContext | undefined, operation: () => T): T;
+export declare function traceContextFromMcpBody(body: unknown): TraceContext | undefined;
+export declare function traceContextFromMcpMeta(meta: unknown): TraceContext | undefined;
+export declare function traceHeaders(): Record<string, string>;

package/dist/observability/trace-context.js

@@ -0,0 +1,138 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+const storage = new AsyncLocalStorage();
+const TRACEPARENT = /^([0-9a-f]{2})-([0-9a-f]{32})-([0-9a-f]{16})-([0-9a-f]{2})((?:-[0-9a-f]+)*)$/;
+export function currentTraceContext() {
+    return storage.getStore();
+}
+export function runWithTraceContext(context, operation) {
+    return context ? storage.run(context, operation) : operation();
+}
+export function traceContextFromMcpBody(body) {
+    if (!body || typeof body !== "object" || Array.isArray(body)) {
+        return undefined;
+    }
+    return traceContextFromMcpMeta(body.params?._meta);
+}
+export function traceContextFromMcpMeta(meta) {
+    if (!meta || typeof meta !== "object" || Array.isArray(meta)) {
+        return undefined;
+    }
+    const values = meta;
+    const traceparent = typeof values.traceparent === "string" ? values.traceparent : undefined;
+    const match = traceparent?.match(TRACEPARENT);
+    if (!match ||
+        match[1] === "ff" ||
+        (match[1] === "00" && match[5] !== "") ||
+        match[2] === "0".repeat(32) ||
+        match[3] === "0".repeat(16)) {
+        return undefined;
+    }
+    const tracestate = validTracestate(values.tracestate);
+    const baggage = validBaggage(values.baggage);
+    return {
+        traceparent: traceparent,
+        traceId: match[2],
+        ...(tracestate ? { tracestate } : {}),
+        ...(baggage ? { baggage } : {}),
+    };
+}
+export function traceHeaders() {
+    const context = currentTraceContext();
+    if (!context)
+        return {};
+    return {
+        traceparent: context.traceparent,
+        ...(context.tracestate ? { tracestate: context.tracestate } : {}),
+        ...(context.baggage ? { baggage: context.baggage } : {}),
+    };
+}
+function boundedHeader(value, maxLength, allowHorizontalTab = false) {
+    if (typeof value !== "string" || value.length > maxLength)
+        return undefined;
+    for (const character of value) {
+        const code = character.charCodeAt(0);
+        if (code > 0x7e ||
+            (code < 0x20 && !(allowHorizontalTab && code === 0x09))) {
+            return undefined;
+        }
+    }
+    return value;
+}
+function validTracestate(value) {
+    const header = boundedHeader(value, 512);
+    if (!header)
+        return undefined;
+    const members = header.split(",");
+    if (members.length > 32)
+        return undefined;
+    const keys = new Set();
+    for (const rawMember of members) {
+        const member = rawMember.trim();
+        const separator = member.indexOf("=");
+        if (separator <= 0)
+            return undefined;
+        const key = member.slice(0, separator);
+        const item = member.slice(separator + 1);
+        if (!/^(?:[a-z][a-z0-9_\-*/]{0,255}|[a-z0-9][a-z0-9_\-*/]{0,240}@[a-z][a-z0-9_\-*/]{0,13})$/.test(key) ||
+            item.length === 0 ||
+            item.length > 256 ||
+            !/^[\x20-\x2b\x2d-\x3c\x3e-\x7e]+$/.test(item) ||
+            item.endsWith(" ") ||
+            keys.has(key)) {
+            return undefined;
+        }
+        keys.add(key);
+    }
+    return header;
+}
+function validBaggage(value) {
+    const header = boundedHeader(value, 8_192, true);
+    if (!header)
+        return undefined;
+    const members = header.split(",");
+    if (members.length > 64)
+        return undefined;
+    for (const rawMember of members) {
+        const segments = rawMember.trim().split(";");
+        const pair = segments.shift();
+        const separator = pair.indexOf("=");
+        const pairValue = separator < 0 ? "" : pair.slice(separator + 1).trim();
+        if (separator <= 0 ||
+            !isHttpToken(pair.slice(0, separator).trim()) ||
+            hasInvalidBaggageCharacters(pairValue, ",")) {
+            return undefined;
+        }
+        for (const rawProperty of segments) {
+            const property = rawProperty.trim();
+            const propertySeparator = property.indexOf("=");
+            const key = propertySeparator < 0
+                ? property
+                : property.slice(0, propertySeparator).trim();
+            const propertyValue = propertySeparator < 0
+                ? ""
+                : property.slice(propertySeparator + 1).trim();
+            if (!isHttpToken(key) ||
+                hasInvalidBaggageCharacters(propertyValue, ";,")) {
+                return undefined;
+            }
+        }
+    }
+    return header;
+}
+function isHttpToken(value) {
+    return /^[!#$%&'*+\-.^_`|~0-9A-Za-z]+$/.test(value);
+}
+function hasInvalidBaggageCharacters(value, delimiters) {
+    for (const character of value) {
+        const code = character.charCodeAt(0);
+        if (code <= 0x20 ||
+            code === 0x22 ||
+            code === 0x5c ||
+            code === 0x7f ||
+            delimiters.includes(character)) {
+            return true;
+        }
+    }
+    return false;
+}
+//# sourceMappingURL=trace-context.js.map