AutoAssembly.swift

import Foundation

public enum Swinject {
  public final class Container {
  }
}

protocol Assembly {
  func assemble(to container: Swinject.Container)
}

@objcMembers public class AutoAssembly: NSObject, Assembly {
  public class Container: NSObject {
    let container: Swinject.Container

    init(container: Swinject.Container) {
      self.container = container
    }
  }

  public func assemble(to container: Swinject.Container) {
    var count: UInt32 = 0
    guard let methods = class_copyMethodList(Self.self, &count) else { return }

    let container = Container(container: container)

    for i in 0..<count {
      let pointer = methods.advanced(by: Int(i))
      let method = pointer.pointee

      let selector = method_getName(method)
      let name = NSStringFromSelector(selector)
      guard name.hasPrefix("register") else { continue }

      let implementation = method_getImplementation(method)

      typealias ObjcRegisterBlock = @convention(c) (Any, Selector, Container) -> Void
      let register = unsafeBitCast(implementation, to: ObjcRegisterBlock.self)
      register(self, selector, container)
    }
  }
}

public class MyAssembly: AutoAssembly {
  func registerFoo(to container: Container) {
    print(#function)
  }

  func registerBar(to container: Container) {
    print(#function)
  }
}

let container = Swinject.Container()
MyAssembly().assemble(to: container)