Pass the onboarding use case to the flow so it can show the already retrieved error

Author: toupper

WooCommerce/Classes/ViewRelated/Dashboard/Settings/In-Person Payments/InPersonPaymentsMenuViewController.swift

@@ -75,41 +75,43 @@ private extension InPersonPaymentsMenuViewController {
             .debounce(for: .milliseconds(100), scheduler: DispatchQueue.main)
             .removeDuplicates()
             .sink(receiveValue: { [weak self] state in
-                guard let self = self else { return }
-
-                self.pluginState = nil
-
-                guard state != .loading else {
-                    self.activityIndicator?.startAnimating()
-                    return
-                }
-
-                switch state {
-                case let .completed(newPluginState):
-                    self.pluginState = newPluginState
-                    self.dismissCardPresentPaymentsOnboardingNoticeIfPresent()
-                case let .selectPlugin(pluginSelectionWasCleared):
-                    // If it was cleared it means that we triggered it manually (e.g by tapping in this view on the plugin selection row)
-                    // No need to show the onboarding notice
-                    if !pluginSelectionWasCleared {
-                        self.showCardPresentPaymentsOnboardingNotice()
-                    }
-                default:
-                    self.showCardPresentPaymentsOnboardingNotice()
-                }
-
-                self.activityIndicator?.stopAnimating()
-                self.configureSections()
-                self.tableView.reloadData()
+                self?.refreshAfterNewOnboardingState(state)
         }).store(in: &cancellables)
     }
 
+    func refreshAfterNewOnboardingState(_ state: CardPresentPaymentOnboardingState) {
+        self.pluginState = nil
+
+        guard state != .loading else {
+            self.activityIndicator?.startAnimating()
+            return
+        }
+
+        switch state {
+        case let .completed(newPluginState):
+            self.pluginState = newPluginState
+            self.dismissCardPresentPaymentsOnboardingNoticeIfPresent()
+        case let .selectPlugin(pluginSelectionWasCleared):
+            // If it was cleared it means that we triggered it manually (e.g by tapping in this view on the plugin selection row)
+            // No need to show the onboarding notice
+            if !pluginSelectionWasCleared {
+                self.showCardPresentPaymentsOnboardingNotice()
+            }
+        default:
+            self.showCardPresentPaymentsOnboardingNotice()
+        }
+
+        self.activityIndicator?.stopAnimating()
+        self.configureSections()
+        self.tableView.reloadData()
+    }
+
     func showCardPresentPaymentsOnboardingNotice() {
         let permanentNotice = PermanentNotice(message: Localization.inPersonPaymentsSetupNotFinishedNotice,
                                               callToActionTitle: Localization.inPersonPaymentsSetupNotFinishedNoticeButtonTitle,
                                               callToActionHandler: { [weak self] in
             ServiceLocator.analytics.track(.paymentsMenuOnboardingErrorTapped)
-            self?.showOnboardingIfRequired()
+            self?.showOnboarding()
         })
 
         permanentNoticePresenter.presentNotice(notice: permanentNotice, from: self)
@@ -119,18 +121,21 @@ private extension InPersonPaymentsMenuViewController {
         permanentNoticePresenter.dismiss()
     }
 
-    func showOnboardingIfRequired() {
+    func showOnboarding() {
         guard let navigationController = self.navigationController else {
             return
         }
 
-        // Recreating it ensures that the readiness state is up to date.
-        // Keeping a reference ensures that the callback closure is retained.
-        cardPresentPaymentsOnboardingPresenter = CardPresentPaymentsOnboardingPresenter()
-
-        cardPresentPaymentsOnboardingPresenter?.showOnboardingIfRequired(from: navigationController) { [weak self] in
-            self?.cardPresentPaymentsOnboardingUseCase.refresh()
+        let onboardingViewModel = InPersonPaymentsViewModel(useCase: cardPresentPaymentsOnboardingUseCase, showMenuOnCompletion: false)
+        onboardingViewModel.onOnboardingCompletion = { [weak self] plugin in
+            self?.refreshAfterNewOnboardingState(.completed(plugin: plugin))
+            if navigationController.visibleViewController is InPersonPaymentsViewController {
+                navigationController.popViewController(animated: true)
+            }
         }
+
+        let onboardingViewController = InPersonPaymentsViewController(viewModel: onboardingViewModel)
+        show(onboardingViewController, sender: self)
     }
 }
 

WooCommerce/Classes/ViewRelated/Dashboard/Settings/In-Person Payments/InPersonPaymentsViewModel.swift

@@ -8,6 +8,7 @@ final class InPersonPaymentsViewModel: ObservableObject {
     var learnMoreURL: URL? = nil
     let showMenuOnCompletion: Bool
     let gatewaySelectionAvailable: Bool
+    var onOnboardingCompletion: ((CardPresentPaymentsPluginState) -> ())?
     private let useCase: CardPresentPaymentsOnboardingUseCase
     let stores: StoresManager
 
@@ -29,6 +30,9 @@ final class InPersonPaymentsViewModel: ObservableObject {
             .debounce(for: .milliseconds(100), scheduler: DispatchQueue.main)
             .removeDuplicates()
             .handleEvents(receiveOutput: { [weak self] result in
+                if case let .completed(plugin) = result {
+                    self?.onOnboardingCompletion?(plugin)
+                }
                 self?.updateLearnMoreURL(state: result)
             })
             .handleEvents(receiveOutput: trackState(_:))