Listen for connectivity changes and handle session state

Author: charlesvien

apps/array/src/renderer/features/sessions/stores/sessionStore.ts

@@ -459,7 +459,11 @@ const useStore = create<SessionStore>()(
       });
 
       if (result) {
-        updateSession(taskRunId, { status: "connected" });
+        // Set isPromptPending true - the resumed agent may still be generating
+        updateSession(taskRunId, {
+          status: "connected",
+          isPromptPending: true,
+        });
       } else {
         unsubscribeFromChannel(taskRunId);
         removeSession(taskRunId);
@@ -873,3 +877,24 @@ useAuthStore.subscribe(
     }
   },
 );
+
+// Listen for connectivity changes and handle session state
+trpcVanilla.connectivity.onStatusChange.subscribe(undefined, {
+  onData: (status) => {
+    if (!status.isOnline) {
+      log.info("Connectivity lost - marking sessions as disconnected");
+
+      useStore.setState((state) => {
+        for (const session of Object.values(state.sessions)) {
+          if (
+            session.status === "connected" ||
+            session.status === "connecting"
+          ) {
+            session.status = "disconnected";
+            session.isPromptPending = false;
+          }
+        }
+      });
+    }
+  },
+});

apps/array/src/renderer/features/task-detail/components/TaskLogsPanel.tsx

@@ -12,6 +12,7 @@ import {
   useWorkspaceStore,
 } from "@features/workspace/stores/workspaceStore";
 import { Box } from "@radix-ui/themes";
+import { useConnectivity } from "@renderer/hooks/useConnectivity";
 import { logger } from "@renderer/lib/logger";
 import { useNavigationStore } from "@renderer/stores/navigationStore";
 import { trpcVanilla } from "@renderer/trpc/client";
@@ -34,6 +35,7 @@ export function TaskLogsPanel({ taskId, task }: TaskLogsPanelProps) {
   const { connectToTask, sendPrompt, cancelPrompt } = useSessionActions();
   const markActivity = useTaskViewedStore((state) => state.markActivity);
   const markAsViewed = useTaskViewedStore((state) => state.markAsViewed);
+  const { isOnline } = useConnectivity();
 
   const isRunning =
     session?.status === "connected" || session?.status === "connecting";
@@ -46,6 +48,7 @@ export function TaskLogsPanel({ taskId, task }: TaskLogsPanelProps) {
   useEffect(() => {
     if (!repoPath) return;
     if (isConnecting.current) return;
+    if (!isOnline) return;
 
     if (session?.status === "connected" || session?.status === "connecting") {
       return;
@@ -75,7 +78,7 @@ export function TaskLogsPanel({ taskId, task }: TaskLogsPanelProps) {
     }).finally(() => {
       isConnecting.current = false;
     });
-  }, [task, repoPath, session, connectToTask, markActivity]);
+  }, [task, repoPath, session, connectToTask, markActivity, isOnline]);
 
   const handleSendPrompt = useCallback(
     async (text: string) => {