🔧 Fix SPImageView bug

README.md

@@ -76,12 +76,12 @@ ScreenProtectKit
 Moreover, it can be easily implemented by using SPImageView. This is particularly useful when images are set asynchronously.
 ```swift
 final class ViewController: UIViewController {
-    @IBOutlet private weak var mainImageView: UIImageView!
+    @IBOutlet private weak var mainImageView: SPImageView!
 
     override func viewDidLoad() {
         super.viewDidLoad()
         DispatchQueue.main.asyncAfter(deadline: .now() + 5) { [weak self] in
-            self?.mainImageView.image = UIImage(name: "sample")
+            self?.mainImageView.image = UIImage(named: "sample")
         }
     }
 }

Source/SPImageView.swift

@@ -15,7 +15,6 @@ public class SPImageView: UIImageView {
     override public var image: UIImage? {
         set {
             super.image = newValue
-            setup()
             if newValue != nil {
                 load.onNext(())
             }
@@ -25,11 +24,18 @@ public class SPImageView: UIImageView {
         }
     }
 
+    override public func layoutSubviews() {
+        super.layoutSubviews()
+        setup()
+    }
+
     private lazy var setup: (() -> Void) = {
-        _ = load
-            .delay(.milliseconds(100), scheduler: MainScheduler.instance)
+        _ = Observable.of(
+            self.rx.isScreenRecord.map { _ in () },
+            load
+            )
+            .merge()
             .withLatestFrom(self.rx.isScreenRecord)
-            .map { !$0 }
             .takeUntil(self.rx.deallocated)
             .bind(to: self.layer.rx.isMosaic)
         return {}