Runtime 替换全局 MJRefreshHeader

实现效果

表头替换后效果

需求

• 自定义带gif表头，类似饿了么app下拉刷新效果
• 替换所有原页面中默认的 MJRefreshNormalHeader 样式
• 自定义表头有较强复用性，可同时用于多个项目

思路

MJRefresh继承关系

方案一
创建 MJRefreshNormalHeader 子类，继承并重写 headerWithRefreshingBlock 方法。可以实现所有功能，但需要对 MJRefreshNormalHeader 进行高度自定义且需要在所有调用的地方导入并修改类名

方案二
直接使用 MJRefreshGifHeader ，替换所有 MJRefreshNormalHeader 表头并进行动画配置，或者直接在 MJRefreshGifHeader 基础上进行封装，需要在所有调用的地方导入并修改类名

方案三
继承 MJRefreshGifHeader 重写初始化方法，在初始化方法中配置相关动画，并使用 runtime 拦截并替换 headerWithRefreshingBlock 方法。该方案自定义内容较少，不用修改任何本类以外代码，工程间可以“即拖即用”。如下图：

方案三.png

实现

RYRefreshGifHeader.m

/*
 * 重写 MJRefreshComponent 刷新控件基类中的初始化方法
 */
- (void)prepare {
    [super prepare];
    
    // 隐藏刷新状态及刷新时间显示
    self.lastUpdatedTimeLabel.hidden = YES;
    self.stateLabel.hidden = YES;
    
    // 加载下拉及刷新状态图片组
    NSArray *idleImageArr = [self configImageArrayWithImagePrefixionString:@""];
    NSArray *refreshingImageArr = [self configImageArrayWithImagePrefixionString:@""];
    
    // 调用 MJRefreshGifHeader 公共方法设置对应状态图片
    [self setImages:idleImageArr duration:idleImageArr.count * 0.1 forState:MJRefreshStateIdle];
    [self setImages:refreshingImageArr duration:refreshingImageArr.count * 0.04 forState:MJRefreshStateRefreshing];
}

/*
 * 重写 MJRefreshGifHeader 加载图片方法，修改图片偏移量以适配刘海屏下加载图片被遮挡问题
 */
- (void)setImages:(NSArray *)images duration:(NSTimeInterval)duration forState:(MJRefreshState)state {
    [super setImages:images duration:duration forState:state];
    BOOL isIphonexSerious = ([UIScreen mainScreen].bounds.size.height == 812 || [UIScreen mainScreen].bounds.size.height == 896);
    self.mj_h += (isIphonexSerious ? kRYRefreshGifHeaderIPhoneXAppendHight : kRYRefreshGifHeaderAppendHight);
}

/*
 * 根据图片名前缀配置图片数组
 */
- (NSArray *)configImageArrayWithImagePrefixionString:(NSString *)prefixionString {
    
    NSMutableArray *imageArr = @[].mutableCopy;
    for (NSInteger index = 0; index < kRYRefreshGifHeaderMaxImageLoopCount; index++) {
        NSString *imageName = [NSString stringWithFormat:@"%@%@", prefixionString, @(index)];
        UIImage *image = [UIImage imageNamed:imageName];
        if (image != nil) {
            
            // MJRefreshGifHeader 暂不支持修改loading图片的比例，所以这里使用 Quart2D 对图片进行等比缩放
            CGFloat screenWidth = [UIScreen mainScreen].bounds.size.width;
            CGFloat screenScale = [NSString stringWithFormat:@"%.2f", screenWidth / 375.f].floatValue;
            CGSize imageTempSize = CGSizeMake(kRYRefreshGifHeaderImageWidth * screenScale, kRYRefreshGifHeaderImageWidth * screenScale);
            // 创建一个bitmap的context
            // 并把它设置成为当前正在使用的context
            UIGraphicsBeginImageContextWithOptions(imageTempSize, NO, [[UIScreen mainScreen] scale]);
            // 绘制改变大小的图片
            [image drawInRect:CGRectMake(0, 0, imageTempSize.width, imageTempSize.height)];
            // 从当前context中创建一个改变大小后的图片
            UIImage *scaledImage = UIGraphicsGetImageFromCurrentImageContext();
            // 使当前的context出堆栈
            UIGraphicsEndImageContext();
            [imageArr addObject:scaledImage];
        } else {
            break;
        }
    }
    return imageArr.copy;
}

/*
 * runtime 拦截到 MJRefreshNormalHeader 的 headerWithRefreshingBlock 方法后会调用该方法进行替换。返回 RYRefreshGifHeader 对象并初始化回调block
 */
+ (instancetype)swizzled_headerWithRefreshingBlock:(MJRefreshComponentRefreshingBlock)refreshingBlock {
    RYRefreshGifHeader *cmp = [[self alloc] init];
    cmp.refreshingBlock = refreshingBlock;
    return cmp;
}    

MJRefreshNormalHeader+Gif.m

/*
 * 修改 headerWithRefreshingBlock 方法的IMP指针指向，使其在调用时执行本类中 swizzled_headerWithRefreshingBlock 的实现
 */
+ (void)load {
    Method normalHeaderMethod = class_getClassMethod(self, @selector(headerWithRefreshingBlock:));
    Method gifHeaderMethod = class_getClassMethod(self, @selector(swizzled_headerWithRefreshingBlock:));
    IMP gifHeaderMethodImplementation = method_getImplementation(gifHeaderMethod);
    method_setImplementation(normalHeaderMethod, gifHeaderMethodImplementation);
}

/*
 * 交换后的方法直接调用 RYRefreshGifHeader 的类方法
 */
+ (instancetype)swizzled_headerWithRefreshingBlock:(MJRefreshComponentRefreshingBlock)refreshingBlock {
    return (MJRefreshNormalHeader *)[RYRefreshGifHeader swizzled_headerWithRefreshingBlock:refreshingBlock];
}

Demo下载

已导入 MJRefresh 情况下将 RYRefreshGifHeader 拖入工程中可直接实现替换