alpha-wallet-ios/Trust/UI/BalanceTitleView.swift

// Copyright SIX DAY LLC. All rights reserved.

import UIKit

enum BalanceMode {
    case short
    case full
}

class BalanceTitleView: UIView {

    let titleLabel = UILabel()
    let subTitleLabel = UILabel()
    var viewModel: BalanceBaseViewModel? {
        didSet {
            guard let viewModel = viewModel else { return }
            configure(viewModel: viewModel)
        }
    }
    private var mode = BalanceMode.short

    override init(frame: CGRect) {
        super.init(frame: .zero)

        titleLabel.translatesAutoresizingMaskIntoConstraints = false
        titleLabel.textAlignment = .center
        titleLabel.adjustsFontSizeToFitWidth = true

        subTitleLabel.translatesAutoresizingMaskIntoConstraints = false
        subTitleLabel.textAlignment = .center
        subTitleLabel.adjustsFontSizeToFitWidth = true

        let stackView = UIStackView(arrangedSubviews: [
            titleLabel,
            subTitleLabel,
        ])
        stackView.translatesAutoresizingMaskIntoConstraints = false
        stackView.axis = .vertical
        stackView.spacing = 2
        addSubview(stackView)

        NSLayoutConstraint.activate([
            stackView.trailingAnchor.constraint(equalTo: trailingAnchor),
            stackView.leadingAnchor.constraint(equalTo: leadingAnchor),
            stackView.topAnchor.constraint(equalTo: topAnchor),
            stackView.bottomAnchor.constraint(equalTo: bottomAnchor),
        ])

        stackView.addGestureRecognizer(
            UITapGestureRecognizer(target: self, action: #selector(self.switchModel))
        )
    }

    private func configure(viewModel: BalanceBaseViewModel) {
        titleLabel.font = UIFont.systemFont(ofSize: 18, weight: UIFont.Weight.semibold)
        titleLabel.textColor = Colors.lightBlack

        subTitleLabel.font = UIFont.systemFont(ofSize: 14, weight: UIFont.Weight.regular)
        subTitleLabel.textColor = Colors.darkGray

        let amount: String
        switch mode {
        case .full:
            amount = viewModel.amountFull
        case .short:
            amount = viewModel.amountShort
        }

        titleLabel.text = "\(amount) \(viewModel.symbol)"
        subTitleLabel.text = viewModel.currencyAmount
    }

    @objc func switchModel() {
        switch mode {
        case .full:
            mode = .short
        case .short:
            mode = .full
        }

        guard let viewModel = viewModel else { return }
        configure(viewModel: viewModel)
    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
}

extension BalanceTitleView {
    static func make(from session: WalletSession, _ transferType: TransferType) -> BalanceTitleView {
        let view = BalanceTitleView(frame: .zero)
        view.translatesAutoresizingMaskIntoConstraints = false
        switch transferType {
        case .ether:
            session.balanceViewModel.subscribe { viewModel in
                guard let viewModel = viewModel else { return }
                view.viewModel = viewModel
            }
        case .token(let token):
            view.viewModel = BalanceTokenViewModel(token: token)
        case .exchange:
            break //
        }
        return view
    }
}