简单的撸一个demo,里面包含枚举、链式语法、语法糖。
这是.h文件
#import <UIKit/UIKit.h>
//枚举
typedef NS_ENUM(NSUInteger,ButtonType){
IMAGETOP = 1, //图片在上,文字在下
IMAGELEFT = 2, //图片在左,文字在右
IMAGERIGHT = 3, //图片在右,文字在左
IMAGEBOTTOM = 4, //图片在下,文字在上
};
@interface ZYButton : UIButton
@property(nonatomic, strong)UILabel *titleL;
@property(nonatomic, strong)UIImageView *imageV;
/*!按钮样式,返回自身self,以便使用点语法。*/
- (ZYButton *(^)(ButtonType type))btnType;
/*!背景颜色*/
- (ZYButton *(^)(UIColor *backgroundColor))backColor;
/*!标题*/
- (ZYButton *(^)(NSString *title))title;
/*!图片*/
- (ZYButton *(^)(NSString *image))image;
@end
.m文件
#import "ZYButton.h"
#import "Masonry.h"
@interface ZYButton()
@end
@implementation ZYButton
- (instancetype)init
{
if (self = [super init]) {
}
return self;
}
- (ZYButton *(^)(ButtonType type))btnType{
return ^(ButtonType type){
switch (type) {
case IMAGETOP:
[self imageTopTitleBottom];
break;
case IMAGELEFT:
[self imageLeftTitleRight];
break;
case IMAGERIGHT:
[self imageRightTitleLeft];
break;
case IMAGEBOTTOM:
[self imageBottomTitleTop];
break;
default:
break;
}
return self;
};
}
- (void)imageTopTitleBottom{
self.imageV = ({
UIImageView *imageView = [[UIImageView alloc] init];
[self addSubview:imageView];
[imageView mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.mas_top).offset(5);
make.left.equalTo(self.mas_left).offset(5);
make.right.equalTo(self.mas_right).offset(-5);
make.bottom.equalTo(self.mas_top).offset(50);
}];
imageView;
});
self.titleL = ({
UILabel *label = [[UILabel alloc] init];
label.textAlignment=NSTextAlignmentCenter;
[self addSubview:label];
[label mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.imageV.mas_bottom).offset(5);
make.left.equalTo(self.mas_left).offset(5);
make.right.equalTo(self.mas_right).offset(-5);
make.bottom.equalTo(self.mas_bottom).offset(-5);
}];
label;
});
}
- (void)imageLeftTitleRight{
self.imageV = ({
UIImageView *imageView = [[UIImageView alloc] init];
[self addSubview:imageView];
[imageView mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.mas_top).offset(5);
make.left.equalTo(self.mas_left).offset(5);
make.right.equalTo(self.mas_left).offset(35);
make.bottom.equalTo(self.mas_bottom).offset(-5);
}];
imageView;
});
self.titleL = ({
UILabel *label = [[UILabel alloc] init];
label.textAlignment=NSTextAlignmentCenter;
[self addSubview:label];
[label mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.mas_top).offset(5);
make.left.equalTo(self.imageV.mas_left).offset(5);
make.right.equalTo(self.mas_right).offset(-5);
make.bottom.equalTo(self.mas_bottom).offset(-5);
}];
label;
});
}
- (void)imageRightTitleLeft{
self.titleL = ({
UILabel *label = [[UILabel alloc] init];
label.textAlignment=NSTextAlignmentCenter;
[self addSubview:label];
[label mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.mas_top).offset(5);
make.left.equalTo(self.mas_left).offset(5);
make.right.equalTo(self.mas_left).offset(50);
make.bottom.equalTo(self.mas_bottom).offset(-5);
}];
label;
});
self.imageV = ({
UIImageView *imageView = [[UIImageView alloc] init];
[self addSubview:imageView];
[imageView mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.mas_top).offset(5);
make.left.equalTo(self.titleL.mas_right).offset(5);
make.right.equalTo(self.mas_right).offset(-5);
make.bottom.equalTo(self.mas_bottom).offset(-5);
}];
imageView;
});
}
- (void)imageBottomTitleTop{
self.titleL = ({
UILabel *label = [[UILabel alloc] init];
label.textAlignment=NSTextAlignmentCenter;
[self addSubview:label];
[label mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.mas_top).offset(5);
make.left.equalTo(self.mas_left).offset(5);
make.right.equalTo(self.mas_right).offset(-5);
make.bottom.equalTo(self.mas_top).offset(35);
}];
label;
});
self.imageV = ({
UIImageView *imageView = [[UIImageView alloc] init];
[self addSubview:imageView];
[imageView mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.titleL.mas_bottom).offset(5);
make.left.equalTo(self.mas_left).offset(5);
make.right.equalTo(self.mas_right).offset(-5);
make.bottom.equalTo(self.mas_bottom).offset(-5);
}];
imageView;
});
}
- (ZYButton *(^)(UIColor *backgroundColor))backColor{
return ^(UIColor *backColor){
self.backgroundColor = backColor;
return self;
};
}
- (ZYButton *(^)(NSString *title))title{
return ^(NSString *title){
self.titleL.text = title;
return self;
};
}
- (ZYButton *(^)(NSString *image))image{
return ^(NSString *image){
self.imageV.image = [UIImage imageNamed:image];
return self;
};
}
@end
使用的时候:
ZYButton *button = [[ZYButton alloc]init];
//链式语法
button.backColor([UIColor blueColor]).btnType(IMAGETOP).image(@"test.png").title(@"测试");
button.titleL.backgroundColor = [UIColor redColor];
[self.view addSubview:button];
[button mas_makeConstraints:^(MASConstraintMaker *make) {
make.top.equalTo(self.view.mas_top).offset(5);
make.left.equalTo(self.view.mas_left).offset(5);
make.size.mas_equalTo(CGSizeMake(100, 100));
}];
只是一个很简单的demo,后期还会以此进行进一步整理.