DeepLink handler update

UpcomingMovies/Helpers/Handlers/DeepLinkHandler/DeepLinkHandler.swift

@@ -6,11 +6,53 @@
 //  Copyright © 2024 Alonso. All rights reserved.
 //
 
-import Foundation
+import UIKit
 
 final class DeepLinkHandler: DeepLinkHandlerProtocol {
 
-    func handleDeepLinkUrl(_ url: URL) {
+    private(set) var rootCoordinators: [RootCoordinator] = []
+
+    func register(rootCoordinators: [RootCoordinator]) {
+        self.rootCoordinators = rootCoordinators
+    }
+
+    func handleDeepLinkUrl(_ url: URL, in window: UIWindow?) {
+        guard let urlHost = url.host, let host = DeepLinkDestination(rawValue: urlHost) else {
+            return
+        }
+        switch host {
+        case .upcomingMovies:
+            changeTabBarToSelectedIndex(RootCoordinatorIdentifier.upcomingMovies, from: window)
+        case .searchMovies:
+            changeTabBarToSelectedIndex(RootCoordinatorIdentifier.searchMovies, from: window)
+        case .favorites:
+            changeTabBarToSelectedIndex(RootCoordinatorIdentifier.account, from: window)
+            let rootCoordinator = rootCoordinators[index(for: RootCoordinatorIdentifier.account)]
+            let unwrappedParentCoordinator = rootCoordinator.childCoordinators.last?.unwrappedParentCoordinator ?? rootCoordinator
+            let coordinator = FavoritesSavedMoviesCoordinator(navigationController: unwrappedParentCoordinator.navigationController)
+            coordinator.parentCoordinator = unwrappedParentCoordinator
+            unwrappedParentCoordinator.childCoordinators.append(coordinator)
+            coordinator.start()
+        }
+    }
+
+    // MARK: - Private
+
+    private func changeTabBarToSelectedIndex(_ rootIdentifier: String, from window: UIWindow?) {
+        let selectedIndex = index(for: rootIdentifier)
+        guard let tabBarController = window?.rootViewController as? MainTabBarController else {
+            return
+        }
+        tabBarController.setSelectedIndex(selectedIndex)
+    }
+
+    private func index(for rootIdentifier: String) -> Int {
+        let coordinatorIdentifiers = rootCoordinators.map { $0.rootIdentifier }
+        guard let indexToSelect = coordinatorIdentifiers.firstIndex(of: rootIdentifier) else {
+            fatalError()
+        }
+
+        return indexToSelect
     }
 
 }

UpcomingMovies/Helpers/Handlers/DeepLinkHandler/DeepLinkHandlerProtocol.swift

@@ -6,10 +6,11 @@
 //  Copyright © 2024 Alonso. All rights reserved.
 //
 
-import Foundation
+import UIKit
 
 protocol DeepLinkHandlerProtocol {
 
-    func handleDeepLinkUrl(_ url: URL)
+    func register(rootCoordinators: [RootCoordinator])
+    func handleDeepLinkUrl(_ url: URL, in window: UIWindow?)
 
 }

UpcomingMovies/Helpers/Handlers/NavigationHandler/NavigationHandler.swift

@@ -11,15 +11,16 @@
 final class NavigationHandler: NavigationHandlerProtocol {
 
     private let deepLinkHandler: DeepLinkHandlerProtocol
+    private let rootCoordinators: [RootCoordinator] = MainTabBarBuilder.buildViewCoordinators()
 
-    private var currentSelectedIndex: Int = 0
-    private var rootCoordinators: [RootCoordinator]
+    private var initialTransitionCompleted: Bool = false
+    private var onInitialTransition: ((UIWindow?) -> Void)?
 
     // MARK: - Initializers
 
     init(deepLinkHandler: DeepLinkHandlerProtocol) {
         self.deepLinkHandler = deepLinkHandler
-        self.rootCoordinators = MainTabBarBuilder.buildViewCoordinators()
+        self.deepLinkHandler.register(rootCoordinators: rootCoordinators)
     }
 
     // MARK: - NavigationHandlerProtocol
@@ -32,35 +33,28 @@
                                     UIView.AnimationOptions.transitionCrossDissolve],
                           animations: {},
                           completion: { _ in
-                            let mainTabBarController = MainTabBarController(coordinators: self.rootCoordinators)
-                            mainTabBarController.setSelectedIndex(self.currentSelectedIndex)
-                            window.rootViewController = mainTabBarController
-                          })
+            let mainTabBarController = MainTabBarController(coordinators: self.rootCoordinators)
+            window.rootViewController = mainTabBarController
+            self.onInitialTransition?(window) ?? mainTabBarController.setSelectedIndex(0)
+            self.initialTransitionCompleted = true
+        })
     }
 
     func handleUrlOpeningNavigation(for url: URL?, and window: UIWindow?) {
-        guard let url = url, let urlHost = url.host else { return }
+        guard let url else { return }
 
         if url.scheme == AppExtension.scheme {
-            guard let host = DeepLinkDestination(rawValue: urlHost) else { return }
-            switch host {
-            case .upcomingMovies:
-                changeTabBarToSelectedIndex(RootCoordinatorIdentifier.upcomingMovies, from: window)
-            case .searchMovies:
-                changeTabBarToSelectedIndex(RootCoordinatorIdentifier.searchMovies, from: window)
-            case .favorites:
-                changeTabBarToSelectedIndex(RootCoordinatorIdentifier.account, from: window)
-                // TODO: - Refactor
-                let rootCoordinator = rootCoordinators[index(for: RootCoordinatorIdentifier.account)]
-                let unwrappedParentCoordinator = rootCoordinator.childCoordinators.last?.unwrappedParentCoordinator ?? rootCoordinator
-                let coordinator = FavoritesSavedMoviesCoordinator(navigationController: unwrappedParentCoordinator.navigationController)
-                coordinator.parentCoordinator = unwrappedParentCoordinator
-                unwrappedParentCoordinator.childCoordinators.append(coordinator)
-                coordinator.start()
+            self.onInitialTransition = { window in self.handleDeepLink(url, and: window) }
+            if initialTransitionCompleted {
+                onInitialTransition?(window)
             }
         }
     }
 
+    private func handleDeepLink(_ url: URL, and window: UIWindow?) {
+        deepLinkHandler.handleDeepLinkUrl(url, in: window)
+    }
+
     func handleShortcutItem(_ shortcutItem: UIApplicationShortcutItem, and window: UIWindow?) {
         guard let shorcut = AppShortcutItem(rawValue: shortcutItem.type) else { return }
         switch shorcut {
@@ -73,7 +67,6 @@
 
     private func changeTabBarToSelectedIndex(_ rootIdentifier: String, from window: UIWindow?) {
         let selectedIndex = index(for: rootIdentifier)
-        currentSelectedIndex = selectedIndex
         guard let tabBarController = window?.rootViewController as? MainTabBarController else {
             return
         }