獲取最高的 UIViewController（如果以模態方式呈現）

Question

我已經使用vc.modalPresentationStyle =.fullScreen模態地呈現了一個 ViewController

但是，當我嘗試使用下面的代碼獲取最高視圖 controller 時，它返回 nil。

func getTopViewController() -> UIViewController? {
    var topController: UIViewController? = UIApplication.shared.keyWindow?.rootViewController
    while topController?.presentedViewController != nil {
        topController = topController?.presentedViewController
    }
    return topController
}

如果我使用vc.modalPresentationStyle =.fullScreen呈現它， topController?.presentedViewController將返回 nil，如果我使用默認呈現樣式，它的工作正常。

有沒有人遇到同樣的問題，任何幫助將不勝感激。

Answer 1

我已經創建了這個擴展以獲得最頂層的 controller

   extension UIViewController {
    var 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
    }
}

extension UIApplication {
    var topMostViewController : UIViewController? {
        return self.keyWindow?.rootViewController?.topMostViewController
    }
}

希望這對你有用