Selector overrides (#239)

Author: davdroman

Sources/IntrospectionSelector.swift

@@ -1,57 +1,77 @@
 @_spi(Internals)
 public struct IntrospectionSelector<Target: PlatformEntity> {
-    private let selector: (IntrospectionPlatformViewController, IntrospectionScope, IntrospectionAnchorID) -> Target?
-
-    static var `default`: Self { .from(Target.self, selector: { $0 }) }
+    @_spi(Internals)
+    public static var `default`: Self { .from(Target.self, selector: { $0 }) }
 
     @_spi(Internals)
     public static func from<Entry: PlatformEntity>(_ entryType: Entry.Type, selector: @escaping (Entry) -> Target?) -> Self {
-        .init { controller, scope, anchorID in
-            guard let entity = { () -> (any PlatformEntity)? in
-                if Entry.Base.self == PlatformView.self {
-                    #if canImport(UIKit)
-                    if let introspectionView = controller.viewIfLoaded {
-                        return introspectionView
-                    }
-                    #elseif canImport(AppKit)
-                    if controller.isViewLoaded {
-                        return controller.view
-                    }
-                    #endif
-                } else if Entry.Base.self == PlatformViewController.self {
-                    return controller
-                }
-                return nil
-            }() else {
-                return nil
-            }
-            if
-                scope.contains(.receiver),
-                let entry = entity.receiver(ofType: Entry.self, anchorID: anchorID),
-                let target = selector(entry)
-            {
-                return target
+        .init(
+            receiverSelector: { controller, anchorID in
+                controller.as(Entry.self)?.receiver(ofType: Entry.self, anchorID: anchorID).flatMap(selector)
+            },
+            ancestorSelector: { controller in
+                controller.as(Entry.self)?.ancestor(ofType: Entry.self).flatMap(selector)
             }
-            if
-                scope.contains(.ancestor),
-                let entry = entity.ancestor(ofType: Entry.self),
-                let target = selector(entry)
-            {
-                return target
-            }
-            return nil
-        }
+        )
+    }
+
+    private var receiverSelector: (IntrospectionPlatformViewController, IntrospectionAnchorID) -> Target?
+    private var ancestorSelector: (IntrospectionPlatformViewController) -> Target?
+
+    private init(
+        receiverSelector: @escaping (IntrospectionPlatformViewController, IntrospectionAnchorID) -> Target?,
+        ancestorSelector: @escaping (IntrospectionPlatformViewController) -> Target?
+    ) {
+        self.receiverSelector = receiverSelector
+        self.ancestorSelector = ancestorSelector
+    }
+
+    @_spi(Internals)
+    public func withReceiverSelector(_ selector: @escaping (PlatformViewController, IntrospectionAnchorID) -> Target?) -> Self {
+        var copy = self
+        copy.receiverSelector = selector
+        return copy
     }
 
-    init(_ selector: @escaping (IntrospectionPlatformViewController, IntrospectionScope, IntrospectionAnchorID) -> Target?) {
-        self.selector = selector
+    @_spi(Internals)
+    public func withAncestorSelector(_ selector: @escaping (PlatformViewController) -> Target?) -> Self {
+        var copy = self
+        copy.ancestorSelector = selector
+        return copy
     }
 
     func callAsFunction(
         _ controller: IntrospectionPlatformViewController,
         _ scope: IntrospectionScope,
         _ anchorID: IntrospectionAnchorID
     ) -> Target? {
-        selector(controller, scope, anchorID)
+        if
+            scope.contains(.receiver),
+            let target = receiverSelector(controller, anchorID)
+        {
+            return target
+        }
+        if
+            scope.contains(.ancestor),
+            let target = ancestorSelector(controller)
+        {
+            return target
+        }
+        return nil
+    }
+}
+
+extension PlatformViewController {
+    func `as`<Entity: PlatformEntity>(_ entityType: Entity.Type) -> (any PlatformEntity)? {
+        if Entity.Base.self == PlatformView.self {
+            #if canImport(UIKit)
+            return viewIfLoaded
+            #elseif canImport(AppKit)
+            return isViewLoaded ? view : nil
+            #endif
+        } else if Entity.Base.self == PlatformViewController.self {
+            return self
+        }
+        return nil
     }
 }

Sources/IntrospectionView.swift

@@ -1,6 +1,7 @@
 import SwiftUI
 
-typealias IntrospectionAnchorID = UUID
+@_spi(Internals)
+public typealias IntrospectionAnchorID = UUID
 
 /// ⚓️
 struct IntrospectionAnchorView: PlatformViewControllerRepresentable {