Merge pull request #91 from iMostfa/feature/add-on-inject-hook

Add After injection hook
krzysztofzablocki · Apr 7, 2024 · 1f368dc · 1f368dc
2 parents d394c53 + 19c7802
commit 1f368dc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Sources/Inject/Integrations/Hosts.swift b/Sources/Inject/Integrations/Hosts.swift
@@ -23,6 +23,7 @@ public typealias ViewHost = _InjectableViewHost
 open class _InjectableViewControllerHost<Hosted: InjectViewControllerType>: InjectViewControllerType {
     public private(set) var instance: Hosted
     let constructor: () -> Hosted
+    public var afterInjectionHook: (() -> Void)?
 
     public init(_ constructor: @autoclosure @escaping () -> Hosted) {
         instance = constructor()
@@ -32,8 +33,9 @@ open class _InjectableViewControllerHost<Hosted: InjectViewControllerType>: Inje
         self.enableInjection()
 
         addAsChild()
-        onInjection { instance in
+        onInjection { [weak self] instance in
             instance.resetHosted()
+            self?.afterInjectionHook?()
         }
     }