mastodon-ios/Mastodon/Scene/Account/View/BadgeButton.swift

//
//  BadgeButton.swift
//  Mastodon
//
//  Created by Cirno MainasuK on 2021-9-16.
//

import UIKit
import MastodonAsset
import MastodonLocalization

final class BadgeButton: UIButton {
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        _init()
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
        _init()
    }
    
}

extension BadgeButton {
    private func _init() {
        titleLabel?.font = UIFontMetrics(forTextStyle: .caption1).scaledFont(for: .systemFont(ofSize: 13, weight: .medium))
        contentEdgeInsets = UIEdgeInsets(top: 6, left: 6, bottom: 6, right: 6)
        setAppearance()
    }
    
    override func traitCollectionDidChange(_ previousTraitCollection: UITraitCollection?) {
        super.traitCollectionDidChange(previousTraitCollection)
        
        setAppearance()
    }
    
    override func layoutSubviews() {
        super.layoutSubviews()
        
        layer.masksToBounds = true
        layer.cornerRadius = frame.height * 0.5
    }
    
    private func setAppearance() {
        setBackgroundColor(Asset.Colors.Label.primary.color, for: .normal)
        setTitleColor(Asset.Colors.Label.primaryReverse.color, for: .normal)
        tintColor = Asset.Colors.Label.primary.color
    }
    
    func setBadge(number: Int) {
        let number = min(99, max(0, number))
        setTitle("\(number)", for: .normal)
        self.isHidden = number == 0
        accessibilityLabel = L10n.A11y.Plural.Count.Unread.notification(number)
    }
}