pmx-canvas 0.6.5 → 0.7.1

package/src/client/state/sse-bridge.ts

@@ -27,10 +27,13 @@ import {
   workbenchConnectionEpoch,
   reconnectAttempt,
   reconnectDelay,
+  resetCanvasInteractionState,
 } from './canvas-store';
+import { applyBoards, loadBoards, setBoardSwitchHandler } from './boards-store';
 import {
   fetchAgentPresence,
   fetchAxSurfaceState,
+  fetchCanvasState,
   reportClientViewportSize,
   requestJson,
   tabAgentKey,
@@ -38,7 +41,7 @@ import {
 import { agentActivity, applyPresenceSnapshot, sessionActive } from './presence-store';
 import { applyHumanSnapshot, startHumanPresence } from './human-store';
 import type { HumanPresenceSnapshot } from '../../shared/human-presence.js';
-import { applySessionReceipt, refreshTimeline } from './session-store';
+import { applySessionReceipt, dismissSessionReceipt, refreshTimeline } from './session-store';
 import { initSessionThemeOverride, themeOverrideActive } from './theme-override';
 import { DEFAULT_POSITIONS, makeNodeState } from './node-factory';
 import { invalidateTokenCache } from '../theme/tokens';
@@ -304,6 +307,7 @@ function isCanvasNodeType(value: unknown): value is CanvasNodeState['type'] {
     value === 'webpage' ||
     value === 'json-render' ||
     value === 'graph' ||
+    value === 'board' ||
     value === 'prompt' ||
     value === 'response' ||
     value === 'status' ||
@@ -456,14 +460,46 @@ function handleAgentSessionEnded(data: Record<string, unknown>): void {
 }
 
 let stopHumanPresence: (() => void) | null = null;
+let boardResyncGeneration = 0;
+let layoutRevision = 0;
+let pinsRevision = 0;
 
 function handleHumanPresence(data: Record<string, unknown>): void {
   applyHumanSnapshot(data as Partial<HumanPresenceSnapshot>);
 }
 
+/**
+ * The open board changed (this tab, another tab, or a restart): the canvas is
+ * a different board now, so take its layout like a fresh connection would —
+ * camera included — and drop what belonged to the old board.
+ */
+function resyncForBoardSwitch(): void {
+  const generation = ++boardResyncGeneration;
+  savedLayout = null;
+  hasInitialServerLayout.value = false;
+  resetIntents();
+  resetCanvasInteractionState();
+  resetAttentionBridge();
+  dismissSessionReceipt();
+  axSurfaceState.value = null;
+  void refreshAxSurface();
+  const layoutAtRequest = layoutRevision;
+  const pinsAtRequest = pinsRevision;
+  void fetchCanvasState().then((layout) => {
+    if (generation === boardResyncGeneration && layoutAtRequest === layoutRevision)
+      handleCanvasLayoutUpdate({ layout });
+  });
+  void requestJson<{ nodeIds?: unknown }>('fetchPinnedContext', '/api/canvas/pinned-context', {}).then((data) => {
+    if (generation === boardResyncGeneration && pinsAtRequest === pinsRevision)
+      handleContextPinsChanged({ nodeIds: data.nodeIds });
+  });
+}
+setBoardSwitchHandler(resyncForBoardSwitch);
+
 function handleConnected(data: Record<string, unknown>): void {
   sessionId.value = (data.sessionId as string) || '';
   connectionStatus.value = 'connected';
+  void loadBoards();
   // Reconnect marker for holders of server-minted URLs (Finding S).
   workbenchConnectionEpoch.value += 1;
   reloadIfServerUpgraded(data.version, data.bundleStamp);
@@ -486,8 +522,7 @@ function handleConnected(data: Record<string, unknown>): void {
   // The AX surface snapshot (work items, gates) likewise — it used to arrive
   // only on the first ax-state-changed, so a fresh load showed an empty
   // session panel while persisted work items existed.
-  void fetchAxSurfaceState().then((state) => {
-    axSurfaceState.value = state;
+  void refreshAxSurface().then(() => {
     if (sessionActive.value) void refreshTimeline();
   });
   // A ?theme= session override (host-default theming) wins over the
@@ -903,6 +938,7 @@ function handleCanvasLayoutUpdate(data: Record<string, unknown>): void {
       }
     | undefined;
   if (!layout?.nodes) return;
+  layoutRevision += 1;
   const shouldApplyViewport = !hasInitialServerLayout.value;
   hasInitialServerLayout.value = true;
 
@@ -993,6 +1029,7 @@ function handleThemeChanged(data: Record<string, unknown>): void {
 }
 
 function handleContextPinsChanged(data: Record<string, unknown>): void {
+  pinsRevision += 1;
   const nodeIds = Array.isArray(data.nodeIds) ? data.nodeIds.filter((id): id is string => typeof id === 'string') : [];
   replaceContextPinsFromServer(nodeIds);
   syncAttentionFromSse({ event: 'context-pins-changed', data });
@@ -1003,13 +1040,21 @@ function handleContextPinsChanged(data: Record<string, unknown>): void {
 // (debounced). The snapshot feeds AX-enabled surfaces (HtmlNode/McpAppNode push it
 // into their iframes), so authored boards reflect the live work queue / focus.
 let axRefreshTimer: ReturnType<typeof setTimeout> | null = null;
+let axRefreshSequence = 0;
+async function refreshAxSurface(): Promise<void> {
+  const sequence = ++axRefreshSequence;
+  const generation = boardResyncGeneration;
+  const state = await fetchAxSurfaceState();
+  if (sequence === axRefreshSequence && generation === boardResyncGeneration) axSurfaceState.value = state;
+}
+
 function handleAxStateChanged(): void {
   if (axRefreshTimer) clearTimeout(axRefreshTimer);
+  // Invalidate any in-flight read before the debounced replacement starts.
+  axRefreshSequence += 1;
   axRefreshTimer = setTimeout(() => {
     axRefreshTimer = null;
-    void fetchAxSurfaceState().then((state) => {
-      axSurfaceState.value = state;
-    });
+    void refreshAxSurface();
     // The session panel's timeline — only worth fetching while it is mounted.
     if (sessionActive.value) void refreshTimeline();
   }, 150);
@@ -1064,6 +1109,7 @@ export const EVENT_HANDLERS: Record<string, (data: Record<string, unknown>) => v
   'ext-app-update': handleExtAppUpdate,
   'ext-app-result': handleExtAppResult,
   'context-pins-changed': handleContextPinsChanged,
+  'boards-changed': applyBoards,
   'canvas-layout-update': handleCanvasLayoutUpdate,
   'canvas-focus-node': handleCanvasFocusNode,
   'canvas-viewport-update': handleCanvasViewportUpdate,