iOS 輕鬆獲取當前控制器的正確方式

背景

在開發過程當中,常常須要獲取當前 window, rootViewController, 以及當前正在顯示的 visibleController 的需求. 若是 .m 實現不是在當前視圖狀況下, 咱們須要快速的獲取到當前控制器, 這種狀況就須要先作好一層封裝,我通常是經過 UIViewController 寫的一個 Category 來實現, 實現起來也很是簡單, 只須要咱們對 控制器幾個方法掌握即可。

獲取根控制器

+ (UIViewController *)jsd_getRootViewController{

    UIWindow* window = [[[UIApplication sharedApplication] delegate] window];
    NSAssert(window, @"The window is empty");
    return window.rootViewController;
}

複製代碼

這裏很簡單, 經過單例獲取到當前 UIApplication 的 delegate 在經過 window 便可輕鬆拿到 rootViewController。

獲取當前頁面控制器

+ (UIViewController *)jsd_findVisibleViewController {
    
    UIViewController* currentViewController = [self jsd_rootViewController];

    BOOL runLoopFind = YES;
    while (runLoopFind) {
        if (currentViewController.presentedViewController) {
            currentViewController = currentViewController.presentedViewController;
        } else {
            if ([currentViewController isKindOfClass:[UINavigationController class]]) {
                currentViewController = ((UINavigationController *)currentViewController).visibleViewController;
            } else if ([currentViewController isKindOfClass:[UITabBarController class]]) {
                currentViewController = ((UITabBarController* )currentViewController).selectedViewController;
            } else {
                break;
            }
        }
    }
    
    return currentViewController;
}


複製代碼

這裏講一下實現思路, 咱們想要與控制器無耦合的狀況下, 想要直接獲取到當前控制器, 基本都是經過 rootViewController 來查找的, 經過上面的方法拿到 rootViewControoler 以後, 咱們先看 presentedViewController, 由於控制器呈現出來的方式有 push 與 present, 咱們先查看它是不是 present 出來的, 若是是則經過此屬性能找到 present 出來的當前控制器, 而後在檢查是否屬於 UINavigationControler 或 UITabBarController ,若是是則經過查找其子控制器裏面最頂層或者其正在選擇的控制器。 最後在判斷當前控制器是否有子控制器的狀況, 若是有則取其子控制器最頂層, 不然當前控制器就是其自己。

這裏主要是查找當前 應用程序基於 UITabBarController 和 UINavigationControler 下管理的視圖控制器, 若是還有其餘控制器則須要添加 if 條件來進行判斷。

方法二: 當咱們有正在呈現的視圖控制器子 View 時, 可經過屬性 nextResponder 遞歸查找

+ (nullable UIViewController *)findBelongViewControllerForView:(UIView *)view {
    UIResponder *responder = view;
    while ((responder = [responder nextResponder]))
        if ([responder isKindOfClass: [UIViewController class]]) {
            return (UIViewController *)responder;
        }
    return nil;
}
複製代碼

presentedViewController

Apple 文檔 presentedViewControlle

經過此方法能夠查找到經過 presented 模態方式(顯示與隱士) 方式推出的當前控制器。 例如: AViewController --> BViewController 經過模態方式推出. 則使用 AViewController.presentedViewController 能獲取到 BViewController。

presentingViewController

Apple 文檔

經過此方法能夠查找到經過 presented 模態方式(顯示與隱士) 方式推出當前控制器的上層控制器。 例如: AViewController --> BViewController 經過模態方式推出. 則使用 BViewController.presentingViewController 能獲取到 AViewController。

modalViewController

查看文檔發現此方法已在 iOS 6 後被棄用, 官方推薦直接使用 presentedViewController 替代便可.

參考資料與Dome

UIViewController的presentedViewController，presentingViewController和parentViewController三個屬性 Dome: 輕鬆獲取當前控制器

但願此篇文章對您有所幫助，若有不對的地方，但願你們能留言指出糾正。謝謝！！！！！ 學習的路上,與君共勉!!!