PoppableVC

PoppableVC.swift

protocol Bindable {
    associatedtype ViewModel
    
    var viewModel: ViewModel! { get set }
    
    func bindToViewModel()
}

protocol PoppableVC {
    associatedtype Parent
}

protocol PoppableVM {
    var action: CocoaAction { get }
}

extension UINavigationControllerDelegate where Self: UIViewController, Self: Bindable & PoppableVC, Self.ViewModel: PoppableVM {
    
    func navigationController(_ navigationController: UINavigationController, didShow viewController: UIViewController, animated: Bool) {
        if viewController is Self.Parent {
            self.viewModel.action.execute(())
        }
    }
}