获取 APP 屏幕最上层的 View Controller（续）

我一直在想，应该还有比之前一篇文章更为简洁的方式去获取 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 添加分类新增是否参与最上层控制器追踪的属性，如此即可。

以上。