From e1a7559b7b95a4121ad9682ea8920ae0871f86cb Mon Sep 17 00:00:00 2001
From: Neil <neil@stably.ai>
Date: Sat, 19 Sep 2026 01:19:33 -0700
Subject: [PATCH] perf(sidebar): cache sleeping worktree status

---
 .../sidebar/use-worktree-sleep-state.ts       | 58 ++++++++++++++++---
 1 file changed, 49 insertions(+), 9 deletions(-)

diff --git a/src/renderer/src/components/sidebar/use-worktree-sleep-state.ts b/src/renderer/src/components/sidebar/use-worktree-sleep-state.ts
index 0db546d8652..1fe5d72063d 100644
--- a/src/renderer/src/components/sidebar/use-worktree-sleep-state.ts
+++ b/src/renderer/src/components/sidebar/use-worktree-sleep-state.ts
@@ -22,7 +22,16 @@ type LiveAgentGeneration = {
   worktreeIds: ReadonlySet<string>
 }
 
+type SleepingWorktreeGeneration = {
+  tabsByWorktree: SleepStateInput['tabsByWorktree']
+  ptyIdsByTabId: SleepStateInput['ptyIdsByTabId']
+  browserTabsByWorktree: SleepStateInput['browserTabsByWorktree']
+  worktreeIdsWithLiveAgent: ReadonlySet<string>
+  sleepingByWorktree: Map<string, boolean>
+}
+
 let liveAgentGeneration: LiveAgentGeneration | null = null
+let sleepingWorktreeGeneration: SleepingWorktreeGeneration | null = null
 
 // Why cached across cards: zustand re-runs every mounted card's selector on every
 // store write, and the live-agent set is a whole-store scan. Keyed on the same
@@ -52,6 +61,44 @@ function selectWorktreeIdsWithLiveAgent(state: SleepStateInput): ReadonlySet<str
   return worktreeIds
 }
 
+/** Cache each worktree's answer until one of the inputs to the predicate changes. */
+export function selectIsSleepingWorktree(state: SleepStateInput, worktreeId: string): boolean {
+  const worktreeIdsWithLiveAgent = selectWorktreeIdsWithLiveAgent(state)
+  const cached = sleepingWorktreeGeneration
+  if (
+    !cached ||
+    cached.tabsByWorktree !== state.tabsByWorktree ||
+    cached.ptyIdsByTabId !== state.ptyIdsByTabId ||
+    cached.browserTabsByWorktree !== state.browserTabsByWorktree ||
+    cached.worktreeIdsWithLiveAgent !== worktreeIdsWithLiveAgent
+  ) {
+    sleepingWorktreeGeneration = {
+      tabsByWorktree: state.tabsByWorktree,
+      ptyIdsByTabId: state.ptyIdsByTabId,
+      browserTabsByWorktree: state.browserTabsByWorktree,
+      worktreeIdsWithLiveAgent,
+      sleepingByWorktree: new Map()
+    }
+  }
+
+  const current = sleepingWorktreeGeneration
+  if (!current) {
+    throw new Error('sleeping worktree cache was not initialized')
+  }
+  if (current.sleepingByWorktree.has(worktreeId)) {
+    return current.sleepingByWorktree.get(worktreeId) === true
+  }
+  const sleeping = isInactiveWorkspace(
+    worktreeId,
+    state.tabsByWorktree,
+    state.ptyIdsByTabId,
+    state.browserTabsByWorktree,
+    worktreeIdsWithLiveAgent
+  )
+  current.sleepingByWorktree.set(worktreeId, sleeping)
+  return sleeping
+}
+
 /**
  * Whether a workspace is asleep: no live terminal, no browser tab, and no live
  * agent holding it awake through a PTY gap.
@@ -65,17 +112,10 @@ function selectWorktreeIdsWithLiveAgent(state: SleepStateInput): ReadonlySet<str
  * are asleep.
  */
 export function useIsSleepingWorktree(worktreeId: string): boolean {
-  return useAppStore((state) =>
-    isInactiveWorkspace(
-      worktreeId,
-      state.tabsByWorktree,
-      state.ptyIdsByTabId,
-      state.browserTabsByWorktree,
-      selectWorktreeIdsWithLiveAgent(state)
-    )
-  )
+  return useAppStore((state) => selectIsSleepingWorktree(state, worktreeId))
 }
 
 export function resetWorktreeSleepStateCacheForTests(): void {
   liveAgentGeneration = null
+  sleepingWorktreeGeneration = null
 }