NavigationHelper.swift

import UIKit

public enum NavigationHelper {
    public static func popToRoot(
        animated: Bool = false
    ) {
        getNavigationController(viewController: UIApplication.shared.connectedScenes.flatMap { ($0 as? UIWindowScene)?.windows ?? [] }.first { $0.isKeyWindow }?.rootViewController)?.popToRootViewController(animated: animated)
    }

    public static func pop(
        animated: Bool = false
    ) {
        getNavigationController(viewController: UIApplication.shared.connectedScenes.flatMap { ($0 as? UIWindowScene)?.windows ?? [] }.first { $0.isKeyWindow }?.rootViewController)?.popViewController(animated: true)
    }

    public static func getNavigationController(viewController: UIViewController?) -> UINavigationController? {
        guard let viewController = viewController else {
            return nil
        }

        if let navigationController = viewController as? UITabBarController {
            return getNavigationController(viewController: navigationController.selectedViewController)
        }

        if let navigationController = viewController as? UINavigationController {
            return navigationController
        }

        for childViewController in viewController.children {
            return getNavigationController(viewController: childViewController)
        }

        return nil
    }
}