Observe if display is off

mx-space · Sep 4, 2023 · 12b0ade · 12b0ade
1 parent 015147f
commit 12b0ade
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/ProcessReporter/Utils/Reporter/Reporter.swift b/ProcessReporter/Utils/Reporter/Reporter.swift
@@ -42,6 +42,22 @@ class Reporter {
     private var disposerList = [Disposable]()
 
     private func sleepHandler() {
+        let distributedNotificationCenter = DistributedNotificationCenter.default()
+
+        distributedNotificationCenter.addObserver(
+            self,
+            selector: #selector(systemWillSleep),
+            name: NSNotification.Name("com.apple.screenIsLocked"),
+            object: nil
+        )
+
+        distributedNotificationCenter.addObserver(
+            self,
+            selector: #selector(systemDidWake),
+            name: NSNotification.Name("com.apple.screenIsUnlocked"),
+            object: nil
+        )
+
         NotificationCenter.default.addObserver(self, selector: #selector(systemWillSleep), name: NSWorkspace.willSleepNotification, object: nil)
         NotificationCenter.default.addObserver(self, selector: #selector(systemDidWake), name: NSWorkspace.didWakeNotification, object: nil)
     }