在之前的一篇文章中已经介绍过, 可以通过 runtime 机制给已有的类"添加"对象属性和非对象属性)[详情]. 那么接下来, 再给系统的 UIButton 添加一个 block 属性(毕竟在 iOS 中, block 是一种 id 类型), 当点击 button 时, 直接触发 block 块中的代码, 这样就可以避免每次使用 UIButton 时都要调用系统的addTarget: action: forControlEvents: 方法.

1. 首先给需要使用的 block 类型定义一个别名:

typedef void(^MHButtonActionCallBack)(UIButton *button);

这样当需要使用这种 bolck 类型的时候, 就不需要把注意力放在 block 的具体内容上了, 即: 不需要考虑 block 的传参和返回值类型了.

2. 利用iOS runtime 应用之给 NSString 添加对象属性和非对象属性中介绍的原理给UIButton"添加"一个MHButtonActionCallBack类型的属性(诚然, 这种"添加"并不是严格意义上的添加, 只不过是添加了一对getter和 setter 方法而已):

- (void)setMHCallBack:(MHButtonActionCallBack)callBack {
    objc_setAssociatedObject(self, &_callBack, callBack, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (MHButtonActionCallBack)getMHCallBack {
     return (MHButtonActionCallBack)objc_getAssociatedObject(self, &_callBack);
}

这样就可以放心大胆的使用UIButton 的setMHCallBack:和getMHCallBack:方法进行赋值和取值操作了.

3. 为了便于调用, 还要把上一步"添加"的"属性", 封装到UIButton 内部, 只暴露出下面的方法:

- (void)addMHClickAction:(MHButtonActionCallBack)callBack;

至此, 当需要给UIButton 添加点击事件的时候, 就可以直接调用这个方法就可以了.

附上完整代码:
UIButton+MHExtra.h

#import <UIKit/UIKit.h>

typedef void(^MHButtonActionCallBack)(UIButton *button);

@interface UIButton (MHExtra)


/**
 *  @brief replace the method 'addTarget:forControlEvents:'
 */
- (void)addMHCallBackAction:(MHButtonActionCallBack)callBack forControlEvents:(UIControlEvents)controlEvents;

/**
 *  @brief replace the method 'addTarget:forControlEvents:UIControlEventTouchUpInside'
 *  the property 'alpha' being 0.5 while 'UIControlEventTouchUpInside'
 */
- (void)addMHClickAction:(MHButtonActionCallBack)callBack;

@end

UIButton+MHExtra.m

#import "UIButton+MHExtra.h"
#import <objc/runtime.h>


/**
 *  @brief add action callback to uibutton
 */
@interface UIButton (MHAddCallBackBlock)

- (void)setMHCallBack:(MHButtonActionCallBack)callBack;
- (MHButtonActionCallBack)getMHCallBack;

@end

@implementation UIButton (MHAddCallBackBlock)

static MHButtonActionCallBack _callBack;

- (void)setMHCallBack:(MHButtonActionCallBack)callBack {
    objc_setAssociatedObject(self, &_callBack, callBack, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (MHButtonActionCallBack)getMHCallBack {
     return (MHButtonActionCallBack)objc_getAssociatedObject(self, &_callBack);
}

@end;


@implementation UIButton (MHExtra)

/**
 *  @brief replace the method 'addTarget:forControlEvents:UIControlEventTouchUpInside'
 *  the property 'alpha' being 0.5 while 'UIControlEventTouchUpInside'
 */
- (void)addMHClickAction:(MHButtonActionCallBack)callBack {
    [self addMHCallBackAction:callBack forControlEvents:(UIControlEventTouchUpInside)];
    [self addTarget:self action:@selector(mhTouchDownAction:) forControlEvents:(UIControlEventTouchDown)];
    [self addTarget:self action:@selector(mhTouchUpAction:) forControlEvents:(UIControlEventTouchUpInside | UIControlEventTouchUpOutside | UIControlEventTouchCancel | UIControlEventTouchDragOutside)];
}

/**
 *  @brief replace the method 'addTarget:forControlEvents:'
 */
- (void)addMHCallBackAction:(MHButtonActionCallBack)callBack forControlEvents:(UIControlEvents)controlEvents{
    [self setMHCallBack:callBack];
    [self addTarget:self action:@selector(mhButtonAction:) forControlEvents:controlEvents];
}

- (void)mhButtonAction:(UIButton *)btn {
    self.getMHCallBack(btn);
}

- (void)mhTouchDownAction:(UIButton *)btn {
    btn.enabled = NO;
    btn.alpha = 0.5f;
}

- (void)mhTouchUpAction:(UIButton *)btn {
    btn.enabled = YES;
    btn.alpha = 1.0f;
}

@end

从现在开始, 当需要给UIBuootn 添加点击事件的时候, 就不需要先调用addTarget: action: forControlEvents:, 然后在实现其中的action 方法了, 直接:

    [btn addMHClickAction:^(UIButton *button) {
        NSLog(@"clicked button");
    }];

就 O 了~

至此, 给UIButton 添加点击事件的 block 就完成了!
如果各位有什么好的思路或者对我的代码有什么优化, 欢迎积极留言哦~