手势action_block

#import "UIGestureRecognizer+YYAdd.h"

#import <objc/runtime.h>

static const int block_key;

@interface _YYUIGestureRecognizerBlockTarget : NSObject

@property (nonatomic, copy) void (^block)(id sender);

- (id)initWithBlock:(void (^)(id sender))block;

- (void)invoke:(id)sender;

@end

@implementation _YYUIGestureRecognizerBlockTarget

- (id)initWithBlock:(void (^)(id sender))block{

    self = [super init];

    if (self) {

        _block = [block copy];

    }

    return self;

}

- (void)invoke:(id)sender {

    if (_block) _block(sender);

}

@end

@implementation UIGestureRecognizer (YYAdd)

- (instancetype)initWithActionBlock:(void (^)(id sender))block {

    self = [selfinit];

    [self addActionBlock:block];

    return self;

}

- (void)addActionBlock:(void (^)(id sender))block {

    _YYUIGestureRecognizerBlockTarget *target = [[_YYUIGestureRecognizerBlockTargetalloc] initWithBlock:block];

    [self addTarget:targetaction:@selector(invoke:)];

    NSMutableArray *targets = [self_yy_allUIGestureRecognizerBlockTargets];

    [targets addObject:target];

}

- (void)removeAllActionBlocks{

    NSMutableArray *targets = [self_yy_allUIGestureRecognizerBlockTargets];

    [targets enumerateObjectsUsingBlock:^(id target,NSUInteger idx, BOOL *stop) {

        [self removeTarget:targetaction:@selector(invoke:)];

    }];

    [targets removeAllObjects];

}

- (NSMutableArray *)_yy_allUIGestureRecognizerBlockTargets {

    NSMutableArray *targets =objc_getAssociatedObject(self, &block_key);

    if (!targets) {

        targets = [NSMutableArray array];

        objc_setAssociatedObject(self, &block_key, targets,OBJC_ASSOCIATION_RETAIN_NONATOMIC);

    }

    return targets;

}