feat: update task run status during local session lifecycle

Author: tatoalo

apps/twig/src/renderer/features/sessions/service/service.test.ts

@@ -61,6 +61,7 @@ const mockAuthStore = vi.hoisted(() => ({
       client: {
         createTaskRun: vi.fn(),
         appendTaskRunLog: vi.fn(),
+        updateTaskRun: vi.fn().mockResolvedValue(undefined),
       },
     })),
   },
@@ -237,6 +238,7 @@ describe("SessionService", () => {
         client: {
           createTaskRun: createTaskRunMock,
           appendTaskRunLog: vi.fn(),
+          updateTaskRun: vi.fn().mockResolvedValue(undefined),
         },
       });
 
@@ -373,6 +375,7 @@ describe("SessionService", () => {
         client: {
           createTaskRun: createTaskRunMock,
           appendTaskRunLog: vi.fn(),
+          updateTaskRun: vi.fn().mockResolvedValue(undefined),
         },
       });
       mockTrpcAgent.start.mutate.mockResolvedValue({

apps/twig/src/renderer/features/sessions/service/service.ts

@@ -390,6 +390,11 @@ export class SessionService {
       throw new Error("Failed to create task run. Please try again.");
     }
 
+    // Signal that the local run is now in progress
+    auth.client
+      .updateTaskRun(taskId, taskRun.id, { status: "in_progress" })
+      .catch((err) => log.warn("Failed to update task run status", { err }));
+
     const result = await trpcVanilla.agent.start.mutate({
       taskId,
       taskRunId: taskRun.id,
@@ -457,6 +462,17 @@ export class SessionService {
     const session = sessionStoreSetters.getSessionByTaskId(taskId);
     if (!session) return;
 
+    // Update remote status based on session state
+    const auth = useAuthStore.getState();
+    if (auth.client) {
+      const status = session.isPromptPending ? "failed" : "completed";
+      auth.client
+        .updateTaskRun(taskId, session.taskRunId, { status })
+        .catch((err) =>
+          log.warn("Failed to update task run status on disconnect", { err }),
+        );
+    }
+
     try {
       await trpcVanilla.agent.cancel.mutate({
         sessionId: session.taskRunId,
@@ -674,6 +690,20 @@ export class SessionService {
         notifyPromptComplete(session.taskTitle, stopReason);
       }
 
+      // Update remote task run status when agent turn completes with no pending work
+      if (session.messageQueue.length === 0) {
+        const auth = useAuthStore.getState();
+        if (auth.client) {
+          auth.client
+            .updateTaskRun(session.taskId, taskRunId, { status: "completed" })
+            .catch((err) =>
+              log.warn("Failed to update task run status to completed", {
+                err,
+              }),
+            );
+        }
+      }
+
       // Process queued messages after turn completes - send all as one prompt
       if (session.messageQueue.length > 0 && session.status === "connected") {
         setTimeout(() => {
@@ -896,6 +926,16 @@ export class SessionService {
       promptStartedAt: Date.now(),
     });
 
+    // Ensure task run status reflects active work
+    const auth = useAuthStore.getState();
+    if (auth.client) {
+      auth.client
+        .updateTaskRun(session.taskId, session.taskRunId, {
+          status: "in_progress",
+        })
+        .catch((err) => log.warn("Failed to update task run status", { err }));
+    }
+
     try {
       const result = await trpcVanilla.agent.prompt.mutate({
         sessionId: session.taskRunId,