Scene+ViewControllerTemplate

Scene+ViewControllerTemplate.swift

import UIKit

extension Scene {
    
    func viewController() -> UIViewController {
        
        switch self {
            
        case .firstScene(let viewModel):
            let nc = UINavigationController(rootViewController: FirstSceneViewController())
            var vc = nc.viewControllers.first as! FirstSceneViewController
            vc.bindViewModel(to: viewModel)
            return nc
            
        case .secondScene(let viewModel):
            var vc = SecondSceneViewController()
            vc.bindViewModel(to: viewModel)
            return vc
        }
    }
}