我一直在想,應該還有比之前一篇文章更為簡潔的方式去獲取 APP 屏幕最上層的 View Controller。當一個 controller 出現(xiàn)在屏幕上,其生命周期的-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 是自定義的一個宏,用于實現(xiàn)方法交換。
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 添加分類新增是否參與最上層控制器追蹤的屬性,如此即可。
以上。