@zvec/zvec-grep 0.1.4 → 0.1.6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +86 -162
- package/README_CN.md +80 -162
- package/dist/authorization/index.d.ts +7 -0
- package/dist/authorization/index.js +7 -0
- package/dist/authorization/index.js.map +1 -0
- package/dist/authorization/manager.d.ts +8 -0
- package/dist/authorization/manager.js +19 -0
- package/dist/authorization/manager.js.map +1 -0
- package/dist/authorization/operation.d.ts +7 -0
- package/dist/authorization/operation.js +50 -0
- package/dist/authorization/operation.js.map +1 -0
- package/dist/authorization/planner.d.ts +21 -0
- package/dist/authorization/planner.js +97 -0
- package/dist/authorization/planner.js.map +1 -0
- package/dist/authorization/prompt.d.ts +11 -0
- package/dist/authorization/prompt.js +62 -0
- package/dist/authorization/prompt.js.map +1 -0
- package/dist/authorization/store.d.ts +20 -0
- package/dist/authorization/store.js +251 -0
- package/dist/authorization/store.js.map +1 -0
- package/dist/authorization/target.d.ts +15 -0
- package/dist/authorization/target.js +49 -0
- package/dist/authorization/target.js.map +1 -0
- package/dist/authorization/types.d.ts +61 -0
- package/dist/authorization/types.js +2 -0
- package/dist/authorization/types.js.map +1 -0
- package/dist/cli/args.d.ts +3 -0
- package/dist/cli/args.js +749 -201
- package/dist/cli/args.js.map +1 -1
- package/dist/cli/auth.d.ts +21 -0
- package/dist/cli/auth.js +212 -0
- package/dist/cli/auth.js.map +1 -0
- package/dist/cli/commands.d.ts +3 -1
- package/dist/cli/commands.js +529 -180
- package/dist/cli/commands.js.map +1 -1
- package/dist/cli/errors.js +7 -6
- package/dist/cli/errors.js.map +1 -1
- package/dist/cli/format/context.d.ts +1 -0
- package/dist/cli/format/context.js +270 -70
- package/dist/cli/format/context.js.map +1 -1
- package/dist/cli/format/debug.js +14 -16
- package/dist/cli/format/debug.js.map +1 -1
- package/dist/cli/format/highlight.js.map +1 -1
- package/dist/cli/format/progress.d.ts +4 -0
- package/dist/cli/format/progress.js +222 -44
- package/dist/cli/format/progress.js.map +1 -1
- package/dist/cli/format/range.js.map +1 -1
- package/dist/cli/format/status.d.ts +77 -6
- package/dist/cli/format/status.js +343 -114
- package/dist/cli/format/status.js.map +1 -1
- package/dist/cli/help.d.ts +1 -1
- package/dist/cli/help.js +368 -84
- package/dist/cli/help.js.map +1 -1
- package/dist/cli/index.js +4 -4
- package/dist/cli/index.js.map +1 -1
- package/dist/cli/install.d.ts +4 -0
- package/dist/cli/install.js +845 -0
- package/dist/cli/install.js.map +1 -0
- package/dist/cli/managed-rg.d.ts +10 -0
- package/dist/cli/managed-rg.js +329 -0
- package/dist/cli/managed-rg.js.map +1 -0
- package/dist/cli/types.d.ts +40 -13
- package/dist/cli/types.js.map +1 -1
- package/dist/cli/version.js +3 -1
- package/dist/cli/version.js.map +1 -1
- package/dist/client/daemon-client.d.ts +19 -0
- package/dist/client/daemon-client.js +180 -0
- package/dist/client/daemon-client.js.map +1 -0
- package/dist/client/mode-router.d.ts +9 -0
- package/dist/client/mode-router.js +28 -0
- package/dist/client/mode-router.js.map +1 -0
- package/dist/client/search-policy.d.ts +11 -0
- package/dist/client/search-policy.js +15 -0
- package/dist/client/search-policy.js.map +1 -0
- package/dist/daemon/backend.d.ts +76 -0
- package/dist/daemon/backend.js +1049 -0
- package/dist/daemon/backend.js.map +1 -0
- package/dist/daemon/change-set.d.ts +25 -0
- package/dist/daemon/change-set.js +101 -0
- package/dist/daemon/change-set.js.map +1 -0
- package/dist/daemon/config.d.ts +24 -0
- package/dist/daemon/config.js +88 -0
- package/dist/daemon/config.js.map +1 -0
- package/dist/daemon/errors.d.ts +5 -0
- package/dist/daemon/errors.js +11 -0
- package/dist/daemon/errors.js.map +1 -0
- package/dist/daemon/http-server.d.ts +30 -0
- package/dist/daemon/http-server.js +287 -0
- package/dist/daemon/http-server.js.map +1 -0
- package/dist/daemon/index-coordinator.d.ts +23 -0
- package/dist/daemon/index-coordinator.js +68 -0
- package/dist/daemon/index-coordinator.js.map +1 -0
- package/dist/daemon/job-scheduler.d.ts +70 -0
- package/dist/daemon/job-scheduler.js +366 -0
- package/dist/daemon/job-scheduler.js.map +1 -0
- package/dist/daemon/logger.d.ts +9 -0
- package/dist/daemon/logger.js +53 -0
- package/dist/daemon/logger.js.map +1 -0
- package/dist/daemon/model-pool.d.ts +45 -0
- package/dist/daemon/model-pool.js +187 -0
- package/dist/daemon/model-pool.js.map +1 -0
- package/dist/daemon/root-lease.d.ts +14 -0
- package/dist/daemon/root-lease.js +188 -0
- package/dist/daemon/root-lease.js.map +1 -0
- package/dist/daemon/root-runtime.d.ts +78 -0
- package/dist/daemon/root-runtime.js +333 -0
- package/dist/daemon/root-runtime.js.map +1 -0
- package/dist/daemon/runtime-manager.d.ts +45 -0
- package/dist/daemon/runtime-manager.js +243 -0
- package/dist/daemon/runtime-manager.js.map +1 -0
- package/dist/daemon/runtime.d.ts +12 -0
- package/dist/daemon/runtime.js +102 -0
- package/dist/daemon/runtime.js.map +1 -0
- package/dist/daemon/server-controller.d.ts +39 -0
- package/dist/daemon/server-controller.js +203 -0
- package/dist/daemon/server-controller.js.map +1 -0
- package/dist/daemon/watch-manager.d.ts +50 -0
- package/dist/daemon/watch-manager.js +293 -0
- package/dist/daemon/watch-manager.js.map +1 -0
- package/dist/daemon/workspace-read-session-cache.d.ts +31 -0
- package/dist/daemon/workspace-read-session-cache.js +115 -0
- package/dist/daemon/workspace-read-session-cache.js.map +1 -0
- package/dist/engine/config.d.ts +50 -0
- package/dist/engine/config.js +321 -0
- package/dist/engine/config.js.map +1 -0
- package/dist/engine/{errors/index.d.ts → errors.d.ts} +6 -1
- package/dist/engine/{errors/details.js → errors.js} +17 -4
- package/dist/engine/errors.js.map +1 -0
- package/dist/engine/extraction/code/adapter.d.ts +1 -1
- package/dist/engine/extraction/code/adapter.js.map +1 -1
- package/dist/engine/extraction/code/extractor.d.ts +5 -11
- package/dist/engine/extraction/code/extractor.js +184 -109
- package/dist/engine/extraction/code/extractor.js.map +1 -1
- package/dist/engine/extraction/code/families/c-family.js +13 -7
- package/dist/engine/extraction/code/families/c-family.js.map +1 -1
- package/dist/engine/extraction/code/families/js-ts.d.ts +1 -1
- package/dist/engine/extraction/code/families/js-ts.js +24 -18
- package/dist/engine/extraction/code/families/js-ts.js.map +1 -1
- package/dist/engine/extraction/code/families/metadata.d.ts +1 -1
- package/dist/engine/extraction/code/families/metadata.js +11 -8
- package/dist/engine/extraction/code/families/metadata.js.map +1 -1
- package/dist/engine/extraction/code/families/name-field.js +5 -5
- package/dist/engine/extraction/code/families/name-field.js.map +1 -1
- package/dist/engine/extraction/code/languages/c.js +1 -4
- package/dist/engine/extraction/code/languages/c.js.map +1 -1
- package/dist/engine/extraction/code/languages/cpp.js.map +1 -1
- package/dist/engine/extraction/code/languages/go.js +1 -3
- package/dist/engine/extraction/code/languages/go.js.map +1 -1
- package/dist/engine/extraction/code/languages/java.js.map +1 -1
- package/dist/engine/extraction/code/languages/javascript.js +1 -3
- package/dist/engine/extraction/code/languages/javascript.js.map +1 -1
- package/dist/engine/extraction/code/languages/python.js +5 -6
- package/dist/engine/extraction/code/languages/python.js.map +1 -1
- package/dist/engine/extraction/code/languages/rust.js +1 -5
- package/dist/engine/extraction/code/languages/rust.js.map +1 -1
- package/dist/engine/extraction/code/languages/typescript.js.map +1 -1
- package/dist/engine/extraction/code/tree-sitter/grammar.js.map +1 -0
- package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/nodes.js +2 -1
- package/dist/engine/extraction/code/tree-sitter/nodes.js.map +1 -0
- package/dist/engine/extraction/code/tree-sitter/parser.js.map +1 -0
- package/dist/engine/extraction/ids.js.map +1 -1
- package/dist/engine/extraction/image/extractor.d.ts +5 -0
- package/dist/engine/extraction/image/extractor.js +35 -0
- package/dist/engine/extraction/image/extractor.js.map +1 -0
- package/dist/engine/extraction/index.d.ts +3 -44
- package/dist/engine/extraction/index.js +1 -280
- package/dist/engine/extraction/index.js.map +1 -1
- package/dist/engine/extraction/markdown/extractor.d.ts +5 -11
- package/dist/engine/extraction/markdown/extractor.js +37 -16
- package/dist/engine/extraction/markdown/extractor.js.map +1 -1
- package/dist/engine/extraction/runtime.d.ts +4 -0
- package/dist/engine/extraction/runtime.js +26 -0
- package/dist/engine/extraction/runtime.js.map +1 -0
- package/dist/engine/extraction/source.d.ts +18 -0
- package/dist/engine/extraction/source.js +22 -0
- package/dist/engine/extraction/source.js.map +1 -0
- package/dist/engine/extraction/text/extractor.d.ts +7 -0
- package/dist/engine/extraction/text/extractor.js +168 -0
- package/dist/engine/extraction/text/extractor.js.map +1 -0
- package/dist/engine/extraction/types.d.ts +4 -0
- package/dist/engine/{models → extraction}/types.js.map +1 -1
- package/dist/engine/{files/file-type.d.ts → file-type.d.ts} +1 -1
- package/dist/engine/file-type.js.map +1 -0
- package/dist/engine/index-status.d.ts +9 -0
- package/dist/engine/index-status.js +38 -0
- package/dist/engine/index-status.js.map +1 -0
- package/dist/engine/manifest.d.ts +12 -0
- package/dist/engine/manifest.js +120 -0
- package/dist/engine/manifest.js.map +1 -0
- package/dist/engine/models/{providers/llama-cpp/embedding.d.ts → backends/llama-cpp.d.ts} +25 -20
- package/dist/engine/models/{providers/llama-cpp/embedding.js → backends/llama-cpp.js} +165 -113
- package/dist/engine/models/backends/llama-cpp.js.map +1 -0
- package/dist/engine/models/backends/model2vec.d.ts +57 -0
- package/dist/engine/models/backends/model2vec.js +332 -0
- package/dist/engine/models/backends/model2vec.js.map +1 -0
- package/dist/engine/models/backends/qwen.d.ts +44 -0
- package/dist/engine/models/{providers/qwen/embedding.js → backends/qwen.js} +168 -102
- package/dist/engine/models/backends/qwen.js.map +1 -0
- package/dist/engine/models/backends/transformers-js.d.ts +87 -0
- package/dist/engine/models/backends/transformers-js.js +316 -0
- package/dist/engine/models/backends/transformers-js.js.map +1 -0
- package/dist/engine/models/catalog.d.ts +188 -23
- package/dist/engine/models/catalog.js +172 -41
- package/dist/engine/models/catalog.js.map +1 -1
- package/dist/engine/models/download-progress.d.ts +15 -0
- package/dist/engine/models/download-progress.js +48 -0
- package/dist/engine/models/download-progress.js.map +1 -0
- package/dist/engine/models/embeddings.d.ts +58 -18
- package/dist/engine/models/embeddings.js +53 -29
- package/dist/engine/models/embeddings.js.map +1 -1
- package/dist/engine/models/factory.d.ts +2 -5
- package/dist/engine/models/factory.js +41 -56
- package/dist/engine/models/factory.js.map +1 -1
- package/dist/engine/models/index.d.ts +4 -6
- package/dist/engine/models/index.js +4 -4
- package/dist/engine/models/index.js.map +1 -1
- package/dist/engine/models/ranking.d.ts +7 -3
- package/dist/engine/models/resolution.d.ts +8 -0
- package/dist/engine/models/resolution.js +22 -0
- package/dist/engine/models/resolution.js.map +1 -0
- package/dist/engine/pipeline/indexing/index.d.ts +9 -7
- package/dist/engine/pipeline/indexing/index.js +399 -147
- package/dist/engine/pipeline/indexing/index.js.map +1 -1
- package/dist/engine/pipeline/indexing/root-paths.js +16 -19
- package/dist/engine/pipeline/indexing/root-paths.js.map +1 -1
- package/dist/engine/pipeline/indexing/scanner/index.d.ts +6 -1
- package/dist/engine/pipeline/indexing/scanner/index.js +318 -66
- package/dist/engine/pipeline/indexing/scanner/index.js.map +1 -1
- package/dist/engine/pipeline/search/index.d.ts +6 -6
- package/dist/engine/pipeline/search/index.js +76 -34
- package/dist/engine/pipeline/search/index.js.map +1 -1
- package/dist/engine/service/index.d.ts +3 -2
- package/dist/engine/service/index.js +1 -1
- package/dist/engine/service/index.js.map +1 -1
- package/dist/engine/service/lexical.d.ts +14 -15
- package/dist/engine/service/lexical.js +78 -119
- package/dist/engine/service/lexical.js.map +1 -1
- package/dist/engine/service/root.d.ts +13 -12
- package/dist/engine/service/root.js +35 -24
- package/dist/engine/service/root.js.map +1 -1
- package/dist/engine/service/structure-enrichment.js +29 -25
- package/dist/engine/service/structure-enrichment.js.map +1 -1
- package/dist/engine/service/types.d.ts +61 -35
- package/dist/engine/service/workspace-index.d.ts +24 -0
- package/dist/engine/service/workspace-index.js +151 -0
- package/dist/engine/service/workspace-index.js.map +1 -0
- package/dist/engine/service/zvec-grep.d.ts +12 -1
- package/dist/engine/service/zvec-grep.js +878 -544
- package/dist/engine/service/zvec-grep.js.map +1 -1
- package/dist/engine/storage/index.d.ts +8 -4
- package/dist/engine/storage/index.js +1 -1
- package/dist/engine/storage/index.js.map +1 -1
- package/dist/engine/storage/zvec.d.ts +10 -33
- package/dist/engine/storage/zvec.js +110 -49
- package/dist/engine/storage/zvec.js.map +1 -1
- package/dist/engine/types.d.ts +29 -23
- package/dist/engine/types.js +1 -1
- package/dist/engine/types.js.map +1 -1
- package/dist/engine/utils/daemon-lease.d.ts +17 -0
- package/dist/engine/utils/daemon-lease.js +175 -0
- package/dist/engine/utils/daemon-lease.js.map +1 -0
- package/dist/engine/utils/file-selection.d.ts +10 -0
- package/dist/engine/utils/file-selection.js +157 -0
- package/dist/engine/utils/file-selection.js.map +1 -0
- package/dist/engine/utils/glob.d.ts +3 -0
- package/dist/engine/utils/glob.js +116 -18
- package/dist/engine/utils/glob.js.map +1 -1
- package/dist/engine/utils/hash.js.map +1 -1
- package/dist/engine/utils/json.d.ts +6 -2
- package/dist/engine/utils/json.js +50 -12
- package/dist/engine/utils/json.js.map +1 -1
- package/dist/engine/utils/lock.js +7 -7
- package/dist/engine/utils/lock.js.map +1 -1
- package/dist/engine/utils/path.js +2 -2
- package/dist/engine/utils/path.js.map +1 -1
- package/dist/engine/utils/timing.js.map +1 -1
- package/dist/index-progress.d.ts +10 -0
- package/dist/index-progress.js +159 -0
- package/dist/index-progress.js.map +1 -0
- package/dist/index.d.ts +6 -6
- package/dist/index.js +3 -3
- package/dist/index.js.map +1 -1
- package/dist/mcp/http-transport.d.ts +27 -0
- package/dist/mcp/http-transport.js +182 -0
- package/dist/mcp/http-transport.js.map +1 -0
- package/dist/mcp/input-normalization.d.ts +38 -0
- package/dist/mcp/input-normalization.js +95 -0
- package/dist/mcp/input-normalization.js.map +1 -0
- package/dist/mcp/progress-heartbeat.d.ts +17 -0
- package/dist/mcp/progress-heartbeat.js +46 -0
- package/dist/mcp/progress-heartbeat.js.map +1 -0
- package/dist/mcp/request-metadata.d.ts +2 -0
- package/dist/mcp/request-metadata.js +11 -0
- package/dist/mcp/request-metadata.js.map +1 -0
- package/dist/mcp/request-state.d.ts +40 -0
- package/dist/mcp/request-state.js +210 -0
- package/dist/mcp/request-state.js.map +1 -0
- package/dist/mcp/result-format.d.ts +13 -0
- package/dist/mcp/result-format.js +75 -0
- package/dist/mcp/result-format.js.map +1 -0
- package/dist/mcp/schemas.d.ts +430 -0
- package/dist/mcp/schemas.js +447 -0
- package/dist/mcp/schemas.js.map +1 -0
- package/dist/mcp/tools.d.ts +150 -0
- package/dist/mcp/tools.js +448 -0
- package/dist/mcp/tools.js.map +1 -0
- package/dist/mcp/toolset.d.ts +5 -0
- package/dist/mcp/toolset.js +12 -0
- package/dist/mcp/toolset.js.map +1 -0
- package/dist/observability/trace-context.d.ts +11 -0
- package/dist/observability/trace-context.js +138 -0
- package/dist/observability/trace-context.js.map +1 -0
- package/docs/01-agents.md +111 -0
- package/docs/02-cli.md +229 -0
- package/docs/03-mcp.md +167 -0
- package/docs/04-pipeline.md +218 -0
- package/docs/05-architecture.md +86 -0
- package/docs/06-server.md +210 -0
- package/docs/07-embedding.md +179 -0
- package/docs/08-roadmap.md +85 -0
- package/docs/README.md +53 -0
- package/package.json +43 -7
- package/skills/zvec-grep/SKILL.md +43 -123
- package/skills/zvec-grep/agents/openai.yaml +2 -2
- package/skills/zvec-grep/references/cli-fallback.md +85 -0
- package/dist/engine/collection/index.d.ts +0 -55
- package/dist/engine/collection/index.js +0 -710
- package/dist/engine/collection/index.js.map +0 -1
- package/dist/engine/errors/details.d.ts +0 -6
- package/dist/engine/errors/details.js.map +0 -1
- package/dist/engine/errors/index.js +0 -16
- package/dist/engine/errors/index.js.map +0 -1
- package/dist/engine/extraction/tree-sitter/grammar.js.map +0 -1
- package/dist/engine/extraction/tree-sitter/nodes.js.map +0 -1
- package/dist/engine/extraction/tree-sitter/parser.js.map +0 -1
- package/dist/engine/files/file-type.js.map +0 -1
- package/dist/engine/models/providers/index.d.ts +0 -2
- package/dist/engine/models/providers/index.js +0 -3
- package/dist/engine/models/providers/index.js.map +0 -1
- package/dist/engine/models/providers/llama-cpp/embedding.js.map +0 -1
- package/dist/engine/models/providers/llama-cpp/index.d.ts +0 -1
- package/dist/engine/models/providers/llama-cpp/index.js +0 -2
- package/dist/engine/models/providers/llama-cpp/index.js.map +0 -1
- package/dist/engine/models/providers/qwen/embedding.d.ts +0 -42
- package/dist/engine/models/providers/qwen/embedding.js.map +0 -1
- package/dist/engine/models/providers/qwen/index.d.ts +0 -1
- package/dist/engine/models/providers/qwen/index.js +0 -2
- package/dist/engine/models/providers/qwen/index.js.map +0 -1
- package/dist/engine/models/types.d.ts +0 -30
- /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/grammar.d.ts +0 -0
- /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/grammar.js +0 -0
- /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/nodes.d.ts +0 -0
- /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/parser.d.ts +0 -0
- /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/parser.js +0 -0
- /package/dist/engine/{models → extraction}/types.js +0 -0
- /package/dist/engine/{files/file-type.js → file-type.js} +0 -0
|
@@ -1,5 +1,6 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
1
|
+
import { randomUUID } from "node:crypto";
|
|
2
|
+
import { chmodSync, mkdirSync, readFileSync, renameSync, unlinkSync, writeFileSync, } from "node:fs";
|
|
3
|
+
import { chmod, mkdir, readFile, rename, unlink, writeFile, } from "node:fs/promises";
|
|
3
4
|
import { dirname } from "node:path";
|
|
4
5
|
export async function readJsonFile(path, fallback) {
|
|
5
6
|
try {
|
|
@@ -25,19 +26,56 @@ export function readJsonFileSync(path, fallback) {
|
|
|
25
26
|
throw error;
|
|
26
27
|
}
|
|
27
28
|
}
|
|
28
|
-
export async function writeJsonFile(path, value) {
|
|
29
|
-
|
|
30
|
-
|
|
29
|
+
export async function writeJsonFile(path, value, options = {}) {
|
|
30
|
+
const directory = dirname(path);
|
|
31
|
+
await mkdir(directory, { recursive: true, mode: options.directoryMode });
|
|
32
|
+
if (options.directoryMode !== undefined) {
|
|
33
|
+
await chmod(directory, options.directoryMode);
|
|
34
|
+
}
|
|
35
|
+
const tmpPath = `${path}.${process.pid}.${randomUUID()}.tmp`;
|
|
31
36
|
const text = `${JSON.stringify(value, null, 2)}\n`;
|
|
32
|
-
|
|
33
|
-
|
|
37
|
+
try {
|
|
38
|
+
await writeFile(tmpPath, text, {
|
|
39
|
+
encoding: "utf8",
|
|
40
|
+
mode: options.fileMode,
|
|
41
|
+
});
|
|
42
|
+
if (options.fileMode !== undefined) {
|
|
43
|
+
await chmod(tmpPath, options.fileMode);
|
|
44
|
+
}
|
|
45
|
+
await rename(tmpPath, path);
|
|
46
|
+
}
|
|
47
|
+
catch (error) {
|
|
48
|
+
await unlink(tmpPath).catch(() => undefined);
|
|
49
|
+
throw error;
|
|
50
|
+
}
|
|
34
51
|
}
|
|
35
|
-
export function writeJsonFileSync(path, value) {
|
|
36
|
-
|
|
37
|
-
|
|
52
|
+
export function writeJsonFileSync(path, value, options = {}) {
|
|
53
|
+
const directory = dirname(path);
|
|
54
|
+
mkdirSync(directory, { recursive: true, mode: options.directoryMode });
|
|
55
|
+
if (options.directoryMode !== undefined) {
|
|
56
|
+
chmodSync(directory, options.directoryMode);
|
|
57
|
+
}
|
|
58
|
+
const tmpPath = `${path}.${process.pid}.${randomUUID()}.tmp`;
|
|
38
59
|
const text = `${JSON.stringify(value, null, 2)}\n`;
|
|
39
|
-
|
|
40
|
-
|
|
60
|
+
try {
|
|
61
|
+
writeFileSync(tmpPath, text, {
|
|
62
|
+
encoding: "utf8",
|
|
63
|
+
mode: options.fileMode,
|
|
64
|
+
});
|
|
65
|
+
if (options.fileMode !== undefined) {
|
|
66
|
+
chmodSync(tmpPath, options.fileMode);
|
|
67
|
+
}
|
|
68
|
+
renameSync(tmpPath, path);
|
|
69
|
+
}
|
|
70
|
+
catch (error) {
|
|
71
|
+
try {
|
|
72
|
+
unlinkSync(tmpPath);
|
|
73
|
+
}
|
|
74
|
+
catch {
|
|
75
|
+
// The temporary file may not have been created.
|
|
76
|
+
}
|
|
77
|
+
throw error;
|
|
78
|
+
}
|
|
41
79
|
}
|
|
42
80
|
function isNodeError(error) {
|
|
43
81
|
return typeof error === "object" && error !== null && "code" in error;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"json.js","sourceRoot":"","sources":["../../../src/engine/utils/json.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"json.js","sourceRoot":"","sources":["../../../src/engine/utils/json.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EACL,SAAS,EACT,SAAS,EACT,YAAY,EACZ,UAAU,EACV,UAAU,EACV,aAAa,GACd,MAAM,SAAS,CAAC;AACjB,OAAO,EACL,KAAK,EACL,KAAK,EACL,QAAQ,EACR,MAAM,EACN,MAAM,EACN,SAAS,GACV,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAOpC,MAAM,CAAC,KAAK,UAAU,YAAY,CAAI,IAAY,EAAE,QAAW;IAC7D,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC1C,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAM,CAAC;IAC/B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,WAAW,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAI,IAAY,EAAE,QAAW;IAC3D,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAM,CAAC;IAC/B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,WAAW,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,IAAY,EACZ,KAAc,EACd,UAA4B,EAAE;IAE9B,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChC,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;IACzE,IAAI,OAAO,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;QACxC,MAAM,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,IAAI,IAAI,OAAO,CAAC,GAAG,IAAI,UAAU,EAAE,MAAM,CAAC;IAC7D,MAAM,IAAI,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC;IAEnD,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE;YAC7B,QAAQ,EAAE,MAAM;YAChB,IAAI,EAAE,OAAO,CAAC,QAAQ;SACvB,CAAC,CAAC;QACH,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACnC,MAAM,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;QACzC,CAAC;QACD,MAAM,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC7C,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,IAAY,EACZ,KAAc,EACd,UAA4B,EAAE;IAE9B,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;IACvE,IAAI,OAAO,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;QACxC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,IAAI,IAAI,OAAO,CAAC,GAAG,IAAI,UAAU,EAAE,MAAM,CAAC;IAC7D,MAAM,IAAI,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC;IAEnD,IAAI,CAAC;QACH,aAAa,CAAC,OAAO,EAAE,IAAI,EAAE;YAC3B,QAAQ,EAAE,MAAM;YAChB,IAAI,EAAE,OAAO,CAAC,QAAQ;SACvB,CAAC,CAAC;QACH,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACnC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC;QACD,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC5B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC;YACH,UAAU,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,gDAAgD;QAClD,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK,CAAC;AACxE,CAAC"}
|
|
@@ -1,8 +1,8 @@
|
|
|
1
1
|
import { randomUUID } from "node:crypto";
|
|
2
|
-
import { existsSync, mkdirSync, readdirSync, readFileSync, rmSync, statSync, writeFileSync } from "node:fs";
|
|
2
|
+
import { existsSync, mkdirSync, readdirSync, readFileSync, rmSync, statSync, writeFileSync, } from "node:fs";
|
|
3
3
|
import { hostname } from "node:os";
|
|
4
4
|
import { dirname, join } from "node:path";
|
|
5
|
-
import { detail, EngineError, errorDetails } from "../errors
|
|
5
|
+
import { detail, EngineError, errorDetails } from "../errors.js";
|
|
6
6
|
const LOCK_INFO_FILE = "lock.json";
|
|
7
7
|
const DEFAULT_STALE_LOCK_MS = 6 * 60 * 60 * 1000;
|
|
8
8
|
function acquireExclusiveDirectoryLock(lockPath, options) {
|
|
@@ -174,11 +174,11 @@ function processIsAlive(pid) {
|
|
|
174
174
|
function readLockInfo(lockPath) {
|
|
175
175
|
try {
|
|
176
176
|
const parsed = JSON.parse(readFileSync(lockInfoPath(lockPath), "utf8"));
|
|
177
|
-
if (typeof parsed.token === "string"
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
177
|
+
if (typeof parsed.token === "string" &&
|
|
178
|
+
typeof parsed.pid === "number" &&
|
|
179
|
+
typeof parsed.hostname === "string" &&
|
|
180
|
+
typeof parsed.startedAt === "number" &&
|
|
181
|
+
typeof parsed.operation === "string") {
|
|
182
182
|
return parsed;
|
|
183
183
|
}
|
|
184
184
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"lock.js","sourceRoot":"","sources":["../../../src/engine/utils/lock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,
|
|
1
|
+
{"version":3,"file":"lock.js","sourceRoot":"","sources":["../../../src/engine/utils/lock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EACL,UAAU,EACV,SAAS,EACT,WAAW,EACX,YAAY,EACZ,MAAM,EACN,QAAQ,EACR,aAAa,GACd,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACnC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAqBjE,MAAM,cAAc,GAAG,WAAW,CAAC;AACnC,MAAM,qBAAqB,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAEjD,SAAS,6BAA6B,CACpC,QAAgB,EAChB,OAAwB;IAExB,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAElD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,SAAS,CAAC,QAAQ,CAAC,CAAC;YACpB,MAAM,IAAI,GAAG,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAChD,aAAa,CACX,YAAY,CAAC,QAAQ,CAAC,EACtB,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EACpC,MAAM,CACP,CAAC;YAEF,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,IAAI;gBACJ,OAAO,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC;aAC/C,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACnD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IACE,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,OAAO,IAAI,qBAAqB,CAAC,EACpE,CAAC;gBACD,SAAS;YACX,CAAC;YAED,MAAM,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,MAAM,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,gCAAgC,CACvC,QAAgB,EAChB,SAAiB;IAEjB,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,OAAO;IACT,CAAC;IAED,IAAI,gBAAgB,CAAC,QAAQ,EAAE,qBAAqB,CAAC,EAAE,CAAC;QACtD,OAAO;IACT,CAAC;IAED,MAAM,aAAa,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,QAAgB,EAChB,IAAsB,EACtB,OAAwB;IAExB,OAAO,IAAI,KAAK,MAAM;QACpB,CAAC,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC;QACpC,CAAC,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,QAAgB,EAAE,SAAiB;IACnE,gCAAgC,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;AACvE,CAAC;AAED,SAAS,eAAe,CAAC,SAAiB;IACxC,OAAO;QACL,KAAK,EAAE,UAAU,EAAE;QACnB,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,QAAQ,EAAE,QAAQ,EAAE;QACpB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,SAAS;KACV,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB,EAAE,OAAwB;IACjE,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,qBAAqB,CAAC;IACzD,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;IAE1C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;QAC7C,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1B,IAAI,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,CAAC;gBACzC,SAAS;YACX,CAAC;YAED,MAAM,aAAa,CAAC,SAAS,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,IAAI,GAAG,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAChD,MAAM,UAAU,GAAG,IAAI,CACrB,eAAe,CAAC,QAAQ,CAAC,EACzB,GAAG,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,KAAK,EAAE,CAC5B,CAAC;QACF,IAAI,CAAC;YACH,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3C,aAAa,CACX,YAAY,CAAC,UAAU,CAAC,EACxB,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EACpC,MAAM,CACP,CAAC;YAEF,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC1B,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;gBAClC,IAAI,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,CAAC;oBACzC,SAAS;gBACX,CAAC;gBAED,MAAM,aAAa,CAAC,SAAS,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;YACpD,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,UAAU;gBAChB,IAAI;gBACJ,OAAO,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC;aACjD,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAe,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAClC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACnD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,aAAa,CAAC,SAAS,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,gBAAgB,CACvB,QAAgB,EAChB,OAAwB;IAExB,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,qBAAqB,CAAC;IACzD,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;IAE1C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;QAC7C,MAAM,IAAI,GAAG,6BAA6B,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC/D,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;YACzC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,MAAM,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;IACvD,CAAC;IAED,MAAM,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB,EAAE,KAAmB;IAC5D,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,OAAO,EAAE,KAAK,KAAK,KAAK,CAAC,KAAK,EAAE,CAAC;QACnC,OAAO;IACT,CAAC;IAED,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgB,EAAE,OAAe;IACzD,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IACpC,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,CAAC,EAAE,CAAC;QAC1C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgB,EAAE,OAAe;IACzD,MAAM,WAAW,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAiB,CAAC;IACtB,IAAI,CAAC;QACH,OAAO,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QAC5C,IAAI,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC,EAAE,CAAC;YAC1C,SAAS;QACX,CAAC;QAED,MAAM,GAAG,IAAI,CAAC;IAChB,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,WAAW,CAClB,QAAgB,EAChB,IAAyB,EACzB,OAAe;IAEf,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,MAAM,SAAS,GAAG,IAAI,EAAE,SAAS,IAAI,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IAClE,MAAM,OAAO,GAAG,GAAG,GAAG,SAAS,GAAG,OAAO,CAAC;IAC1C,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QAC9D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,cAAc,CAAC,GAAW;IACjC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC;QACvC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,WAAW,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,CAAC;IACtD,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,QAAgB;IACpC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CACvB,YAAY,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,CACpB,CAAC;QAC3B,IACE,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;YAChC,OAAO,MAAM,CAAC,GAAG,KAAK,QAAQ;YAC9B,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ;YACnC,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ;YACpC,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EACpC,CAAC;YACD,OAAO,MAAsB,CAAC;QAChC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,kBAAkB,CAAC,QAAgB;IAC1C,IAAI,CAAC;QACH,OAAO,QAAQ,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;IACpC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CACpB,QAAgB,EAChB,kBAA0B;IAE1B,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAErC,OAAO,IAAI,WAAW,CAAC,mBAAmB,EAAE;QAC1C,IAAI,EAAE,4BAA4B;QAClC,OAAO,EAAE,YAAY,CAAC;YACpB,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC;YACxB,MAAM,CAAC,WAAW,EAAE,kBAAkB,CAAC;YACvC,MAAM,CAAC,gBAAgB,EAAE,KAAK,EAAE,SAAS,CAAC;YAC1C,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC;YAC9B,MAAM,CAAC,WAAW,EAAE,KAAK,EAAE,QAAQ,CAAC;SACrC,CAAC;KACH,CAAC,CAAC;AACL,CAAC;AAED,SAAS,iBAAiB,CACxB,QAAgB,EAChB,kBAA0B;IAE1B,MAAM,KAAK,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAE9C,OAAO,IAAI,WAAW,CAAC,mBAAmB,EAAE;QAC1C,IAAI,EAAE,4BAA4B;QAClC,OAAO,EAAE,YAAY,CAAC;YACpB,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,WAAW,EAAE,kBAAkB,CAAC;YACvC,MAAM,CAAC,gBAAgB,EAAE,KAAK,EAAE,SAAS,CAAC;YAC1C,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC;YAC9B,MAAM,CAAC,WAAW,EAAE,KAAK,EAAE,QAAQ,CAAC;SACrC,CAAC;KACH,CAAC,CAAC;AACL,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAgB;IAC7C,MAAM,WAAW,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC;QACH,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC;YACpD,IAAI,IAAI,EAAE,CAAC;gBACT,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,QAAgB;IACpC,OAAO,IAAI,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;AACxC,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB;IACrC,OAAO,GAAG,QAAQ,QAAQ,CAAC;AAC7B,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB;IACvC,OAAO,GAAG,QAAQ,UAAU,CAAC;AAC/B,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK,CAAC;AACxE,CAAC"}
|
|
@@ -12,7 +12,7 @@ export function toDisplayPath(path) {
|
|
|
12
12
|
export function isPathInside(parent, child) {
|
|
13
13
|
const normalizedParent = normalizePath(parent);
|
|
14
14
|
const normalizedChild = normalizePath(child);
|
|
15
|
-
return normalizedChild === normalizedParent
|
|
16
|
-
|
|
15
|
+
return (normalizedChild === normalizedParent ||
|
|
16
|
+
normalizedChild.startsWith(`${normalizedParent}${sep}`));
|
|
17
17
|
}
|
|
18
18
|
//# sourceMappingURL=path.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"path.js","sourceRoot":"","sources":["../../../src/engine/utils/path.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;
|
|
1
|
+
{"version":3,"file":"path.js","sourceRoot":"","sources":["../../../src/engine/utils/path.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAEzC,MAAM,UAAU,WAAW;IACzB,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,OAAO,CAAC,OAAO,EAAE,EAAE,YAAY,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,IAAY;IACxC,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,IAAY;IACxC,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,MAAc,EAAE,KAAa;IACxD,MAAM,gBAAgB,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;IAC/C,MAAM,eAAe,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;IAE7C,OAAO,CACL,eAAe,KAAK,gBAAgB;QACpC,eAAe,CAAC,UAAU,CAAC,GAAG,gBAAgB,GAAG,GAAG,EAAE,CAAC,CACxD,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"timing.js","sourceRoot":"","sources":["../../../src/engine/utils/timing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"timing.js","sourceRoot":"","sources":["../../../src/engine/utils/timing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAS9C,MAAM,OAAO,eAAe;IACT,aAAa,GAAG,IAAI,GAAG,EAA8B,CAAC;IAEvE,KAAK,CAAC,IAAI,CAAI,IAAY,EAAE,IAAsB;QAChD,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,EAAE,CAAC;QACtB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAED,QAAQ,CAAI,IAAY,EAAE,IAAa;QACrC,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;QAChB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAED,GAAG,CAAC,IAAY,EAAE,UAAkB,EAAE,KAAK,GAAG,CAAC;QAC7C,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,cAAc,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YACpD,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACxD,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;YAC/C,QAAQ,CAAC,KAAK,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YACrC,OAAO;QACT,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,EAAE;YACrC,IAAI,EAAE,cAAc;YACpB,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC;SAC1B,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,OAA2C,EAAE,MAAM,GAAG,EAAE;QACjE,KAAK,MAAM,KAAK,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;YAClC,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,GAAG,KAAK,CAAC,IAAI,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IAED,OAAO;QACL,OAAO,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACtD,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC;YACxC,GAAG,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACnD,CAAC,CAAC,CAAC;IACN,CAAC;CACF;AAED,MAAM,OAAO,gBAAgB;IAKR;IACA;IALX,MAAM,GAAG,CAAC,CAAC;IACX,SAAS,GAAG,CAAC,CAAC;IAEtB,YACmB,SAA0B,EAC1B,IAAY;QADZ,cAAS,GAAT,SAAS,CAAiB;QAC1B,SAAI,GAAJ,IAAI,CAAQ;IAC5B,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAI,IAAsB;QAClC,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,EAAE,CAAC;QACtB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK;QACX,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAEO,IAAI;QACV,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,CAAC;QACd,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;YAClE,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { IndexProgress } from "./engine/types.js";
|
|
2
|
+
export type IndexProgressUpdate = {
|
|
3
|
+
phase: IndexProgress["phase"];
|
|
4
|
+
line: string;
|
|
5
|
+
progress: IndexProgress;
|
|
6
|
+
};
|
|
7
|
+
export declare function formatIndexProgressLine(progress: IndexProgress): string | undefined;
|
|
8
|
+
export declare function formatModelDownloadProgress(progress: IndexProgress): string | undefined;
|
|
9
|
+
export declare function indexProgressMessage(progress: IndexProgress): string | undefined;
|
|
10
|
+
export declare function indexProgressFromMessage(message: string | undefined): IndexProgressUpdate | undefined;
|
|
@@ -0,0 +1,159 @@
|
|
|
1
|
+
const MCP_INDEX_PROGRESS_PREFIX = "zvec-grep:index-progress:";
|
|
2
|
+
export function formatIndexProgressLine(progress) {
|
|
3
|
+
if (progress.phase === "scanning") {
|
|
4
|
+
return progress.detail;
|
|
5
|
+
}
|
|
6
|
+
if (progress.phase === "indexing") {
|
|
7
|
+
const modelDownload = formatModelDownloadProgress(progress);
|
|
8
|
+
if (modelDownload) {
|
|
9
|
+
return modelDownload;
|
|
10
|
+
}
|
|
11
|
+
const indexed = progress.filesIndexed ?? 0;
|
|
12
|
+
const total = progress.filesTotal ?? 0;
|
|
13
|
+
const detail = progress.detail ? ` ${truncate(progress.detail, 100)}` : "";
|
|
14
|
+
const failed = formatFailedProgress(progress);
|
|
15
|
+
const embedding = formatEmbeddingProgress(progress);
|
|
16
|
+
return total > 0
|
|
17
|
+
? `Indexing files: ${indexed}/${total}${detail}${failed}${embedding}`
|
|
18
|
+
: `Indexing files...${detail}${failed}${embedding}`;
|
|
19
|
+
}
|
|
20
|
+
return progress.detail ?? "Indexing complete";
|
|
21
|
+
}
|
|
22
|
+
export function formatModelDownloadProgress(progress) {
|
|
23
|
+
const embedding = progress.embedding;
|
|
24
|
+
if (!embedding?.model) {
|
|
25
|
+
return undefined;
|
|
26
|
+
}
|
|
27
|
+
if (embedding.stage === "preparing") {
|
|
28
|
+
return `Preparing ${embedding.model}`;
|
|
29
|
+
}
|
|
30
|
+
if (embedding.stage === "warning") {
|
|
31
|
+
return `zvec-grep warning: ${singleLineText(embedding.message ?? "Embedding model warning")}`;
|
|
32
|
+
}
|
|
33
|
+
if (embedding.stage === "ready") {
|
|
34
|
+
return `Model ready: ${embedding.model}`;
|
|
35
|
+
}
|
|
36
|
+
if (embedding.stage !== "downloading") {
|
|
37
|
+
return undefined;
|
|
38
|
+
}
|
|
39
|
+
const parts = [`Downloading ${embedding.model}`];
|
|
40
|
+
const downloadedBytes = embedding.downloadedBytes;
|
|
41
|
+
const totalBytes = embedding.totalBytes;
|
|
42
|
+
if (typeof downloadedBytes === "number" &&
|
|
43
|
+
typeof totalBytes === "number" &&
|
|
44
|
+
totalBytes > 0) {
|
|
45
|
+
const percent = Math.min(100, Math.max(0, Math.round((downloadedBytes / totalBytes) * 100)));
|
|
46
|
+
parts.push(`${percent}%`);
|
|
47
|
+
parts.push(`${formatByteCount(downloadedBytes)}/${formatByteCount(totalBytes)}`);
|
|
48
|
+
}
|
|
49
|
+
else if (typeof downloadedBytes === "number") {
|
|
50
|
+
parts.push(formatByteCount(downloadedBytes));
|
|
51
|
+
}
|
|
52
|
+
return parts.join(" · ");
|
|
53
|
+
}
|
|
54
|
+
export function indexProgressMessage(progress) {
|
|
55
|
+
const line = formatIndexProgressLine(progress);
|
|
56
|
+
return line
|
|
57
|
+
? `${MCP_INDEX_PROGRESS_PREFIX}${JSON.stringify({ line, progress })}`
|
|
58
|
+
: undefined;
|
|
59
|
+
}
|
|
60
|
+
export function indexProgressFromMessage(message) {
|
|
61
|
+
if (!message?.startsWith(MCP_INDEX_PROGRESS_PREFIX))
|
|
62
|
+
return undefined;
|
|
63
|
+
const payload = message.slice(MCP_INDEX_PROGRESS_PREFIX.length);
|
|
64
|
+
if (payload.startsWith("{")) {
|
|
65
|
+
try {
|
|
66
|
+
const decoded = JSON.parse(payload);
|
|
67
|
+
if (typeof decoded.line === "string" &&
|
|
68
|
+
isIndexProgress(decoded.progress)) {
|
|
69
|
+
return {
|
|
70
|
+
phase: decoded.progress.phase,
|
|
71
|
+
line: decoded.line,
|
|
72
|
+
progress: decoded.progress,
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
catch {
|
|
77
|
+
return undefined;
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
// Backward-compatible decoding for progress emitted by an older daemon.
|
|
81
|
+
const separator = payload.indexOf(":");
|
|
82
|
+
if (separator < 0)
|
|
83
|
+
return undefined;
|
|
84
|
+
const phase = payload.slice(0, separator);
|
|
85
|
+
if (phase !== "scanning" && phase !== "indexing" && phase !== "done") {
|
|
86
|
+
return undefined;
|
|
87
|
+
}
|
|
88
|
+
const line = payload.slice(separator + 1);
|
|
89
|
+
return { phase, line, progress: { phase, detail: line } };
|
|
90
|
+
}
|
|
91
|
+
function isIndexProgress(value) {
|
|
92
|
+
if (!value || typeof value !== "object")
|
|
93
|
+
return false;
|
|
94
|
+
const progress = value;
|
|
95
|
+
return (progress.phase === "scanning" ||
|
|
96
|
+
progress.phase === "indexing" ||
|
|
97
|
+
progress.phase === "done");
|
|
98
|
+
}
|
|
99
|
+
function formatFailedProgress(progress) {
|
|
100
|
+
const failed = progress.filesFailed ?? 0;
|
|
101
|
+
return failed > 0 ? ` [failed=${failed}]` : "";
|
|
102
|
+
}
|
|
103
|
+
function formatEmbeddingProgress(progress) {
|
|
104
|
+
const embedding = progress.embedding;
|
|
105
|
+
if (!embedding) {
|
|
106
|
+
return "";
|
|
107
|
+
}
|
|
108
|
+
const parts = [];
|
|
109
|
+
if (typeof embedding.concurrency === "number") {
|
|
110
|
+
parts.push(`concurrency=${embedding.concurrency}`);
|
|
111
|
+
}
|
|
112
|
+
if (typeof embedding.retryableFailures === "number" &&
|
|
113
|
+
embedding.retryableFailures > 0) {
|
|
114
|
+
parts.push(`retries=${embedding.retryableFailures}`);
|
|
115
|
+
}
|
|
116
|
+
return parts.length > 0 ? ` [embed ${parts.join(" ")}]` : "";
|
|
117
|
+
}
|
|
118
|
+
function truncate(value, maxLength) {
|
|
119
|
+
if (value.length <= maxLength) {
|
|
120
|
+
return value;
|
|
121
|
+
}
|
|
122
|
+
return `${value.slice(0, Math.max(0, maxLength - 1))}...`;
|
|
123
|
+
}
|
|
124
|
+
function formatByteCount(value) {
|
|
125
|
+
const bytes = Math.max(0, value);
|
|
126
|
+
const units = ["B", "KiB", "MiB", "GiB"];
|
|
127
|
+
let amount = bytes;
|
|
128
|
+
let unitIndex = 0;
|
|
129
|
+
while (amount >= 1024 && unitIndex < units.length - 1) {
|
|
130
|
+
amount /= 1024;
|
|
131
|
+
unitIndex++;
|
|
132
|
+
}
|
|
133
|
+
const digits = unitIndex === 0 || amount >= 10 ? 0 : 1;
|
|
134
|
+
return `${amount.toFixed(digits)} ${units[unitIndex]}`;
|
|
135
|
+
}
|
|
136
|
+
function singleLineText(value) {
|
|
137
|
+
let result = "";
|
|
138
|
+
for (let index = 0; index < value.length; index++) {
|
|
139
|
+
const code = value.charCodeAt(index);
|
|
140
|
+
if (code === 0x1b && value.charCodeAt(index + 1) === 0x5b) {
|
|
141
|
+
index += 2;
|
|
142
|
+
while (index < value.length) {
|
|
143
|
+
const ansiCode = value.charCodeAt(index);
|
|
144
|
+
if (ansiCode >= 0x40 && ansiCode <= 0x7e) {
|
|
145
|
+
break;
|
|
146
|
+
}
|
|
147
|
+
index++;
|
|
148
|
+
}
|
|
149
|
+
continue;
|
|
150
|
+
}
|
|
151
|
+
if (code < 0x20 || (code >= 0x7f && code <= 0x9f)) {
|
|
152
|
+
result += " ";
|
|
153
|
+
continue;
|
|
154
|
+
}
|
|
155
|
+
result += value[index];
|
|
156
|
+
}
|
|
157
|
+
return result.replace(/\s+/g, " ").trim();
|
|
158
|
+
}
|
|
159
|
+
//# sourceMappingURL=index-progress.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index-progress.js","sourceRoot":"","sources":["../src/index-progress.ts"],"names":[],"mappings":"AAEA,MAAM,yBAAyB,GAAG,2BAA2B,CAAC;AAQ9D,MAAM,UAAU,uBAAuB,CACrC,QAAuB;IAEvB,IAAI,QAAQ,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;QAClC,OAAO,QAAQ,CAAC,MAAM,CAAC;IACzB,CAAC;IAED,IAAI,QAAQ,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;QAClC,MAAM,aAAa,GAAG,2BAA2B,CAAC,QAAQ,CAAC,CAAC;QAC5D,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,MAAM,OAAO,GAAG,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC;QAC3C,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,IAAI,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3E,MAAM,MAAM,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;QACpD,OAAO,KAAK,GAAG,CAAC;YACd,CAAC,CAAC,mBAAmB,OAAO,IAAI,KAAK,GAAG,MAAM,GAAG,MAAM,GAAG,SAAS,EAAE;YACrE,CAAC,CAAC,oBAAoB,MAAM,GAAG,MAAM,GAAG,SAAS,EAAE,CAAC;IACxD,CAAC;IAED,OAAO,QAAQ,CAAC,MAAM,IAAI,mBAAmB,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,QAAuB;IAEvB,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC;QACtB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;QACpC,OAAO,aAAa,SAAS,CAAC,KAAK,EAAE,CAAC;IACxC,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAClC,OAAO,sBAAsB,cAAc,CACzC,SAAS,CAAC,OAAO,IAAI,yBAAyB,CAC/C,EAAE,CAAC;IACN,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,KAAK,OAAO,EAAE,CAAC;QAChC,OAAO,gBAAgB,SAAS,CAAC,KAAK,EAAE,CAAC;IAC3C,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,KAAK,aAAa,EAAE,CAAC;QACtC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,eAAe,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC;IACjD,MAAM,eAAe,GAAG,SAAS,CAAC,eAAe,CAAC;IAClD,MAAM,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC;IACxC,IACE,OAAO,eAAe,KAAK,QAAQ;QACnC,OAAO,UAAU,KAAK,QAAQ;QAC9B,UAAU,GAAG,CAAC,EACd,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CACtB,GAAG,EACH,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,eAAe,GAAG,UAAU,CAAC,GAAG,GAAG,CAAC,CAAC,CAC9D,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC;QAC1B,KAAK,CAAC,IAAI,CACR,GAAG,eAAe,CAAC,eAAe,CAAC,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CACrE,CAAC;IACJ,CAAC;SAAM,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;QAC/C,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,QAAuB;IAEvB,MAAM,IAAI,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IAC/C,OAAO,IAAI;QACT,CAAC,CAAC,GAAG,yBAAyB,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE;QACrE,CAAC,CAAC,SAAS,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,OAA2B;IAE3B,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,yBAAyB,CAAC;QAAE,OAAO,SAAS,CAAC;IACtE,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC;IAChE,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAGjC,CAAC;YACF,IACE,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ;gBAChC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,EACjC,CAAC;gBACD,OAAO;oBACL,KAAK,EAAE,OAAO,CAAC,QAAQ,CAAC,KAAK;oBAC7B,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ;iBAC3B,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAED,wEAAwE;IACxE,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACvC,IAAI,SAAS,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IACpC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IAC1C,IAAI,KAAK,KAAK,UAAU,IAAI,KAAK,KAAK,UAAU,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;QACrE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;IAC1C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;AAC5D,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,QAAQ,GAAG,KAAgC,CAAC;IAClD,OAAO,CACL,QAAQ,CAAC,KAAK,KAAK,UAAU;QAC7B,QAAQ,CAAC,KAAK,KAAK,UAAU;QAC7B,QAAQ,CAAC,KAAK,KAAK,MAAM,CAC1B,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,QAAuB;IACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,WAAW,IAAI,CAAC,CAAC;IACzC,OAAO,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;AACjD,CAAC;AAED,SAAS,uBAAuB,CAAC,QAAuB;IACtD,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,OAAO,SAAS,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;QAC9C,KAAK,CAAC,IAAI,CAAC,eAAe,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC;IACrD,CAAC;IAED,IACE,OAAO,SAAS,CAAC,iBAAiB,KAAK,QAAQ;QAC/C,SAAS,CAAC,iBAAiB,GAAG,CAAC,EAC/B,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,WAAW,SAAS,CAAC,iBAAiB,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;AAC/D,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa,EAAE,SAAiB;IAChD,IAAI,KAAK,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC;AAC5D,CAAC;AAED,SAAS,eAAe,CAAC,KAAa;IACpC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IACzC,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,OAAO,MAAM,IAAI,IAAI,IAAI,SAAS,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtD,MAAM,IAAI,IAAI,CAAC;QACf,SAAS,EAAE,CAAC;IACd,CAAC;IACD,MAAM,MAAM,GAAG,SAAS,KAAK,CAAC,IAAI,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACvD,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,cAAc,CAAC,KAAa;IACnC,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,IAAI,KAAK,IAAI,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YAC1D,KAAK,IAAI,CAAC,CAAC;YACX,OAAO,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBACzC,IAAI,QAAQ,IAAI,IAAI,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;oBACzC,MAAM;gBACR,CAAC;gBACD,KAAK,EAAE,CAAC;YACV,CAAC;YACD,SAAS;QACX,CAAC;QACD,IAAI,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,IAAI,GAAG,CAAC;YACd,SAAS;QACX,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AAC5C,CAAC"}
|
package/dist/index.d.ts
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
export { createZvecGrep
|
|
2
|
-
export type { CreateZvecGrepOptions, ZvecGrep,
|
|
3
|
-
export {
|
|
4
|
-
export { isImageContent, isTextContent
|
|
5
|
-
export type { CodeEntityMetadata, CodeEntityModifier, CodeSymbolType,
|
|
6
|
-
export type {
|
|
1
|
+
export { createZvecGrep } from "./engine/service/index.js";
|
|
2
|
+
export type { CreateZvecGrepOptions, ZvecGrep, ZvecGrepContent, ZvecGrepContextWorkspaceIndex, ZvecGrepContextContainer, ZvecGrepContextCoverage, ZvecGrepContextDiagnostics, ZvecGrepContextFile, ZvecGrepContextItem, ZvecGrepContextItemKind, ZvecGrepContextOptions, ZvecGrepContextResult, ZvecGrepContextRoute, ZvecGrepContextSource, ZvecGrepStructureEnrichmentDiagnostics, ZvecGrepIndexDiagnostics, ZvecGrepInfoOptions, ZvecGrepInfoResult, ZvecGrepIndexOptions, ZvecGrepRgDiagnostics, ZvecGrepSearchOptions, } from "./engine/service/index.js";
|
|
3
|
+
export { createEmbeddingModel, EmbeddingPurpose, } from "./engine/models/index.js";
|
|
4
|
+
export { isImageContent, isTextContent } from "./engine/types.js";
|
|
5
|
+
export type { CodeEntityMetadata, CodeEntityModifier, CodeSymbolType, WorkspaceIndexEmbeddingSchema, WorkspaceIndexPolicy, WorkspaceIndexStatus, WorkspaceIndexInfo, Content, ContentKind, EntityMetadata, FileFormat, FileIndexStatus, FileInfo, FileKind, ImageContent, ImageFormat, IndexOptions, IndexProgress, IndexProgressPhase, IndexResult, MarkdownEntityMetadata, RootPath, SearchFinalTrace, SearchHitTrace, SearchRecallTrace, SearchStageTrace, TextContent, TimingEntry, } from "./engine/types.js";
|
|
6
|
+
export type { CreateEmbeddingModelOptions, EmbeddingModel, EmbeddingModelProgress, EmbeddingModelInfo, EmbeddingOptions, EmbeddingResult, } from "./engine/models/index.js";
|
package/dist/index.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
export { createZvecGrep
|
|
2
|
-
export {
|
|
3
|
-
export { isImageContent, isTextContent
|
|
1
|
+
export { createZvecGrep } from "./engine/service/index.js";
|
|
2
|
+
export { createEmbeddingModel, EmbeddingPurpose, } from "./engine/models/index.js";
|
|
3
|
+
export { isImageContent, isTextContent } from "./engine/types.js";
|
|
4
4
|
//# sourceMappingURL=index.js.map
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAwB3D,OAAO,EACL,oBAAoB,EACpB,gBAAgB,GACjB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import type { IncomingMessage, ServerResponse } from "node:http";
|
|
2
|
+
import type { ZvecGrepDaemonBackend, ZvecGrepMcpServerOptions } from "./tools.js";
|
|
3
|
+
export type McpHttpEndpointOptions = {
|
|
4
|
+
legacySessionIdleTtlMs?: number;
|
|
5
|
+
maxLegacySessions?: number;
|
|
6
|
+
modernOnly?: boolean;
|
|
7
|
+
};
|
|
8
|
+
export declare class McpHttpEndpoint {
|
|
9
|
+
private readonly backend;
|
|
10
|
+
private readonly version;
|
|
11
|
+
private readonly sessions;
|
|
12
|
+
private readonly initializing;
|
|
13
|
+
private readonly serverOptions;
|
|
14
|
+
private readonly modern;
|
|
15
|
+
private readonly modernNodeHandler;
|
|
16
|
+
private readonly legacySessionIdleTtlMs;
|
|
17
|
+
private readonly maxLegacySessions;
|
|
18
|
+
private readonly modernOnly;
|
|
19
|
+
constructor(backend: ZvecGrepDaemonBackend, version: string, serverOptions?: Readonly<ZvecGrepMcpServerOptions>, endpointOptions?: Readonly<McpHttpEndpointOptions>);
|
|
20
|
+
handlePost(request: IncomingMessage, response: ServerResponse, body: unknown): Promise<void>;
|
|
21
|
+
handleSessionRequest(request: IncomingMessage, response: ServerResponse): Promise<void>;
|
|
22
|
+
close(): Promise<void>;
|
|
23
|
+
private handleLegacyPost;
|
|
24
|
+
private expireIdleSessions;
|
|
25
|
+
private withActiveSession;
|
|
26
|
+
private forget;
|
|
27
|
+
}
|
|
@@ -0,0 +1,182 @@
|
|
|
1
|
+
import { randomUUID } from "node:crypto";
|
|
2
|
+
import { createMcpHandler, isInitializeRequest, isLegacyRequest, } from "@modelcontextprotocol/server";
|
|
3
|
+
import { NodeStreamableHTTPServerTransport, toNodeHandler, toWebRequest, } from "@modelcontextprotocol/node";
|
|
4
|
+
import { createZvecGrepMcpServer } from "./tools.js";
|
|
5
|
+
import { InMemoryRemoteEmbeddingRequestStateReplayGuard } from "./request-state.js";
|
|
6
|
+
const LEGACY_SESSION_IDLE_TTL_MS = 30 * 60 * 1_000;
|
|
7
|
+
const MAX_LEGACY_SESSIONS = 256;
|
|
8
|
+
export class McpHttpEndpoint {
|
|
9
|
+
backend;
|
|
10
|
+
version;
|
|
11
|
+
sessions = new Map();
|
|
12
|
+
initializing = new Set();
|
|
13
|
+
serverOptions;
|
|
14
|
+
modern;
|
|
15
|
+
modernNodeHandler;
|
|
16
|
+
legacySessionIdleTtlMs;
|
|
17
|
+
maxLegacySessions;
|
|
18
|
+
modernOnly;
|
|
19
|
+
constructor(backend, version, serverOptions = {}, endpointOptions = {}) {
|
|
20
|
+
this.backend = backend;
|
|
21
|
+
this.version = version;
|
|
22
|
+
this.serverOptions = Object.freeze({
|
|
23
|
+
...serverOptions,
|
|
24
|
+
requestStateReplayGuard: serverOptions.requestStateReplayGuard ??
|
|
25
|
+
new InMemoryRemoteEmbeddingRequestStateReplayGuard(),
|
|
26
|
+
});
|
|
27
|
+
this.legacySessionIdleTtlMs =
|
|
28
|
+
endpointOptions.legacySessionIdleTtlMs ?? LEGACY_SESSION_IDLE_TTL_MS;
|
|
29
|
+
this.maxLegacySessions =
|
|
30
|
+
endpointOptions.maxLegacySessions ?? MAX_LEGACY_SESSIONS;
|
|
31
|
+
this.modernOnly = endpointOptions.modernOnly ?? false;
|
|
32
|
+
if (!Number.isFinite(this.legacySessionIdleTtlMs) ||
|
|
33
|
+
this.legacySessionIdleTtlMs <= 0 ||
|
|
34
|
+
!Number.isInteger(this.maxLegacySessions) ||
|
|
35
|
+
this.maxLegacySessions <= 0) {
|
|
36
|
+
throw new RangeError("Legacy MCP session limits must be positive.");
|
|
37
|
+
}
|
|
38
|
+
this.modern = createMcpHandler(() => createZvecGrepMcpServer(this.backend, this.version, this.serverOptions), { legacy: "reject" });
|
|
39
|
+
this.modernNodeHandler = toNodeHandler(this.modern);
|
|
40
|
+
}
|
|
41
|
+
async handlePost(request, response, body) {
|
|
42
|
+
if (this.modernOnly) {
|
|
43
|
+
await this.modernNodeHandler(request, response, body);
|
|
44
|
+
return;
|
|
45
|
+
}
|
|
46
|
+
const webRequest = await toWebRequest(request, body);
|
|
47
|
+
if (!(await isLegacyRequest(webRequest, body))) {
|
|
48
|
+
await this.modernNodeHandler(request, response, body);
|
|
49
|
+
return;
|
|
50
|
+
}
|
|
51
|
+
await this.handleLegacyPost(request, response, body);
|
|
52
|
+
}
|
|
53
|
+
async handleSessionRequest(request, response) {
|
|
54
|
+
if (this.modernOnly) {
|
|
55
|
+
await this.modernNodeHandler(request, response);
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
this.expireIdleSessions();
|
|
59
|
+
const sessionId = requestSessionId(request);
|
|
60
|
+
if (!sessionId) {
|
|
61
|
+
response.statusCode = request.method === "GET" ? 405 : 400;
|
|
62
|
+
response.end();
|
|
63
|
+
return;
|
|
64
|
+
}
|
|
65
|
+
const session = this.sessions.get(sessionId);
|
|
66
|
+
if (!session) {
|
|
67
|
+
writeMcpError(response, 404, "Unknown or expired MCP session.");
|
|
68
|
+
return;
|
|
69
|
+
}
|
|
70
|
+
await this.withActiveSession(session, () => session.transport.handleRequest(request, response));
|
|
71
|
+
}
|
|
72
|
+
async close() {
|
|
73
|
+
const sessions = new Set([
|
|
74
|
+
...this.sessions.values(),
|
|
75
|
+
...this.initializing.values(),
|
|
76
|
+
]);
|
|
77
|
+
this.sessions.clear();
|
|
78
|
+
this.initializing.clear();
|
|
79
|
+
await Promise.all([
|
|
80
|
+
this.modern.close(),
|
|
81
|
+
...[...sessions].map((session) => session.server.close().catch(() => undefined)),
|
|
82
|
+
]);
|
|
83
|
+
}
|
|
84
|
+
async handleLegacyPost(request, response, body) {
|
|
85
|
+
this.expireIdleSessions();
|
|
86
|
+
const sessionId = requestSessionId(request);
|
|
87
|
+
if (sessionId) {
|
|
88
|
+
const session = this.sessions.get(sessionId);
|
|
89
|
+
if (!session) {
|
|
90
|
+
writeMcpError(response, 404, "Unknown or expired MCP session.");
|
|
91
|
+
return;
|
|
92
|
+
}
|
|
93
|
+
await this.withActiveSession(session, () => session.transport.handleRequest(request, response, body));
|
|
94
|
+
return;
|
|
95
|
+
}
|
|
96
|
+
if (!isInitializeRequest(body)) {
|
|
97
|
+
writeMcpError(response, 400, "An MCP initialize request is required before other requests.");
|
|
98
|
+
return;
|
|
99
|
+
}
|
|
100
|
+
if (this.sessions.size + this.initializing.size >= this.maxLegacySessions) {
|
|
101
|
+
writeMcpError(response, 503, "MCP legacy session limit reached.");
|
|
102
|
+
return;
|
|
103
|
+
}
|
|
104
|
+
const holder = {};
|
|
105
|
+
const transport = new NodeStreamableHTTPServerTransport({
|
|
106
|
+
sessionIdGenerator: () => randomUUID(),
|
|
107
|
+
onsessioninitialized: (id) => {
|
|
108
|
+
const session = holder.session;
|
|
109
|
+
if (!session)
|
|
110
|
+
return;
|
|
111
|
+
session.id = id;
|
|
112
|
+
session.lastAccessedAt = Date.now();
|
|
113
|
+
this.sessions.set(id, session);
|
|
114
|
+
},
|
|
115
|
+
});
|
|
116
|
+
const server = createZvecGrepMcpServer(this.backend, this.version, this.serverOptions);
|
|
117
|
+
const session = {
|
|
118
|
+
activeRequests: 0,
|
|
119
|
+
server,
|
|
120
|
+
transport,
|
|
121
|
+
lastAccessedAt: Date.now(),
|
|
122
|
+
};
|
|
123
|
+
holder.session = session;
|
|
124
|
+
this.initializing.add(session);
|
|
125
|
+
transport.onclose = () => this.forget(session);
|
|
126
|
+
try {
|
|
127
|
+
await server.connect(transport);
|
|
128
|
+
await transport.handleRequest(request, response, body);
|
|
129
|
+
}
|
|
130
|
+
catch (error) {
|
|
131
|
+
if (!session.id) {
|
|
132
|
+
await server.close().catch(() => undefined);
|
|
133
|
+
}
|
|
134
|
+
throw error;
|
|
135
|
+
}
|
|
136
|
+
finally {
|
|
137
|
+
this.initializing.delete(session);
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
expireIdleSessions(now = Date.now()) {
|
|
141
|
+
for (const session of this.sessions.values()) {
|
|
142
|
+
if (session.activeRequests > 0 ||
|
|
143
|
+
now - session.lastAccessedAt <= this.legacySessionIdleTtlMs) {
|
|
144
|
+
continue;
|
|
145
|
+
}
|
|
146
|
+
this.forget(session);
|
|
147
|
+
void session.server.close().catch(() => undefined);
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
async withActiveSession(session, operation) {
|
|
151
|
+
session.activeRequests += 1;
|
|
152
|
+
session.lastAccessedAt = Date.now();
|
|
153
|
+
try {
|
|
154
|
+
await operation();
|
|
155
|
+
}
|
|
156
|
+
finally {
|
|
157
|
+
session.activeRequests -= 1;
|
|
158
|
+
session.lastAccessedAt = Date.now();
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
forget(session) {
|
|
162
|
+
if (session.id && this.sessions.get(session.id) === session) {
|
|
163
|
+
this.sessions.delete(session.id);
|
|
164
|
+
}
|
|
165
|
+
this.initializing.delete(session);
|
|
166
|
+
}
|
|
167
|
+
}
|
|
168
|
+
function requestSessionId(request) {
|
|
169
|
+
const value = request.headers["mcp-session-id"];
|
|
170
|
+
return typeof value === "string" && value.length > 0 ? value : undefined;
|
|
171
|
+
}
|
|
172
|
+
function writeMcpError(response, status, message) {
|
|
173
|
+
response.statusCode = status;
|
|
174
|
+
response.setHeader("Content-Type", "application/json");
|
|
175
|
+
response.setHeader("Cache-Control", "no-store");
|
|
176
|
+
response.end(JSON.stringify({
|
|
177
|
+
jsonrpc: "2.0",
|
|
178
|
+
error: { code: -32000, message },
|
|
179
|
+
id: null,
|
|
180
|
+
}));
|
|
181
|
+
}
|
|
182
|
+
//# sourceMappingURL=http-transport.js.map
|