From ce0e60fddd8d18a76dab76f8935ebeca59411191 Mon Sep 17 00:00:00 2001
From: Jordan Baird <jordanbaird.dev@gmail.com>
Date: Sun, 25 Aug 2024 08:11:02 -0600
Subject: [PATCH] Attempt to fix continuation resume crash

---
 Ice/MenuBar/MenuBarItemManager.swift       | 37 ++++++++++++----------
 Ice/Utilities/EventMonitors/EventTap.swift | 13 +++-----
 2 files changed, 26 insertions(+), 24 deletions(-)

diff --git a/Ice/MenuBar/MenuBarItemManager.swift b/Ice/MenuBar/MenuBarItemManager.swift
index 1cfad7b8..5b429acc 100644
--- a/Ice/MenuBar/MenuBarItemManager.swift
+++ b/Ice/MenuBar/MenuBarItemManager.swift
@@ -475,13 +475,6 @@ extension MenuBarItemManager {
         }
     }
 
-    /// Delays an event tap callback from returning.
-    private func delayEventTapCallback() {
-        // Small delay to prevent timeouts when running alongside certain event tapping apps.
-        // TODO: Try to find a better solution for this.
-        Thread.sleep(forTimeInterval: 0.015)
-    }
-
     /// Posts an event to the given event tap location and waits until it is
     /// received before returning.
     ///
@@ -491,6 +484,7 @@ extension MenuBarItemManager {
     ///   - item: The menu bar item that the event affects.
     private func postEventAndWaitToReceive(_ event: CGEvent, to location: EventTap.Location, item: MenuBarItem) async throws {
         return try await withCheckedThrowingContinuation { continuation in
+            var resumed = false
             let eventTap = EventTap(
                 options: .defaultTap,
                 location: location,
@@ -523,16 +517,21 @@ extension MenuBarItemManager {
                 Logger.itemManager.debug("Received \(type.logString) at \(location.logString) (item: \(item.logString))")
 
                 proxy.disable()
-                continuation.resume()
-                delayEventTapCallback()
+                if !resumed {
+                    resumed = true
+                    continuation.resume()
+                }
 
                 return rEvent
             }
 
-            eventTap.enable(timeout: .milliseconds(50)) {
+            eventTap.enable(timeout: 0.05) {
                 Logger.itemManager.error("Event tap \"\(eventTap.label)\" timed out (item: \(item.logString))")
                 eventTap.disable()
-                continuation.resume()
+                if !resumed {
+                    resumed = true
+                    continuation.resume()
+                }
             }
 
             postEvent(event, to: location)
@@ -554,6 +553,7 @@ extension MenuBarItemManager {
         item: MenuBarItem
     ) async throws {
         return try await withCheckedThrowingContinuation { continuation in
+            var resumed = false
             let eventTap = EventTap(
                 options: .defaultTap,
                 location: initialLocation,
@@ -588,16 +588,21 @@ extension MenuBarItemManager {
                 postEvent(event, to: forwardedLocation)
 
                 proxy.disable()
-                continuation.resume()
-                delayEventTapCallback()
+                if !resumed {
+                    resumed = true
+                    continuation.resume()
+                }
 
                 return rEvent
             }
 
-            eventTap.enable(timeout: .milliseconds(50)) {
+            eventTap.enable(timeout: 0.05) {
                 Logger.itemManager.error("Event tap \"\(eventTap.label)\" timed out (item: \(item.logString))")
                 eventTap.disable()
-                continuation.resume()
+                if !resumed {
+                    resumed = true
+                    continuation.resume()
+                }
             }
 
             postEvent(event, to: initialLocation)
@@ -1125,7 +1130,7 @@ extension MenuBarItemManager {
                 continue
             }
             do {
-                try await move(item: item, to: context.returnDestination)
+                try await slowMove(item: item, to: context.returnDestination)
             } catch {
                 Logger.itemManager.error("Failed to rehide \(item.logString) (error: \(error))")
                 failedContexts.append(context)
diff --git a/Ice/Utilities/EventMonitors/EventTap.swift b/Ice/Utilities/EventMonitors/EventTap.swift
index 54594e6d..27626129 100644
--- a/Ice/Utilities/EventMonitors/EventTap.swift
+++ b/Ice/Utilities/EventMonitors/EventTap.swift
@@ -75,7 +75,7 @@ class EventTap {
         }
 
         /// Enables the event tap with the given timeout.
-        func enable(timeout: Duration, onTimeout: @escaping () -> Void) {
+        func enable(timeout: TimeInterval, onTimeout: @escaping () -> Void) {
             tap.enable(timeout: timeout, onTimeout: onTimeout)
         }
 
@@ -231,17 +231,14 @@ class EventTap {
     }
 
     /// Enables the event tap with the given timeout.
-    func enable(timeout: Duration, onTimeout: @escaping () -> Void) {
+    func enable(timeout: TimeInterval, onTimeout: @escaping () -> Void) {
         enable()
-        Task { [weak self] in
-            try await Task.sleep(for: timeout)
+        queue.asyncAfter(deadline: .now() + timeout) { [weak self] in
             guard let self else {
                 return
             }
-            queue.async {
-                if self.isEnabled {
-                    onTimeout()
-                }
+            if isEnabled {
+                onTimeout()
             }
         }
     }