@herbertgao/pi-extensions 2026.9.0 → 2026.9.2

package/node_modules/pi-lens/dist/clients/mcp/session.js

@@ -20,7 +20,7 @@ import * as crypto from "node:crypto";
 import * as fs from "node:fs";
 import * as path from "node:path";
 import { AstGrepClient } from "../ast-grep-client.js";
-import { loadBootstrapClients } from "../bootstrap.js";
+import { loadBootstrapClients, residentBootstrapAccess, } from "../bootstrap.js";
 import { CacheManager, MCP_TURN_STATE_OWNER_ID, } from "../cache-manager.js";
 import { resetDispatchBaselines } from "../dispatch/integration.js";
 import { resetFormatService } from "../format-service.js";
@@ -96,23 +96,12 @@ export async function runSessionStart(cwd) {
         dbg: noop,
         log: noop,
         runtime: ctx.runtime,
-        metricsClient: ctx.clients.metricsClient,
         cacheManager: ctx.cacheManager,
-        todoScanner: ctx.clients.todoScanner,
         astGrepClient: ctx.astGrepClient,
-        biomeClient: ctx.clients.biomeClient,
-        ruffClient: ctx.clients.ruffClient,
-        knipClient: ctx.clients.knipClient,
-        jscpdClient: ctx.clients.jscpdClient,
-        deadCodeClients: ctx.clients.deadCodeClients,
-        govulncheckClient: ctx.clients.govulncheckClient,
-        gitleaksClient: ctx.clients.gitleaksClient,
-        trivyClient: ctx.clients.trivyClient,
-        opengrepClient: ctx.clients.opengrepClient,
-        depChecker: ctx.clients.depChecker,
-        testRunnerClient: ctx.clients.testRunnerClient,
-        goClient: ctx.clients.goClient,
-        rustClient: ctx.clients.rustClient,
+        // The MCP server has already loaded the analyzer graph (an MCP call has
+        // nothing to defer to), so it hands the handler the same one-shape seam
+        // every other caller uses rather than a second, field-by-field shape.
+        bootstrap: residentBootstrapAccess(ctx.clients),
         ensureTool: async (name) => (await import("../installer/index.js")).ensureTool(name),
         cleanStaleTsBuildInfo: () => [],
         resetDispatchBaselines: (cwdArg) => resetDispatchBaselines(cwdArg ?? cwd),

package/node_modules/pi-lens/dist/clients/memory-sampler.js

@@ -106,6 +106,13 @@ export function toMemoryProcessUsage(mem, resourceUsage) {
  * process-global singletons) but every individual read is a `.size`/`.length`
  * access — see the module docstring's hard constraint.
  */
+/** Byte attribution for `dispatchCaches`, from measured per-entry costs: three
+ *  isolated-store runs put a recently-clean neighbor entry at 320 bytes, and
+ *  #2282 measured a 500-file batch's 1,000 session-fact entries at 439,560
+ *  bytes (~440 each). O(1) — the cache key and Map bookkeeping are included. */
+export function estimateDispatchCacheBytes(stats) {
+    return (stats.recentlyCleanNeighborCacheSize * 320 + stats.sessionFactEntries * 440);
+}
 export function collectMemorySampleSubsystems(wordIndex) {
     const reviewGraph = getReviewGraphWorkspaceCacheSnapshot();
     const treeSitterClient = getSharedTreeSitterClient();
@@ -149,9 +156,7 @@ export function collectMemorySampleSubsystems(wordIndex) {
         treeSitter,
         dispatchCaches: {
             ...dispatchCaches,
-            // Three isolated-store runs measured 320 bytes per production entry.
-            // Keep this O(1); the cache key and Map bookkeeping are included.
-            estimatedBytes: dispatchCaches.recentlyCleanNeighborCacheSize * 320,
+            estimatedBytes: estimateDispatchCacheBytes(dispatchCaches),
         },
     };
 }

package/node_modules/pi-lens/dist/clients/metrics-history.js

@@ -93,7 +93,7 @@ export function loadHistory() {
 /**
  * Save history to disk
  */
-export function saveHistory(history) {
+function saveHistory(history) {
     const historyDir = getProjectDataDir(process.cwd());
     if (!fs.existsSync(historyDir)) {
         fs.mkdirSync(historyDir, { recursive: true });
@@ -161,10 +161,35 @@ export function captureSnapshot(filePath, metrics) {
     }, SAVE_DEBOUNCE_MS);
     saveTimer.unref?.();
 }
+// --- Trend Analysis ---
 /**
- * Capture snapshots for multiple files (explicit, immediate save)
- * Used by /lens-metrics for batch capture
+ * Compute trend direction from history snapshots
+ * Uses last 3 snapshots for stability (or 2 if only 2 available)
  */
+function computeTrend(history) {
+    if (history.length < 2)
+        return "stable";
+    const recent = history.slice(-3);
+    const first = recent[0];
+    const last = recent[recent.length - 1];
+    // Use MI as primary indicator, cognitive as secondary
+    const miDelta = last.mi - first.mi;
+    const cogDelta = last.cognitive - first.cognitive;
+    // Thresholds (MI changes < 2 are noise)
+    if (miDelta > 2)
+        return "improving";
+    if (miDelta < -2)
+        return "regressing";
+    // If MI is stable, check cognitive
+    if (cogDelta < -10)
+        return "improving";
+    if (cogDelta > 10)
+        return "regressing";
+    return "stable";
+}
+// --- Technical Debt Index (TDI) ---
+// Test-reached through a child-process `require` string
+/** @public — reached through a child-process `require` string in tests/clients/metrics-history-stderr.test.ts; knip cannot see it. */
 export function captureSnapshots(files) {
     const history = loadHistory();
     for (const file of files) {
@@ -200,110 +225,6 @@ export function captureSnapshots(files) {
     saveHistory(history);
     return history;
 }
-// --- Trend Analysis ---
-/**
- * Compute trend direction from history snapshots
- * Uses last 3 snapshots for stability (or 2 if only 2 available)
- */
-export function computeTrend(history) {
-    if (history.length < 2)
-        return "stable";
-    const recent = history.slice(-3);
-    const first = recent[0];
-    const last = recent[recent.length - 1];
-    // Use MI as primary indicator, cognitive as secondary
-    const miDelta = last.mi - first.mi;
-    const cogDelta = last.cognitive - first.cognitive;
-    // Thresholds (MI changes < 2 are noise)
-    if (miDelta > 2)
-        return "improving";
-    if (miDelta < -2)
-        return "regressing";
-    // If MI is stable, check cognitive
-    if (cogDelta < -10)
-        return "improving";
-    if (cogDelta > 10)
-        return "regressing";
-    return "stable";
-}
-/**
- * Get delta between current snapshot and previous
- */
-export function getDelta(history) {
-    if (!history || history.history.length < 2)
-        return null;
-    const current = history.history[history.history.length - 1];
-    const previous = history.history[history.history.length - 2];
-    return {
-        mi: Math.round((current.mi - previous.mi) * 10) / 10,
-        cognitive: current.cognitive - previous.cognitive,
-        trend: history.trend,
-    };
-}
-/**
- * Get trend emoji for display
- */
-export function getTrendEmoji(trend) {
-    switch (trend) {
-        case "improving":
-            return "📈";
-        case "regressing":
-            return "📉";
-        default:
-            return "➡️";
-    }
-}
-/**
- * Get trend summary across all files
- */
-export function getTrendSummary(history) {
-    let improving = 0;
-    let regressing = 0;
-    let stable = 0;
-    const regressions = [];
-    for (const [file, fileHistory] of Object.entries(history.files)) {
-        switch (fileHistory.trend) {
-            case "improving":
-                improving++;
-                break;
-            case "regressing": {
-                regressing++;
-                const delta = getDelta(fileHistory);
-                if (delta) {
-                    regressions.push({ file, miDelta: delta.mi });
-                }
-                break;
-            }
-            default:
-                stable++;
-        }
-    }
-    // Sort regressions by MI delta (worst first)
-    regressions.sort((a, b) => a.miDelta - b.miDelta);
-    return {
-        improving,
-        regressing,
-        stable,
-        worstRegressions: regressions.slice(0, 5),
-    };
-}
-/**
- * Format trend for metrics table
- */
-export function formatTrendCell(filePath, history) {
-    const relativePath = path.relative(process.cwd(), filePath);
-    const fileHistory = history.files[relativePath];
-    if (!fileHistory || fileHistory.history.length < 2) {
-        return "—"; // No history
-    }
-    const delta = getDelta(fileHistory);
-    if (!delta)
-        return "—";
-    const emoji = getTrendEmoji(delta.trend);
-    const miSign = delta.mi > 0 ? "+" : "";
-    const miColor = delta.mi > 0 ? "🟢" : delta.mi < 0 ? "🔴" : "⚪";
-    return `${emoji} ${miColor}${miSign}${delta.mi}`;
-}
 /**
  * Calculate Technical Debt Index for the project.
  * Score: 0 = perfect, 100 = maximum debt.

package/node_modules/pi-lens/dist/clients/middle-man-analysis.js

@@ -27,6 +27,7 @@
  * slice can add them once middle-man analysis has an AST-node entry point
  * rather than this text-based one.
  */
+import { escapeRegExp } from "./string-utils.js";
 /** Self-reference token + member-access separator, per languageId. Both the
  * self→field and field→method hops use the SAME separator in every language
  * covered here (true even for PHP's `$this->field->method()`). */
@@ -65,9 +66,6 @@ const MIN_CANDIDATE_METHODS = 2;
  * "mostly" — a class with a couple of forwarding convenience methods among
  * real logic must NOT flag (the exact distinction #325 asks for). */
 const DELEGATION_RATIO_THRESHOLD = 0.9;
-function escapeRegExp(s) {
-    return s.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
-}
 function buildForwardRegex(token, sep) {
     const t = escapeRegExp(token);
     const s = escapeRegExp(sep);
@@ -98,7 +96,7 @@ function paramNamesFromSignature(signature, languageId) {
         .map((raw) => {
         let name = raw;
         // Rest/spread params forward as themselves textually (`...args`).
-        const isRest = /^\.\.\./.test(name);
+        const isRest = name.startsWith("...");
         name = name.replace(/^\.\.\./, "");
         name = name.split("=")[0].trim(); // strip default value
         name = name.split(":")[0].trim(); // strip type annotation
@@ -161,7 +159,7 @@ function isAccessorLine(rawLine) {
  * extracted data — no re-parsing, no file I/O beyond the lines already read
  * by the caller.
  */
-export function analyzeMiddleMan(classEntry, lines, languageId) {
+function analyzeMiddleMan(classEntry, lines, languageId) {
     if (!languageId)
         return undefined;
     const selfSpec = SELF_TOKEN[languageId];

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

@@ -41,63 +41,36 @@ 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-
 // lsp.ts) to be re-homed in #236, where LSP writes provenance-tagged edges INTO
 // the review graph (computed once, persisted) so this path just reads them.
 /** Hard payload bound for the per-symbol who-uses-this section. */
-export const MAX_MODULE_REPORT_REFS = 100;
+const MAX_MODULE_REPORT_REFS = 100;
 function normalizeMaxRefsPerSymbol(value) {
     if (value === undefined || !Number.isFinite(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++) {