快速获取 topmostViewController 避免 UIAlertController

Question

我想获得 topMostVC 所以我做了这个功能：

    func topMostViewController() -> UIViewController {
        
        if let presented = self.presentedViewController {
            return presented.topMostViewController()
        }
        
        if let navigation = self as? UINavigationController {
            return navigation.visibleViewController?.topMostViewController() ?? navigation
        }
        
        if let tab = self as? UITabBarController {
            return tab.selectedViewController?.topMostViewController() ?? tab
        }
        
        return self
    }

然后使用：

AppDelegate.shared.window?.rootViewController?.topMostViewController()

但我的问题是当 UIAlertController 在屏幕上显示时，它返回 UIAlertController，而不是我期望的 viewcontroller

谁能告诉我如何避免 UIAlertController，谢谢

Answer 1

再添加一个条件

if let presented = self.presentedViewController,
   !(presented is UIAlertController) { // <== Here or use !presented.isKind(of: UIAlertController.self)
    return presented.topMostViewController()
}