GradientLoadingBar/Example/GradientLoadingBar/ViewController.swift

//
//  ViewController.swift
//  GradientLoadingBar
//
//  Created by fxm90 on 09/30/2017.
//  Copyright (c) 2017 fxm90. All rights reserved.
//

import GradientLoadingBar
import UIKit

class ViewController: UIViewController {
    @IBOutlet var showButton: UIButton!
    @IBOutlet var toggleButton: UIButton!
    @IBOutlet var hideButton: UIButton!

    @IBOutlet var customSuperviewButton: UIButton!
    @IBOutlet var customContraintsButton: UIButton!

    @IBOutlet var navigationBarButton: UIButton!

    var customSuperviewLoadingBar: GradientLoadingBar?
    var customConstraintsLoadingBar: BottomGradientLoadingBar?
    var navigationLoadingBar: BottomGradientLoadingBar?

    override func viewDidLoad() {
        super.viewDidLoad()
        // Do any additional setup after loading the view, typically from a nib.

        // Style buttons
        let basicButtonList = [showButton, toggleButton, hideButton, navigationBarButton]
        basicButtonList.forEach { (button: UIButton!) in
            button.layer.cornerRadius = 4.0
            button.tintColor = UIColor.white
            button.backgroundColor = UIColor.aquaBlue
        }

        let customButtonList = [customSuperviewButton, customContraintsButton]
        customButtonList.forEach { (button: UIButton!) in
            button.layer.borderColor = UIColor.aquaBlue.cgColor
            button.layer.borderWidth = 1.0
            button.layer.cornerRadius = 4.0

            button.tintColor = UIColor.aquaBlue
            button.backgroundColor = UIColor.clear
        }

        // Add gradient loading bar to button.
        customSuperviewLoadingBar = GradientLoadingBar(onView: customSuperviewButton)
        customSuperviewButton.clipsToBounds = true

        // Add custom gradient loading bar to button.
        customConstraintsLoadingBar = BottomGradientLoadingBar(onView: customContraintsButton)
        customContraintsButton.clipsToBounds = true

        // Add custom gradient loading bar to navigation bar
        if let navigationBar = navigationController?.navigationBar {
            navigationLoadingBar = BottomGradientLoadingBar(onView: navigationBar)
        }
    }

    // MARK: - User actions

    @IBAction func onShowButtonTouchUpInside(_: Any) {
        GradientLoadingBar.shared.show()
    }

    @IBAction func onToggleButtonTouchUpInside(_: Any) {
        GradientLoadingBar.shared.toggle()
    }

    @IBAction func onHideButtonTouchUpInside(_: Any) {
        GradientLoadingBar.shared.hide()
    }

    @IBAction func onCustomSuperviewButtonTouchUpInside(_: Any) {
        guard let customSuperviewLoadingBar = customSuperviewLoadingBar else { return }

        customSuperviewLoadingBar.toggle()
    }

    @IBAction func onCustomContraintsButtonTouchUpInside(_: Any) {
        guard let customConstraintsLoadingBar = customConstraintsLoadingBar else { return }

        customConstraintsLoadingBar.toggle()
    }

    @IBAction func onNavigationBarButtonTouchUpInside(_: Any) {
        guard let navigationLoadingBar = navigationLoadingBar else { return }

        navigationLoadingBar.toggle()
    }
}

// MARK: - Custom Gradient Loading Bar

class BottomGradientLoadingBar: GradientLoadingBar {
    override func setupConstraints() {
        guard let superview = superview else { return }

        NSLayoutConstraint.activate([
            gradientView.leadingAnchor.constraint(equalTo: superview.leadingAnchor),
            gradientView.trailingAnchor.constraint(equalTo: superview.trailingAnchor),

            gradientView.bottomAnchor.constraint(equalTo: superview.bottomAnchor),
            gradientView.heightAnchor.constraint(equalToConstant: CGFloat(height))
        ])
    }
}

// MARK: - Helper

fileprivate extension UIColor {
    static let aquaBlue = UIColor(hex: "#4990e2")
}