iOS 获取当前的ViewController

刺骨的言语ヽ痛彻心扉 2022-09-24 00:21 191阅读 0赞

Swift版本：

//获取当前屏幕显示的viewController
        class func getCurrentViewController() -> UIViewController {
            var result = UIViewController()
            
            var window = UIApplication.shared.keyWindow
            if window?.windowLevel != UIWindowLevelNormal {
                let windows = UIApplication.shared.windows
                for temp in windows {
                    if temp.windowLevel == UIWindowLevelNormal {
                        window = temp
                        break
                    }
                }
            }
            
            if let appRootVC = window?.rootViewController {
                if let frontView = window?.subviews.first {
                    if var nextResponder = frontView.next  {
                        if let _ = appRootVC.presentedViewController {
                            nextResponder = appRootVC.presentedViewController!
                        }
                        
                        if nextResponder is UITabBarController {
                            let tabbar = nextResponder as! UITabBarController
                            let nav = tabbar.viewControllers![tabbar.selectedIndex] as! UINavigationController
                            result = nav.childViewControllers.last!
                        } else if nextResponder is UINavigationController {
                            let nav = nextResponder as! UINavigationController
                            result = nav.childViewControllers.last!
                        } else {
                            result = nextResponder as! UIViewController
                        }
                    }
                }
            }
            return result
        }

OC版本：

+ (UIViewController *)getCurrentViewController {
        UIViewController * result = [[UIViewController alloc] init];
        UIWindow * window = [UIApplication sharedApplication].keyWindow;
        if (window.windowLevel != UIWindowLevelNormal) {
            NSArray * windows = [UIApplication sharedApplication].windows;
            for (UIWindow * temp in windows) {
                if (temp.windowLevel == UIWindowLevelNormal) {
                    window = temp;
                    break;
                }
            }
        }
        
        UIViewController * rootVC = window.rootViewController;
        if (rootVC) {
            UIView * frontView = window.subviews.firstObject;
            if (frontView) {
                UIResponder * nextResponder = frontView.nextResponder;
                if (rootVC.presentedViewController) {
                    nextResponder = rootVC.presentedViewController;
                }
                if ([nextResponder isKindOfClass:[UITabBarController class]]) {
                    UITabBarController * tabbar = (UITabBarController *)nextResponder;
                    UINavigationController * nav = (UINavigationController *)tabbar.viewControllers[tabbar.selectedIndex];
                    result = nav.childViewControllers.lastObject;
                }else if ([nextResponder isKindOfClass:[UINavigationController class]]) {
                    UINavigationController * nav = (UINavigationController *)nextResponder;
                    result = nav.childViewControllers.lastObject;
                }else {
                    if ([nextResponder isKindOfClass:[UIView class]]) {
                        
                    }else {
                        result = (UIViewController *)nextResponder;
                    }
                }
            }
        }
        return result;
    }