push栈中删除指定的controller视图控制器

当A视图控制器push到C视图控制器时，受条件影响，需要先push到B视图控制器，然后再由B视图控制器push到C视图控制器；最后C视图控制器pop回时，直接pop到A视图控制器，由下图所示：

通常返回上一级视图控制器时，会使用到如下两个方法

// 返回到上一级视图控制器
[self.navigationController popToRootViewControllerAnimated:YES];

/// 返回到指定的上一级视图控制器
[self.navigationController popToViewController:xxx animated:YES];

很不巧，这两种实现方法都不适合我的需求，所以我只有根据自己的需求去实现，于是，使用UINavigationController的category重写了push方法，并在方法中实现自己的业务逻辑，包括跳转前的条件判断，和返回时的视图控制器控制。
代码示例：

.h文件

#import 

@interface UINavigationController (Utils)

/// 根据登录状态进行页面跳转
- (void)pushViewControllerDidLogin:(UIViewController *)viewController animated:(BOOL)animated handle:(void (^)(BOOL success))handle;

/// 移除视图
- (void)removeController;
@end

.m文件

#import "UINavigationController+Utils.h"
#import "LoginVC.h"

@implementation UINavigationController (Utils)

/// 根据登录状态进行页面跳转
- (void)pushViewControllerDidLogin:(UIViewController *)viewController animated:(BOOL)animated handle:(void (^)(BOOL success))handle
{
    if (kCurrentUser.didLogin) {
 	    // 已登录
        [self pushViewController:viewController animated:animated];
    } else {
       // 未登录
        LoginVC *loginVC = [[LoginVC alloc] init];
        __weak typeof(self) weakSelf = self;
        loginVC.loginBlock = ^(BOOL success){
            __strong __typeof__(self) strongSelf = weakSelf;
            [strongSelf pushViewController:viewController animated:animated];
            [strongSelf removeController];
            if (handle) {
                handle(success);
			}
        };
        [self pushViewController:loginVC animated:YES];
    }
}

/// 移除登录或密码设置视图
- (void)removeController
{
    NSMutableArray *viewControllers = [NSMutableArray arrayWithArray:[self.viewControllers copy]];
    
    for (UIViewController *controller in self.viewControllers) {
        if ([controller isKindOfClass:LoginVC.class]) {
            [viewControllers removeObject:controller];
        }
    }
    self.viewControllers = [viewControllers copy];
}

@end