CGWindowListScreenCapturer — michelf#23 Move unhideOnNextDisplay to a…

… single queue
importRyan · Aug 23, 2021 · 7bf5c79 · 7bf5c79
1 parent f086d0a
commit 7bf5c79
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Metal/ScreenCapture/CGWindowListScreenCapturer.swift b/Metal/ScreenCapture/CGWindowListScreenCapturer.swift
@@ -134,10 +134,8 @@ private extension CGWindowListScreenCapturer {
 
     func afterCapturingUnhideOnNextDisplayIfNeeded() {
         guard unhideOnNextDisplay else { return }
+        unhideOnNextDisplay = false
         DispatchQueue.main.async { [weak self] in
-            // recheck, because the state could have changed since the dispatch
-            guard self?.unhideOnNextDisplay == true else { return }
-            self?.unhideOnNextDisplay = false
             self?.view?.isHidden = false
         }
     }
@@ -254,7 +252,9 @@ private extension CGWindowListScreenCapturer {
             if let link = displayLink { CVDisplayLinkStart(link) }
             capturingDisabled = false
             captureImmediately()
-            unhideOnNextDisplay = true
+            queue?.async { [weak self]  in
+                self?.unhideOnNextDisplay = true
+            }
         }
     }
 }