ViewControllerDismissAction.swift

import UIKit

extension UIViewController {
    @IBAction func dismiss(sender: AnyObject?) {
        if let nc = self.navigationController {
            nc.popViewControllerAnimated(true)
        } else {
            self.dismissViewControllerAnimated(true, completion: nil)
        }
    }
}