Nutrivurv/Nutrivurv/TabBarItemBadgeView.swift
//
// TabBarItemBadgeView.swift
// Nutrivurv
//
// Created by Dillon on 8/5/20.
// Copyright © 2020 Lambda School. All rights reserved.
//
import UIKit
open class TabBarItemBadgeView: UIView {
public static var defaultBadgeColor = UIColor(red: 255.0/255.0, green: 59.0/255.0, blue: 48.0/255.0, alpha: 1.0)
/// Badge color
open var badgeColor: UIColor? = defaultBadgeColor {
didSet {
imageView.backgroundColor = badgeColor
}
}
/// Badge value, supprot nil, "", "1", "someText". Hidden when nil. Show Little dot style when "".
open var badgeValue: String? {
didSet {
badgeLabel.text = badgeValue
}
}
open var imageView: UIImageView = {
let imageView = UIImageView.init(frame: CGRect.zero)
imageView.backgroundColor = .clear
return imageView
}()
open var badgeLabel: UILabel = {
let badgeLabel = UILabel.init(frame: CGRect.zero)
badgeLabel.backgroundColor = .clear
badgeLabel.textColor = .white
badgeLabel.font = UIFont.systemFont(ofSize: 13.0)
badgeLabel.textAlignment = .center
return badgeLabel
}()
public override init(frame: CGRect) {
super.init(frame: frame)
self.addSubview(imageView)
self.addSubview(badgeLabel)
self.imageView.backgroundColor = badgeColor
}
public required init?(coder aDecoder: NSCoder) {
fatalError("init(coder:) has not been implemented")
}
open override func layoutSubviews() {
super.layoutSubviews()
guard let badgeValue = badgeValue else {
imageView.isHidden = true
badgeLabel.isHidden = true
return
}
imageView.isHidden = false
badgeLabel.isHidden = false
if badgeValue == "" {
imageView.frame = CGRect.init(origin: CGPoint.init(x: (bounds.size.width - 8.0) / 2.0, y: (bounds.size.height - 8.0) / 2.0), size: CGSize.init(width: 8.0, height: 8.0))
} else {
imageView.frame = bounds
}
imageView.layer.cornerRadius = imageView.bounds.size.height / 2.0
badgeLabel.sizeToFit()
badgeLabel.center = imageView.center
}
open override func sizeThatFits(_ size: CGSize) -> CGSize {
guard let _ = badgeValue else {
return CGSize.init(width: 18.0, height: 18.0)
}
let textSize = badgeLabel.sizeThatFits(CGSize.init(width: CGFloat.greatestFiniteMagnitude, height: CGFloat.greatestFiniteMagnitude))
return CGSize.init(width: max(18.0, textSize.width + 10.0), height: 18.0)
}
}