From dcd003fbd67edc78f0364ea1cecf1dcc7d078f23 Mon Sep 17 00:00:00 2001
From: harshitsinghbhandari <24b4506@iitb.ac.in>
Date: Mon, 20 Apr 2026 13:18:49 +0530
Subject: [PATCH] fix: restore dead orchestrators from session detail

---
 packages/web/src/components/SessionDetail.tsx | 24 +++++++--------
 .../__tests__/SessionDetail.desktop.test.tsx  | 29 +++++++++++++++++++
 2 files changed, 40 insertions(+), 13 deletions(-)

diff --git a/packages/web/src/components/SessionDetail.tsx b/packages/web/src/components/SessionDetail.tsx
index 1e3b57ed2..5d1d91774 100644
--- a/packages/web/src/components/SessionDetail.tsx
+++ b/packages/web/src/components/SessionDetail.tsx
@@ -594,25 +594,23 @@ export function SessionDetail({
             </div>
           ) : null}
 
-          {/* Kill / Restore — only for non-orchestrator worker sessions */}
-          {!isOrchestrator && (
-            isRestorable ? (
-              <button type="button" className="dashboard-app-btn" onClick={handleRestore}>
-                <svg className="h-3 w-3" fill="none" stroke="currentColor" strokeWidth="2" viewBox="0 0 24 24">
-                  <polyline points="1 4 1 10 7 10" />
-                  <path d="M3.51 15a9 9 0 1 0 2.13-9.36L1 10" />
-                </svg>
-                <span className="topbar-btn-label">Restore</span>
-              </button>
-            ) : !terminalEnded ? (
+          {/* Restore is available for any restorable session; Kill stays worker-only. */}
+          {isRestorable ? (
+            <button type="button" className="dashboard-app-btn" onClick={handleRestore}>
+              <svg className="h-3 w-3" fill="none" stroke="currentColor" strokeWidth="2" viewBox="0 0 24 24">
+                <polyline points="1 4 1 10 7 10" />
+                <path d="M3.51 15a9 9 0 1 0 2.13-9.36L1 10" />
+              </svg>
+              <span className="topbar-btn-label">Restore</span>
+            </button>
+          ) : !isOrchestrator && !terminalEnded ? (
               <button type="button" className="dashboard-app-btn dashboard-app-btn--danger" onClick={handleKill}>
                 <svg className="h-3.5 w-3.5" fill="none" stroke="currentColor" strokeWidth="2" viewBox="0 0 24 24">
                   <path d="M3 6h18M19 6v14a2 2 0 0 1-2 2H7a2 2 0 0 1-2-2V6m3 0V4a2 2 0 0 1 2-2h4a2 2 0 0 1 2 2v2" />
                 </svg>
                 <span className="topbar-btn-label">Kill</span>
               </button>
-            ) : null
-          )}
+          ) : null}
 
           {!isOrchestrator && orchestratorHref ? (
             <a
diff --git a/packages/web/src/components/__tests__/SessionDetail.desktop.test.tsx b/packages/web/src/components/__tests__/SessionDetail.desktop.test.tsx
index dd78c4004..f35995c4d 100644
--- a/packages/web/src/components/__tests__/SessionDetail.desktop.test.tsx
+++ b/packages/web/src/components/__tests__/SessionDetail.desktop.test.tsx
@@ -195,6 +195,35 @@ describe("SessionDetail desktop layout", () => {
     expect(screen.queryByTestId("direct-terminal")).not.toBeInTheDocument();
   });
 
+  it("shows restore for restorable orchestrator sessions", () => {
+    render(
+      <SessionDetail
+        session={makeSession({
+          id: "my-app-orchestrator",
+          projectId: "my-app",
+          status: "terminated",
+          activity: "exited",
+          summary: "Project orchestrator",
+          pr: null,
+        })}
+        isOrchestrator
+        orchestratorZones={{
+          merge: 1,
+          respond: 0,
+          review: 0,
+          pending: 0,
+          working: 2,
+          done: 3,
+        }}
+        projectOrchestratorId="my-app-orchestrator"
+        projects={[{ id: "my-app", name: "My App", path: "/tmp/my-app" }]}
+      />,
+    );
+
+    expect(within(screen.getByRole("banner")).getByRole("button", { name: "Restore" })).toBeInTheDocument();
+    expect(within(screen.getByRole("banner")).queryByRole("button", { name: "Kill" })).not.toBeInTheDocument();
+  });
+
   it("hides the desktop orchestrator button on orchestrator session pages", () => {
     render(
       <SessionDetail