我一直在想，应该还有比之前一篇文章更为简洁的方式去获取 APP 屏幕最上层的 View Controller。当一个 controller 出现在屏幕上，其生命周期的-viewDidAppear:方法会被调用，因此只需要在该方法中记录当前的 controller 即可。为了减少对原有代码的侵入性，考虑使用 Swizzing 对 -viewDidAppear: 方法挂钩。
// TopVC.h
@interface TopVC : NSObject
@property (nonatomic, weak, readonly) UIViewController *top;
+ (instancetype)shared;
@end
// TopVC.m
#define swizzing(a, b) ...
@interface TopVC ()
@property (nonatomic, weak, readwrite) UIViewController *top;
@end
@implementation TopVC
+ (instancetype)shared {
static TopVC *instance = nil;
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
instance = [[self alloc] init];
});
return instance;
}
@end
@implementation UIViewController (VCS)
+ (void)load {
@autoreleasepool {
// swizzing 是自定义的一个宏，用于实现方法交换。
swizzing(@"viewDidAppear:", @"vcs_viewDidAppear:");
}
}
- (void)vcs_viewDidAppear:(BOOL)animated {
[TopVC shared].top = self;
[self vcs_viewDidAppear:animated];
}
@end
#undef swizzing
将上述代两个文件放入项目中，直接使用[TopVC shared].top获取即可。项目地址：https://github.com/zhwayne/TopVC.
2017-10-23 更新：
有些时候我们并不希望当前添加的 UIViewController 影响到最上层视图控制器的追踪，比如某些 ChildViewController 对用户不可见，或者 ViewController 为半透明弹窗（如 UIAlertViewController ），这时建议为 UIViewController 添加分类新增是否参与最上层控制器追踪的属性，如此即可。
以上。
