UIViewController -Extension

// MARK:UIViewController Extension

extension UIViewController{
    // 获取 ViewController 最高层的显示的 视图控制器
    var topPresentedViewController:UIViewController?{
        var target:UIViewController? = self
        while (target?.presentedViewController != nil) {
            target = target?.presentedViewController
        }
        return target
    }
    
    //从 是视图控制器的栈中 获取VisibleViewController 在显示的层级上
    
    var topVisibleViewController:UIViewController?{
        
        if let navigation = self as?UINavigationController{
            if let visibleViewController = navigation.visibleViewController {
                return visibleViewController.topVisibleViewController
            }
        }
        
        if let tab = self as?UITabBarController{
            if let selectedViewController = tab.selectedViewController{
                return selectedViewController.topVisibleViewController
            }
        }
        
        return self
    }
    
    //通过 获取,正在显示的视图控制器,获得 可见的试图控制器
    var topMostViewController:UIViewController?{
        return self.topPresentedViewController?.topVisibleViewController
    }
}

你可能感兴趣的:(UIViewController -Extension)