//
//  IBDesginAddition.swift
//  PaiAi
//
//  Created by mac on 16/7/21.
//  Copyright © 2016年 FFIB. All rights reserved.
//

import UIKit

var badgeLabelKey = "badgeLabelKey"
extension UIView {
    var badgeLabel: UILabel {
        get {
            return associatedObject(key: &badgeLabelKey, initialiser: { () -> UILabel in
                return UILabel()
            })
        }

        set {
            associateObject(key: &badgeLabelKey, value: newValue)
        }
    }

    func setBageLabel(badge: Int, badgeColor: UIColor) {
        self.subviews.filter {$0.isMember(of: UILabel.classForCoder())}.forEach {$0.removeFromSuperview()}
        let digits = badge < 10 ? 12 : 20
        badgeLabel = UILabel(frame: CGRect(x: 0, y: 0, width: digits, height: 12))
        badgeLabel.text = "\(badge)"
        self.clipsToBounds = false
        badgeLabel.font = UIFont.systemFont(ofSize: 12)
        badgeLabel.backgroundColor = badgeColor
        badgeLabel.textColor = UIColor.white
        badgeLabel.textAlignment = .center
        badgeLabel.cornerRadius = 6
        self.addSubview(badgeLabel)
    }

    func updateBageLabel(badge: Int) {
        badgeLabel.text = "\(badge)"
    }
}