UIAlertView 加Block 扩展

// 定义block

typedef void(^UIAlertView_block_self_index)(UIAlertView *alertView, NSInteger btnIndex);

typedef void(^UIAlertView_block_self)(UIAlertView *alertView);

typedef BOOL(^UIAlertView_block_shouldEnableFirstOtherButton)(UIAlertView *alertView);

@interface UIAlertView (LL)

// 根据意思理解吧  不多说

- (void)handlerClickedButton:(UIAlertView_block_self_index)aBlock;

- (void)handlerCancel:(UIAlertView_block_self)aBlock;

- (void)handlerWillPresent:(UIAlertView_block_self)aBlock;

- (void)handlerDidPresent:(UIAlertView_block_self)aBlock;

- (void)handlerWillDismiss:(UIAlertView_block_self_index)aBlock;

- (void)handlerDidDismiss:(UIAlertView_block_self_index)aBlock;

- (void)handlerShouldEnableFirstOtherButton:(UIAlertView_block_shouldEnableFirstOtherButton)aBlock;

// 延时消失

- (void)showWithDuration:(NSTimeInterval)i;

@end

#undef UIAlertView_key_clicked

#define UIAlertView_key_clicked "UIAlertView.clicked"

#undef UIAlertView_key_cancel

#define UIAlertView_key_cancel "UIAlertView.cancel"

#undef UIAlertView_key_willPresent

#define UIAlertView_key_willPresent "UIAlertView.willPresent"

#undef UIAlertView_key_didPresent

#define UIAlertView_key_didPresent "UIAlertView.didPresent"

#undef UIAlertView_key_willDismiss

#define UIAlertView_key_willDismiss "UIAlertView.willDismiss"

#undef UIAlertView_key_didDismiss

#define UIAlertView_key_didDismiss "UIAlertView.didDismiss"

#undef UIAlertView_key_shouldEnableFirstOtherButton

#define UIAlertView_key_shouldEnableFirstOtherButton "UIAlertView.SEFOB"

@implementation UIAlertView (LL)

- (void)handlerClickedButton:(UIAlertView_block_self_index)aBlock

{

    self.delegate = self;

    objc_setAssociatedObject(self, UIAlertView_key_clicked, aBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);

}

- (void)handlerCancel:(void (^)(UIAlertView *alertView))aBlock

{

    self.delegate = self;

    objc_setAssociatedObject(self, UIAlertView_key_cancel, aBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);

}

- (void)handlerWillPresent:(void (^)(UIAlertView *alertView))aBlock

{

    self.delegate = self;

    objc_setAssociatedObject(self, UIAlertView_key_willPresent, aBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);

}

- (void)handlerDidPresent:(void (^)(UIAlertView *alertView))aBlock

{

    self.delegate = self;

    objc_setAssociatedObject(self, UIAlertView_key_didPresent, aBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);

}

- (void)handlerWillDismiss:(UIAlertView_block_self_index)aBlock

{

    self.delegate = self;

    objc_setAssociatedObject(self, UIAlertView_key_willDismiss, aBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);

}

- (void)handlerDidDismiss:(UIAlertView_block_self_index)aBlock

{

    self.delegate = self;

    objc_setAssociatedObject(self, UIAlertView_key_didDismiss, aBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);

}

- (void)handlerShouldEnableFirstOtherButton:(UIAlertView_block_shouldEnableFirstOtherButton)aBlock

{

    self.delegate = self;

    objc_setAssociatedObject(self, UIAlertView_key_shouldEnableFirstOtherButton, aBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);

}

#pragma mark - UIAlertViewDelegate

- (void)alertView:(UIAlertView *)alertView clickedButtonAtIndex:(NSInteger)buttonIndex

{

    UIAlertView_block_self_index block = objc_getAssociatedObject(self, UIAlertView_key_clicked);

    

    if (block)

        block(alertView, buttonIndex);

}

- (void)alertViewCancel:(UIAlertView *)alertView

{

    UIAlertView_block_self block = objc_getAssociatedObject(self, UIAlertView_key_cancel);

    

    if (block)

        block(alertView);

}

- (void)willPresentAlertView:(UIAlertView *)alertView{

    UIAlertView_block_self block = objc_getAssociatedObject(self, UIAlertView_key_willPresent);

    

    if (block)

        block(alertView);

}

- (void)didPresentAlertView:(UIAlertView *)alertView

{

    UIAlertView_block_self block = objc_getAssociatedObject(self, UIAlertView_key_didPresent);

    

    if (block)

        block(alertView);

}

- (void)alertView:(UIAlertView *)alertView willDismissWithButtonIndex:(NSInteger)buttonIndex

{

    UIAlertView_block_self_index block = objc_getAssociatedObject(self, UIAlertView_key_willDismiss);

    

    if (block)

        block(alertView,buttonIndex);

}

- (void)alertView:(UIAlertView *)alertView didDismissWithButtonIndex:(NSInteger)buttonIndex

{

    UIAlertView_block_self_index block = objc_getAssociatedObject(self, UIAlertView_key_didDismiss);

    

    if (block)

        block(alertView, buttonIndex);

}

- (BOOL)alertViewShouldEnableFirstOtherButton:(UIAlertView *)alertView

{

    UIAlertView_block_shouldEnableFirstOtherButton block = objc_getAssociatedObject(self, UIAlertView_key_shouldEnableFirstOtherButton);

    

    if (block)

        return block(alertView);

    return YES;

}

- (void)showWithDuration:(NSTimeInterval)i

{

    [NSTimer scheduledTimerWithTimeInterval:i

                                     target:self

                                   selector:@selector(llDismiss)

                                   userInfo:self

                                    repeats:NO];

    [self show];

}

- (void)llDismiss

{

    [self dismissWithClickedButtonIndex:0 animated:YES];

}