@herbertgao/pi-extensions 2026.9.0 → 2026.9.1

package/node_modules/pi-lens/dist/clients/module-report.js

@@ -41,7 +41,7 @@ import { loadCallGraph, } from "./call-graph.js";
 import { resolveImportToFiles } from "./review-graph/import-resolvers.js";
 import { buildSymbolId } from "./review-graph/symbol-id.js";
 import { getSharedTreeSitterClient, resolveTreeSitterLanguage, } from "./tree-sitter-shared.js";
-import { TreeSitterSymbolExtractor, } from "./tree-sitter-symbol-extractor.js";
+import { symbolExtractionGrammar, TreeSitterSymbolExtractor, } from "./tree-sitter-symbol-extractor.js";
 // NOTE: live-LSP enrichment is NO LONGER called on this read path (#256). Firing
 // LSP per read — speculatively, on the agent's fan-out path — repeatedly OOM'd
 // pi. The enrichment logic is kept as a separate module (clients/module-report-
@@ -54,50 +54,23 @@ function normalizeMaxRefsPerSymbol(value) {
         return 10;
     return Math.min(MAX_MODULE_REPORT_REFS, Math.max(1, Math.floor(value)));
 }
-// kind -> tree-sitter languageId. The languageId keys BOTH the grammar map
-// (tree-sitter-client) and SYMBOL_QUERIES (tree-sitter-symbol-extractor), so it
-// must match a key present in both. jsts/cxx are extension-split kinds resolved
-// through the SHARED ext→grammar resolver below (never a local extension map —
-// #887). Using these gives the primary languages the same rich outline
-// (classes/interfaces/types/signatures) as every other language, not the
-// functions-only FunctionSummary.
-const KIND_TO_TS_LANG = {
-    python: "python",
-    go: "go",
-    rust: "rust",
-    ruby: "ruby",
-    java: "java",
-    kotlin: "kotlin",
-    dart: "dart",
-    elixir: "elixir",
-    csharp: "csharp",
-    php: "php",
-    swift: "swift",
-    lua: "lua",
-    ocaml: "ocaml",
-    zig: "zig",
-    shell: "bash",
-    // cxx resolved by extension below (c vs cpp)
-};
 export function tsLangForFile(filePath, kind) {
-    // jsts/cxx are extension-split kinds: resolve them through the shared
-    // ext→grammar authority (tree-sitter-shared.ts EXT_TO_LANG) — the SAME
-    // resolver the dispatch tree-sitter runner, fact providers, project scanner
-    // and read expansion use — so a file is parsed and TreeCache-keyed
-    // (`languageId:path`) under exactly one grammar process-wide. #887: this
-    // used to hand-roll a local map that sent .js/.mjs/.cjs to the typescript
-    // grammar and .jsx to tsx, so every plain-JS file was parsed and cached
-    // twice under two grammars and ran TS-grammar symbol queries on JS trees.
-    // Extensions the shared map does not cover (.svelte/.vue for jsts; the
-    // module-interface/Objective-C/CUDA tail for cxx) keep the historical
-    // kind default.
+    // jsts is an extension-split kind with no single grammar: resolve it through
+    // the SHARED ext -> grammar authority (language-registry via
+    // tree-sitter-shared) so a file is parsed and TreeCache-keyed
+    // (`languageId:path`) under exactly one grammar process-wide. #887: this used
+    // to hand-roll a local map that sent .js/.mjs/.cjs to the typescript grammar
+    // and .jsx to tsx, so every plain-JS file was parsed and cached twice under
+    // two grammars and ran TS-grammar symbol queries on JS trees. .svelte/.vue
+    // have no grammar wiring and keep the historical kind default.
     if (kind === "jsts") {
         return resolveTreeSitterLanguage(filePath) ?? "typescript";
     }
-    if (kind === "cxx") {
-        return resolveTreeSitterLanguage(filePath) ?? "cpp";
-    }
-    return kind ? KIND_TO_TS_LANG[kind] : undefined;
+    // Every other kind goes through the registry-derived resolver, which handles
+    // the cxx `.c`/`.h`-vs-cpp split and only answers for grammars that have
+    // symbol queries. #2424 deleted the hand-kept kind -> languageId map that
+    // used to live here alongside review-graph's copy of the same thing.
+    return symbolExtractionGrammar(kind, filePath);
 }
 // Per-language extractor cache — extractors are cheap once their queries are
 // compiled. The shared TreeSitterClient (which memoizes grammar init) is obtained
@@ -867,7 +840,7 @@ const pythonCallbackRules = {
                 include: true,
             };
         }
-        if (/\.add_done_callback$/.test(callName)) {
+        if (callName.endsWith(".add_done_callback")) {
             return {
                 kind: "future_callback",
                 flags: withFlag(base.flags, "future completion"),
@@ -1012,7 +985,7 @@ const javaCallbackRules = {
         const base = classifyGenericCallback(ctx);
         const obj = ancestorOfType(ctx.node, new Set(["object_creation_expression"]), 3);
         const created = obj?.children?.find((c) => c.type === "type_identifier");
-        if (created && /Thread$/.test(created.text)) {
+        if (created && created.text.endsWith("Thread")) {
             return {
                 kind: "task",
                 flags: withFlag(base.flags, "thread"),
@@ -1734,8 +1707,10 @@ function levenshteinDistance(a, b) {
         return bl;
     if (bl === 0)
         return al;
-    let prev = new Array(bl + 1);
-    let curr = new Array(bl + 1);
+    let prev = [];
+    prev.length = bl + 1;
+    let curr = [];
+    curr.length = bl + 1;
     for (let j = 0; j <= bl; j++)
         prev[j] = j;
     for (let i = 1; i <= al; i++) {