feat: propagate flow outcome to presenting vc

wireapp · Dec 2, 2024 · ae1c608 · ae1c608
1 parent 98144e2
commit ae1c608
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 7 deletions.
diff --git a/WireUI/Package.swift b/WireUI/Package.swift
@@ -24,7 +24,6 @@ let package = Package(
     ],
     dependencies: [
         .package(url: "https://github.com/swiftlang/swift-docc-plugin", from: "1.1.0"),
-        .package(path: "../SourceryPlugin"),
         .package(name: "WireDomainPackage", path: "../WireDomain"),
         .package(name: "WireFoundation", path: "../WireFoundation"),
         .package(path: "../WirePlugins")

diff --git a/WireUI/Sources/WireIndividualToTeamMigrationUI/IndividualToTeamMigrationViewController.swift b/WireUI/Sources/WireIndividualToTeamMigrationUI/IndividualToTeamMigrationViewController.swift
@@ -23,6 +23,12 @@ import WireDomainAPI
 import WireFoundation
 
 public class IndividualToTeamMigrationViewController: UIViewController {
+    public enum Action: Sendable {
+        case cancel
+        case completionGoToApp
+        case completionGoToTeamManagement
+    }
+
     enum Step: Sendable {
         case teamPlanSelection(features: [TeamPlanFeature])
         case teamName
@@ -76,12 +82,18 @@ public class IndividualToTeamMigrationViewController: UIViewController {
         case toTeamManagement
     }
 
+    let actionCallback: @Sendable (Action) -> Void
     let childController: UINavigationController
     var currentStep: Step
     let features: [TeamPlanFeature]
     let useCase: any IndividualToTeamMigrationUseCase
 
-    public init(features: [TeamPlanFeature], useCase: any IndividualToTeamMigrationUseCase) {
+    public init(
+        features: [TeamPlanFeature],
+        useCase: any IndividualToTeamMigrationUseCase,
+        actionCallback: @escaping @Sendable (Action) -> Void
+    ) {
+        self.actionCallback = actionCallback
         self.currentStep = .teamPlanSelection(features: features)
         self.childController = UINavigationController()
         self.features = features
@@ -109,7 +121,7 @@ public class IndividualToTeamMigrationViewController: UIViewController {
         case .toCancellationAlert:
             let alert = cancellationSheetFactory(
                 onLeave: { [weak self] in
-                    self?.transition(to: .toApp)
+                    self?.actionCallback(.cancel)
                 }, onContinue: { [weak self] in
                     self?.transition(to: .dismissCancellationAlert)
                 })
@@ -149,9 +161,9 @@ public class IndividualToTeamMigrationViewController: UIViewController {
             )
             childController.pushViewController(vc, animated: true)
         case .toApp:
-            break
+            actionCallback(.completionGoToApp)
         case .toTeamManagement:
-            break
+            actionCallback(.completionGoToTeamManagement)
         }
     }
 }

diff --git a/WireUI/Sources/WireIndividualToTeamMigrationUI/Views/Previews/_FeaturePreview.swift b/WireUI/Sources/WireIndividualToTeamMigrationUI/Views/Previews/_FeaturePreview.swift
@@ -32,10 +32,11 @@ fileprivate struct FeaturePreviewContainer: UIViewControllerRepresentable {
     let features: [TeamPlanFeature]
 
     func makeUIViewController(context: Context) -> IndividualToTeamMigrationViewController {
-        // Return MyViewController instance
+
         return IndividualToTeamMigrationViewController(
             features: features,
-            useCase: MockUseCase()
+            useCase: MockUseCase(),
+            actionCallback: { _ in }
         )
     }