第四十一篇Masonry利用Block实现链式编程

Posted 2020-08-12 久冬不雨

一直都觉得使用Masonry的时候语法特别优雅,很早的时候就想尝试下怎么实现,

一直都没弄明白,直到最近看见一篇叫block实现链式编程的

使用实例：

 UIButton * btn = [UIButton button].normalTitle(@"普通状态").normalImage([UIImage imageNamed:@"icon_help"]).selectedTitle(@"选中状态");

 

/// 声明文件
@interface UIButton (JQExtension)
/// 连接符,没啥实际意义,蛋疼的时候可以无限.set....
- (instancetype)set;
 
+ (instancetype)button;
/// 设置普通状态标题
- (UIButton *(^)(NSString *))normalTitle;
 
/// 设置选中状态标题
- (UIButton *(^)(NSString *))selectedTitle;
 
/// 设置高亮状态标题
- (UIButton *(^)(NSString *))highlightTitle;
 
/// 设置普通状态图片
- (UIButton *(^)(UIImage *))normalImage;
 
/// 设置选中状态图片
- (UIButton *(^)(UIImage *))selectedImage;
 
/// 设置高亮状态图片
- (UIButton *(^)(UIImage *))highlightedImage;
 
/// 设置普通状态背景图片
- (UIButton *(^)(UIImage *))normalBackgroundImage;
 
/// 设置高亮状态背景图片
- (UIButton *(^)(UIImage *))highlightedBackgroundImage;
 
/// 设置选中状态背景图片
- (UIButton *(^)(UIImage *))selectedBackgroundImage;
 
@end
 
 
 
实现文件
#import "UIButton+GKExtension.h"
 
@implementation UIButton (GKExtension)
- (instancetype)set {
    return self;
}
 
+ (instancetype)button {
    return [self buttonWithType:UIButtonTypeCustom];
}
 
- (UIButton *(^)(NSString *))normalTitle {
    return ^(NSString *title){
        [self setTitle:title forState:UIControlStateNormal];
        return self;
    };
}
 
- (UIButton *(^)(NSString *))selectedTitle {
    return ^(NSString *title){
        [self setTitle:title forState:UIControlStateNormal];
        return self;
    };
}
 
- (UIButton *(^)(NSString *))highlightTitle {
    return ^(NSString *title){
        [self setTitle:title forState:UIControlStateNormal];
        return self;
    };
}
 
- (UIButton *(^)(UIImage *))normalImage {
    return ^(UIImage *image){
        [self setImage:image forState:UIControlStateNormal];
        return self;
    };
}
- (UIButton *(^)(UIImage *))normalBackgroundImage {
    return ^(UIImage *image){
        [self setBackgroundImage:image forState:UIControlStateNormal];
        return self;
    };
}
- (UIButton *(^)(UIImage *))selectedImage {
    return ^(UIImage *image){
        [self setImage:image forState:UIControlStateSelected];
        return self;
    };
}
 
- (UIButton *(^)(UIImage *))selectedBackgroundImage {
    return ^(UIImage *image){
        [self setBackgroundImage:image forState:UIControlStateSelected];
        return self;
    };
}
 
- (UIButton *(^)(UIImage *))highlightedImage {
    return ^(UIImage *image){
        [self setImage:image forState:UIControlStateHighlighted];
        return self;
    };
}
 
- (UIButton *(^)(UIImage *))highlightedBackgroundImage {
    return ^(UIImage *image){
        [self setBackgroundImage:image forState:UIControlStateHighlighted];
        return self;
    };
}