@herbertgao/pi-extensions 2026.8.5 → 2026.8.7

package/node_modules/pi-lens/dist/mcp/server.js

@@ -29,6 +29,7 @@ import { acknowledgeTurnEnd, analyzeFile, analyzeFileFresh, canRebuildPiLens, cr
 import { createAstGrepReplaceTool } from "../tools/ast-grep-replace.js";
 import { createAstGrepSearchTool } from "../tools/ast-grep-search.js";
 import { createLensDiagnosticsTool } from "../tools/lens-diagnostics.js";
+import { peekMcpSessionRuntime } from "../clients/mcp/session.js";
 import { createLspDiagnosticsTool } from "../tools/lsp-diagnostics.js";
 import { createLspNavigationTool } from "../tools/lsp-navigation.js";
 import { computeBuildStamp, STALE_SERVED_BY_FRESH, STALE_WARN_ONLY, StalenessGate, stalenessCheckEnabled, } from "./build-staleness.js";
@@ -411,7 +412,12 @@ const cacheManager = new CacheManager();
 // nothing — a no-op dressed as a fix, not a real parity gap. The 4th arg is left
 // at its default (`async () => {}`, already a no-op) rather than importing and
 // wiring a flush with nothing to flush.
-const lensDiagnosticsTool = createLensDiagnosticsTool(cacheManager, () => DEFAULT_CWD);
+const lensDiagnosticsTool = createLensDiagnosticsTool(cacheManager, () => DEFAULT_CWD, undefined, undefined, undefined, undefined, 
+// #1413 surface parity: validate cached test-runner findings against the
+// same session identity the in-process path uses. Resolved lazily — until
+// an MCP session context exists there is no session to compare against and
+// validation skips the check, which is the honest classification.
+() => peekMcpSessionRuntime());
 const astGrepClient = new AstGrepClient();
 const astGrepSearchTool = createAstGrepSearchTool(astGrepClient);
 const astGrepReplaceTool = createAstGrepReplaceTool(astGrepClient);
@@ -544,7 +550,7 @@ const ALL_TOOLS = [
             properties: {
                 query: {
                     type: "string",
-                    description: "Identifier-ish query, e.g. 'authenticate user'.",
+                    description: "Identifier-ish query, e.g. 'authenticate user'. Mix in composable prefix filters: lang:<kind> (e.g. lang:jsts, lang:python — kinds from file-kinds.ts), file:<substr> (path substring), ext:<ext> (e.g. ext:ts or ext:.ts), each optionally negated with a leading '-' (-file:test). Filters apply before ranking; e.g. 'lang:jsts file:clients/ -file:test rank'. Unknown prefixes/kinds error with the supported list.",
                 },
                 cwd: { type: "string" },
                 limit: {

package/node_modules/pi-lens/dist/tools/activate-tools.js

@@ -18,7 +18,7 @@
  * instead, so calling this tool is harmless, just unnecessary.
  */
 import { Type } from "../clients/deps/typebox.js";
-export function createActivateToolsTool(pi, lazyTools) {
+export function createActivateToolsTool(pi, lazyTools, options = {}) {
     const lazyNames = lazyTools.map((t) => t.name);
     const lazyNameSet = new Set(lazyNames);
     const catalog = lazyTools.map((t) => `${t.name} — ${t.summary}`).join("\n");
@@ -35,7 +35,7 @@ export function createActivateToolsTool(pi, lazyTools) {
                 description: "Names of situational tools to activate (see this tool's description for the catalog).",
             }),
         }),
-        async execute(_toolCallId, params, _signal, _onUpdate) {
+        async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
             const requested = Array.isArray(params.tools)
                 ? params.tools.filter((t) => typeof t === "string" && lazyNameSet.has(t))
                 : [];
@@ -53,10 +53,22 @@ export function createActivateToolsTool(pi, lazyTools) {
             }
             // Additive only, per the docs' contract: never drop currently active
             // tools in the same call.
+            // Remember every requested tool, not just the newly-added ones: a
+            // tool that is already active still has to survive the next
+            // fork/reload/resume restore.
+            options.onActivated?.(requested);
             const active = typeof pi.getActiveTools === "function" ? pi.getActiveTools() : [];
-            const merged = [...new Set([...active, ...requested])];
-            if (typeof pi.setActiveTools === "function") {
+            const activeSet = new Set(active);
+            const added = requested.filter((name) => !activeSet.has(name));
+            const merged = [...new Set([...active, ...added])];
+            if (added.length > 0 && typeof pi.setActiveTools === "function") {
                 pi.setActiveTools(merged);
+                options.onMutation?.({
+                    addedCount: added.length,
+                    removedCount: 0,
+                    reason: "lazy_activation",
+                    deferralApplies: options.deferredToolSupport?.(ctx) ?? false,
+                });
             }
             return {
                 content: [
@@ -65,7 +77,7 @@ export function createActivateToolsTool(pi, lazyTools) {
                         text: `Activated: ${requested.join(", ")}. Available starting next turn.`,
                     },
                 ],
-                details: { matches: requested, added: requested },
+                details: { matches: requested, added },
             };
         },
     };

package/node_modules/pi-lens/dist/tools/ast-grep-replace.js

@@ -67,7 +67,12 @@ export function createAstGrepReplaceTool(astGrepClient) {
             }),
             paths: Type.Optional(Type.Array(Type.String(), { description: "Specific files/folders" })),
             insideKind: Type.Optional(Type.String({ description: "Restrict matches to nodes inside an ancestor of this AST node kind. Synthesizes a YAML rule." })),
-            hasKind: Type.Optional(Type.String({ description: "Restrict matches to nodes that contain a descendant of this AST node kind." })),
+            hasKind: Type.Optional(Type.String({
+                description: 'Restrict matches to nodes whose immediate child has this AST node kind (ast-grep default stopBy: neighbor). Example: `hasKind: "await_expression"`.',
+            })),
+            hasDescendantKind: Type.Optional(Type.String({
+                description: "Restrict matches to nodes containing this AST node kind anywhere in their descendants. Explicit recursive form (`stopBy: end`); use this instead of `hasKind` when nesting is not immediate.",
+            })),
             follows: Type.Optional(Type.String({ description: "Restrict matches to nodes that immediately follow a sibling matching this pattern." })),
             precedes: Type.Optional(Type.String({ description: "Restrict matches to nodes that immediately precede a sibling matching this pattern." })),
             apply: Type.Optional(Type.Boolean({ description: "Apply changes (default: false)" })),
@@ -78,7 +83,7 @@ export function createAstGrepReplaceTool(astGrepClient) {
         }),
         async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
             const startedAt = Date.now();
-            const { pattern, rewrite, paths, apply, strictness, insideKind, hasKind, follows, precedes } = params;
+            const { pattern, rewrite, paths, apply, strictness, insideKind, hasKind, hasDescendantKind, follows, precedes } = params;
             const lang = (params.lang ?? "").replace(/^"|"$/g, "");
             const pathsCount = paths?.length ?? 1;
             const applyFlag = apply ?? false;
@@ -122,10 +127,10 @@ export function createAstGrepReplaceTool(astGrepClient) {
             }
             const searchPaths = paths?.length ? paths : [ctx.cwd || "."];
             // Phase 3: structural-intent params → synthesize YAML with fix: field
-            if (hasStructuralIntent({ insideKind, hasKind, follows, precedes })) {
+            if (hasStructuralIntent({ insideKind, hasKind, hasDescendantKind, follows, precedes })) {
                 let ruleYaml;
                 try {
-                    ruleYaml = synthesizeReplaceRule({ pattern, lang, rewrite, insideKind, hasKind, follows, precedes });
+                    ruleYaml = synthesizeReplaceRule({ pattern, lang, rewrite, insideKind, hasKind, hasDescendantKind, follows, precedes });
                 }
                 catch (err) {
                     logOutcome("error", { errorRaw: String(err) });

package/node_modules/pi-lens/dist/tools/lens-diagnostic-mark.js

@@ -162,7 +162,10 @@ function verifyLine(content, absPath, tool, rule, message, callerLine) {
         };
     }
 }
-export function createLensDiagnosticMarkTool(getCwd) {
+export function createLensDiagnosticMarkTool(getCwd, 
+/** Runtime telemetry identity, when known (#1448 class sweep) — attributed
+ * onto the disposition log alongside the mark. */
+getIdentity) {
     return {
         name: "lens_diagnostic_mark",
         label: "Mark Diagnostic",
@@ -326,7 +329,7 @@ export function createLensDiagnosticMarkTool(getCwd) {
                 message,
                 line: verifiedLine,
                 content,
-            }, disposition, reason);
+            }, disposition, reason, getIdentity?.());
             const verb = disposition === "suppress"
                 ? `suppressed (inline pi-lens-ignore comment written above line ${verifiedLine})`
                 : disposition === "defer"