Attractor/SBrowser/Titlebar and URL Bar/ToolbarViewController.swift

//
//  ToolbarViewController.swift
//  SBrowser
//
//  Created by James Magahern on 7/23/20.
//

import UIKit

class ToolbarButtonView: UIView
{
    private var buttonPadding = CGFloat(24.0)
    private var buttonViews: [UIView] = []
    
    func addButtonView(_ button: UIView) {
        buttonViews.append(button)
        addSubview(button)
        setNeedsLayout()
    }
    
    func removeAllButtonViews() {
        buttonViews.forEach { $0.removeFromSuperview() }
        buttonViews.removeAll()
        setNeedsLayout()
    }
    
    override func sizeThatFits(_ size: CGSize) -> CGSize {
        let width: CGFloat = buttonViews.reduce(0.0) { (result, button) -> CGFloat in
            return result + button.sizeThatFits(size).width + buttonPadding
        }
        
        return CGSize(width: width, height: size.height)
    }
    
    override func layoutSubviews() {
        var buttonRect = CGRect(origin: .zero, size: CGSize(width: 0, height: bounds.height))
        buttonRect.origin.x = layoutMargins.left
        
        for button in buttonViews {
            let buttonSize = button.sizeThatFits(bounds.size)
            buttonRect.size = CGSize(width: buttonSize.width, height: bounds.height)
            button.frame = buttonRect
            
            buttonRect.origin.x += buttonRect.width + buttonPadding
        }
    }
}

class ToolbarView: UIView
{
    var urlBar: URLBar? { didSet { containerView.addSubview(urlBar!) } }
    
    var cancelButtonVisible: Bool = false { didSet { layoutSubviews() } }
    
    let containerView = UIView(frame: .zero)
    let backgroundView = UIVisualEffectView(effect: UIBlurEffect(style: .systemThickMaterial))
    let buttonsView = ToolbarButtonView(frame: .zero)
    let cancelButton = UIButton(type: .system)
    
    convenience init()
    {
        self.init(frame: .zero)
        addSubview(backgroundView)
        addSubview(containerView)
        
        containerView.addSubview(buttonsView)
        
        cancelButton.setTitle("Cancel", for: .normal)
        containerView.addSubview(cancelButton)
        
        layer.masksToBounds = false
        layer.shadowColor = UIColor.black.cgColor
        layer.shadowOpacity = 0.2
        layer.shadowOffset = CGSize(width: 0.0, height: 1.0)
        layer.shadowRadius = 1.5
    }
    
    override func sizeThatFits(_ size: CGSize) -> CGSize
    {
        return CGSize(width: size.width, height: 44.0)
    }
    
    override func layoutSubviews()
    {
        super.layoutSubviews()

        backgroundView.frame = bounds
        
        var containerBounds = bounds
        containerBounds.size.height -= safeAreaInsets.bottom
        containerView.frame = containerBounds
        containerView.frame = containerView.frame.insetBy(dx: 8.0, dy: 4.0)
        
        // Cancel button
        let urlBarPadding: CGFloat = 8.0
        let cancelButtonSize = cancelButton.sizeThatFits(containerView.bounds.size)
        cancelButton.frame = CGRect(origin: CGPoint(x: (containerView.bounds.maxX - cancelButtonSize.width), y: 0),
                                    size: CGSize(width: cancelButtonSize.width, height: containerView.bounds.height))
        
        // Toolbar buttons
        let buttonContainerSize = buttonsView.sizeThatFits(containerView.bounds.size)
        buttonsView.frame = CGRect(origin: CGPoint(x: (containerView.bounds.maxX - buttonContainerSize.width) + urlBarPadding, y: 0), size: buttonContainerSize)
        
        var avoidingSize: CGSize = .zero
        if cancelButtonVisible {
            cancelButton.alpha = 1.0
            buttonsView.alpha = 0.0
            
            avoidingSize = cancelButtonSize
        } else {
            cancelButton.alpha = 0.0
            buttonsView.alpha = 1.0
            
            avoidingSize = buttonContainerSize
        }
        
        if let urlBar = urlBar {
            urlBar.frame = CGRect(origin: .zero, size: CGSize(width: containerView.bounds.width - avoidingSize.width - urlBarPadding, height: containerView.bounds.height))
        }
    }
}

class ToolbarViewController: UIViewController
{
    let urlBar = URLBar()
    let toolbarView = ToolbarView()
    let scriptControllerIconView = ScriptControllerIconView()
    let shareButton = UIButton(frame: .zero)
    let darkModeButton = UIButton(frame: .zero)
    let windowButton = UIButton(frame: .zero)
    
    var darkModeEnabled: Bool = false {
        didSet {
            if darkModeEnabled {
                darkModeButton.setImage(darkModeEnabledImage, for: .normal)
            } else {
                darkModeButton.setImage(darkModeDisabledImage, for: .normal)
            }
        }
    }
    
    private let darkModeDisabledImage = UIImage(systemName: "moon.circle")
    private let darkModeEnabledImage = UIImage(systemName: "moon.circle.fill")
    
    init() {
        super.init(nibName: nil, bundle: nil)
        
        toolbarView.urlBar = urlBar
        
        // Dark mode button
        darkModeButton.setImage(darkModeDisabledImage, for: .normal)
        
        // Share button
        shareButton.setImage(UIImage(systemName: "square.and.arrow.up"), for: .normal)
        
        // Window button
        windowButton.setImage(UIImage(systemName: "rectangle.on.rectangle"), for: .normal)
        
        let toolbarAnimationDuration: TimeInterval = 0.3
        urlBar.textField.addAction(.init(handler: { [weak self] _ in
            guard let self = self else { return }
            UIView.animate(withDuration: toolbarAnimationDuration) {
                self.toolbarView.cancelButtonVisible = self.urlBar.textField.isFirstResponder
            }
        }), for: [ .editingDidBegin, .editingDidEnd ])
        
        toolbarView.cancelButton.addAction(.init(handler: { [weak self] action in
            self?.urlBar.textField.resignFirstResponder()
        }), for: .touchUpInside)
        
        traitCollectionDidChange(nil)
    }
    
    override func traitCollectionDidChange(_ previousTraitCollection: UITraitCollection?) {
        super.traitCollectionDidChange(previousTraitCollection)
        
        toolbarView.buttonsView.removeAllButtonViews()
        
        // Setup toolbar based on trait collection
        if traitCollection.horizontalSizeClass == .compact {
            toolbarView.buttonsView.addButtonView(darkModeButton)
            toolbarView.buttonsView.addButtonView(scriptControllerIconView)
            toolbarView.buttonsView.addButtonView(windowButton)
        } else {
            toolbarView.buttonsView.addButtonView(darkModeButton)
            toolbarView.buttonsView.addButtonView(shareButton)
            toolbarView.buttonsView.addButtonView(scriptControllerIconView)
            toolbarView.buttonsView.addButtonView(windowButton)
        }
    }
    
    override func loadView() {
        self.view = toolbarView
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
}