欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

iOS 获取当前正在显示的视图控制器ViewController(最全)

程序员文章站 2024-01-15 08:53:46
...

一、前言:
本来不怎么想写这篇文章的, 因为这种东西网上随便搜一大堆。但是还是发现项目中有些同学,没怎么理解好,直接照搬网上的,殊不知其本身虽处理了大部分情况,却仍有一些情况是漏掉的。所以就有了下面这篇文章。
二、场景:
在处理 URL Router 跳转的时候,经常需要得到“当前最上层的视图控制器”来进行视图跳转。
三、注意点:
网上比较多遗漏的场景是:A present B, B present C,在A中查找当前显示的视图控制器的情况。
四、直接快速使用方法:
使用方法很简单:
①、Podfile添加pod ‘CJBaseHelper/UIViewControllerCJHelper’,并pod update或pod install
②、库引入成功后,直接使用如下方法即可。即:

UIViewController *vc = [UIViewControllerCJHelper findCurrentShowingViewController];
或
UIViewController *vc = [UIViewControllerCJHelper findCurrentShowingViewControllerFrom:self];

2、Podfile添加pod 'CJBaseHelper/UIViewControllerCJHelper’后,执行pod install即可使用该库中的UIViewControllerCJHelper,即。
即通过调用以下两个方法中的任意一个即可获取到当前显示的视图控制器。

///获取Window当前显示的视图控制器ViewController
+ (UIViewController *)findCurrentShowingViewController;

/**
 *  获取Window当前显示的视图控制器ViewController
 *
 *  @param vc   从哪个界面开始分析
 *
 *  @return 当前显示的视图控制器ViewController
 */
+ (UIViewController *)findCurrentShowingViewControllerFrom:(UIViewController *)vc;

3、实际使用过程

    //获取Window当前显示的视图控制器ViewController方法①
    UIViewController *currentShowViewController1 = [UIViewControllerCJHelper findCurrentShowingViewController];
    NSLog(@"currentShowViewController1 = %@", NSStringFromClass([currentShowViewController1 class]));
    
    //获取Window当前显示的视图控制器ViewController方法②
    UIViewController *currentShowViewController2 = [UIViewControllerCJHelper findCurrentShowingViewControllerFrom:self];
    NSLog(@"currentShowViewController2 = %@", NSStringFromClass([currentShowViewController2 class]));

五、上述使用的CJBaseHelper/UIViewControllerCJHelper库的代码解析
代码中包括两种方法,一种是遍历方法,一种是递归方法,两种方法都能正确查找到当前正在显示的视图控制器。废话不多说,直接上代码。

+ (UIViewController *)findCurrentShowingViewController {
    //获得当前活动窗口的根视图
    UIViewController *vc = [UIApplication sharedApplication].keyWindow.rootViewController;
    UIViewController *currentShowingVC = [self findCurrentShowingViewControllerFrom:vc];
    return currentShowingVC;
}

//注意考虑几种特殊情况:①A present B, B present C,参数vc为A时候的情况
/* 完整的描述请参见文件头部 */
+ (UIViewController *)findCurrentShowingViewControllerFrom:(UIViewController *)vc
{
    //方法1:递归方法 Recursive method
    UIViewController *currentShowingVC;
    if ([vc presentedViewController]) { //注要优先判断vc是否有弹出其他视图,如有则当前显示的视图肯定是在那上面
        // 当前视图是被presented出来的
        UIViewController *nextRootVC = [vc presentedViewController];
        currentShowingVC = [self findCurrentShowingViewControllerFrom:nextRootVC];
        
    } else if ([vc isKindOfClass:[UITabBarController class]]) {
        // 根视图为UITabBarController
        UIViewController *nextRootVC = [(UITabBarController *)vc selectedViewController];
        currentShowingVC = [self findCurrentShowingViewControllerFrom:nextRootVC];
        
    } else if ([vc isKindOfClass:[UINavigationController class]]){
        // 根视图为UINavigationController
        UIViewController *nextRootVC = [(UINavigationController *)vc visibleViewController];
        currentShowingVC = [self findCurrentShowingViewControllerFrom:nextRootVC];
        
    } else {
        // 根视图为非导航类
        currentShowingVC = vc;
    }
    
    return currentShowingVC;
    
    /*
    //方法2:遍历方法
    while (1)
    {
        if (vc.presentedViewController) {
            vc = vc.presentedViewController;
            
        } else if ([vc isKindOfClass:[UITabBarController class]]) {
            vc = ((UITabBarController*)vc).selectedViewController;
            
        } else if ([vc isKindOfClass:[UINavigationController class]]) {
            vc = ((UINavigationController*)vc).visibleViewController;
            
        //} else if (vc.childViewControllers.count > 0) {
        //    //如果是普通控制器,找childViewControllers最后一个
        //    vc = [vc.childViewControllers lastObject];
        } else {
            break;
        }
    }
    return vc;
    //*/
}

六、其他补充:
2、如何在多次presentViewController后直接返回到指定层
场景:如果多个控制器都通过 present 的方式跳转呢?比如从A跳转到B,从B跳转到C,从C跳转到D,如何由D直接返回到A呢?
答:可以通过 presentingViewController 一直找到A控制器,然后调用A控制器的 dismissViewControllerAnimated 方法。方法如下:

UIViewController *controller = self;
while(controller.presentingViewController != nil){
    controller = controller.presentingViewController;
}
[controller dismissViewControllerAnimated:YES completion:nil];

PS:如果不是想直接返回到A控制器,比如想回到B控制器,while循环的终止条件可以通过控制器的类来判断。

3、presentedViewController 与 presentingViewController

假设从A控制器通过present的方式跳转到了B控制器,那么 A.presentedViewController 就是B控制器; B.presentingViewController 就是A控制器。

4、如何通过视图(view)获取该视图所在的控制器(viewController)

+ (nullable UIViewController *)findBelongViewControllerForView:(UIView *)view {
    UIResponder *responder = view;
    while ((responder = [responder nextResponder]))
        if ([responder isKindOfClass: [UIViewController class]]) {
            return (UIViewController *)responder;
        }
    return nil;
}

感谢dvlproad的分享,贴出原链接:
https://juejin.im/post/5c8125f6f265da2dae514f00#heading-12