//
//  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 {
    UIView *badgeView = [self viewWithTag:badgeTag];
    if (badgeView) {
        [badgeView removeFromSuperview];
        badgeView = nil;
    }
    
    badgeView = [[UIView alloc]initWithFrame:CGRectMake(CGRectGetWidth(self.bounds)-5,-5, 10, 10)];
    badgeView.backgroundColor = [UIColor colorWithHexString:@"0xc92228"];
    badgeView.layer.cornerRadius = 5;
    badgeView.tag = badgeTag;
    [self addSubview:badgeView];
}

- (void)clearBadgeView {
    UIView *badgeView = [self viewWithTag:badgeTag];
    if (badgeView) {
        [badgeView removeFromSuperview];
        badgeView = nil;
    }
}


@end