pi-extensions-tool-display 0.2.2 → 1.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-extensions-tool-display",
-  "version": "0.2.2",
+  "version": "1.0.1",
   "description": "Pi tool display host and shared result-rendering protocol for extensions",
   "type": "module",
   "main": "./index.ts",

package/src/result-render-middleware.ts

@@ -66,10 +66,14 @@ export function registerResultRenderMiddleware(
   const api = getApi();
   if (typeof api?.registerResultRenderMiddleware === "function") {
     api.registerResultRenderMiddleware(registration);
-  } else {
-    queueRegistration(registration);
   }
 
+  // Keep the registration in the shared protocol queue even when a host is
+  // already available. The host can be replaced during /reload or when a new
+  // session creates a new ExtensionAPI instance; the new host must be able to
+  // rehydrate registrations that were attached to the previous API object.
+  queueRegistration(registration);
+
   return () => {
     getApi()?.unregisterResultRenderMiddleware?.(registration.id);
     removeQueuedRegistration(registration.id);

package/src/tool-overrides.ts

@@ -1525,8 +1525,11 @@ function drainPendingResultRenderMiddlewares(api: ToolDisplayApi): void {
   const pending = globalWithApi[TOOL_DISPLAY_PENDING_RESULT_MIDDLEWARES_KEY];
   if (!Array.isArray(pending) || pending.length === 0) return;
 
-  const entries = pending.splice(0);
-  for (const entry of entries) {
+  // This queue is the cross-host registry for external middleware, not a
+  // one-shot bootstrap queue. Keep it populated so a replacement host during
+  // /reload or a new session can rehydrate middleware registered on the old
+  // API instance.
+  for (const entry of pending) {
     if (!entry?.id || !entry.toolName || typeof entry.middleware !== "function") continue;
     api.registerResultRenderMiddleware(entry);
   }

package/tool-display-api-consumer.js

@@ -53,19 +53,25 @@ export function registerToolResultRenderMiddleware(toolName, middleware, options
   }
 
   const id = options.id || `external-result-middleware-${++nextResultMiddlewareId}`;
+  const registration = { id, toolName, middleware };
   const api = getToolDisplayApi();
   if (typeof api?.registerResultRenderMiddleware === "function") {
-    return api.registerResultRenderMiddleware({ id, toolName, middleware });
+    const registeredId = api.registerResultRenderMiddleware(registration);
+    queueResultRenderMiddleware(registration);
+    return registeredId;
   }
 
+  queueResultRenderMiddleware(registration);
+  return id;
+}
+
+function queueResultRenderMiddleware(registration) {
   const existing = globalThis[TOOL_DISPLAY_PENDING_RESULT_MIDDLEWARES_KEY];
   const queue = Array.isArray(existing) ? existing : [];
-  const registration = { id, toolName, middleware };
-  const index = queue.findIndex((entry) => entry?.id === id);
+  const index = queue.findIndex((entry) => entry?.id === registration.id);
   if (index >= 0) queue[index] = registration;
   else queue.push(registration);
   globalThis[TOOL_DISPLAY_PENDING_RESULT_MIDDLEWARES_KEY] = queue;
-  return id;
 }
 
 export function unregisterToolResultRenderMiddleware(id) {