Badge.swift

@IBDesignable class Badge: UILabel {
    @IBInspectable var topInset: CGFloat = 5.0
    @IBInspectable var bottomInset: CGFloat = 5.0
    @IBInspectable var leftInset: CGFloat = 7.0
    @IBInspectable var rightInset: CGFloat = 7.0
    @IBInspectable var round: CGFloat = 7.0 {
        didSet {
            layer.cornerRadius = round
            clipsToBounds = true
        }
    }
    @IBInspectable
    var type: Int = 0 {
        didSet {
            _type = BadgeType(rawValue: type) ?? .primary
        }
    }
    
    var _type: BadgeType = .primary {
        didSet {
            setup()
        }
    }

    override func drawText(in rect: CGRect) {
        let insets = UIEdgeInsets(top: topInset, left: leftInset, bottom: bottomInset, right: rightInset)
        super.drawText(in: rect.inset(by: insets))
    }

    override var intrinsicContentSize: CGSize {
        get {
            var contentSize = super.intrinsicContentSize
            contentSize.height += topInset + bottomInset
            contentSize.width += leftInset + rightInset
            return contentSize
        }
    }
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        
        setup()
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
        
        setup()
    }
    
    func setup() {
        layer.cornerRadius = round
        clipsToBounds = true
        textColor = _type.textColor
        backgroundColor = _type.backgroundColor
        font = UIFont.systemFont(ofSize: 12, weight: .bold)
    }
}

BadgeType.swift

enum BadgeType: Int {
    case primary
    case secondary
    case success
    case danger
    case warning
    case info
    
    var backgroundColor: UIColor {
        switch self {
        case .primary:
            return .systemBlue
        case .secondary:
            return .lightGray
        case .success:
            return .systemGreen
        case .danger:
            return .systemRed
        case .warning:
            return .systemYellow
        case .info:
            return .systemTeal
            
        }
    }
    
    var textColor: UIColor {
        switch self {
        case .warning:
            return .darkGray
        default:
            return .white
        }
    }
}