stanwood/Stanwood_Debugger_iOS

//
//  FilterUIButton.swift
//
//  The MIT License (MIT)
//
//  Copyright (c) 2018 Stanwood GmbH (www.stanwood.io)
//
//  Permission is hereby granted, free of charge, to any person obtaining a copy
//  of this software and associated documentation files (the "Software"), to deal
//  in the Software without restriction, including without limitation the rights
//  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
//  copies of the Software, and to permit persons to whom the Software is
//  furnished to do so, subject to the following conditions:
//
//  The above copyright notice and this permission notice shall be included in
//  all copies or substantial portions of the Software.
//
//  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
//  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
//  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
//  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
//  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
//  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
//  THE SOFTWARE.
//

import Foundation
import UIKit

class FilterUIButton: UIButton {
    
    private(set) var filter: DebuggerFilterView.DebuggerFilter?
    private var badgeLabel: UILabel!
    private var badgeWidthAnchor: NSLayoutConstraint!
    private var badgeLeadingAnchor: NSLayoutConstraint!
    
    override func awakeFromNib() {
        super.awakeFromNib()
        
        badgeLabel = UILabel(frame: CGRect(x: 0, y: 0, width: 20, height: 20))
        badgeLabel.layer.masksToBounds = true
        badgeLabel.layer.borderWidth = 1
        badgeLabel.layer.borderColor = UIColor.black.withAlphaComponent(0.15).cgColor
        
        badgeLabel.translatesAutoresizingMaskIntoConstraints = false
        badgeLabel.textAlignment = .center
        badgeLabel.font = UIFont.systemFont(ofSize: 13)
        badgeLabel.text = "0"
        badgeLabel.isHidden = true
        
        translatesAutoresizingMaskIntoConstraints = false
        superview?.addSubview(badgeLabel)
        
        badgeWidthAnchor = badgeLabel.widthAnchor.constraint(equalToConstant: 20)
        badgeLeadingAnchor = badgeLabel.leadingAnchor.constraint(equalTo: trailingAnchor, constant: -18)
        NSLayoutConstraint.activate([
            badgeLeadingAnchor,
            badgeLabel.topAnchor.constraint(equalTo: topAnchor, constant: -5),
            badgeLabel.heightAnchor.constraint(equalToConstant: 20),
            badgeWidthAnchor
            ])
        
        badgeLabel.layer.cornerRadius = badgeLabel.frame.width / 2
        
        NotificationCenter.default.addObservers(self, observers: Observer(selector: #selector(didAddNewItem(_:)), name: Notification.Name.DebuggerDidAddDebuggerItem))
    }
    
    func setTitle(_ filter: DebuggerFilterView.DebuggerFilter?) {
        self.filter = filter
        [UIControl.State.normal, UIControl.State.highlighted].forEach({ setTitle(filter?.label, for: $0) })
    }
    
    func set(selected: Bool) {
        switch selected {
        case true:
            setTitleColor(.white, for: .normal)
            backgroundColor = Harlow.Style.tintColor
            badgeLabel.backgroundColor = .white
            badgeLabel.textColor = Harlow.Style.tintColor
        case false:
            setTitleColor(Harlow.Style.tintColor, for: .normal)
            backgroundColor = .white
            badgeLabel.backgroundColor = Harlow.Style.tintColor
            badgeLabel.textColor = .white
        }
    }
    
    @objc private func didAddNewItem(_ notification: Notification) {
        
        guard let addedItems = notification.object as? [AddedItem],
            let addedItem = addedItems.filter({ $0.type.label == self.filter?.label }).first else { return }
        
        UIView.animate(withDuration: 0.3,  animations: {
            self.badgeLabel.transform = CGAffineTransform(scaleX: 0.01, y: 0.01)
        }, completion: { _ in
            
            let width: CGFloat = addedItem.count < 10 ? 20 : addedItem.count < 100 ? 25 : 30
            self.badgeWidthAnchor.constant = width
            let leading: CGFloat = addedItem.count > 99 ? -23 : -18
            self.badgeLeadingAnchor.constant = leading
            
            UIView.performWithoutAnimation {
                self.badgeLabel.layoutIfNeeded()
            }
            self.badgeLabel.text = "\(addedItem.count)"
            self.badgeLabel.isHidden = addedItem.count == 0
            
            UIView.animate(withDuration: 0.4, delay: 0, usingSpringWithDamping: 0.3, initialSpringVelocity: 0, options: .curveEaseOut, animations: {
                self.badgeLabel.transform = .identity
            }, completion: nil)
            
        })
    }
}