Merge pull request #387 from Planetable/fixes-vapor-api-sleep

Monitor system sleep/awake event.
Planetable · Oct 15, 2024 · a7f32ab · a7f32ab
2 parents d9b133c + f4cf342
commit a7f32ab
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/Planet/API/PlanetAPIController.swift b/Planet/API/PlanetAPIController.swift
@@ -111,6 +111,12 @@ class PlanetAPIController: NSObject, ObservableObject {
             }
         }
     }
+
+    func pauseServerForSleep() {
+        globalApp?.shutdown()
+        globalApp = nil
+        stopBonjourService()
+    }
 
     func startBonjourService() {
         if bonjourService == nil {

diff --git a/Planet/PlanetAppDelegate.swift b/Planet/PlanetAppDelegate.swift
@@ -134,6 +134,25 @@ class PlanetAppDelegate: NSObject, NSApplicationDelegate {
                 await PlanetStore.shared.aggregate()
             }
         }
+
+        // Notify API server if system is going to sleep / awake
+        NotificationCenter.default.addObserver(forName: NSWorkspace.willSleepNotification, object: nil, queue: nil) { _ in
+            if UserDefaults.standard.bool(forKey: .settingsAPIEnabled) {
+                Task { @MainActor in
+                    PlanetAPIController.shared.pauseServerForSleep()
+                }
+            }
+        }
+        NotificationCenter.default.addObserver(forName: NSWorkspace.didWakeNotification, object: nil, queue: nil) { _ in
+            if UserDefaults.standard.bool(forKey: .settingsAPIEnabled) {
+                Task.detached(priority: .utility) {
+                    try? await Task.sleep(nanoseconds: 1_000_000_000)
+                    Task { @MainActor in
+                        PlanetAPIController.shared.startServer()
+                    }
+                }
+            }
+        }
     }
 
     func applicationShouldTerminateAfterLastWindowClosed(_ sender: NSApplication) -> Bool {

diff --git a/Planet/versioning.xcconfig b/Planet/versioning.xcconfig
@@ -1 +1 @@
-CURRENT_PROJECT_VERSION = 2276
+CURRENT_PROJECT_VERSION = 2277