add each over classes reference without recursive fatall error

Author: Hadevs

SwiftInjector/SwiftInjector/AppDelegate.swift

@@ -16,10 +16,9 @@ class AppDelegate: UIResponder, UIApplicationDelegate {
 
 
   func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
-    let newVc: ViewController? = container.resolve(name: "123")
+    let newVc: ViewController? = self.container.resolve(name: "123")
     let testClass = newVc?.testClass
 
-
     print(testClass)
     return true
   }

SwiftInjector/SwiftInjector/Source/Containerable.swift

@@ -12,10 +12,15 @@ protocol Containerable: class {
   typealias Object = AnyObject
   typealias ServiceName = String
   typealias Service = (() -> Object)
+  typealias RelationIn = String
+  typealias RelationOut = String
+
   var container: Containerable? { get set }
   var dispatchRegistrationGroup: DispatchGroup { get }
 
   var services: [ServiceName: [ContainerObject]] { get set }
+  var relations: [RelationIn: [RelationOut]] { get set }
+  var recursiveNotResolvedObjects: [Object] { get set }
 }
 
 struct ContainerObject {
@@ -30,7 +35,6 @@ struct ContainerObject {
 
 extension Containerable {
   private func resolveAny(typeString: String, name: String? = nil) -> Object? {
-    // TODO: replace .first with more smart condition
     let array: [ContainerObject]? = {
       if let filterName = name {
         return services[typeString]?.filter { $0.name == filterName }
@@ -40,19 +44,22 @@ extension Containerable {
     }()
 
     if (array?.count ?? 0) > 1 {
-      SILogger("Warning in \(typeString) resolving. You registered two different object for this type. Try to provide \"name\" argument while registering your object. But you will receive first object of all services you registered.").log()
+      SILogger("Warning in \(typeString) resolving. You registered two different objects for this type. Try to provide \"name\" argument while registering your object. But you will receive first object of all services you registered.").log(priority: .high)
     }
 
     if let object = array?.first?.object {
-      autoresolve(on: object)
       return object
     } else if let object = array?.first?.registration() {
+
       var objects = services[typeString]
       if objects?.count ?? 0 > 0 {
         objects?[0].object = object
       }
+
       services[typeString] = objects ?? []
       autoresolve(on: object)
+      finishRegistrations()
+
       return object
     } else {
       return nil
@@ -66,14 +73,33 @@ extension Containerable {
     return object
   }
 
+  func finishRegistrations() {
+    relations = [:]
+    for object in recursiveNotResolvedObjects {
+      autoresolve(on: object)
+    }
+    recursiveNotResolvedObjects.removeAll()
+  }
+
   private func autoresolve<T: Object>(on object: T) {
     let mirror = Mirror(reflecting: object)
+
     for attr in mirror.children {
       if let property_name = attr.label {
         let objectType = type(of: attr.value)
         let typeString = String(describing: objectType).replacingOccurrences(of: "Optional<", with: "").replacingOccurrences(of: ">", with: "")
+
+        // MARK: MUTATE VARIABLE (DANGER OBJC RUNTIME ZONE)
         if let ivar = class_getInstanceVariable(type(of: object), property_name) {
-          print(type(of: object), mirror.children.map { $0.label })
+          let typeStringOfMyObject = formattedString(of: object)
+          recordRelations(relationIn: typeString, relationOut: typeStringOfMyObject)
+          if relations[typeStringOfMyObject]?.contains(typeString) ?? false {
+            relations[typeString]?.removeAll(where: { (string) -> Bool in
+              return string == typeStringOfMyObject
+            })
+            recursiveNotResolvedObjects.append(object)
+            return
+          }
           if let value = resolveAny(typeString: typeString) {
             object_setIvar(object, ivar, value)
           }
@@ -82,6 +108,61 @@ extension Containerable {
     }
   }
 
+  private func memmoryAddress(_ object: Object) -> String {
+    return "\(Unmanaged.passUnretained(object).toOpaque())"
+  }
+
+  private func recordRelations(relationIn: RelationIn, relationOut: RelationOut) {
+    if let relationsOut = relations[relationIn], !relationsOut.contains(relationOut) {
+      var newRelationsOut = relationsOut
+      newRelationsOut.append(relationOut)
+      relations[relationIn] = newRelationsOut
+    } else {
+      relations[relationIn] = [relationOut]
+    }
+  }
+
+  private func formattedString<T: Object>(of object: T) -> String {
+    return String(describing: type(of: object)).replacingOccurrences(of: "Optional<", with: "").replacingOccurrences(of: ">", with: "")
+  }
+
+  private func doesObjectHas(object: Object, propertyType: String) -> Bool {
+    let mirror = Mirror(reflecting: object)
+    for attr in mirror.children {
+      let objectType = type(of: attr.value)
+      let typeString = String(describing: objectType).replacingOccurrences(of: "Optional<", with: "").replacingOccurrences(of: ">", with: "")
+      if typeString == typeString {
+        return true
+      }
+    }
+
+    return false
+  }
+
+  private func property<T>(object: Object, propertyName: String) -> T? {
+    let mirror = Mirror(reflecting: object)
+    for attr in mirror.children {
+      if attr.label == propertyName {
+        return attr.value as? T
+      }
+    }
+
+    return nil
+  }
+
+  private func propertyName(by typeString: String, in object: Object) -> String? {
+    let mirror = Mirror(reflecting: object)
+    for attr in mirror.children {
+      let propertyType = type(of: attr.value)
+      let typeStringInside = String(describing: propertyType).replacingOccurrences(of: "Optional<", with: "").replacingOccurrences(of: ">", with: "")
+      if typeStringInside == typeString {
+        return attr.label
+      }
+    }
+
+    return nil
+  }
+
   func register<T: Object>(_ registration: @escaping (() -> T), name: String? = nil) {
 
     dispatchRegistrationGroup.enter()

SwiftInjector/SwiftInjector/Source/RootContainer.swift

@@ -9,15 +9,22 @@
 import Foundation
 
 class RootContainer: Containerable {
-  var services: [ServiceName : [ContainerObject]] = [:]
+  var recursiveNotResolvedObjects: [Object] = []
+
+  var relations: [RelationIn: [RelationOut]] = [:]
+  var services: [ServiceName: [ContainerObject]] = [:]
 
   var dispatchRegistrationGroup: DispatchGroup = DispatchGroup()
 
   var container: Containerable?
 
   init() {
+    let date = Date()
     dispatchRegistrationGroup.notify(queue: .main) {
-      print("All properties have been successfully injected.")
+
+      self.finishRegistrations()
+      let seconds = Date().timeIntervalSince(date)
+      print("All properties have been successfully injected for \(seconds) seconds.")
     }
   }
 }

SwiftInjector/SwiftInjector/TestClass.swift

@@ -10,4 +10,5 @@ import Foundation
 
 class TestClass {
   var name: String = "123"
+  weak var viewController: ViewController?
 }