首页 热点资讯 义务教育 高等教育 出国留学 考研考公
您的当前位置:首页正文

获取 APP 屏幕最上层的 View Controller

2024-12-16 来源:花图问答

有一个 APP 里有若干个 View Controller,有些 Controller 里的 View 涉及到用户敏感信息,比如显示用户存款金额和社保号。程序进入后台后,我需要在这类 Controller 上 present 一个模态的 View Controller 进行遮挡(典型的用法就是弹出一个手势密码,或者对这个 Controller 上的 View 进行高斯模糊),但是在其他不涉及用户隐私的 Controller 上,我不能这样做。那么在程序进入后台时,我如何知道屏幕最上层的一个Controller 是什么鬼(拿到这个 Controller 的实例),并选择性的将其遮挡?

//获取当前屏幕显示的 View Controller
- (UIViewController *)getCurrentVC
{
    UIViewController *result = nil;
    UIWindow * window        = [[UIApplication sharedApplication] keyWindow];
    if (window.windowLevel != UIWindowLevelNormal) {
        NSArray *windows = [[UIApplication sharedApplication] windows];
        for(UIWindow * tmpWin in windows) {
            if (tmpWin.windowLevel == UIWindowLevelNormal) {
                window = tmpWin;
                break;
            }
        }
    }
    
    UIView *frontView = [[window subviews] objectAtIndex:0];
    id nextResponder  = [frontView nextResponder];
    
    if ([nextResponder isKindOfClass:[UIViewController class]])
        result = nextResponder;
    else
        result = window.rootViewController;
    
    return result;
}

分析了一下这段源码,前半部分是获取 APP 的普通层级 window,我们的视图一般会显示在这个 window 上。后部分源码就有点不靠谱了,断点打印[window subviews]的值发现这个数组里只有一个元素:

屏幕快照 2015-12-30 下午5.37.04.png

这个UILayoutContainerView就是 window 上的第一个 subview,那么他的 nextResponder 自然是这个 window ,所以结果总是返回rootViewController。而 rootViewController 可能是 TabBarController 也可能是 NavigationController,假设我们的目标控制器有着下面这种层级关系:

Window -> TabBarController -> NavigationController -> ViewController1 -> Target ViewController

上面这种方法只能拿到 TabBarController

那么我们换一种思路。若要获取最上层的 view controller 可以采取【顺藤摸瓜】模式,从 root view controller 开始递归遍历所有的控制器,如果找到的控制器是 UITabBarController 的子类,就找它的selectedViewController继续向递归,如果找到的控制器是 UINavigationController 的子类,就找它的visibleViewController继续递归。这样一路找上来总会到头,最头上的控制器就是我们的目标控制器。

这里我将其写成一个UIApplication的分类方便使用:

@implementation UIApplication (ActivityViewController)

- (UIViewController *)activityViewController {
    __block UIWindow *normalWindow = [self.delegate window];
    if (normalWindow.windowLevel != UIWindowLevelNormal) {
        [self.windows enumerateObjectsUsingBlock:^(__kindof UIWindow * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
            if (obj.windowLevel == UIWindowLevelNormal) {
                normalWindow = obj;
                *stop        = YES;
            }
        }];
    }
    
    return [self p_nextTopForViewController:normalWindow.rootViewController];
}

- (UIViewController *)p_nextTopForViewController:(UIViewController *)inViewController {
    while (inViewController.presentedViewController) {
        inViewController = inViewController.presentedViewController;
    }
    
    if ([inViewController isKindOfClass:[UITabBarController class]]) {
        UIViewController *selectedVC = [self p_nextTopForViewController:((UITabBarController *)inViewController).selectedViewController];
        return selectedVC;
    } else if ([inViewController isKindOfClass:[UINavigationController class]]) {
        UIViewController *selectedVC = [self p_nextTopForViewController:((UINavigationController *)inViewController).visibleViewController];
        return selectedVC;
    } else {
        return inViewController;
    }
}

@end

在需要的时候直接使用[[UIApplication sharedApplication] activityViewController]就可以返回最上层的 view controller 了:)。

2017-02-06 更新

显示全文