Show initial-agent workspaces as Working

packages/app/src/hooks/use-status-mode-workspaces.ts

@@ -1,21 +1,47 @@
 import { useMemo } from "react";
-import { useSessionStore, type WorkspaceDescriptor } from "@/stores/session-store";
+import { useCreateFlowStore } from "@/stores/create-flow-store";
+import { useSessionStore } from "@/stores/session-store";
 import {
   createSidebarWorkspaceEntry,
+  type SidebarProjectEntry,
   type SidebarWorkspaceEntry,
 } from "./use-sidebar-workspaces-list";
 
-export function useHydratedWorkspaceEntries(serverId: string | null): SidebarWorkspaceEntry[] {
+const EMPTY_WORKSPACES: SidebarWorkspaceEntry[] = [];
+
+export function useStatusModeWorkspaceEntries(input: {
+  serverId: string | null;
+  projects: SidebarProjectEntry[];
+}): SidebarWorkspaceEntry[] {
   const workspaces = useSessionStore((state) =>
-    serverId ? state.sessions[serverId]?.workspaces : undefined,
+    input.serverId ? state.sessions[input.serverId]?.workspaces : undefined,
   );
+  const agents = useSessionStore((state) =>
+    input.serverId ? state.sessions[input.serverId]?.agents : undefined,
+  );
+  const pendingCreateAttempts = useCreateFlowStore((state) => state.pendingByDraftId);
 
   return useMemo(() => {
-    if (!serverId || !workspaces || workspaces.size === 0) return [];
-    return Array.from(workspaces.values()).map((workspace: WorkspaceDescriptor) =>
-      createSidebarWorkspaceEntry({ serverId, workspace }),
-    );
-  }, [serverId, workspaces]);
+    if (!input.serverId || input.projects.length === 0 || !workspaces) {
+      return EMPTY_WORKSPACES;
+    }
+
+    const entries: SidebarWorkspaceEntry[] = [];
+    for (const placedWorkspace of input.projects.flatMap((project) => project.workspaces)) {
+      const workspace = workspaces.get(placedWorkspace.workspaceId);
+      entries.push(
+        workspace
+          ? createSidebarWorkspaceEntry({
+              serverId: input.serverId,
+              workspace,
+              pendingCreateAttempts,
+              agents,
+            })
+          : placedWorkspace,
+      );
+    }
+    return entries;
+  }, [agents, input.projects, input.serverId, pendingCreateAttempts, workspaces]);
 }
 
 export function useProjectNamesMap(serverId: string | null): Map<string, string> {