fix: unmask when removed from hierarchy

PostHog · Dec 5, 2024 · f62f070 · f62f070
1 parent 74dd9a9
commit f62f070
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/PostHog/PostHogMaskViewModifier.swift b/PostHog/PostHogMaskViewModifier.swift
@@ -41,16 +41,24 @@
     }
 
     private class PostHogMaskViewTaggerView: UIView {
-        override func didMoveToSuperview() {
-            super.didMoveToSuperview()
+        weak private var maskedView: UIView?
+        override func layoutSubviews() {
+            super.layoutSubviews()
             // ### Why grandparent view?
             //
             // Because of SwiftUI-to-UIKit view bridging:
             //     OriginalView (SwiftUI) <- we tag here
             //       L PostHogMaskViewTagger (ViewRepresentable)
             //           L PostHogMaskViewTaggerView (UIView) <- we are here
+            maskedView = superview?.superview
             superview?.superview?.postHogNoCapture = true
         }
+
+        override func removeFromSuperview() {
+            super.removeFromSuperview()
+            maskedView?.postHogNoCapture = false
+            maskedView = nil
+        }
     }
 
     extension UIView {