iOS获取当前UIViewController

391 阅读1分钟
    UIViewController *rootViewController = [UIApplication sharedApplication].keyWindow.rootViewController;
 
    UIViewController *currentVC  = [self getCurrentVCFrom:rootViewController];
- (UIViewController *)getCurrentVCFrom:(UIViewController *)rootVC
{
    UIViewController *currentVC;
    
    if ([rootVC presentedViewController]) {
        // 视图是被presented出来的
        
        rootVC = [rootVC presentedViewController];
    }
    
    if ([rootVC isKindOfClass:[UITabBarController class]]) {
        // 根视图为UITabBarController
        
        currentVC = [self getCurrentVCFrom:[(UITabBarController *)rootVC selectedViewController]];
        
    } else if ([rootVC isKindOfClass:[UINavigationController class]]){
        // 根视图为UINavigationController
        
        currentVC = [self getCurrentVCFrom:[(UINavigationController *)rootVC visibleViewController]];
        
    } else {
        // 根视图为非导航类
        
        currentVC = rootVC;
    }
    
    return currentVC;
}

换成静态方法也可

#import <Foundation/Foundation.h>
#import <UIKit/UIKit.h>


@interface HTVGetCurrentVCManager : NSObject
+ (UIViewController *)getCurrentVCFrom:(UIViewController *)rootVC;

@end
@implementation HTVGetCurrentVCManager
+ (UIViewController *)getCurrentVCFrom:(UIViewController *)rootVC
{
    UIViewController *currentVC;
    
    if ([rootVC presentedViewController]) {
        // 视图是被presented出来的
        
        rootVC = [rootVC presentedViewController];
    }
    
    if ([rootVC isKindOfClass:[UITabBarController class]]) {
        // 根视图为UITabBarController
        
        currentVC = [HTVGetCurrentVCManager getCurrentVCFrom:[(UITabBarController *)rootVC selectedViewController]];
        
    } else if ([rootVC isKindOfClass:[UINavigationController class]]){
        // 根视图为UINavigationController
        
        currentVC = [HTVGetCurrentVCManager getCurrentVCFrom:[(UINavigationController *)rootVC visibleViewController]];
        
    } else {
        // 根视图为非导航类
        
        currentVC = rootVC;
    }
    
    return currentVC;
}

调用方式为

 UIViewController *rootViewController = [UIApplication sharedApplication].keyWindow.rootViewController;
[HTVGetCurrentVCManager getCurrentVCFrom:rootViewController];