ItemViewModel BuilderContainer extension

ItemViewModelBuilderContainer.swift

struct ColorItemViewModel: ItemViewModel {

  var reuseIdentifier: String
  
  var color: UIColor
}

extension ColorItemViewModel: BuilderContainer {
    
    func getBuilder<Context>(_ contextType: Context.Type) -> Builder<Context>? {
        return ColorBuilder(color: self.color) as? Builder<contextType>  
    }
}

class ColorBuilder<ApplicationContext>: Builder<Context> {
  
    var color: UIColor
    
    open override func build(_ context: ApplicationContext) -> UIViewController? {
        let controller = UIViewController()
        controller.view.backgroundColor = self.color
        return controller
    }
}