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

背景

在開發過程中,經常需要獲取當前 window, rootViewController, 以及當前 ViewController 的需求. 如果 .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_getCurrentViewController{

    UIViewController* currentViewController = [self jsd_getRootViewController];
    BOOL runLoopFind = YES;
    while (runLoopFind) {
        if (currentViewController.presentedViewController) {

            currentViewController = currentViewController.presentedViewController;
        } else if ([currentViewController isKindOfClass:[UINavigationController class]]) {

          UINavigationController* navigationController = (UINavigationController* )currentViewController;
            currentViewController = [navigationController.childViewControllers lastObject];

        } else if ([currentViewController isKindOfClass:[UITabBarController class]]) {

          UITabBarController* tabBarController = (UITabBarController* )currentViewController;
            currentViewController = tabBarController.selectedViewController;
        } else {

            NSUInteger childViewControllerCount = currentViewController.childViewControllers.count;
            if (childViewControllerCount > 0) {

                currentViewController = currentViewController.childViewControllers.lastObject;

                return currentViewController;
            } else {

                return currentViewController;
            }
        }

    }
    return currentViewController;
}

這裡講一下實現思路, 我們想要與控制器無耦合的情況下, 想要直接獲取到當前控制器, 基本都是通過 rootViewController 來查詢的, 通過上面的方法拿到 rootViewControoler 之後, 我們先看 presentedViewController, 因為控制器呈現出來的方式有 push 與 present, 我們先檢視它是否是 present 出來的, 如果是則通過此屬效能找到 present 出來的當前控制器, 然後在檢查是否屬於 UINavigationControler 或 UITabBarController ,如果是則通過查詢其子控制器裡面最頂層或者其正在選擇的控制器。
最後在判斷當前控制器是否有子控制器的情況, 如果有則取其子控制器最頂層, 否則當前控制器就是其本身。

這裡主要是查詢當前 應用程式基於 UITabBarController 和 UINavigationControler 下管理的檢視控制器, 如果還有其他控制器則需要新增 if 條件來進行判斷。

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: 輕鬆獲取當前控制器

希望此篇文章對您有所幫助，如有不對的地方，希望大家能留言指出糾正。謝謝！！！！！
學習的路上,與君共勉!!!