@intentic/iq-engine 1.15.1
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/LICENSE +21 -0
- package/README.md +10 -0
- package/dist/embed/embedder.d.ts +9 -0
- package/dist/embed/embedder.d.ts.map +1 -0
- package/dist/embed/embedder.js +25 -0
- package/dist/embed/embedder.js.map +1 -0
- package/dist/embed/reranker.d.ts +6 -0
- package/dist/embed/reranker.d.ts.map +1 -0
- package/dist/embed/reranker.js +24 -0
- package/dist/embed/reranker.js.map +1 -0
- package/dist/engines/astq.d.ts +3 -0
- package/dist/engines/astq.d.ts.map +1 -0
- package/dist/engines/astq.js +34 -0
- package/dist/engines/astq.js.map +1 -0
- package/dist/engines/bm25.d.ts +6 -0
- package/dist/engines/bm25.d.ts.map +1 -0
- package/dist/engines/bm25.js +65 -0
- package/dist/engines/bm25.js.map +1 -0
- package/dist/engines/files.d.ts +4 -0
- package/dist/engines/files.d.ts.map +1 -0
- package/dist/engines/files.js +58 -0
- package/dist/engines/files.js.map +1 -0
- package/dist/engines/git.d.ts +23 -0
- package/dist/engines/git.d.ts.map +1 -0
- package/dist/engines/git.js +154 -0
- package/dist/engines/git.js.map +1 -0
- package/dist/engines/lexical.d.ts +13 -0
- package/dist/engines/lexical.d.ts.map +1 -0
- package/dist/engines/lexical.js +68 -0
- package/dist/engines/lexical.js.map +1 -0
- package/dist/engines/semantic.d.ts +6 -0
- package/dist/engines/semantic.d.ts.map +1 -0
- package/dist/engines/semantic.js +44 -0
- package/dist/engines/semantic.js.map +1 -0
- package/dist/engines/symbols.d.ts +15 -0
- package/dist/engines/symbols.d.ts.map +1 -0
- package/dist/engines/symbols.js +91 -0
- package/dist/engines/symbols.js.map +1 -0
- package/dist/features.d.ts +5 -0
- package/dist/features.d.ts.map +1 -0
- package/dist/features.js +24 -0
- package/dist/features.js.map +1 -0
- package/dist/index.d.ts +22 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +104 -0
- package/dist/index.js.map +1 -0
- package/dist/indexer/chunker.d.ts +3 -0
- package/dist/indexer/chunker.d.ts.map +1 -0
- package/dist/indexer/chunker.js +51 -0
- package/dist/indexer/chunker.js.map +1 -0
- package/dist/indexer/indexer.d.ts +14 -0
- package/dist/indexer/indexer.d.ts.map +1 -0
- package/dist/indexer/indexer.js +55 -0
- package/dist/indexer/indexer.js.map +1 -0
- package/dist/indexer/languages.d.ts +13 -0
- package/dist/indexer/languages.d.ts.map +1 -0
- package/dist/indexer/languages.js +87 -0
- package/dist/indexer/languages.js.map +1 -0
- package/dist/indexer/symbols.d.ts +3 -0
- package/dist/indexer/symbols.d.ts.map +1 -0
- package/dist/indexer/symbols.js +115 -0
- package/dist/indexer/symbols.js.map +1 -0
- package/dist/plan/classify.d.ts +3 -0
- package/dist/plan/classify.d.ts.map +1 -0
- package/dist/plan/classify.js +24 -0
- package/dist/plan/classify.js.map +1 -0
- package/dist/plan/fuse.d.ts +10 -0
- package/dist/plan/fuse.d.ts.map +1 -0
- package/dist/plan/fuse.js +85 -0
- package/dist/plan/fuse.js.map +1 -0
- package/dist/render/budget.d.ts +2 -0
- package/dist/render/budget.d.ts.map +1 -0
- package/dist/render/budget.js +2 -0
- package/dist/render/budget.js.map +1 -0
- package/dist/render/cursor.d.ts +19 -0
- package/dist/render/cursor.d.ts.map +1 -0
- package/dist/render/cursor.js +35 -0
- package/dist/render/cursor.js.map +1 -0
- package/dist/render/text.d.ts +31 -0
- package/dist/render/text.d.ts.map +1 -0
- package/dist/render/text.js +134 -0
- package/dist/render/text.js.map +1 -0
- package/dist/store/db.d.ts +10 -0
- package/dist/store/db.d.ts.map +1 -0
- package/dist/store/db.js +95 -0
- package/dist/store/db.js.map +1 -0
- package/dist/store/index-store.d.ts +27 -0
- package/dist/store/index-store.d.ts.map +1 -0
- package/dist/store/index-store.js +47 -0
- package/dist/store/index-store.js.map +1 -0
- package/dist/testing.d.ts +5 -0
- package/dist/testing.d.ts.map +1 -0
- package/dist/testing.js +31 -0
- package/dist/testing.js.map +1 -0
- package/dist/types.d.ts +100 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/verbs/context.d.ts +13 -0
- package/dist/verbs/context.d.ts.map +1 -0
- package/dist/verbs/context.js +68 -0
- package/dist/verbs/context.js.map +1 -0
- package/dist/verbs/dispatch.d.ts +18 -0
- package/dist/verbs/dispatch.d.ts.map +1 -0
- package/dist/verbs/dispatch.js +486 -0
- package/dist/verbs/dispatch.js.map +1 -0
- package/dist/workspace/floor.d.ts +3 -0
- package/dist/workspace/floor.d.ts.map +1 -0
- package/dist/workspace/floor.js +3 -0
- package/dist/workspace/floor.js.map +1 -0
- package/dist/workspace/glob.d.ts +2 -0
- package/dist/workspace/glob.d.ts.map +1 -0
- package/dist/workspace/glob.js +53 -0
- package/dist/workspace/glob.js.map +1 -0
- package/dist/workspace/scan.d.ts +7 -0
- package/dist/workspace/scan.d.ts.map +1 -0
- package/dist/workspace/scan.js +104 -0
- package/dist/workspace/scan.js.map +1 -0
- package/package.json +68 -0
- package/scripts/fetch-model.mjs +18 -0
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
import { Lang, parse, registerDynamicLanguage } from "@ast-grep/napi";
|
|
2
|
+
import langGo from "@ast-grep/lang-go";
|
|
3
|
+
import langJava from "@ast-grep/lang-java";
|
|
4
|
+
import langPython from "@ast-grep/lang-python";
|
|
5
|
+
import langRust from "@ast-grep/lang-rust";
|
|
6
|
+
let registered = false;
|
|
7
|
+
const ensureRegistered = () => {
|
|
8
|
+
if (registered) {
|
|
9
|
+
return;
|
|
10
|
+
}
|
|
11
|
+
registerDynamicLanguage({ python: langPython, go: langGo, rust: langRust, java: langJava });
|
|
12
|
+
registered = true;
|
|
13
|
+
};
|
|
14
|
+
export const LANGUAGES = {
|
|
15
|
+
ts: {
|
|
16
|
+
astLang: Lang.TypeScript,
|
|
17
|
+
rules: [
|
|
18
|
+
{ kind: "function_declaration", symbolKind: "fn" },
|
|
19
|
+
{ kind: "class_declaration", symbolKind: "class" },
|
|
20
|
+
{ kind: "interface_declaration", symbolKind: "type" },
|
|
21
|
+
{ kind: "type_alias_declaration", symbolKind: "type" },
|
|
22
|
+
{ kind: "enum_declaration", symbolKind: "type" },
|
|
23
|
+
{ kind: "method_definition", symbolKind: "method" },
|
|
24
|
+
],
|
|
25
|
+
},
|
|
26
|
+
tsx: {
|
|
27
|
+
astLang: Lang.Tsx,
|
|
28
|
+
rules: [
|
|
29
|
+
{ kind: "function_declaration", symbolKind: "fn" },
|
|
30
|
+
{ kind: "class_declaration", symbolKind: "class" },
|
|
31
|
+
{ kind: "interface_declaration", symbolKind: "type" },
|
|
32
|
+
{ kind: "type_alias_declaration", symbolKind: "type" },
|
|
33
|
+
{ kind: "method_definition", symbolKind: "method" },
|
|
34
|
+
],
|
|
35
|
+
},
|
|
36
|
+
js: {
|
|
37
|
+
astLang: Lang.JavaScript,
|
|
38
|
+
rules: [
|
|
39
|
+
{ kind: "function_declaration", symbolKind: "fn" },
|
|
40
|
+
{ kind: "class_declaration", symbolKind: "class" },
|
|
41
|
+
{ kind: "method_definition", symbolKind: "method" },
|
|
42
|
+
],
|
|
43
|
+
},
|
|
44
|
+
python: {
|
|
45
|
+
astLang: "python",
|
|
46
|
+
rules: [
|
|
47
|
+
{ kind: "function_definition", symbolKind: "fn" },
|
|
48
|
+
{ kind: "class_definition", symbolKind: "class" },
|
|
49
|
+
],
|
|
50
|
+
},
|
|
51
|
+
go: {
|
|
52
|
+
astLang: "go",
|
|
53
|
+
rules: [
|
|
54
|
+
{ kind: "function_declaration", symbolKind: "fn" },
|
|
55
|
+
{ kind: "method_declaration", symbolKind: "method" },
|
|
56
|
+
{ kind: "type_spec", symbolKind: "type" },
|
|
57
|
+
],
|
|
58
|
+
},
|
|
59
|
+
rust: {
|
|
60
|
+
astLang: "rust",
|
|
61
|
+
rules: [
|
|
62
|
+
{ kind: "function_item", symbolKind: "fn" },
|
|
63
|
+
{ kind: "struct_item", symbolKind: "type" },
|
|
64
|
+
{ kind: "enum_item", symbolKind: "type" },
|
|
65
|
+
{ kind: "trait_item", symbolKind: "type" },
|
|
66
|
+
{ kind: "const_item", symbolKind: "const" },
|
|
67
|
+
],
|
|
68
|
+
},
|
|
69
|
+
java: {
|
|
70
|
+
astLang: "java",
|
|
71
|
+
rules: [
|
|
72
|
+
{ kind: "method_declaration", symbolKind: "method" },
|
|
73
|
+
{ kind: "class_declaration", symbolKind: "class" },
|
|
74
|
+
{ kind: "interface_declaration", symbolKind: "type" },
|
|
75
|
+
{ kind: "enum_declaration", symbolKind: "type" },
|
|
76
|
+
],
|
|
77
|
+
},
|
|
78
|
+
};
|
|
79
|
+
export const parseLang = (lang, content) => {
|
|
80
|
+
const config = LANGUAGES[lang];
|
|
81
|
+
if (config === undefined) {
|
|
82
|
+
return undefined;
|
|
83
|
+
}
|
|
84
|
+
ensureRegistered();
|
|
85
|
+
return parse(config.astLang, content);
|
|
86
|
+
};
|
|
87
|
+
//# sourceMappingURL=languages.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"languages.js","sourceRoot":"","sources":["../../src/indexer/languages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,uBAAuB,EAAe,MAAM,gBAAgB,CAAC;AACnF,OAAO,MAAM,MAAM,mBAAmB,CAAC;AACvC,OAAO,QAAQ,MAAM,qBAAqB,CAAC;AAC3C,OAAO,UAAU,MAAM,uBAAuB,CAAC;AAC/C,OAAO,QAAQ,MAAM,qBAAqB,CAAC;AAe3C,IAAI,UAAU,GAAG,KAAK,CAAC;AAEvB,MAAM,gBAAgB,GAAG,GAAS,EAAE;IAChC,IAAI,UAAU,EAAE,CAAC;QACb,OAAO;IACX,CAAC;IACD,uBAAuB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAmD,CAAC,CAAC;IAC7I,UAAU,GAAG,IAAI,CAAC;AACtB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAmC;IACrD,EAAE,EAAE;QACA,OAAO,EAAE,IAAI,CAAC,UAAU;QACxB,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,sBAAsB,EAAE,UAAU,EAAE,IAAI,EAAE;YAClD,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,OAAO,EAAE;YAClD,EAAE,IAAI,EAAE,uBAAuB,EAAE,UAAU,EAAE,MAAM,EAAE;YACrD,EAAE,IAAI,EAAE,wBAAwB,EAAE,UAAU,EAAE,MAAM,EAAE;YACtD,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,EAAE;YAChD,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,QAAQ,EAAE;SACtD;KACJ;IACD,GAAG,EAAE;QACD,OAAO,EAAE,IAAI,CAAC,GAAG;QACjB,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,sBAAsB,EAAE,UAAU,EAAE,IAAI,EAAE;YAClD,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,OAAO,EAAE;YAClD,EAAE,IAAI,EAAE,uBAAuB,EAAE,UAAU,EAAE,MAAM,EAAE;YACrD,EAAE,IAAI,EAAE,wBAAwB,EAAE,UAAU,EAAE,MAAM,EAAE;YACtD,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,QAAQ,EAAE;SACtD;KACJ;IACD,EAAE,EAAE;QACA,OAAO,EAAE,IAAI,CAAC,UAAU;QACxB,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,sBAAsB,EAAE,UAAU,EAAE,IAAI,EAAE;YAClD,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,OAAO,EAAE;YAClD,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,QAAQ,EAAE;SACtD;KACJ;IACD,MAAM,EAAE;QACJ,OAAO,EAAE,QAAQ;QACjB,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,qBAAqB,EAAE,UAAU,EAAE,IAAI,EAAE;YACjD,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,EAAE,OAAO,EAAE;SACpD;KACJ;IACD,EAAE,EAAE;QACA,OAAO,EAAE,IAAI;QACb,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,sBAAsB,EAAE,UAAU,EAAE,IAAI,EAAE;YAClD,EAAE,IAAI,EAAE,oBAAoB,EAAE,UAAU,EAAE,QAAQ,EAAE;YACpD,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE;SAC5C;KACJ;IACD,IAAI,EAAE;QACF,OAAO,EAAE,MAAM;QACf,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,eAAe,EAAE,UAAU,EAAE,IAAI,EAAE;YAC3C,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,EAAE;YAC3C,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE;YACzC,EAAE,IAAI,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,EAAE;YAC1C,EAAE,IAAI,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE;SAC9C;KACJ;IACD,IAAI,EAAE;QACF,OAAO,EAAE,MAAM;QACf,KAAK,EAAE;YACH,EAAE,IAAI,EAAE,oBAAoB,EAAE,UAAU,EAAE,QAAQ,EAAE;YACpD,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,OAAO,EAAE;YAClD,EAAE,IAAI,EAAE,uBAAuB,EAAE,UAAU,EAAE,MAAM,EAAE;YACrD,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,EAAE;SACnD;KACJ;CACJ,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,IAAY,EAAE,OAAe,EAAsB,EAAE;IAC3E,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;IAC/B,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACvB,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,gBAAgB,EAAE,CAAC;IACnB,OAAO,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,OAAO,CAAC,CAAC;AAClD,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"symbols.d.ts","sourceRoot":"","sources":["../../src/indexer/symbols.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAgG7C,eAAO,MAAM,cAAc,GAAI,MAAM,MAAM,EAAE,MAAM,MAAM,GAAG,SAAS,EAAE,SAAS,MAAM,KAAG,SAAS,EAiCjG,CAAC"}
|
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
import { LANGUAGES, parseLang } from "./languages.js";
|
|
2
|
+
const SIGNATURE_MAX = 160;
|
|
3
|
+
const signatureOf = (node) => {
|
|
4
|
+
const first = node.text().split("\n", 1)[0].trim();
|
|
5
|
+
return first.length > SIGNATURE_MAX ? `${first.slice(0, SIGNATURE_MAX)}…` : first;
|
|
6
|
+
};
|
|
7
|
+
const isExported = (node, lang, name) => {
|
|
8
|
+
if (lang === "go") {
|
|
9
|
+
return /^[A-Z]/.test(name);
|
|
10
|
+
}
|
|
11
|
+
if (lang === "python") {
|
|
12
|
+
return !name.startsWith("_");
|
|
13
|
+
}
|
|
14
|
+
if (lang === "rust") {
|
|
15
|
+
return node.children().some((child) => child.kind() === "visibility_modifier");
|
|
16
|
+
}
|
|
17
|
+
for (let current = node, depth = 0; current !== null && depth < 4; current = current.parent(), depth++) {
|
|
18
|
+
if (current.kind() === "export_statement") {
|
|
19
|
+
return true;
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
return false;
|
|
23
|
+
};
|
|
24
|
+
const isTestPath = (path) => /(^|\/)((__tests__|tests?)\/|test_[^/]*$)|\.(test|spec)\.[^/.]+$/.test(path);
|
|
25
|
+
const row = (path, lang, node, name, kind) => {
|
|
26
|
+
const range = node.range();
|
|
27
|
+
return {
|
|
28
|
+
name,
|
|
29
|
+
kind: isTestPath(path) && (kind === "fn" || kind === "const") ? "test" : kind,
|
|
30
|
+
line: range.start.line + 1,
|
|
31
|
+
endLine: range.end.line + 1,
|
|
32
|
+
signature: signatureOf(node),
|
|
33
|
+
exported: isExported(node, lang, name),
|
|
34
|
+
heuristic: false,
|
|
35
|
+
};
|
|
36
|
+
};
|
|
37
|
+
const declaratorRows = (path, lang, root) => {
|
|
38
|
+
const rows = [];
|
|
39
|
+
for (const declarator of root.findAll({ rule: { kind: "variable_declarator" } })) {
|
|
40
|
+
const statementParent = declarator.parent()?.parent()?.kind();
|
|
41
|
+
if (statementParent !== "program" && statementParent !== "export_statement") {
|
|
42
|
+
continue;
|
|
43
|
+
}
|
|
44
|
+
const name = declarator.field("name")?.text();
|
|
45
|
+
if (name === undefined) {
|
|
46
|
+
continue;
|
|
47
|
+
}
|
|
48
|
+
const value = declarator.field("value")?.kind();
|
|
49
|
+
const statement = statementParent === "export_statement" ? declarator.parent().parent() : declarator.parent();
|
|
50
|
+
rows.push(row(path, lang, statement, name, value === "arrow_function" || value === "function_expression" ? "fn" : "const"));
|
|
51
|
+
}
|
|
52
|
+
return rows;
|
|
53
|
+
};
|
|
54
|
+
const HEURISTIC_PATTERNS = [
|
|
55
|
+
{ re: /^\s*(?:export\s+)?(?:default\s+)?(?:async\s+)?function\s+([A-Za-z_$][\w$]*)/, kind: "fn" },
|
|
56
|
+
{ re: /^\s*(?:export\s+)?(?:abstract\s+)?class\s+([A-Za-z_$][\w$]*)/, kind: "class" },
|
|
57
|
+
{ re: /^\s*(?:async\s+)?def\s+([A-Za-z_]\w*)/, kind: "fn" },
|
|
58
|
+
{ re: /^\s*func\s+(?:\([^)]*\)\s*)?([A-Za-z_]\w*)/, kind: "fn" },
|
|
59
|
+
];
|
|
60
|
+
const heuristicSymbols = (path, content) => {
|
|
61
|
+
const rows = [];
|
|
62
|
+
content.split(/\r?\n/).forEach((text, i) => {
|
|
63
|
+
for (const { re, kind } of HEURISTIC_PATTERNS) {
|
|
64
|
+
const match = re.exec(text);
|
|
65
|
+
if (match !== null) {
|
|
66
|
+
rows.push({
|
|
67
|
+
name: match[1],
|
|
68
|
+
kind: isTestPath(path) && kind === "fn" ? "test" : kind,
|
|
69
|
+
line: i + 1,
|
|
70
|
+
endLine: i + 1,
|
|
71
|
+
signature: text.trim().slice(0, SIGNATURE_MAX),
|
|
72
|
+
exported: !match[1].startsWith("_"),
|
|
73
|
+
heuristic: true,
|
|
74
|
+
});
|
|
75
|
+
return;
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
});
|
|
79
|
+
return rows;
|
|
80
|
+
};
|
|
81
|
+
const NON_CODE = /\.(md|mdx|rst|txt|json|jsonc|ya?ml|toml|ini|lock|csv|html|xml|svg|css|scss)$/i;
|
|
82
|
+
export const extractSymbols = (path, lang, content) => {
|
|
83
|
+
if (NON_CODE.test(path)) {
|
|
84
|
+
return [];
|
|
85
|
+
}
|
|
86
|
+
const config = lang !== undefined ? LANGUAGES[lang] : undefined;
|
|
87
|
+
if (lang === undefined || config === undefined) {
|
|
88
|
+
return heuristicSymbols(path, content);
|
|
89
|
+
}
|
|
90
|
+
const root = parseLang(lang, content)?.root();
|
|
91
|
+
if (root === undefined) {
|
|
92
|
+
return heuristicSymbols(path, content);
|
|
93
|
+
}
|
|
94
|
+
const rows = [];
|
|
95
|
+
for (const rule of config.rules) {
|
|
96
|
+
for (const node of root.findAll({ rule: { kind: rule.kind } })) {
|
|
97
|
+
const name = node.field("name")?.text();
|
|
98
|
+
if (name !== undefined && name !== "") {
|
|
99
|
+
rows.push(row(path, lang, node, name, rule.symbolKind));
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
if (lang === "ts" || lang === "tsx" || lang === "js") {
|
|
104
|
+
rows.push(...declaratorRows(path, lang, root));
|
|
105
|
+
}
|
|
106
|
+
if (lang === "python") {
|
|
107
|
+
for (const item of rows) {
|
|
108
|
+
if (item.kind === "fn" && rows.some((other) => other.kind === "class" && other.line < item.line && other.endLine >= item.endLine)) {
|
|
109
|
+
rows[rows.indexOf(item)] = { ...item, kind: "method" };
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
return rows.toSorted((a, b) => a.line - b.line);
|
|
114
|
+
};
|
|
115
|
+
//# sourceMappingURL=symbols.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"symbols.js","sourceRoot":"","sources":["../../src/indexer/symbols.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAEtD,MAAM,aAAa,GAAG,GAAG,CAAC;AAE1B,MAAM,WAAW,GAAG,CAAC,IAAY,EAAU,EAAE;IACzC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,IAAI,EAAE,CAAC;IACpD,OAAO,KAAK,CAAC,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC;AACtF,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,CAAC,IAAY,EAAE,IAAY,EAAE,IAAY,EAAW,EAAE;IACrE,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAChB,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IACD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IACjC,CAAC;IACD,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,qBAAqB,CAAC,CAAC;IACnF,CAAC;IACD,KAAK,IAAI,OAAO,GAAkB,IAAI,EAAE,KAAK,GAAG,CAAC,EAAE,OAAO,KAAK,IAAI,IAAI,KAAK,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;QACpH,IAAI,OAAO,CAAC,IAAI,EAAE,KAAK,kBAAkB,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,CAAC,IAAY,EAAW,EAAE,CAAC,iEAAiE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAE3H,MAAM,GAAG,GAAG,CAAC,IAAY,EAAE,IAAY,EAAE,IAAY,EAAE,IAAY,EAAE,IAAuB,EAAa,EAAE;IACvG,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,OAAO;QACH,IAAI;QACJ,IAAI,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;QAC7E,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC;QAC1B,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC;QAC3B,SAAS,EAAE,WAAW,CAAC,IAAI,CAAC;QAC5B,QAAQ,EAAE,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC;QACtC,SAAS,EAAE,KAAK;KACnB,CAAC;AACN,CAAC,CAAC;AAGF,MAAM,cAAc,GAAG,CAAC,IAAY,EAAE,IAAY,EAAE,IAAY,EAAe,EAAE;IAC7E,MAAM,IAAI,GAAgB,EAAE,CAAC;IAC7B,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,qBAAqB,EAAE,EAAE,CAAC,EAAE,CAAC;QAC/E,MAAM,eAAe,GAAG,UAAU,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC;QAC9D,IAAI,eAAe,KAAK,SAAS,IAAI,eAAe,KAAK,kBAAkB,EAAE,CAAC;YAC1E,SAAS;QACb,CAAC;QACD,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC;QAC9C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACrB,SAAS;QACb,CAAC;QACD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC;QAEhD,MAAM,SAAS,GAAG,eAAe,KAAK,kBAAkB,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,EAAG,CAAC,MAAM,EAAG,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,EAAG,CAAC;QACjH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,KAAK,gBAAgB,IAAI,KAAK,KAAK,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;IAChI,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,kBAAkB,GAAuD;IAC3E,EAAE,EAAE,EAAE,6EAA6E,EAAE,IAAI,EAAE,IAAI,EAAE;IACjG,EAAE,EAAE,EAAE,8DAA8D,EAAE,IAAI,EAAE,OAAO,EAAE;IACrF,EAAE,EAAE,EAAE,uCAAuC,EAAE,IAAI,EAAE,IAAI,EAAE;IAC3D,EAAE,EAAE,EAAE,4CAA4C,EAAE,IAAI,EAAE,IAAI,EAAE;CACnE,CAAC;AAGF,MAAM,gBAAgB,GAAG,CAAC,IAAY,EAAE,OAAe,EAAe,EAAE;IACpE,MAAM,IAAI,GAAgB,EAAE,CAAC;IAC7B,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;QACvC,KAAK,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,kBAAkB,EAAE,CAAC;YAC5C,MAAM,KAAK,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5B,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;gBACjB,IAAI,CAAC,IAAI,CAAC;oBACN,IAAI,EAAE,KAAK,CAAC,CAAC,CAAE;oBACf,IAAI,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;oBACvD,IAAI,EAAE,CAAC,GAAG,CAAC;oBACX,OAAO,EAAE,CAAC,GAAG,CAAC;oBACd,SAAS,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC;oBAC9C,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,UAAU,CAAC,GAAG,CAAC;oBACpC,SAAS,EAAE,IAAI;iBAClB,CAAC,CAAC;gBACH,OAAO;YACX,CAAC;QACL,CAAC;IACL,CAAC,CAAC,CAAC;IACH,OAAO,IAAI,CAAC;AAChB,CAAC,CAAC;AAGF,MAAM,QAAQ,GAAG,+EAA+E,CAAC;AAEjG,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,IAAY,EAAE,IAAwB,EAAE,OAAe,EAAe,EAAE;IACnG,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACtB,OAAO,EAAE,CAAC;IACd,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChE,IAAI,IAAI,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC7C,OAAO,gBAAgB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IACD,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC;IAC9C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACrB,OAAO,gBAAgB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IACD,MAAM,IAAI,GAAgB,EAAE,CAAC;IAC7B,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAC9B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC;YAC7D,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC;YACxC,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;gBACpC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;YAC5D,CAAC;QACL,CAAC;IACL,CAAC;IACD,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QACnD,IAAI,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;QACpB,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YACtB,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBAChI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAC3D,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;AACpD,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"classify.d.ts","sourceRoot":"","sources":["../../src/plan/classify.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,YAAY,GAAG,OAAO,GAAG,SAAS,CAAC;AAepE,eAAO,MAAM,QAAQ,GAAI,OAAO,MAAM,KAAG,SAaxC,CAAC"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
const IDENTIFIER = /^[A-Za-z_$][\w$]*([.#][\w$]+)*$/;
|
|
2
|
+
const REGEX_META = /[\\^$.|+()[\]{]/;
|
|
3
|
+
const compiles = (pattern) => {
|
|
4
|
+
try {
|
|
5
|
+
return new RegExp(pattern) instanceof RegExp;
|
|
6
|
+
}
|
|
7
|
+
catch {
|
|
8
|
+
return false;
|
|
9
|
+
}
|
|
10
|
+
};
|
|
11
|
+
export const classify = (query) => {
|
|
12
|
+
const trimmed = query.trim();
|
|
13
|
+
if (!/\s/.test(trimmed) && (trimmed.includes("/") || /[*?]/.test(trimmed)) && !REGEX_META.test(trimmed.replace(/[*?.]/g, ""))) {
|
|
14
|
+
return "path";
|
|
15
|
+
}
|
|
16
|
+
if (IDENTIFIER.test(trimmed)) {
|
|
17
|
+
return "identifier";
|
|
18
|
+
}
|
|
19
|
+
if (!/\s/.test(trimmed) && REGEX_META.test(trimmed) && compiles(trimmed)) {
|
|
20
|
+
return "regex";
|
|
21
|
+
}
|
|
22
|
+
return "natural";
|
|
23
|
+
};
|
|
24
|
+
//# sourceMappingURL=classify.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"classify.js","sourceRoot":"","sources":["../../src/plan/classify.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,GAAG,iCAAiC,CAAC;AACrD,MAAM,UAAU,GAAG,iBAAiB,CAAC;AAErC,MAAM,QAAQ,GAAG,CAAC,OAAe,EAAW,EAAE;IAC1C,IAAI,CAAC;QACD,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,MAAM,CAAC;IACjD,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,KAAK,CAAC;IACjB,CAAC;AACL,CAAC,CAAC;AAIF,MAAM,CAAC,MAAM,QAAQ,GAAG,CAAC,KAAa,EAAa,EAAE;IACjD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAE7B,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;QAC5H,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,IAAI,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,OAAO,YAAY,CAAC;IACxB,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACvE,OAAO,OAAO,CAAC;IACnB,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { EngineResult, RankedGroup } from "../types.js";
|
|
2
|
+
export interface FuseContext {
|
|
3
|
+
readonly queryTokens: readonly string[];
|
|
4
|
+
readonly mtimes: ReadonlyMap<string, number>;
|
|
5
|
+
readonly now: number;
|
|
6
|
+
readonly boosts: boolean;
|
|
7
|
+
}
|
|
8
|
+
export declare const fuse: (results: readonly EngineResult[], context: FuseContext) => RankedGroup[];
|
|
9
|
+
export declare const queryTokens: (query: string) => string[];
|
|
10
|
+
//# sourceMappingURL=fuse.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fuse.d.ts","sourceRoot":"","sources":["../../src/plan/fuse.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAa,MAAM,aAAa,CAAC;AAOxE,MAAM,WAAW,WAAW;IAExB,QAAQ,CAAC,WAAW,EAAE,SAAS,MAAM,EAAE,CAAC;IACxC,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IAErB,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;CAC5B;AAkBD,eAAO,MAAM,IAAI,GAAI,SAAS,SAAS,YAAY,EAAE,EAAE,SAAS,WAAW,KAAG,WAAW,EAmExF,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,MAAM,KAAG,MAAM,EAAsE,CAAC"}
|
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
const RRF_K = 60;
|
|
2
|
+
const DEF_BOOST = 1.5;
|
|
3
|
+
const PATH_BOOST = 1.25;
|
|
4
|
+
const RECENCY_HALF_LIFE_DAYS = 14;
|
|
5
|
+
const TAG_ORDER = ["def", "path", "fuzzy", "rerank", "sem", "bm25", "import", "call", "type", "write", "text", "heuristic"];
|
|
6
|
+
const dedupeTags = (tags) => {
|
|
7
|
+
const seen = new Map();
|
|
8
|
+
for (const tag of tags) {
|
|
9
|
+
const existing = seen.get(tag.kind);
|
|
10
|
+
if (existing === undefined || (tag.score ?? 0) > (existing.score ?? 0)) {
|
|
11
|
+
seen.set(tag.kind, tag);
|
|
12
|
+
}
|
|
13
|
+
}
|
|
14
|
+
return [...seen.values()].toSorted((a, b) => TAG_ORDER.indexOf(a.kind) - TAG_ORDER.indexOf(b.kind));
|
|
15
|
+
};
|
|
16
|
+
export const fuse = (results, context) => {
|
|
17
|
+
const byKey = new Map();
|
|
18
|
+
for (const result of results) {
|
|
19
|
+
result.hits.forEach((hit, rank) => {
|
|
20
|
+
const key = `${hit.path}:${hit.line}`;
|
|
21
|
+
const existing = byKey.get(key);
|
|
22
|
+
const contribution = 1 / (RRF_K + rank + 1);
|
|
23
|
+
if (existing === undefined) {
|
|
24
|
+
byKey.set(key, {
|
|
25
|
+
path: hit.path,
|
|
26
|
+
line: hit.line,
|
|
27
|
+
text: hit.text,
|
|
28
|
+
...(hit.start !== undefined ? { start: hit.start } : {}),
|
|
29
|
+
...(hit.end !== undefined ? { end: hit.end } : {}),
|
|
30
|
+
tags: [...hit.tags],
|
|
31
|
+
score: contribution,
|
|
32
|
+
});
|
|
33
|
+
return;
|
|
34
|
+
}
|
|
35
|
+
existing.score += contribution;
|
|
36
|
+
existing.tags.push(...hit.tags);
|
|
37
|
+
if (existing.start === undefined && hit.start !== undefined) {
|
|
38
|
+
existing.text = hit.text;
|
|
39
|
+
existing.start = hit.start;
|
|
40
|
+
existing.end = hit.end ?? hit.start;
|
|
41
|
+
}
|
|
42
|
+
});
|
|
43
|
+
}
|
|
44
|
+
const hits = [];
|
|
45
|
+
for (const hit of byKey.values()) {
|
|
46
|
+
let score = hit.score;
|
|
47
|
+
if (context.boosts) {
|
|
48
|
+
if (hit.tags.some((tag) => tag.kind === "def")) {
|
|
49
|
+
score *= DEF_BOOST;
|
|
50
|
+
}
|
|
51
|
+
const pathLower = hit.path.toLowerCase();
|
|
52
|
+
if (context.queryTokens.some((token) => pathLower.includes(token))) {
|
|
53
|
+
score *= PATH_BOOST;
|
|
54
|
+
}
|
|
55
|
+
const mtime = context.mtimes.get(hit.path);
|
|
56
|
+
if (mtime !== undefined) {
|
|
57
|
+
const ageDays = Math.max(0, (context.now - mtime) / 86_400_000);
|
|
58
|
+
score *= 1 + 0.2 * 2 ** (-ageDays / RECENCY_HALF_LIFE_DAYS);
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
hits.push({ ...hit, tags: dedupeTags(hit.tags), score });
|
|
62
|
+
}
|
|
63
|
+
const byPath = new Map();
|
|
64
|
+
for (const hit of hits) {
|
|
65
|
+
const list = byPath.get(hit.path);
|
|
66
|
+
if (list === undefined) {
|
|
67
|
+
byPath.set(hit.path, [hit]);
|
|
68
|
+
}
|
|
69
|
+
else {
|
|
70
|
+
list.push(hit);
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
const groups = [];
|
|
74
|
+
for (const [path, groupHits] of byPath) {
|
|
75
|
+
const best = Math.max(...groupHits.map((hit) => hit.score));
|
|
76
|
+
groups.push({
|
|
77
|
+
path,
|
|
78
|
+
score: best * (1 + 0.05 * Math.log(groupHits.length)),
|
|
79
|
+
hits: groupHits.toSorted((a, b) => a.line - b.line),
|
|
80
|
+
});
|
|
81
|
+
}
|
|
82
|
+
return groups.toSorted((a, b) => b.score - a.score || (a.path < b.path ? -1 : 1));
|
|
83
|
+
};
|
|
84
|
+
export const queryTokens = (query) => [...new Set(query.toLowerCase().match(/[a-z0-9_$]{3,}/g) ?? [])];
|
|
85
|
+
//# sourceMappingURL=fuse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fuse.js","sourceRoot":"","sources":["../../src/plan/fuse.ts"],"names":[],"mappings":"AAGA,MAAM,KAAK,GAAG,EAAE,CAAC;AACjB,MAAM,SAAS,GAAG,GAAG,CAAC;AACtB,MAAM,UAAU,GAAG,IAAI,CAAC;AACxB,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAWlC,MAAM,SAAS,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;AAE5H,MAAM,UAAU,GAAG,CAAC,IAA0B,EAAwB,EAAE;IACpE,MAAM,IAAI,GAAG,IAAI,GAAG,EAA8B,CAAC;IACnD,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,QAAQ,KAAK,SAAS,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,CAAC,EAAE,CAAC;YACrE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QAC5B,CAAC;IACL,CAAC;IAED,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;AACxG,CAAC,CAAC;AAIF,MAAM,CAAC,MAAM,IAAI,GAAG,CAAC,OAAgC,EAAE,OAAoB,EAAiB,EAAE;IAC1F,MAAM,KAAK,GAAG,IAAI,GAAG,EAAiI,CAAC;IACvJ,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YAC9B,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;YACtC,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChC,MAAM,YAAY,GAAG,CAAC,GAAG,CAAC,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC;YAC5C,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBACzB,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE;oBACX,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,GAAG,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACxD,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAClD,IAAI,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC;oBACnB,KAAK,EAAE,YAAY;iBACtB,CAAC,CAAC;gBACH,OAAO;YACX,CAAC;YACD,QAAQ,CAAC,KAAK,IAAI,YAAY,CAAC;YAC/B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC;YAEhC,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,IAAI,GAAG,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBAC1D,QAAQ,CAAC,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;gBACzB,QAAQ,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;gBAC3B,QAAQ,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC;YACxC,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IACD,MAAM,IAAI,GAAgB,EAAE,CAAC;IAC7B,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;QAC/B,IAAI,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;QACtB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACjB,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBAC7C,KAAK,IAAI,SAAS,CAAC;YACvB,CAAC;YACD,MAAM,SAAS,GAAG,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACzC,IAAI,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjE,KAAK,IAAI,UAAU,CAAC;YACxB,CAAC;YACD,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC3C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACtB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,GAAG,KAAK,CAAC,GAAG,UAAU,CAAC,CAAC;gBAChE,KAAK,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,OAAO,GAAG,sBAAsB,CAAC,CAAC;YAChE,CAAC;QACL,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC9C,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACrB,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAClC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACrB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QAChC,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;IACL,CAAC;IACD,MAAM,MAAM,GAAkB,EAAE,CAAC;IACjC,KAAK,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,MAAM,EAAE,CAAC;QACrC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QAC5D,MAAM,CAAC,IAAI,CAAC;YACR,IAAI;YACJ,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACrD,IAAI,EAAE,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC;SACtD,CAAC,CAAC;IACP,CAAC;IACD,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACtF,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,KAAa,EAAY,EAAE,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"budget.d.ts","sourceRoot":"","sources":["../../src/render/budget.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,cAAc,GAAI,MAAM,MAAM,KAAG,MAAoC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"budget.js","sourceRoot":"","sources":["../../src/render/budget.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,IAAY,EAAU,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { RankedGroup } from "../types.js";
|
|
2
|
+
export interface Spool {
|
|
3
|
+
readonly generation: number;
|
|
4
|
+
readonly createdAt: number;
|
|
5
|
+
readonly echo: string;
|
|
6
|
+
readonly unit: string;
|
|
7
|
+
readonly style: "hits" | "paths" | "plain";
|
|
8
|
+
readonly showTags: boolean;
|
|
9
|
+
readonly groups: readonly RankedGroup[];
|
|
10
|
+
}
|
|
11
|
+
export declare const cursorId: (echo: string, scopeKey: string) => string;
|
|
12
|
+
export declare const encodeCursor: (id: string, offset: number) => string;
|
|
13
|
+
export declare const decodeCursor: (cursor: string) => {
|
|
14
|
+
id: string;
|
|
15
|
+
offset: number;
|
|
16
|
+
} | undefined;
|
|
17
|
+
export declare const writeSpool: (indexDir: string, id: string, spool: Spool) => void;
|
|
18
|
+
export declare const readSpool: (indexDir: string, id: string) => Spool | undefined;
|
|
19
|
+
//# sourceMappingURL=cursor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cursor.d.ts","sourceRoot":"","sources":["../../src/render/cursor.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAK/C,MAAM,WAAW,KAAK;IAClB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC;IAC3C,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,MAAM,EAAE,SAAS,WAAW,EAAE,CAAC;CAC3C;AAGD,eAAO,MAAM,QAAQ,GAAI,MAAM,MAAM,EAAE,UAAU,MAAM,KAAG,MAC+B,CAAC;AAE1F,eAAO,MAAM,YAAY,GAAI,IAAI,MAAM,EAAE,QAAQ,MAAM,KAAG,MAAuC,CAAC;AAElG,eAAO,MAAM,YAAY,GAAI,QAAQ,MAAM,KAAG;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,SAK9E,CAAC;AAIF,eAAO,MAAM,UAAU,GAAI,UAAU,MAAM,EAAE,IAAI,MAAM,EAAE,OAAO,KAAK,KAAG,IAYvE,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,UAAU,MAAM,EAAE,IAAI,MAAM,KAAG,KAAK,GAAG,SAMhE,CAAC"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import { createHash } from "node:crypto";
|
|
2
|
+
import { mkdirSync, readdirSync, readFileSync, statSync, unlinkSync, writeFileSync } from "node:fs";
|
|
3
|
+
import { join } from "node:path";
|
|
4
|
+
const ID_LENGTH = 8;
|
|
5
|
+
const SPOOL_MAX_AGE_MS = 24 * 60 * 60 * 1000;
|
|
6
|
+
export const cursorId = (echo, scopeKey) => createHash("sha256").update(`${echo}\0${scopeKey}`).digest("hex").slice(0, ID_LENGTH);
|
|
7
|
+
export const encodeCursor = (id, offset) => `${id}${offset.toString(36)}`;
|
|
8
|
+
export const decodeCursor = (cursor) => {
|
|
9
|
+
if (cursor.length <= ID_LENGTH || !/^[0-9a-f]{8}[0-9a-z]+$/.test(cursor)) {
|
|
10
|
+
return undefined;
|
|
11
|
+
}
|
|
12
|
+
return { id: cursor.slice(0, ID_LENGTH), offset: parseInt(cursor.slice(ID_LENGTH), 36) };
|
|
13
|
+
};
|
|
14
|
+
const spoolDir = (indexDir) => join(indexDir, "spool");
|
|
15
|
+
export const writeSpool = (indexDir, id, spool) => {
|
|
16
|
+
const dir = spoolDir(indexDir);
|
|
17
|
+
mkdirSync(dir, { recursive: true });
|
|
18
|
+
for (const name of readdirSync(dir)) {
|
|
19
|
+
const path = join(dir, name);
|
|
20
|
+
const age = Date.now() - (statSync(path, { throwIfNoEntry: false })?.mtimeMs ?? Date.now());
|
|
21
|
+
if (age > SPOOL_MAX_AGE_MS) {
|
|
22
|
+
unlinkSync(path);
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
writeFileSync(join(dir, `${id}.json`), JSON.stringify(spool));
|
|
26
|
+
};
|
|
27
|
+
export const readSpool = (indexDir, id) => {
|
|
28
|
+
try {
|
|
29
|
+
return JSON.parse(readFileSync(join(spoolDir(indexDir), `${id}.json`), "utf8"));
|
|
30
|
+
}
|
|
31
|
+
catch {
|
|
32
|
+
return undefined;
|
|
33
|
+
}
|
|
34
|
+
};
|
|
35
|
+
//# sourceMappingURL=cursor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cursor.js","sourceRoot":"","sources":["../../src/render/cursor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACpG,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAGjC,MAAM,SAAS,GAAG,CAAC,CAAC;AACpB,MAAM,gBAAgB,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAa7C,MAAM,CAAC,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAE,QAAgB,EAAU,EAAE,CAC/D,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,GAAG,IAAI,KAAK,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;AAE1F,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,EAAU,EAAE,MAAc,EAAU,EAAE,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;AAElG,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,MAAc,EAA8C,EAAE;IACvF,IAAI,MAAM,CAAC,MAAM,IAAI,SAAS,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACvE,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;AAC7F,CAAC,CAAC;AAEF,MAAM,QAAQ,GAAG,CAAC,QAAgB,EAAU,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;AAEvE,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,QAAgB,EAAE,EAAU,EAAE,KAAY,EAAQ,EAAE;IAC3E,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC/B,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAEpC,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QAC5F,IAAI,GAAG,GAAG,gBAAgB,EAAE,CAAC;YACzB,UAAU,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;IACL,CAAC;IACD,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;AAClE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,QAAgB,EAAE,EAAU,EAAqB,EAAE;IACzE,IAAI,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,CAAU,CAAC;IAC7F,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,SAAS,CAAC;IACrB,CAAC;AACL,CAAC,CAAC"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import type { WorkspaceSearchFreshness } from "@intentic/sandbox-contract";
|
|
2
|
+
import type { RankedGroup } from "../types.js";
|
|
3
|
+
export interface RenderRequest {
|
|
4
|
+
readonly verb: string;
|
|
5
|
+
readonly echo: string;
|
|
6
|
+
readonly unit: string;
|
|
7
|
+
readonly style: "hits" | "paths" | "plain";
|
|
8
|
+
readonly showTags: boolean;
|
|
9
|
+
readonly groups: readonly RankedGroup[];
|
|
10
|
+
readonly offset: number;
|
|
11
|
+
readonly freshness: WorkspaceSearchFreshness;
|
|
12
|
+
readonly budget: number;
|
|
13
|
+
readonly limit?: number;
|
|
14
|
+
readonly filesOnly?: boolean;
|
|
15
|
+
readonly count?: boolean;
|
|
16
|
+
readonly headerNote?: string;
|
|
17
|
+
readonly hint?: string;
|
|
18
|
+
readonly related?: readonly string[];
|
|
19
|
+
readonly candidates?: readonly string[];
|
|
20
|
+
readonly cursorId: string;
|
|
21
|
+
}
|
|
22
|
+
export interface Rendered {
|
|
23
|
+
readonly text: string;
|
|
24
|
+
readonly shownGroups: number;
|
|
25
|
+
readonly shownHits: number;
|
|
26
|
+
readonly truncated: boolean;
|
|
27
|
+
readonly cursor?: string;
|
|
28
|
+
readonly exitCode: 0 | 1;
|
|
29
|
+
}
|
|
30
|
+
export declare const renderText: (request: RenderRequest) => Rendered;
|
|
31
|
+
//# sourceMappingURL=text.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"text.d.ts","sourceRoot":"","sources":["../../src/render/text.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,wBAAwB,EAAsB,MAAM,4BAA4B,CAAC;AAC/F,OAAO,KAAK,EAAE,WAAW,EAAa,MAAM,aAAa,CAAC;AAI1D,MAAM,WAAW,aAAa;IAC1B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAGtB,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC;IAC3C,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,MAAM,EAAE,SAAS,WAAW,EAAE,CAAC;IAExC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,SAAS,EAAE,wBAAwB,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC;IAC7B,QAAQ,CAAC,KAAK,CAAC,EAAE,OAAO,CAAC;IACzB,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAEvB,QAAQ,CAAC,OAAO,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IAGrC,QAAQ,CAAC,UAAU,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IAExC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,QAAQ;IACrB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;IAC5B,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC;CAC5B;AA8BD,eAAO,MAAM,UAAU,GAAI,SAAS,aAAa,KAAG,QAuHnD,CAAC"}
|
|
@@ -0,0 +1,134 @@
|
|
|
1
|
+
import { estimateTokens } from "./budget.js";
|
|
2
|
+
import { encodeCursor } from "./cursor.js";
|
|
3
|
+
const tagText = (tags) => tags.map((tag) => (tag.score === undefined ? `[${tag.kind}]` : `[${tag.kind} ${tag.score.toFixed(2)}]`)).join(" ");
|
|
4
|
+
const freshnessText = (freshness) => {
|
|
5
|
+
if (freshness.state === "building") {
|
|
6
|
+
return `index building ${Math.round((freshness.progress ?? 0) * 100)}%`;
|
|
7
|
+
}
|
|
8
|
+
if (freshness.state === "stale") {
|
|
9
|
+
return "index stale";
|
|
10
|
+
}
|
|
11
|
+
return `index fresh (${((freshness.ageMs ?? 0) / 1000).toFixed(1)}s)`;
|
|
12
|
+
};
|
|
13
|
+
const hitLine = (hit, showTags, plain) => {
|
|
14
|
+
const tags = showTags && hit.tags.length > 0 ? ` ${tagText(hit.tags)}` : "";
|
|
15
|
+
const context = hit.context !== undefined ? ` ⟨in ${hit.context}⟩` : "";
|
|
16
|
+
return plain ? ` ${hit.text}${tags}` : ` ${hit.line}: ${hit.text}${tags}${context}`;
|
|
17
|
+
};
|
|
18
|
+
const pathLine = (group, showTags) => {
|
|
19
|
+
const tags = showTags ? (group.hits[0]?.tags ?? []) : [];
|
|
20
|
+
const summary = group.hits[0] !== undefined && group.hits[0].text !== group.path ? ` ${group.hits[0].text}` : "";
|
|
21
|
+
return ` ${group.path}${summary}${tags.length > 0 ? ` ${tagText(tags)}` : ""}`;
|
|
22
|
+
};
|
|
23
|
+
export const renderText = (request) => {
|
|
24
|
+
const { groups, offset, unit, style } = request;
|
|
25
|
+
const totalHits = style === "paths" ? groups.length : groups.reduce((sum, group) => sum + group.hits.length, 0);
|
|
26
|
+
const totalFiles = groups.length;
|
|
27
|
+
const pending = groups.slice(offset, request.limit !== undefined ? offset + request.limit : undefined);
|
|
28
|
+
const header = (shown, note) => {
|
|
29
|
+
const scope = style === "paths" ? `${totalHits} ${unit}` : `${totalHits} ${unit} in ${totalFiles} files`;
|
|
30
|
+
const noteText = note !== undefined ? ` · ${note}` : "";
|
|
31
|
+
return `iq: ${request.echo} — ${scope} · ${freshnessText(request.freshness)}${noteText} · showing ${shown}/${totalHits}`;
|
|
32
|
+
};
|
|
33
|
+
const footer = (remainingHits, remainingFiles, cursor) => style === "paths"
|
|
34
|
+
? `──── truncated: ${remainingHits} more ${unit} · next: iq ${request.echo} --after ${cursor} ────`
|
|
35
|
+
: `──── truncated: ${remainingHits} ${unit} in ${remainingFiles} more files · next: iq ${request.echo} --after ${cursor} ────`;
|
|
36
|
+
const worstCursor = encodeCursor(request.cursorId, offset + pending.length);
|
|
37
|
+
const relatedLines = (request.related ?? []).map((line) => `related: ${line}`);
|
|
38
|
+
const candidateLine = request.candidates !== undefined && request.candidates.length > 0 ? `candidates: ${request.candidates.join(" · ")}` : undefined;
|
|
39
|
+
let remaining = request.budget -
|
|
40
|
+
estimateTokens(header(totalHits, request.headerNote)) -
|
|
41
|
+
estimateTokens(footer(totalHits, totalFiles, worstCursor)) -
|
|
42
|
+
relatedLines.reduce((sum, line) => sum + estimateTokens(line), 0) -
|
|
43
|
+
(candidateLine !== undefined ? estimateTokens(candidateLine) : 0) -
|
|
44
|
+
(request.hint !== undefined ? estimateTokens(`hint: ${request.hint}`) : 0);
|
|
45
|
+
const groupCap = Math.max(200, Math.floor(request.budget / 6));
|
|
46
|
+
const bodyLines = [];
|
|
47
|
+
let shownGroups = 0;
|
|
48
|
+
let shownHits = 0;
|
|
49
|
+
for (const group of pending) {
|
|
50
|
+
const lines = [];
|
|
51
|
+
let groupTokens = 0;
|
|
52
|
+
if (style === "paths") {
|
|
53
|
+
const line = pathLine(group, request.showTags);
|
|
54
|
+
lines.push(line);
|
|
55
|
+
groupTokens = estimateTokens(line);
|
|
56
|
+
}
|
|
57
|
+
else if (request.filesOnly || request.count) {
|
|
58
|
+
const line = request.count ? ` ${group.path}: ${group.hits.length}` : ` ${group.path} (${group.hits.length})`;
|
|
59
|
+
lines.push(line);
|
|
60
|
+
groupTokens = estimateTokens(line);
|
|
61
|
+
}
|
|
62
|
+
else {
|
|
63
|
+
const head = `════ ${group.path} (${group.hits.length}) ════`;
|
|
64
|
+
lines.push(head);
|
|
65
|
+
groupTokens = estimateTokens(head);
|
|
66
|
+
for (let i = 0; i < group.hits.length; i++) {
|
|
67
|
+
const hit = group.hits[i];
|
|
68
|
+
const line = hitLine(hit, request.showTags, style === "plain");
|
|
69
|
+
const lineTokens = estimateTokens(line);
|
|
70
|
+
const elided = group.hits.length - i;
|
|
71
|
+
if (groupTokens + lineTokens > groupCap && i > 0) {
|
|
72
|
+
const elision = ` … ${elided} more — iq context ${group.path}:${hit.line}`;
|
|
73
|
+
lines.push(elision);
|
|
74
|
+
groupTokens += estimateTokens(elision);
|
|
75
|
+
break;
|
|
76
|
+
}
|
|
77
|
+
lines.push(line);
|
|
78
|
+
groupTokens += lineTokens;
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
if (groupTokens > remaining && shownGroups > 0) {
|
|
82
|
+
break;
|
|
83
|
+
}
|
|
84
|
+
if (groupTokens > remaining) {
|
|
85
|
+
const trimmed = [];
|
|
86
|
+
let used = 0;
|
|
87
|
+
for (const line of lines) {
|
|
88
|
+
const lineTokens = estimateTokens(line);
|
|
89
|
+
if (used + lineTokens > remaining) {
|
|
90
|
+
break;
|
|
91
|
+
}
|
|
92
|
+
trimmed.push(line);
|
|
93
|
+
used += lineTokens;
|
|
94
|
+
}
|
|
95
|
+
if (trimmed.length === 0) {
|
|
96
|
+
break;
|
|
97
|
+
}
|
|
98
|
+
bodyLines.push(...trimmed);
|
|
99
|
+
remaining -= used;
|
|
100
|
+
shownGroups++;
|
|
101
|
+
shownHits += style === "paths" ? 1 : Math.max(0, trimmed.length - 1);
|
|
102
|
+
break;
|
|
103
|
+
}
|
|
104
|
+
bodyLines.push(...lines);
|
|
105
|
+
remaining -= groupTokens;
|
|
106
|
+
shownGroups++;
|
|
107
|
+
shownHits += style === "paths" ? 1 : Math.min(group.hits.length, lines.length - 1);
|
|
108
|
+
}
|
|
109
|
+
const truncated = offset + shownGroups < groups.length;
|
|
110
|
+
const cursor = truncated ? encodeCursor(request.cursorId, offset + shownGroups) : undefined;
|
|
111
|
+
const shownTotal = style === "paths" ? shownGroups : shownHits;
|
|
112
|
+
const parts = [header(shownTotal, request.headerNote), ...bodyLines];
|
|
113
|
+
if (truncated && cursor !== undefined) {
|
|
114
|
+
const remainingFiles = groups.length - offset - shownGroups;
|
|
115
|
+
const remainingHits = style === "paths" ? remainingFiles : groups.slice(offset + shownGroups).reduce((sum, group) => sum + group.hits.length, 0);
|
|
116
|
+
parts.push(footer(remainingHits, remainingFiles, cursor));
|
|
117
|
+
}
|
|
118
|
+
parts.push(...relatedLines);
|
|
119
|
+
if (candidateLine !== undefined) {
|
|
120
|
+
parts.push(candidateLine);
|
|
121
|
+
}
|
|
122
|
+
if (request.hint !== undefined) {
|
|
123
|
+
parts.push(`hint: ${request.hint}`);
|
|
124
|
+
}
|
|
125
|
+
return {
|
|
126
|
+
text: `${parts.join("\n")}\n`,
|
|
127
|
+
shownGroups,
|
|
128
|
+
shownHits: shownTotal,
|
|
129
|
+
truncated,
|
|
130
|
+
...(cursor !== undefined ? { cursor } : {}),
|
|
131
|
+
exitCode: totalHits > 0 ? 0 : 1,
|
|
132
|
+
};
|
|
133
|
+
};
|
|
134
|
+
//# sourceMappingURL=text.js.map
|