Don't allow AnyScreen to wrap itself (#264)

square · Apr 11, 2019 · 3beb41b · 3beb41b
1 parent 45c8b4a
commit 3beb41b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/swift/WorkflowUI/Sources/Screen/AnyScreen/AnyScreen.swift b/swift/WorkflowUI/Sources/Screen/AnyScreen/AnyScreen.swift
@@ -5,6 +5,10 @@ public struct AnyScreen: Screen {
     private let viewControllerBuilder: (ViewRegistry) -> AnyScreenViewController.WrappedViewController
 
     public init<T: Screen>(_ screen: T) {
+        if let anyScreen = screen as? AnyScreen {
+            self = anyScreen
+            return
+        }
         self.wrappedScreen = screen
         self.viewControllerBuilder = { viewRegistry in
             return viewRegistry.provideView(for: screen)