pi-extensions-tool-display 1.1.0 → 1.1.2

package/README.md

@@ -14,6 +14,8 @@ It is also a standalone Pi extension. Install or include this package in Pi's pa
 
 This package owns the display protocol and generic component composition only. Each feature extension continues to own its domain-specific audit extraction, copy, status, and layout.
 
+Built-in overrides only replace tools that Pi has already activated; `registerToolOverrides` does not enable inactive built-in tools.
+
 ## Upstream attribution
 
 The display integration is designed to work with the MIT-licensed [`MasuRii/pi-tool-display`](https://github.com/MasuRii/pi-tool-display), the original full-featured Pi tool-display project. We thank MasuRii for that work.

package/README.zh-CN.md

@@ -14,6 +14,8 @@
 
 这个包负责实际工具展示、展示协议和通用组件组合。具体业务的审计数据提取、文案、状态和布局仍由调用方维护，避免把不同扩展耦合在一起。
 
+内建工具覆盖只替换 Pi 当前已经激活的工具；`registerToolOverrides` 不会启用原本未激活的内建工具。
+
 ## 上游引用
 
 本包的展示集成设计用于兼容 MIT 许可的 [`MasuRii/pi-tool-display`](https://github.com/MasuRii/pi-tool-display)，这是 Pi 生态中原始的完整工具展示项目。感谢 MasuRii 的工作。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-extensions-tool-display",
-  "version": "1.1.0",
+  "version": "1.1.2",
   "description": "Pi tool display host and shared result-rendering protocol for extensions",
   "type": "module",
   "main": "./index.ts",
@@ -48,8 +48,8 @@
     "tool-display"
   ],
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": ">=0.80.0 <0.81.0",
-    "@earendil-works/pi-tui": ">=0.80.0 <0.81.0"
+    "@earendil-works/pi-coding-agent": ">=0.80.0",
+    "@earendil-works/pi-tui": ">=0.80.0"
   },
   "pi": {
     "extensions": [
@@ -60,8 +60,8 @@
     ]
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "0.80.10",
-    "@earendil-works/pi-tui": "0.80.10",
+    "@earendil-works/pi-coding-agent": "0.85.1",
+    "@earendil-works/pi-tui": "0.85.1",
     "@types/node": "24.12.4",
     "tsx": "4.23.1",
     "typescript": "5.9.3"

package/src/index.ts

@@ -13,6 +13,7 @@ import {
   type ToolDisplayCapabilities,
 } from "./capabilities.js";
 import { registerToolDisplayOverrides } from "./tool-overrides.js";
+import { logToolDisplayDebug } from "./debug-logger.js";
 import { disposeAll, resetDisposed } from "./disposable.js";
 import { registerThinkingLabeling } from "./thinking-label.js";
 import registerNativeUserMessageBox from "./user-message-box-native.js";
@@ -24,6 +25,12 @@ import {
 
 export * from "./result-render-middleware.js";
 
+const TOOL_DISPLAY_RELOAD_ACTIVE_TOOLS_KEY = Symbol.for("pi-tool-display.reloadActiveTools.v1");
+const BUILT_IN_TOOL_OVERRIDE_NAME_SET = new Set<string>(BUILT_IN_TOOL_OVERRIDE_NAMES);
+type GlobalWithToolDisplayReloadState = typeof globalThis & {
+  [TOOL_DISPLAY_RELOAD_ACTIVE_TOOLS_KEY]?: string[];
+};
+
 function ownershipChanged(
   previous: ToolDisplayConfig,
   next: ToolDisplayConfig,
@@ -65,10 +72,21 @@ export function ensureToolDisplayHost(
   if (initializedHosts.has(pi)) return;
   initializedHosts.add(pi);
 
+  const globalReloadState = globalThis as GlobalWithToolDisplayReloadState;
+  const activeBuiltInToolNamesBeforeReload = globalReloadState[TOOL_DISPLAY_RELOAD_ACTIVE_TOOLS_KEY];
+  delete globalReloadState[TOOL_DISPLAY_RELOAD_ACTIVE_TOOLS_KEY];
+
   resetDisposed();
 
   pi.on("session_shutdown", (event: { reason: string }) => {
     if (event.reason === "reload") {
+      try {
+        globalReloadState[TOOL_DISPLAY_RELOAD_ACTIVE_TOOLS_KEY] = pi
+          .getActiveTools()
+          .filter((toolName) => BUILT_IN_TOOL_OVERRIDE_NAME_SET.has(toolName));
+      } catch (error) {
+        logToolDisplayDebug("Failed to preserve active built-in tools for reload.", error);
+      }
       disposeAll();
       initializedHosts.delete(pi);
     }
@@ -112,7 +130,7 @@ export function ensureToolDisplayHost(
   };
 
   if (initial.config.enabled) {
-    registerToolDisplayOverrides(pi, getEffectiveConfig);
+    registerToolDisplayOverrides(pi, getEffectiveConfig, activeBuiltInToolNamesBeforeReload);
     registerNativeUserMessageBox(pi, getConfig);
     registerThinkingLabeling(pi);
   }

package/src/tool-overrides.ts

@@ -1668,9 +1668,20 @@ function tryGetAllTools(pi: ExtensionAPI, debugMessage: string): unknown[] | und
   }
 }
 
+// Return undefined until Pi binds the extension action methods.
+function tryGetActiveToolNames(pi: ExtensionAPI, debugMessage: string): string[] | undefined {
+  try {
+    return pi.getActiveTools();
+  } catch (error) {
+    logToolDisplayDebug(debugMessage, error);
+    return undefined;
+  }
+}
+
 export function registerToolDisplayOverrides(
   pi: ExtensionAPI,
   getConfig: ConfigGetter,
+  activeToolNamesDuringLoad?: readonly string[],
 ): void {
   clearBuiltInToolCache();
   const toolDisplayApi = installToolDisplayApi(getConfig);
@@ -1688,7 +1699,7 @@ export function registerToolDisplayOverrides(
   const registeredBuiltInToolOverrides = new Set<BuiltInToolOverrideName>();
 
   const isExternallyOwnedBuiltInTool = (toolName: BuiltInToolOverrideName): boolean => {
-    const allTools = tryGetAllTools(pi, "Built-in tool override ownership discovery unavailable during extension load; registering renderer for pre-bind history rendering.");
+    const allTools = tryGetAllTools(pi, "Built-in tool override ownership discovery unavailable; skipping conflicting owners.");
     if (!allTools) {
       return false;
     }
@@ -1780,7 +1791,37 @@ export function registerToolDisplayOverrides(
     return { scope: getSearchScope(args), limitSuffix: args.limit !== undefined ? ` (limit ${args.limit})` : "" };
   };
 
-  registerIfOwned("read", () => {
+  const builtInToolRegistrations = new Map<BuiltInToolOverrideName, () => void>();
+
+  // Record each built-in replacement until Pi exposes its active tool selection.
+  const recordBuiltInToolRegistration = (
+    toolName: BuiltInToolOverrideName,
+    register: () => void,
+  ): void => {
+    builtInToolRegistrations.set(toolName, register);
+  };
+
+  // Retry deferred replacements against Pi's current active tool selection.
+  const registerActiveBuiltInToolOverrides = (
+    knownActiveToolNames?: readonly string[],
+  ): void => {
+    const activeToolNames = knownActiveToolNames ?? tryGetActiveToolNames(
+      pi,
+      "Built-in tool activation discovery unavailable; skipping display overrides.",
+    );
+    if (!activeToolNames) {
+      return;
+    }
+
+    const activeTools = new Set(activeToolNames);
+    for (const [toolName, register] of builtInToolRegistrations) {
+      if (activeTools.has(toolName)) {
+        registerIfOwned(toolName, register);
+      }
+    }
+  };
+
+  recordBuiltInToolRegistration("read", () => {
     registerRuntimeTool(pi, {
       name: "read",
       label: "read",
@@ -1795,7 +1836,7 @@ export function registerToolDisplayOverrides(
     });
   });
 
-  registerIfOwned("grep", () => {
+  recordBuiltInToolRegistration("grep", () => {
     registerRuntimeTool(pi, {
       name: "grep",
     label: "grep",
@@ -1816,7 +1857,7 @@ export function registerToolDisplayOverrides(
     });
   });
 
-  registerIfOwned("find", () => {
+  recordBuiltInToolRegistration("find", () => {
     registerRuntimeTool(pi, {
       name: "find",
     label: "find",
@@ -1834,7 +1875,7 @@ export function registerToolDisplayOverrides(
     });
   });
 
-  registerIfOwned("ls", () => {
+  recordBuiltInToolRegistration("ls", () => {
     registerRuntimeTool(pi, {
       name: "ls",
     label: "ls",
@@ -1850,7 +1891,7 @@ export function registerToolDisplayOverrides(
     });
   });
 
-  registerIfOwned("edit", () => {
+  recordBuiltInToolRegistration("edit", () => {
     registerRuntimeTool(pi, {
       name: "edit",
     label: "edit",
@@ -1872,7 +1913,7 @@ export function registerToolDisplayOverrides(
     });
   });
 
-  registerIfOwned("write", () => {
+  recordBuiltInToolRegistration("write", () => {
     registerRuntimeTool(pi, {
       name: "write",
     label: "write",
@@ -1945,7 +1986,7 @@ export function registerToolDisplayOverrides(
     });
   });
 
-  registerIfOwned("bash", () => {
+  recordBuiltInToolRegistration("bash", () => {
     registerRuntimeTool(pi, {
       name: "bash",
     label: "bash",
@@ -2024,6 +2065,8 @@ export function registerToolDisplayOverrides(
     });
   });
 
+  registerActiveBuiltInToolOverrides(activeToolNamesDuringLoad);
+
   const wrappedCustomToolNames = new Set<string>();
   registerCleanup(() => wrappedCustomToolNames.clear());
 
@@ -2229,11 +2272,13 @@ export function registerToolDisplayOverrides(
 
   pi.on("session_start", async () => {
     clearWriteExecutionMeta(writeExecutionMetaByToolCallId);
+    registerActiveBuiltInToolOverrides();
     registerMcpToolOverrides();
     scheduleMcpToolOverrideDiscovery();
   });
   pi.on("before_agent_start", async () => {
     clearWriteExecutionMeta(writeExecutionMetaByToolCallId);
+    registerActiveBuiltInToolOverrides();
     registerMcpToolOverrides();
     scheduleMcpToolOverrideDiscovery();
   });