【Swift】NavigationControllerでpopかpushをいい感じにしてくれるカスタムfuncを作る方法

2019年12月25日

NavigationControllerで次に遷移するViewControllerのClassがViewControllerスタックの中に既に存在している場合はそこまでpopし、無ければpushする方法を書いていきます。

popかpushするカスタムfunc

以下をHelperクラスか何かに追加します。

extension UINavigationController {
    
    func popOrPushToViewController(ofKind kind: AnyClass, pushController: UIViewController, animated: Bool) {
        if let vc = viewControllers.last(where: { $0.isKind(of: kind)}) {
            popToViewController(vc, animated: animated)
        } else {
            pushViewController(pushController, animated: animated)
        }
    }
    
}

extension UINavigationController {

func popOrPushToViewController(ofKind kind: AnyClass, pushController: UIViewController, animated: Bool) {

if let vc = viewControllers.last(where: { $0.isKind(of: kind)}) {

popToViewController(vc, animated: animated)

} else {

pushViewController(pushController, animated: animated)

}

使い方はこんな感じです。

let className = String(describing: CustomViewController.self)
let customVC = UIStoryboard(name: "Main", bundle: nil).instantiateViewController(withIdentifier: className) as! CustomViewController
self.navigationController?.popOrPushToViewController(ofKind: CustomViewController.self, pushController: customVC, animated: true)