Add Keep Mac Awake control on Mac and iOS

.github/swift-file-length-budget.tsv

@@ -2,7 +2,7 @@
 # Format: max_lines<TAB>relative path
 # Reduce counts as files shrink. CI fails if tracked files exceed this budget.
 35600	CLI/cmux.swift
-18229	Sources/AppDelegate.swift
+18231	Sources/AppDelegate.swift
 16443	Sources/ContentView.swift
 15163	Sources/TerminalController.swift
 13232	Sources/Workspace.swift
@@ -39,7 +39,7 @@
 2563	Sources/Panels/CmuxWebView.swift
 2546	cmuxTests/WorkspaceManualUnreadTests.swift
 2524	cmuxTests/CommandPaletteSearchEngineTests.swift
-2404	Sources/Mobile/MobileHostService.swift
+2443	Sources/Mobile/MobileHostService.swift
 2328	cmuxTests/CJKIMEInputTests.swift
 2242	Sources/TerminalNotificationStore.swift
 2229	Sources/TerminalWindowPortal.swift
@@ -71,7 +71,7 @@
 1363	Sources/CMUXInstalledExtensionSidebarHostView.swift
 1360	Sources/Feed/FeedButtonStyleDebugWindowController.swift
 1317	Sources/FileExplorerStore.swift
-1295	cmuxTests/MobileHostAuthorizationTests.swift
+1315	cmuxTests/MobileHostAuthorizationTests.swift
 1291	Packages/macOS/CmuxTerminalCore/Sources/CmuxTerminalCore/Config/GhosttyConfig.swift
 1290	cmuxTests/TextBoxSubmitActionTests.swift
 1285	cmuxUITests/SidebarHelpMenuUITests.swift

Packages/macOS/CmuxSettings/Sources/CmuxSettings/Keys/PowerCatalogSection.swift

@@ -0,0 +1,21 @@
+import Foundation
+
+/// Power management settings for keeping the Mac reachable during work.
+public struct PowerCatalogSection: SettingCatalogSection {
+    /// Keeps the Mac from idle system sleep while cmux observes running agent sessions.
+    public let preventSleepWhileAgentsRunning = DefaultsKey<Bool>(
+        id: "power.preventSleepWhileAgentsRunning.enabled",
+        defaultValue: false,
+        userDefaultsKey: "power.preventSleepWhileAgentsRunning.enabled"
+    )
+
+    /// Keeps the Mac from idle system sleep while an iPhone or iPad is connected.
+    public let preventSleepWhileMobileConnected = DefaultsKey<Bool>(
+        id: "power.preventSleepWhileMobileConnected.enabled",
+        defaultValue: true,
+        userDefaultsKey: "power.preventSleepWhileMobileConnected.enabled"
+    )
+
+    /// Creates the Power settings catalog section.
+    public init() {}
+}

Packages/macOS/CmuxSettings/Sources/CmuxSettings/Keys/SettingCatalog.swift

@@ -42,6 +42,8 @@ public struct SettingCatalog: SettingCatalogSection {
     public let fileEditor = FileEditorCatalogSection()
     /// Settings for Mobile pairing and sync.
     public let mobile = MobileCatalogSection()
+    /// Settings for keeping the Mac awake while work or mobile clients are active.
+    public let power = PowerCatalogSection()
     public let betaFeatures = BetaFeaturesCatalogSection()
     /// Settings for custom (user/agent-authored) sidebars (the `customSidebars.*` keys).
     public let customSidebars = CustomSidebarsCatalogSection()

Packages/macOS/CmuxSettingsUI/Sources/CmuxSettingsUI/Navigation/CuratedSettingEntry+Default.swift

@@ -163,6 +163,8 @@ extension Array where Element == CuratedSettingEntry {
             .init(section: .mobile, id: "iOSPairingHost", title: "iOS Pairing", synonyms: "ios iphone ipad mobile pairing local network permission sync"),
             .init(section: .mobile, id: "iOSPairingPort", title: String(localized: "settings.mobile.port", defaultValue: "Pairing Port"), synonyms: "mobile.iOSPairingHost.port ios iphone mobile pairing port tcp listener firewall conflict"),
             .init(section: .mobile, id: "iOSPairingDisplayName", title: String(localized: "settings.mobile.displayName", defaultValue: "Display Name"), synonyms: "mobile.iOSPairingHost.displayName ios iphone mobile pairing display name mac hostname device label"),
+            .init(section: .mobile, id: "preventSleepWhileAgentsRunning", title: String(localized: "settings.mobile.preventSleepWhileAgentsRunning", defaultValue: "Prevent sleep while agents are running"), synonyms: "power.preventSleepWhileAgentsRunning.enabled agents running keep awake caffeinate prevent sleep insomnia system sleep idle"),
+            .init(section: .mobile, id: "preventSleepWhileMobileConnected", title: String(localized: "settings.mobile.preventSleepWhileMobileConnected", defaultValue: "Prevent sleep while iPhone is connected"), synonyms: "power.preventSleepWhileMobileConnected.enabled iphone mobile connected keep awake caffeinate prevent sleep insomnia system sleep idle"),
 
             // Custom Sidebars
             .init(section: .customSidebars, id: "enabled", title: String(localized: "settings.customSidebars.enabled", defaultValue: "Show Custom Sidebars"), synonyms: "custom sidebars enable show vibe swift json interpreted picker beta"),

Packages/macOS/CmuxSettingsUI/Sources/CmuxSettingsUI/Navigation/SettingsSectionID.swift

@@ -89,7 +89,7 @@ public enum SettingsSectionID: String, CaseIterable, Identifiable, Sendable, Has
         case .terminal: return "scrollbar copy on select agent resume hibernation"
         case .textBox: return "textbox text box rich input prompt default new terminal workspace split tab focus show beta"
         case .sleepyMode: return "sleepy mode screensaver caffeinate keep awake lock touch id battery wifi clock mascot theme glow pixel"
-        case .mobile: return "ios iphone ipad mobile pairing local network sync"
+        case .mobile: return "ios iphone ipad mobile pairing local network sync keep awake caffeinate prevent sleep insomnia"
         case .sidebarAppearance: return "sidebar details branches material terminal background"
         case .customSidebars: return "custom sidebars vibe swift json interpreted renderer in-process remote worker isolated"
         case .betaFeatures: return "beta experimental unstable feed dock right sidebar"

Packages/macOS/CmuxSettingsUI/Sources/CmuxSettingsUI/Sections/MobileSection.swift

@@ -11,6 +11,8 @@ public struct MobileSection: View {
     @State private var iOSPairingHost: DefaultsValueModel<Bool>
     @State private var port: DefaultsValueModel<Int>
     @State private var displayName: DefaultsValueModel<String>
+    @State private var preventSleepWhileAgentsRunning: DefaultsValueModel<Bool>
+    @State private var preventSleepWhileMobileConnected: DefaultsValueModel<Bool>
     @State private var status: MobilePairingStatusModel
 
     /// The user's in-progress port edit, or `nil` when the field should track
@@ -45,6 +47,8 @@ public struct MobileSection: View {
         _iOSPairingHost = State(initialValue: DefaultsValueModel(store: defaultsStore, key: catalog.mobile.iOSPairingHost))
         _port = State(initialValue: DefaultsValueModel(store: defaultsStore, key: catalog.mobile.iOSPairingPort))
         _displayName = State(initialValue: DefaultsValueModel(store: defaultsStore, key: catalog.mobile.iOSPairingDisplayName))
+        _preventSleepWhileAgentsRunning = State(initialValue: DefaultsValueModel(store: defaultsStore, key: catalog.power.preventSleepWhileAgentsRunning))
+        _preventSleepWhileMobileConnected = State(initialValue: DefaultsValueModel(store: defaultsStore, key: catalog.power.preventSleepWhileMobileConnected))
         _status = State(initialValue: MobilePairingStatusModel(hostActions: hostActions))
         self.hostActions = hostActions
     }
@@ -78,6 +82,11 @@ public struct MobileSection: View {
                 boundPortStatusRow
                 SettingsCardDivider()
                 displayNameRow
+                SettingsCardDivider()
+                SettingsCardNote(String(localized: "settings.mobile.keepAwake", defaultValue: "Keep Awake"))
+                preventSleepWhileAgentsRunningRow
+                SettingsCardDivider()
+                preventSleepWhileMobileConnectedRow
                 if iOSPairingHost.current {
                     SettingsCardDivider()
                     diagnostics
@@ -96,6 +105,8 @@ public struct MobileSection: View {
             iOSPairingHost,
             port,
             displayName,
+            preventSleepWhileAgentsRunning,
+            preventSleepWhileMobileConnected,
             status,
         ]
         models.forEach { $0.startObserving() }
@@ -135,6 +146,36 @@ public struct MobileSection: View {
         }
     }
 
+    @ViewBuilder
+    private var preventSleepWhileAgentsRunningRow: some View {
+        SettingsCardRow(
+            configurationReview: .settingsOnly,
+            searchAnchorID: "setting:mobile:preventSleepWhileAgentsRunning",
+            String(localized: "settings.mobile.preventSleepWhileAgentsRunning", defaultValue: "Prevent sleep while agents are running"),
+            subtitle: String(localized: "settings.mobile.preventSleepWhileAgentsRunning.subtitle", defaultValue: "Keeps system sleep off while agents run; the display can still sleep.")
+        ) {
+            Toggle("", isOn: Binding(get: { preventSleepWhileAgentsRunning.current }, set: { preventSleepWhileAgentsRunning.set($0) }))
+                .labelsHidden()
+                .controlSize(.small)
+                .accessibilityIdentifier("SettingsMobilePreventSleepWhileAgentsRunningToggle")
+        }
+    }
+
+    @ViewBuilder
+    private var preventSleepWhileMobileConnectedRow: some View {
+        SettingsCardRow(
+            configurationReview: .settingsOnly,
+            searchAnchorID: "setting:mobile:preventSleepWhileMobileConnected",
+            String(localized: "settings.mobile.preventSleepWhileMobileConnected", defaultValue: "Prevent sleep while iPhone is connected"),
+            subtitle: String(localized: "settings.mobile.preventSleepWhileMobileConnected.subtitle", defaultValue: "Keeps system sleep off while a mobile client is connected; the display can still sleep.")
+        ) {
+            Toggle("", isOn: Binding(get: { preventSleepWhileMobileConnected.current }, set: { preventSleepWhileMobileConnected.set($0) }))
+                .labelsHidden()
+                .controlSize(.small)
+                .accessibilityIdentifier("SettingsMobilePreventSleepWhileMobileConnectedToggle")
+        }
+    }
+
     @ViewBuilder
     private var portRow: some View {
         SettingsCardRow(

Packages/macOS/CmuxSettingsUI/Tests/CmuxSettingsUITests/SettingsRowAnchorResolutionTests.swift

@@ -145,6 +145,8 @@ struct SettingsRowAnchorResolutionTests {
         "setting:mobile:iOSPairingHost",
         "setting:mobile:iOSPairingPort",
         "setting:mobile:iOSPairingDisplayName",
+        "setting:mobile:preventSleepWhileAgentsRunning",
+        "setting:mobile:preventSleepWhileMobileConnected",
         "setting:betaFeatures:feed",
         "setting:betaFeatures:dock",
         "setting:betaFeatures:customSidebars",

Resources/Localizable.xcstrings

@@ -159000,6 +159000,23 @@
         }
       }
     },
+    "settings.mobile.keepAwake": {
+      "extractionState": "manual",
+      "localizations": {
+        "en": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "Keep Awake"
+          }
+        },
+        "ja": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "スリープ防止"
+          }
+        }
+      }
+    },
     "settings.mobile.pairDevice": {
       "extractionState": "manual",
       "localizations": {
@@ -159051,6 +159068,74 @@
         }
       }
     },
+    "settings.mobile.preventSleepWhileAgentsRunning": {
+      "extractionState": "manual",
+      "localizations": {
+        "en": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "Prevent sleep while agents are running"
+          }
+        },
+        "ja": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "エージェント実行中はスリープを防止"
+          }
+        }
+      }
+    },
+    "settings.mobile.preventSleepWhileAgentsRunning.subtitle": {
+      "extractionState": "manual",
+      "localizations": {
+        "en": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "Keeps system sleep off while agents run; the display can still sleep."
+          }
+        },
+        "ja": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "エージェントの実行中はシステムスリープを防止します。ディスプレイはスリープできます。"
+          }
+        }
+      }
+    },
+    "settings.mobile.preventSleepWhileMobileConnected": {
+      "extractionState": "manual",
+      "localizations": {
+        "en": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "Prevent sleep while iPhone is connected"
+          }
+        },
+        "ja": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "iPhone 接続中はスリープを防止"
+          }
+        }
+      }
+    },
+    "settings.mobile.preventSleepWhileMobileConnected.subtitle": {
+      "extractionState": "manual",
+      "localizations": {
+        "en": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "Keeps system sleep off while a mobile client is connected; the display can still sleep."
+          }
+        },
+        "ja": {
+          "stringUnit": {
+            "state": "translated",
+            "value": "モバイルクライアントの接続中はシステムスリープを防止します。ディスプレイはスリープできます。"
+          }
+        }
+      }
+    },
     "settings.mobile.port": {
       "extractionState": "manual",
       "localizations": {

Sources/AppDelegate.swift

@@ -2008,6 +2008,7 @@ final class AppDelegate: NSObject, NSApplicationDelegate, UNUserNotificationCent
         CloudVMActionLauncher.shared.terminateAll()
         CmuxSSHURLProcessLauncher.shared.terminateAll()
         MobileHostService.shared.stop()
+        PreventSleepManager.shared.stop()
         TerminalController.shared.stop()
         GhosttyApp.terminalPasteboard.cleanupAllOwnedTemporaryImageFiles()
         VSCodeServeWebController.shared.stop()
@@ -2065,6 +2066,7 @@ final class AppDelegate: NSObject, NSApplicationDelegate, UNUserNotificationCent
         MobileTerminalRenderObserver.shared.start()
         agentChatTranscriptService.start()
         installMobileHostSettingsObserver()
+        PreventSleepManager.shared.start()
         scheduleGhosttyCrashBreadcrumbIfNeeded(notificationStore: notificationStore)
         startPaneMemoryGuardrailIfNeeded()
         disableSuddenTerminationIfNeeded()

Sources/Mobile/MobileHostService.swift

@@ -250,6 +250,12 @@ struct MobileHostServiceStatus {
     let usesEphemeralFallback: Bool
     let routes: [CmxAttachRoute]
     let activeConnectionCount: Int
+    /// Connections that have completed at least one authorized request.
+    /// `activeConnectionCount` counts raw accepted TCP sessions, which include
+    /// peers that never authenticate (`mobile.host.status` is auth-exempt), so
+    /// consumers gating behavior on "a phone is really connected" — like the
+    /// keep-awake power assertion — must use this count instead.
+    let authenticatedConnectionCount: Int
     let lastErrorDescription: String?
 
     var payload: [String: Any] {
@@ -260,6 +266,7 @@ struct MobileHostServiceStatus {
             "uses_ephemeral_fallback": usesEphemeralFallback,
             "routes": routes.map(\.mobileHostJSONObject),
             "active_connection_count": activeConnectionCount,
+            "authenticated_connection_count": authenticatedConnectionCount,
             "last_error": lastErrorDescription ?? NSNull()
         ]
     }
@@ -941,10 +948,17 @@ final class MobileHostService {
             usesEphemeralFallback: isRunning && listenerUsesEphemeralFallback,
             routes: routes,
             activeConnectionCount: MobileHostConnectionRegistry.shared.count,
+            authenticatedConnectionCount: authenticatedConnectionCount,
             lastErrorDescription: lastErrorDescription
         )
     }
 
+    /// Lean accessor for ``MobileHostServiceStatus/authenticatedConnectionCount``:
+    /// callers that only need the count (keep-awake reconciliation on every
+    /// settings/agent event) must not pay ``statusSnapshot()``'s route
+    /// resolution, which enumerates network interfaces on each call.
+    var authenticatedConnectionCount: Int { clientIDsByConnectionID.count }
+
     /// Reconcile the live listener with current settings (enable/disable and
     /// preferred-port changes). Safe to call on any settings change: it no-ops
     /// unless the enabled state or the configured port actually changed, so an
@@ -1024,6 +1038,14 @@ final class MobileHostService {
                     return await MobileHostService.shared.authorizationError(for: request)
                 },
                 onAuthorizedRequest: { request in
+                    // Auth-exempt verbs (mobile.host.status) reach this hook
+                    // too, because authorizeRequest returns nil for them.
+                    // Recording their client_id would let an unauthenticated
+                    // peer count as an authenticated connection and hold the
+                    // keep-awake assertion.
+                    guard Self.requiresAuthorization(method: request.method) else {
+                        return
+                    }
                     guard let clientID = Self.clientID(from: request.params) else {
                         return
                     }
@@ -1144,6 +1166,12 @@ final class MobileHostService {
                 await MobileHostService.shared.authorizationError(for: request)
             },
             onAuthorizedRequest: { request in
+                // Same guard as the network listener above: authorizationError
+                // exempts mobile.host.status internally, so this hook also
+                // fires for unauthenticated status requests.
+                guard Self.requiresAuthorization(method: request.method) else {
+                    return
+                }
                 if let clientID = Self.clientID(from: request.params) {
                     await MobileHostService.shared.recordClientID(clientID, for: id)
                 }
@@ -1211,14 +1239,25 @@ final class MobileHostService {
                 clientIDs: clientIDs,
                 reason: "mobile.connection.closed"
             )
+            // The registry's own remove() post can be delivered before this
+            // main-actor cleanup runs (onClose posts from the network callback
+            // thread), so re-post after the authenticated count actually
+            // dropped or keep-awake would hold until an unrelated event.
+            NotificationCenter.default.post(name: .mobileHostStatusDidChange, object: nil)
         }
         MobileHostRequestActivity.endConnection()
     }
 
     private func recordClientID(_ clientID: String, for connectionID: UUID) {
         var clientIDs = clientIDsByConnectionID[connectionID] ?? []
+        let isFirstAuthorizedRequestForConnection = clientIDs.isEmpty
         clientIDs.insert(clientID)
         clientIDsByConnectionID[connectionID] = clientIDs
+        if isFirstAuthorizedRequestForConnection {
+            // The connection just transitioned unauthenticated -> authenticated;
+            // authenticated-count consumers (keep-awake) resync on this signal.
+            NotificationCenter.default.post(name: .mobileHostStatusDidChange, object: nil)
+        }
     }
 
     private nonisolated static func clientID(from params: [String: Any]) -> String? {