make produceDirect work in unwind scenarios (#21)

trafi · Nov 6, 2023 · f5de659 · f5de659
1 parent c7f5c56
commit f5de659
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/Sources/StoryFlow/Flow/ImplicitFlow/ImplicitFlow.swift b/Sources/StoryFlow/Flow/ImplicitFlow/ImplicitFlow.swift
@@ -66,7 +66,18 @@ extension Flow {
                 return
             }
 
-            // MARK: Direct
+            // MARK: Direct Unwind
+
+            if let to = from.unwindVc(for: Swift.type(of: value)) {
+                to.handleUpdate(value, type)
+                let transition = TransitionInfo(from: from, producedType: output.type, receivedType: type, to: to, isUnwind: true)
+                if CustomTransition.attempt(transition) == false {
+                    Transition.unwind().go(from, to)
+                }
+                return
+            }
+
+            // MARK: Direct Input
 
             if let to = Flow<Any>.destination(for: (value, Swift.type(of: value))) {
                 let transition = TransitionInfo(from: from, producedType: Swift.type(of: value), receivedType: type, to: to, isUnwind: false)