@blackbelt-technology/pi-agent-dashboard 0.5.2 → 0.5.4

package/packages/server/src/browser-gateway.ts

@@ -66,10 +66,10 @@ import { createViewedSessionTracker, type ViewedSessionTracker } from "./viewed-
 import type { TerminalManager } from "./terminal-manager.js";
 import type { BrowserHandlerContext } from "./browser-handlers/handler-context.js";
 import { handleSubscribe } from "./browser-handlers/subscription-handler.js";
-import { handleSendPrompt, handleResumeSession, handleSpawnSession, handleShutdown, handleAbort, handleFlowControl, handleForceKill, handleKillProcess } from "./browser-handlers/session-action-handler.js";
+import { handleSendPrompt, handleResumeSession, handleSpawnSession, handleShutdown, handleAbort, handleFlowControl, handleForceKill, handleKillProcess, handleClearSteeringQueue, handleClearFollowupSlot, handleEditFollowupSlot, handlePromoteFollowupEntry, handleRemoveFollowupEntry, handleEditFollowupEntry } from "./browser-handlers/session-action-handler.js";
 import { handleRenameSession, handleHideSession, handleUnhideSession, handleAttachProposal, handleDetachProposal, handleFetchContent, handleListSessions } from "./browser-handlers/session-meta-handler.js";
 import { handleCreateTerminal, handleKillTerminal, handleRenameTerminal } from "./browser-handlers/terminal-handler.js";
-import { handlePinDirectory, handleUnpinDirectory, handleReorderPinnedDirs, handleReorderSessions, handleOpenSpecRefresh, handleOpenSpecBulkArchive, handleExtensionUiResponse, handlePiGatewayForward } from "./browser-handlers/directory-handler.js";
+import { handlePinDirectory, handleUnpinDirectory, handleReorderPinnedDirs, handleReorderSessions, handleOpenSpecRefresh, handleOpenSpecBulkArchive, handleExtensionUiResponse, handlePiGatewayForward, handleCreateWorkspace, handleRenameWorkspace, handleDeleteWorkspace, handleSetWorkspaceCollapsed, handleAddFolderToWorkspace, handleRemoveFolderFromWorkspace, handleReorderWorkspaceFolders, handleReorderWorkspaces } from "./browser-handlers/directory-handler.js";
 
 
 
@@ -111,6 +111,17 @@ export interface BrowserGateway {
   onConnect?: (ws: WebSocket) => void;
   /** Broadcast a message to all connected clients */
   broadcast(msg: ServerToBrowserMessage): void;
+  /**
+   * Register a handler for a Browser→Server message type the gateway does
+   * not natively handle. Used by plugins to receive `plugin_action`
+   * messages without modifying the gateway's switch statement.
+   * See change: adopt-server-driven-intent-rendering.
+   */
+  registerHandler(
+    type: string,
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    handler: (msg: any, ws: WebSocket) => void,
+  ): void;
 }
 
 export function createBrowserGateway(
@@ -131,6 +142,14 @@ export function createBrowserGateway(
 ): BrowserGateway {
   const wss = new WebSocketServer({ noServer: true });
 
+  /**
+   * Plugin-registered handlers for custom Browser→Server message types.
+   * Lives outside subscriptions because handlers are global, not per-WS.
+   * See change: adopt-server-driven-intent-rendering.
+   */
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  const customHandlers = new Map<string, (msg: any, ws: WebSocket) => void>();
+
   // Track subscriptions: ws → Set<sessionId>
   const subscriptions = new Map<WebSocket, Set<string>>();
   // Track which sessions are mid-replay per WebSocket (suppress live events)
@@ -274,6 +293,12 @@ export function createBrowserGateway(
     // Send pinned directories on connect
     if (preferencesStore) {
       sendTo(ws, { type: "pinned_dirs_updated", paths: preferencesStore.getPinnedDirectories() });
+      // Send current workspaces snapshot. See change: folder-workspaces.
+      // Guarded with `typeof` so old PreferencesStore stubs in tests that
+      // predate workspaces still work — they simply get no workspace snapshot.
+      if (typeof preferencesStore.getWorkspaces === "function") {
+        sendTo(ws, { type: "workspaces_updated", workspaces: preferencesStore.getWorkspaces() });
+      }
     }
 
     // Send OpenSpec data for every known directory — exactly one
@@ -358,6 +383,24 @@ export function createBrowserGateway(
           case "abort":
             handleAbort(msg, ctx);
             break;
+          case "clear_steering_queue":
+            handleClearSteeringQueue(msg, ctx);
+            break;
+          case "clear_followup_slot":
+            handleClearFollowupSlot(msg, ctx);
+            break;
+          case "edit_followup_slot":
+            handleEditFollowupSlot(msg, ctx);
+            break;
+          case "promote_followup_entry":
+            handlePromoteFollowupEntry(msg, ctx);
+            break;
+          case "remove_followup_entry":
+            handleRemoveFollowupEntry(msg, ctx);
+            break;
+          case "edit_followup_entry":
+            handleEditFollowupEntry(msg, ctx);
+            break;
           case "force_kill":
             await handleForceKill(msg, ctx);
             break;
@@ -409,6 +452,30 @@ export function createBrowserGateway(
           case "reorder_pinned_dirs":
             handleReorderPinnedDirs(msg, ctx);
             break;
+          case "create_workspace":
+            handleCreateWorkspace(msg, ctx);
+            break;
+          case "rename_workspace":
+            handleRenameWorkspace(msg, ctx);
+            break;
+          case "delete_workspace":
+            handleDeleteWorkspace(msg, ctx);
+            break;
+          case "set_workspace_collapsed":
+            handleSetWorkspaceCollapsed(msg, ctx);
+            break;
+          case "add_folder_to_workspace":
+            handleAddFolderToWorkspace(msg, ctx);
+            break;
+          case "remove_folder_from_workspace":
+            handleRemoveFolderFromWorkspace(msg, ctx);
+            break;
+          case "reorder_workspace_folders":
+            handleReorderWorkspaceFolders(msg, ctx);
+            break;
+          case "reorder_workspaces":
+            handleReorderWorkspaces(msg, ctx);
+            break;
           case "openspec_refresh":
             handleOpenSpecRefresh(msg, ctx);
             break;
@@ -532,10 +599,17 @@ export function createBrowserGateway(
             viewedSessionTracker.unview(msg.sessionId, ws);
             break;
           }
-          default:
-            // Forward simple pi-gateway commands
-            handlePiGatewayForward(msg, ctx);
+          default: {
+            // Plugin-registered custom handler takes precedence over pi-gateway forward.
+            const type = (msg as { type?: string } | undefined)?.type;
+            if (type && customHandlers.has(type)) {
+              customHandlers.get(type)!(msg, ws);
+            } else {
+              // Forward simple pi-gateway commands
+              handlePiGatewayForward(msg, ctx);
+            }
             break;
+          }
         }
       } catch (err) {
         const type = (msg as { type?: string } | undefined)?.type ?? "unknown";
@@ -568,6 +642,10 @@ export function createBrowserGateway(
       broadcast(msg);
     },
 
+    registerHandler(type, handler) {
+      customHandlers.set(type, handler);
+    },
+
     broadcastEvent(sessionId: string, seq: number, event: any) {
       const subscribers = getSubscribers(sessionId);
       const msg: ServerToBrowserMessage = {

package/packages/server/src/browser-handlers/directory-handler.ts

@@ -75,6 +75,75 @@ export function handleReorderPinnedDirs(
   }
 }
 
+// ── folder-workspaces handlers ──────────────────────────────────
+//
+// Each handler dispatches to PreferencesStore which returns true on
+// mutation. Broadcast `workspaces_updated` only on actual mutation
+// (no broadcast for no-op / invalid / unknown-id calls). See spec
+// folder-workspaces.
+
+function broadcastWorkspaces(ctx: BrowserHandlerContext): void {
+  if (!ctx.preferencesStore) return;
+  ctx.broadcast({ type: "workspaces_updated", workspaces: ctx.preferencesStore.getWorkspaces() });
+}
+
+export function handleCreateWorkspace(
+  msg: Extract<BrowserToServerMessage, { type: "create_workspace" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  const ws = ctx.preferencesStore?.createWorkspace(msg.name);
+  if (ws) broadcastWorkspaces(ctx);
+}
+
+export function handleRenameWorkspace(
+  msg: Extract<BrowserToServerMessage, { type: "rename_workspace" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (ctx.preferencesStore?.renameWorkspace(msg.id, msg.name)) broadcastWorkspaces(ctx);
+}
+
+export function handleDeleteWorkspace(
+  msg: Extract<BrowserToServerMessage, { type: "delete_workspace" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (ctx.preferencesStore?.deleteWorkspace(msg.id)) broadcastWorkspaces(ctx);
+}
+
+export function handleSetWorkspaceCollapsed(
+  msg: Extract<BrowserToServerMessage, { type: "set_workspace_collapsed" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (ctx.preferencesStore?.setWorkspaceCollapsed(msg.id, msg.collapsed)) broadcastWorkspaces(ctx);
+}
+
+export function handleAddFolderToWorkspace(
+  msg: Extract<BrowserToServerMessage, { type: "add_folder_to_workspace" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (ctx.preferencesStore?.addFolderToWorkspace(msg.id, msg.path)) broadcastWorkspaces(ctx);
+}
+
+export function handleRemoveFolderFromWorkspace(
+  msg: Extract<BrowserToServerMessage, { type: "remove_folder_from_workspace" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (ctx.preferencesStore?.removeFolderFromWorkspace(msg.id, msg.path)) broadcastWorkspaces(ctx);
+}
+
+export function handleReorderWorkspaceFolders(
+  msg: Extract<BrowserToServerMessage, { type: "reorder_workspace_folders" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (ctx.preferencesStore?.reorderWorkspaceFolders(msg.id, msg.paths)) broadcastWorkspaces(ctx);
+}
+
+export function handleReorderWorkspaces(
+  msg: Extract<BrowserToServerMessage, { type: "reorder_workspaces" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (ctx.preferencesStore?.reorderWorkspaces(msg.ids)) broadcastWorkspaces(ctx);
+}
+
 export function handleReorderSessions(
   msg: Extract<BrowserToServerMessage, { type: "reorder_sessions" }>,
   ctx: BrowserHandlerContext,

package/packages/server/src/browser-handlers/session-action-handler.ts

@@ -256,6 +256,7 @@ export async function handleSendPrompt(
       sessionId: msg.sessionId,
       text: msg.text,
       images: msg.images,
+      delivery: msg.delivery,
     });
     if (!sent) {
       console.error(`[dashboard] send_prompt failed: no bridge connection for session ${msg.sessionId}`);
@@ -523,6 +524,94 @@ export function handleAbort(
   ctx.piGateway.sendToSession(msg.sessionId, { type: "abort", sessionId: msg.sessionId });
 }
 
+/**
+ * Clear pi's steering queue for the given session. Forwards to bridge,
+ * which calls `pi.clearSteeringQueue()`. Idempotent. Drops silently when
+ * the session is unknown. See change: add-followup-edit-and-steer-cancel.
+ */
+export function handleClearSteeringQueue(
+  msg: Extract<BrowserToServerMessage, { type: "clear_steering_queue" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (!ctx.sessionManager.get(msg.sessionId)) return;
+  ctx.piGateway.sendToSession(msg.sessionId, { type: "clear_steering_queue", sessionId: msg.sessionId });
+}
+
+/**
+ * Clear pi's follow-up slot for the given session. Forwards to bridge,
+ * which calls `pi.clearFollowUpQueue()`. Idempotent.
+ * See change: add-followup-edit-and-steer-cancel.
+ */
+export function handleClearFollowupSlot(
+  msg: Extract<BrowserToServerMessage, { type: "clear_followup_slot" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (!ctx.sessionManager.get(msg.sessionId)) return;
+  ctx.piGateway.sendToSession(msg.sessionId, { type: "clear_followup_slot", sessionId: msg.sessionId });
+}
+
+/**
+ * Atomic replace of pi's follow-up slot (clear-then-send). Forwards text +
+ * optional images to bridge, which calls clearFollowUpQueue then sendUserMessage.
+ * See change: add-followup-edit-and-steer-cancel.
+ */
+export function handleEditFollowupSlot(
+  msg: Extract<BrowserToServerMessage, { type: "edit_followup_slot" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (!ctx.sessionManager.get(msg.sessionId)) return;
+  ctx.piGateway.sendToSession(msg.sessionId, {
+    type: "edit_followup_slot",
+    sessionId: msg.sessionId,
+    text: msg.text,
+    images: msg.images,
+  });
+}
+
+/**
+ * Promote the follow-up entry at `index` to the head (position 0) of the queue.
+ * See change: add-followup-edit-and-steer-cancel (task 13.3).
+ */
+export function handlePromoteFollowupEntry(
+  msg: Extract<BrowserToServerMessage, { type: "promote_followup_entry" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (!ctx.sessionManager.get(msg.sessionId)) return;
+  ctx.piGateway.sendToSession(msg.sessionId, {
+    type: "promote_followup_entry",
+    sessionId: msg.sessionId,
+    index: msg.index,
+  });
+}
+
+/** Remove the follow-up entry at `index` from the queue. */
+export function handleRemoveFollowupEntry(
+  msg: Extract<BrowserToServerMessage, { type: "remove_followup_entry" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (!ctx.sessionManager.get(msg.sessionId)) return;
+  ctx.piGateway.sendToSession(msg.sessionId, {
+    type: "remove_followup_entry",
+    sessionId: msg.sessionId,
+    index: msg.index,
+  });
+}
+
+/** Replace the follow-up entry at `index` with new text. */
+export function handleEditFollowupEntry(
+  msg: Extract<BrowserToServerMessage, { type: "edit_followup_entry" }>,
+  ctx: BrowserHandlerContext,
+): void {
+  if (!ctx.sessionManager.get(msg.sessionId)) return;
+  ctx.piGateway.sendToSession(msg.sessionId, {
+    type: "edit_followup_entry",
+    sessionId: msg.sessionId,
+    index: msg.index,
+    text: msg.text,
+    images: msg.images,
+  });
+}
+
 export function handleFlowControl(
   msg: Extract<BrowserToServerMessage, { type: "flow_control" }>,
   ctx: BrowserHandlerContext,

package/packages/server/src/browser-handlers/subscription-handler.ts

@@ -5,6 +5,7 @@ import type { WebSocket } from "ws";
 import type { ServerToBrowserMessage, BrowserToServerMessage } from "@blackbelt-technology/pi-dashboard-shared/browser-protocol.js";
 import type { BrowserHandlerContext } from "./handler-context.js";
 import { extractStatsFromEvents } from "../event-status-extraction.js";
+import { pluginIntentCache } from "../plugin-intent-cache.js";
 import type { StoredEvent } from "../memory-event-store.js";
 
 const REPLAY_BATCH_SIZE = 50;
@@ -94,6 +95,28 @@ export function replayUiState(
       sendTo(ws, { type: "ext_ui_decorator", sessionId, descriptor } as any);
     }
   }
+
+  // Replay cached plugin intents for this session (per-session AND global).
+  // See change: adopt-server-driven-intent-rendering.
+  for (const entry of pluginIntentCache.getForSession(sessionId)) {
+    sendTo(ws, {
+      type: "plugin_intents",
+      pluginId: entry.pluginId,
+      sessionId: entry.sessionId,
+      slot: entry.slot,
+      intent: entry.intent,
+    } as any);
+  }
+  // Also replay global (sessionId === null) intents.
+  for (const entry of pluginIntentCache.getForSession(null)) {
+    sendTo(ws, {
+      type: "plugin_intents",
+      pluginId: entry.pluginId,
+      sessionId: null,
+      slot: entry.slot,
+      intent: entry.intent,
+    } as any);
+  }
 }
 
 /**

package/packages/server/src/changelog-parser.ts

@@ -1,7 +1,7 @@
 /**
  * Pure parser for Keep-a-Changelog-style markdown files.
  *
- * Pi (`@mariozechner/pi-coding-agent`) ships a `CHANGELOG.md` whose
+ * Pi (`@earendil-works/pi-coding-agent` (formerly `@mariozechner/pi-coding-agent`)) ships a `CHANGELOG.md` whose
  * format is mechanically reliable:
  *   - H2 release headers: `## [<version>] - <date>`
  *   - H3 sub-section headers: `### Breaking Changes`, `### New Features`,