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

爱被打了一巴掌 2022-11-30 01:38 172阅读 0赞

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

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdGF0bzUxMg_size_16_color_FFFFFF_t_70_pic_center]

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

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

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

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

.h文件

#import <UIKit/UIKit.h>
    
    @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

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BvdGF0bzUxMg_size_16_color_FFFFFF_t_70_pic_center]: /images/20221124/6eec967711454cdea8d6fea6d2ce5037.png