Get top most UIViewController if its presented modally

Question

I have presented a ViewController modally using vc.modalPresentationStyle =.fullScreen

But when I try to get the top most view controller using the below code, it returns nil.

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

Here topController?.presentedViewController returns nil, if I present it using vc.modalPresentationStyle =.fullScreen and its working fine if I use default presentation style.

Did anyone faced the same issue, any help would be appreciated.

Answer 1

I have created this extension for getting topmost 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
    }
}

Hope this will work for you