有一个 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]
的值发现这个数组里只有一个元素:
这个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 了:)。