fix(workbench): sync engine bridge task statuses

Author: bb-connor

apps/workbench/src/features/swarm/hooks/__tests__/use-engine-board-bridge.test.ts

@@ -358,4 +358,127 @@ describe("useEngineBoardBridge", () => {
       vi.useRealTimers();
     }
   });
+
+  it("refreshes the restore status when guard evaluation re-triggers after a status change", () => {
+    vi.useFakeTimers();
+
+    const events = new TypedEventEmitter<SwarmEngineEventMap>();
+    const engine = {
+      getState: () => makeEngineState(),
+      getEvents: () => events,
+    };
+
+    const { unmount } = renderHook(() =>
+      useEngineBoardBridge(engine as any),
+    );
+
+    try {
+      act(() => {
+        events.emit("agent.status_changed", {
+          agentId: "agt_pool_1",
+          newStatus: "running",
+        } as any);
+      });
+
+      act(() => {
+        events.emit("guard.evaluated", {
+          action: { agentId: "agt_pool_1" },
+          result: {
+            verdict: "allow",
+            guardResults: [],
+            receipt: {
+              signature: "beef".repeat(32),
+              publicKey: "1234".repeat(16),
+            },
+          },
+        } as any);
+      });
+
+      act(() => {
+        events.emit("agent.status_changed", {
+          agentId: "agt_pool_1",
+          newStatus: "idle",
+        } as any);
+      });
+
+      act(() => {
+        events.emit("guard.evaluated", {
+          action: { agentId: "agt_pool_1" },
+          result: {
+            verdict: "allow",
+            guardResults: [],
+            receipt: {
+              signature: "cafe".repeat(32),
+              publicKey: "5678".repeat(16),
+            },
+          },
+        } as any);
+      });
+
+      act(() => {
+        vi.advanceTimersByTime(2000);
+      });
+
+      expect(
+        useSwarmBoardStore.getState().nodes.find((node) => node.id === "agt_pool_1")?.data.status,
+      ).toBe("idle");
+    } finally {
+      unmount();
+      vi.useRealTimers();
+    }
+  });
+
+  it("maps created tasks from engine status updates instead of forcing them to running", () => {
+    const events = new TypedEventEmitter<SwarmEngineEventMap>();
+    const engine = {
+      getState: () => makeEngineState(),
+      getEvents: () => events,
+    };
+
+    const { unmount } = renderHook(() =>
+      useEngineBoardBridge(engine as any),
+    );
+
+    try {
+      act(() => {
+        events.emit("task.created", {
+          task: {
+            id: "tsk_1",
+            name: "Queue scan",
+            type: "analysis",
+            status: "created",
+            assignedTo: null,
+          },
+        } as any);
+      });
+
+      expect(
+        useSwarmBoardStore.getState().nodes.find((node) => node.data.taskId === "tsk_1")?.data.status,
+      ).toBe("idle");
+
+      act(() => {
+        events.emit("task.status_changed", {
+          taskId: "tsk_1",
+          newStatus: "paused",
+        } as any);
+      });
+
+      expect(
+        useSwarmBoardStore.getState().nodes.find((node) => node.data.taskId === "tsk_1")?.data.status,
+      ).toBe("blocked");
+
+      act(() => {
+        events.emit("task.status_changed", {
+          taskId: "tsk_1",
+          newStatus: "cancelled",
+        } as any);
+      });
+
+      expect(
+        useSwarmBoardStore.getState().nodes.find((node) => node.data.taskId === "tsk_1")?.data.status,
+      ).toBe("completed");
+    } finally {
+      unmount();
+    }
+  });
 });

apps/workbench/src/features/swarm/hooks/use-engine-board-bridge.ts

@@ -17,14 +17,24 @@ const DEFAULT_LAYOUT_VIEWPORT = { width: 1200, height: 800 };
 
 function mapEngineStatus(engineStatus: string): SessionStatus {
   switch (engineStatus) {
+    case "created":
+    case "queued":
+    case "assigned":
+      return "idle";
     case "running":
       return "running";
     case "idle":
       return "idle";
     case "busy":
       return "running";
+    case "paused":
+      return "blocked";
     case "terminated":
       return "completed";
+    case "cancelled":
+      return "completed";
+    case "timeout":
+      return "failed";
     case "offline":
       return "failed";
     case "completed":
@@ -290,7 +300,7 @@ export function useEngineBoardBridge(engine: SwarmOrchestrator | null): void {
           data: {
             nodeType: "terminalTask",
             title: event.task.type ?? event.task.name ?? "Task",
-            status: "running",
+            status: mapEngineStatus(event.task.status),
             taskId: event.task.id,
             agentId: event.task.assignedTo,
             engineManaged: true,
@@ -309,6 +319,21 @@ export function useEngineBoardBridge(engine: SwarmOrchestrator | null): void {
       }),
     );
 
+    unsubs.push(
+      events.on("task.status_changed", (event: any) => {
+        const { nodes, actions } = store();
+
+        const taskNode = nodes.find(
+          (n: Node<SwarmBoardNodeData>) => n.data.taskId === event.taskId,
+        );
+        if (!taskNode) return;
+
+        actions.updateNode(taskNode.id, {
+          status: mapEngineStatus(event.newStatus),
+        });
+      }),
+    );
+
     unsubs.push(
       events.on("task.completed", (event: any) => {
         const { nodes, actions } = store();
@@ -350,7 +375,9 @@ export function useEngineBoardBridge(engine: SwarmOrchestrator | null): void {
           const existingTimeout = timeouts.get(nodeId);
           if (existingTimeout != null) {
             clearTimeout(existingTimeout);
-          } else {
+          }
+
+          if (currentStatus !== "evaluating" || !restoreStatuses.has(nodeId)) {
             restoreStatuses.set(
               nodeId,
               currentStatus === "evaluating" ? "running" : currentStatus,