fix(desktop): snapshot running work after the window probes and count queued browser actions

- Read terminal, computer-use, browser and agent activity after the unsaved-edits probes finish, so work started meanwhile still prompts
- A browser action waiting in the queue counts as running
- A window whose probe errors counts as possibly unsaved, like one that times out

Author: waleedlatif1

apps/desktop/src/main/browser-agent/driver.test.ts

@@ -1150,11 +1150,15 @@ describe('executeTool', () => {
     }
   })
 
-  it('reports a browser action as running from admission until it settles', async () => {
+  it('reports a browser action as running from queueing until it settles', async () => {
     vi.useFakeTimers()
     try {
       await driver.executeTool('chat-test', 'browser_open_tab', {})
       expect(driver.isBrowserToolRunning()).toBe(false)
+      const queued = driver.captureBrowserToolQueueBoundary('chat-test')
+      expect(driver.isBrowserToolRunning()).toBe(true)
+      if (queued) driver.releaseBrowserToolQueueBoundary(queued)
+      expect(driver.isBrowserToolRunning()).toBe(false)
       const contents = session.requireTab().view.webContents
       vi.mocked(contents.getURL).mockReturnValue('https://example.com/')
       vi.mocked(contents.executeJavaScript).mockImplementation(() => new Promise<never>(() => {}))

apps/desktop/src/main/browser-agent/driver.ts

@@ -459,7 +459,7 @@ function releaseBrowserToolAdmission(state: DriverScopeState, admission: symbol)
 
 /** Whether any chat has a browser action queued or running in this process. */
 export function isBrowserToolRunning(): boolean {
-  return activeBrowserToolAdmissions.size > 0
+  return activeBrowserToolAdmissions.size > 0 || pendingBrowserToolQueueBoundaries.size > 0
 }
 
 function retireDriverScopeState(state: DriverScopeState): void {

apps/desktop/src/main/in-flight-work.test.ts

@@ -0,0 +1,26 @@
+import type { WebContents } from 'electron'
+import { describe, expect, it, vi } from 'vitest'
+import { countWindowsWithUnsavedChanges } from '@/main/in-flight-work'
+
+vi.mock('electron', () => import('@/test/electron-mock'))
+
+function windowAnswering(answer: () => Promise<unknown>, state = { destroyed: false }) {
+  return {
+    isDestroyed: () => state.destroyed,
+    isCrashed: () => false,
+    executeJavaScript: vi.fn(answer),
+  } as unknown as WebContents
+}
+
+describe('countWindowsWithUnsavedChanges', () => {
+  it('never reports a window it could not ask as clean', async () => {
+    const counts = await countWindowsWithUnsavedChanges([
+      windowAnswering(() => Promise.resolve(true)),
+      windowAnswering(() => Promise.resolve(false)),
+      windowAnswering(() => Promise.reject(new Error('Script failed to execute'))),
+      windowAnswering(() => Promise.resolve(true), { destroyed: true }),
+    ])
+
+    expect(counts).toEqual({ unsavedWindowCount: 1, unresponsiveWindowCount: 1 })
+  })
+})

apps/desktop/src/main/in-flight-work.ts

@@ -41,7 +41,7 @@ async function askAboutUnsavedChanges(contents: WebContents): Promise<UnloadAnsw
     logger.warn('Could not ask a window about unsaved changes', {
       message: getErrorMessage(error, 'unknown'),
     })
-    return 'clean'
+    return 'unresponsive'
   }
 }
 

apps/desktop/src/main/index.ts

@@ -1167,13 +1167,17 @@ function main(): void {
         rememberWindowsForRelaunch(pending)
       },
       describeInFlightWork: async () => {
+        // Native activity is read after the window probes so work started meanwhile still counts.
+        const windows = await countWindowsWithUnsavedChanges(
+          getWindows().map((win) => win.webContents)
+        )
         const computerUseActivity = computerUse.getActivity()
         return describeInFlightWork({
           terminalCommands: terminal.runningCommands(),
           computerUseApp: computerUseActivity ? (computerUseActivity.appName ?? '') : null,
           browserToolRunning: isBrowserToolRunning(),
           agentWorkRunning,
-          ...(await countWindowsWithUnsavedChanges(getWindows().map((win) => win.webContents))),
+          ...windows,
         })
       },
       beforeInstall: async () => {