display(macOS): update clock with QEMU GA when waking from sleep

Resolves #5252
utmapp · May 29, 2023 · 6f07ed0 · 6f07ed0
1 parent 3a9c8b4
commit 6f07ed0
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/Platform/macOS/Display/VMDisplayWindowController.swift b/Platform/macOS/Display/VMDisplayWindowController.swift
@@ -64,6 +64,11 @@ class VMDisplayWindowController: NSWindowController {
         self.init(window: nil)
         self.vm = vm
         self.onClose = onClose
+        NSWorkspace.shared.notificationCenter.addObserver(self, selector: #selector(didWake), name: NSWorkspace.didWakeNotification, object: nil)
+    }
+
+    deinit {
+        NSWorkspace.shared.notificationCenter.removeObserver(self, name: NSWorkspace.didWakeNotification, object: nil)
     }
 
     @IBAction func stopButtonPressed(_ sender: Any) {
@@ -374,3 +379,12 @@ extension VMDisplayWindowController: UTMVirtualMachineDelegate {
         }
     }
 }
+
+// MARK: - Computer wakeup
+extension VMDisplayWindowController {
+    @objc private func didWake(_ notification: NSNotification) {
+        if let qemuVM = vm as? UTMQemuVirtualMachine, let ga = qemuVM.guestAgent {
+            ga.guestSetTime(NSDate.now.timeIntervalSince1970)
+        }
+    }
+}
diff --git a/Platform/macOS/VMHeadlessSessionState.swift b/Platform/macOS/VMHeadlessSessionState.swift
@@ -36,6 +36,11 @@ import IOKit.pwr_mgt
         self.onStop = onStop
         super.init()
         vm.delegate = self
+        NSWorkspace.shared.notificationCenter.addObserver(self, selector: #selector(didWake), name: NSWorkspace.didWakeNotification, object: nil)
+    }
+
+    deinit {
+        NSWorkspace.shared.notificationCenter.removeObserver(self, name: NSWorkspace.didWakeNotification, object: nil)
     }
 }
 
@@ -96,3 +101,12 @@ extension Notification.Name {
     static let vmSessionEnded = Self("VMSessionEnded")
     static let vmSessionError = Self("VMSessionError")
 }
+
+// MARK: - Computer wakeup
+extension VMHeadlessSessionState {
+    @objc private func didWake(_ notification: NSNotification) {
+        if let qemuVM = vm as? UTMQemuVirtualMachine, let ga = qemuVM.guestAgent {
+            ga.guestSetTime(NSDate.now.timeIntervalSince1970)
+        }
+    }
+}