feat: Enable/disable keyboard event handling with a global hotkey (Ct…

…rl+Opt+Cmd+K) (#12)
inseven · Dec 3, 2022 · 97b9e22 · 97b9e22
1 parent 688f911
commit 97b9e22
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 20 deletions.
diff --git a/macos/TinyBoard/Model/ApplicationModel.swift b/macos/TinyBoard/Model/ApplicationModel.swift
@@ -18,6 +18,7 @@
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 // SOFTWARE.
 
+import Carbon
 import Combine
 import SwiftUI
 
@@ -29,8 +30,7 @@ class ApplicationModel: NSObject, ObservableObject {
     @AppStorage("TrustedDevices") var trustedDevices: Set<UUID> = []
 
     let deviceManager = DeviceManager()
-
-    private let eventTap: EventTap
+    private let eventTap = EventTap()
     private var cancellables: Set<AnyCancellable> = []
 
     private lazy var aboutWindow: NSWindow = {
@@ -54,19 +54,9 @@ class ApplicationModel: NSObject, ObservableObject {
     }()
 
     override init() {
-        eventTap = EventTap(deviceManager: deviceManager)
         super.init()
-        $isEnabled
-            .receive(on: DispatchQueue.main)
-            .sink { isEnabled in
-                switch isEnabled {
-                case true:
-                    self.eventTap.enableTap()
-                case false:
-                    self.eventTap.disableTap()
-                }
-            }
-            .store(in: &cancellables)
+        eventTap.delegate = self
+        eventTap.start()
         deviceManager.delegate = self
     }
 
@@ -99,3 +89,25 @@ extension ApplicationModel: DeviceManagerDelegate {
     }
 
 }
+
+extension ApplicationModel: EventTapDelegate {
+
+    func eventTap(_ eventTap: EventTap, handleEvent event: CGEvent) -> Bool {
+        if let nsEvent = NSEvent(cgEvent: event) {
+            let deviceIndependentModifiers = nsEvent.modifierFlags.intersection(.deviceIndependentFlagsMask)
+            if deviceIndependentModifiers == [.control, .option, .command] && nsEvent.keyCode == kVK_ANSI_K {
+                if nsEvent.type == .keyDown {
+                    isEnabled = !isEnabled
+                }
+                return true
+            }
+        }
+
+        guard isEnabled else {
+            return false
+        }
+        deviceManager.sendEvent(event)
+        return true
+    }
+
+}
diff --git a/macos/TinyBoard/Model/EventTap.swift b/macos/TinyBoard/Model/EventTap.swift
@@ -33,13 +33,19 @@ private func eventTapCallback(proxy: CGEventTapProxy,
     return eventTap.handleEvent(proxy: proxy, type: type, event: event)
 }
 
+protocol EventTapDelegate: AnyObject {
+
+    func eventTap(_ eventTap: EventTap, handleEvent event: CGEvent) -> Bool
+
+}
+
 class EventTap {
 
-    let deviceManager: DeviceManager
     var eventTap: CFMachPort? = nil
 
-    init(deviceManager: DeviceManager) {
-        self.deviceManager = deviceManager
+    weak var delegate: EventTapDelegate?
+
+    init() {
     }
 
     func createEventTapIfNecessry() {
@@ -61,7 +67,7 @@ class EventTap {
         self.eventTap = eventTap
     }
 
-    func enableTap() {
+    func start() {
         createEventTapIfNecessry()
         guard let eventTap = eventTap else {
             print("No event tap to disable")
@@ -70,7 +76,7 @@ class EventTap {
         CGEvent.tapEnable(tap: eventTap, enable: true)
     }
 
-    func disableTap() {
+    func stop() {
         print("disableTap")
         guard let eventTap = eventTap else {
             return
@@ -79,7 +85,9 @@ class EventTap {
     }
 
     func handleEvent(proxy: CGEventTapProxy, type: CGEventType, event: CGEvent) -> Unmanaged<CGEvent>? {
-        deviceManager.sendEvent(event)
+        guard delegate?.eventTap(self, handleEvent: event) ?? false else {
+            return Unmanaged.passRetained(event)
+        }
         return nil
     }