UIViewController+Extensions.swift

UIViewController+Extensions.swift

import UIKit

enum UIViewControllerDismissType {
    case pop
    case dismiss
}

extension UIViewController {

    func presentAlertWithTitle(title: String = "Oops!", message: String, actions: [UIAlertAction] = [], preferredStyle: UIAlertControllerStyle = .alert,  onComplete: ((KeystoneResultSimple) -> Void)? = nil) {
        let alertController = UIAlertController(title: title, message: message, preferredStyle: preferredStyle)
        
        if actions.isEmpty {
            let alertAction = UIAlertAction(title: "OK", style: .default, handler: { (alert) in
                onComplete?(.success)
            })
            
            alertController.addAction(alertAction)
        }
        else {
            actions.forEach({alertController.addAction($0)})
        }
        
        present(alertController, animated: true, completion: nil)
    }
    
}