不带标识符的shouldPerformSegue的替代方案

class ConditionalSegue: UIStoryboardSegue {
    override func perform() {
        if shouldPerform() {
            super.perform()
        }
    }

    func shouldPerform() -> Bool {
        // 在这里实现你的条件判断逻辑
        return true // 或者 false
    }
}

protocol SegueHandlerDelegate: AnyObject {
    func shouldPerformSegue(withIdentifier identifier: String) -> Bool
}

class MyViewController: UIViewController, SegueHandlerDelegate {
    override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
        if let identifier = segue.identifier {
            if !shouldPerformSegue(withIdentifier: identifier) {
                segue.source?.navigationController?.popViewController(animated: true)
            }
        }
    }

    func shouldPerformSegue(withIdentifier identifier: String) -> Bool {
        // 在这里实现你的条件判断逻辑
        return true // 或者 false
    }
}

class MyViewController: UIViewController {
    var shouldPerformSegueClosure: ((String?) -> Bool)?

    override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
        if let identifier = segue.identifier, !shouldPerformSegueClosure?(identifier) {
            segue.source?.navigationController?.popViewController(animated: true)
        }
    }
}

let viewController = MyViewController()
viewController.shouldPerformSegueClosure = { [weak self] identifier in
    // 在这里实现你的条件判断逻辑
    return true // 或者 false
}

extension MyViewController: UINavigationControllerDelegate {
    func navigationController(_ navigationController: UINavigationController, willShow viewController: UIViewController, animated: Bool) {
        if let identifier = navigationController.topViewController?.restorationIdentifier, !shouldPerformSegue(withIdentifier: identifier) {
            navigationController.popViewController(animated: true)
        }
    }

    func shouldPerformSegue(withIdentifier identifier: String) -> Bool {
        // 在这里实现你的条件判断逻辑
        return true // 或者 false
    }
}

let navigationController = UINavigationController(rootViewController: MyViewController())
navigationController.delegate = MyViewController()