有一個 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 了:)。
2017-02-06 更新
該方法太過繁瑣，可參考另一種方案。
