//
//  UIButton+BadgeView.m
//  ThePaperHD
//
//  Created by scar1900 on 15/3/29.
//  Copyright (c) 2015年 scar1900. All rights reserved.
//

#import "UIButton+BadgeView.h"

#define badgeTag 5000

@implementation UIButton (BadgeView)
- (void)addBadgeView:(UIEdgeInsets)inset {
    UIView *badgeView = [self viewWithTag:badgeTag];
    if (badgeView) {
        [badgeView removeFromSuperview];
        badgeView = nil;
    }
    
    CGSize imageSize = self.currentImage.size;
    
    badgeView = [[UIView alloc]initWithFrame:CGRectMake(inset.left+imageSize.width/2+5,inset.top-2.5, 10, 10)];
    
    badgeView.backgroundColor = [UIColor colorWithHexString:@"0xc92228"];
    badgeView.layer.cornerRadius = 5;
    badgeView.tag = badgeTag;
    [UIView animateWithDuration:0.25 animations:^{
        [self addSubview:badgeView];
    }];
}

- (void)addBadgeView {
    UIView *badgeView = [self viewWithTag:badgeTag];
    if (badgeView) {
        [badgeView removeFromSuperview];
        badgeView = nil;
    }
    
    UIEdgeInsets inset = self.imageEdgeInsets;
    CGSize imageSize = self.currentImage.size;
    
    badgeView = [[UIView alloc]initWithFrame:CGRectMake(inset.left+imageSize.width/2+5,inset.top-2.5, 10, 10)];
    
    badgeView.backgroundColor = [UIColor colorWithHexString:@"0xc92228"];
    badgeView.layer.cornerRadius = 5;
    badgeView.tag = badgeTag;
    [UIView animateWithDuration:0.25 animations:^{
        [self addSubview:badgeView];
    }];
}

- (void)clearBadgeView {
    UIView *badgeView = [self viewWithTag:badgeTag];
    __block UIView *view = badgeView;
    if (badgeView) {
        [UIView animateWithDuration:0.25 animations:^{
            [view removeFromSuperview];
        }completion:^(BOOL finished) {
            view = nil;
        }];
    }
}


@end