feat: DI is aware of context that starts it

Infomaniak · Feb 26, 2024 · d2573fa · d2573fa
1 parent 01bb3b5
commit d2573fa
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 5 deletions.
diff --git a/kDrive/AppDelegate.swift b/kDrive/AppDelegate.swift
@@ -36,7 +36,7 @@ import VersionChecker
 @main
 final class AppDelegate: UIResponder, UIApplicationDelegate, AccountManagerDelegate {
     /// Making sure the DI is registered at a very early stage of the app launch.
-    private let dependencyInjectionHook = EarlyDIHook()
+    private let dependencyInjectionHook = EarlyDIHook(context: .app)
 
     private var reachabilityListener: ReachabilityListener!
     private static let currentStateVersion = 4

diff --git a/kDrive/Utils/AppFactoryService.swift b/kDrive/Utils/AppFactoryService.swift
@@ -21,15 +21,24 @@ import InfomaniakDI
 import kDriveCore
 import os.log
 
+public enum AppContext {
+    case app
+    case actionExtension
+    case fileProviderExtension
+    case shareExtension
+}
+
 /// Something that loads the DI on init
 public struct EarlyDIHook {
-    public init() {
+    public init(context: AppContext) {
         // setup DI ASAP
 
         let navigationManagerFactory = Factory(type: NavigationManageable.self) { _, _ in
             NavigationManager()
         }
 
+        // TODO: Insert context in dedicated DI object
+
         os_log("EarlyDIHook")
         FactoryService.setupDependencyInjection(other: [navigationManagerFactory])
     }

diff --git a/kDriveActionExtension/ActionNavigationController.swift b/kDriveActionExtension/ActionNavigationController.swift
@@ -25,7 +25,7 @@ import VersionChecker
 
 final class ActionNavigationController: TitleSizeAdjustingNavigationController {
     /// Making sure the DI is registered at a very early stage of the app launch.
-    private let dependencyInjectionHook = EarlyDIHook()
+    private let dependencyInjectionHook = EarlyDIHook(context: .actionExtension)
 
     @LazyInjectService var accountManager: AccountManageable
 

diff --git a/kDriveFileProvider/FileProviderExtension.swift b/kDriveFileProvider/FileProviderExtension.swift
@@ -26,7 +26,7 @@ import RealmSwift
 
 final class FileProviderExtension: NSFileProviderExtension {
     /// Making sure the DI is registered at a very early stage of the app launch.
-    private let dependencyInjectionHook = EarlyDIHook()
+    private let dependencyInjectionHook = EarlyDIHook(context: .fileProviderExtension)
 
     /// Something to enqueue async await tasks in a serial manner.
     let asyncAwaitQueue = TaskQueue()

diff --git a/kDriveShareExtension/ShareNavigationViewController.swift b/kDriveShareExtension/ShareNavigationViewController.swift
@@ -25,7 +25,7 @@ import VersionChecker
 
 final class ShareNavigationViewController: TitleSizeAdjustingNavigationController {
     /// Making sure the DI is registered at a very early stage of the app launch.
-    private let dependencyInjectionHook = EarlyDIHook()
+    private let dependencyInjectionHook = EarlyDIHook(context: .shareExtension)
 
     @LazyInjectService var accountManager: AccountManageable