/*
 * Copyright (C) 2015 - 2017, Daniel Dahan and CosmicMind, Inc. <http://cosmicmind.com>.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *	*	Redistributions of source code must retain the above copyright notice, this
 *		list of conditions and the following disclaimer.
 *
 *	*	Redistributions in binary form must reproduce the above copyright notice,
 *		this list of conditions and the following disclaimer in the documentation
 *		and/or other materials provided with the distribution.
 *
 *	*	Neither the name of CosmicMind nor the names of its
 *		contributors may be used to endorse or promote products derived from
 *		this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

import UIKit

@objc(TabBarLineAlignment)
public enum TabBarLineAlignment: Int {
	case top
	case bottom
}

@objc(TabBarDelegate)
public protocol TabBarDelegate {
    /**
     A delegation method that is executed when the button will trigger the
     animation to the next tab.
     - Parameter tabBar: A TabBar.
     - Parameter button: A UIButton.
     */
    @objc
    optional func tabBar(tabBar: TabBar, willSelect button: UIButton)
    
    /**
     A delegation method that is executed when the button did complete the
     animation to the next tab.
     - Parameter tabBar: A TabBar.
     - Parameter button: A UIButton.
     */
    @objc
    optional func tabBar(tabBar: TabBar, didSelect button: UIButton)
}

@objc(TabBarStyle)
public enum TabBarStyle: Int {
    case normal
    case scrollable
}

open class TabBar: Bar {
    /// A boolean indicating if the TabBar line is in an animation state.
    open fileprivate(set) var isAnimating = false
    
    /// Enables and disables bouncing when swiping.
    open var isBounceEnabled: Bool {
        get {
            return contentView.bounces
        }
        set(value) {
            contentView.bounces = value
        }
    }
    
    /// An enum that determines the tab bar style.
    open var tabBarStyle = TabBarStyle.scrollable {
        didSet {
            layoutSubviews()
        }
    }
    
    /// A delegation reference.
    open weak var delegate: TabBarDelegate?
    
    /// The currently selected button.
    open fileprivate(set) var selected: UIButton?
    
    /// A preset wrapper around contentEdgeInsets.
    open override var contentEdgeInsetsPreset: EdgeInsetsPreset {
        get {
            return contentView.grid.contentEdgeInsetsPreset
        }
        set(value) {
            contentView.grid.contentEdgeInsetsPreset = value
        }
    }
    
    /// A reference to EdgeInsets.
    @IBInspectable
    open override var contentEdgeInsets: EdgeInsets {
        get {
            return contentView.grid.contentEdgeInsets
        }
        set(value) {
            contentView.grid.contentEdgeInsets = value
        }
    }
    
    /// A preset wrapper around interimSpace.
    open override var interimSpacePreset: InterimSpacePreset {
        get {
            return contentView.grid.interimSpacePreset
        }
        set(value) {
            contentView.grid.interimSpacePreset = value
        }
    }
    
    /// A wrapper around contentView.grid.interimSpace.
    @IBInspectable
    open override var interimSpace: InterimSpace {
        get {
            return contentView.grid.interimSpace
        }
        set(value) {
            contentView.grid.interimSpace = value
        }
    }
    
	/// Buttons.
	open var buttons = [UIButton]() {
		didSet {
			for b in oldValue {
                b.removeFromSuperview()
            }
			
			layoutSubviews()
		}
	}
    
    /// A boolean to animate the line when touched.
    @IBInspectable
    open var isLineAnimated = true {
        didSet {
            for b in buttons {
                if isLineAnimated {
                    prepareLineAnimationHandler(button: b)
                } else {
                    removeLineAnimationHandler(button: b)
                }
            }
        }
    }
    
    /// A reference to the line UIView.
    open let line = UIView()
    
    /// The line color.
    open var lineColor: UIColor? {
        get {
            return line.backgroundColor
        }
        set(value) {
            line.backgroundColor = value
        }
    }
    
    /// A value for the line alignment.
    open var lineAlignment = TabBarLineAlignment.bottom {
        didSet {
            layoutSubviews()
        }
    }
    
    /// The line height.
    open var lineHeight: CGFloat {
        get {
            return line.height
        }
        set(value) {
            line.height = value
        }
    }
    
    open override func layoutSubviews() {
		super.layoutSubviews()
        guard willLayout else {
            return
        }
        
        guard 0 < buttons.count else {
            return
        }
            
        for b in buttons {
            b.grid.columns = 0
            b.cornerRadius = 0
            b.contentEdgeInsets = .zero
            
            if isLineAnimated {
                prepareLineAnimationHandler(button: b)
            }
        }
        
        if .scrollable == tabBarStyle {
            var w: CGFloat = 0
            
            for b in buttons {
                let width = b.sizeThatFits(CGSize(width: CGFloat.greatestFiniteMagnitude, height: contentView.height)).width + interimSpace
                contentView.addSubview(b)
                b.height = contentView.height
                b.width = width
                b.x = w
                w += width
            }
            
            if w > contentView.width {
                contentView.contentSize.width = w
            }
        } else {
            contentView.grid.axis.columns = buttons.count
            centerViews = buttons
        }
        
        if nil == selected {
            selected = buttons.first
        }
            
        line.frame = CGRect(x: selected!.x, y: .bottom == lineAlignment ? height - lineHeight : 0, width: selected!.width, height: lineHeight)
	}
	
    /**
     Prepares the view instance when intialized. When subclassing,
     it is recommended to override the prepare method
     to initialize property values and other setup operations.
     The super.prepare method should always be called immediately
     when subclassing.
     */
    open override func prepare() {
        super.prepare()
        isBounceEnabled = true
        contentEdgeInsetsPreset = .none
        interimSpacePreset = .interimSpace5
        prepareLine()
        prepareDivider()
    }
}

extension TabBar {
    // Prepares the line.
    fileprivate func prepareLine() {
        line.zPosition = 6000
        lineColor = Color.blue.base
        lineHeight = 3
        addSubview(line)
    }
    
    /// Prepares the divider.
    fileprivate func prepareDivider() {
        dividerAlignment = .top
    }
    
    /**
     Prepares the line animation handlers.
     - Parameter button: A UIButton.
     */
    fileprivate func prepareLineAnimationHandler(button: UIButton) {
        removeLineAnimationHandler(button: button)
        button.addTarget(self, action: #selector(handleLineAnimation(button:)), for: .touchUpInside)
    }
    
    /**
     Removes the line animation handlers.
     - Parameter button: A UIButton.
     */
    fileprivate func removeLineAnimationHandler(button: UIButton) {
        button.removeTarget(self, action: #selector(handleLineAnimation(button:)), for: .touchUpInside)
    }
}

extension TabBar {
    /// Handles the button touch event.
    @objc
    fileprivate func handleLineAnimation(button: UIButton) {
        animate(to: button, isTriggeredByUserInteraction: true)
    }
}

extension TabBar {
    /**
     Selects a given index from the buttons array.
     - Parameter at index: An Int.
     - Paramater completion: An optional completion block.
     */
    open func select(at index: Int, completion: ((UIButton) -> Void)? = nil) {
        guard -1 < index, index < buttons.count else {
            return
        }
        animate(to: buttons[index], isTriggeredByUserInteraction: false, completion: completion)
    }
    
    /**
     Animates to a given button.
     - Parameter to button: A UIButton.
     - Parameter completion: An optional completion block.
     */
    open func animate(to button: UIButton, completion: ((UIButton) -> Void)? = nil) {
        animate(to: button, isTriggeredByUserInteraction: false, completion: completion)
    }
    
    /**
     Animates to a given button.
     - Parameter to button: A UIButton.
     - Parameter isTriggeredByUserInteraction: A boolean indicating whether the
     state was changed by a user interaction, true if yes, false otherwise.
     - Parameter completion: An optional completion block.
     */
    fileprivate func animate(to button: UIButton, isTriggeredByUserInteraction: Bool, completion: ((UIButton) -> Void)? = nil) {
        if isTriggeredByUserInteraction {
            delegate?.tabBar?(tabBar: self, willSelect: button)
        }
        
        selected = button
        isAnimating = true
        
        UIView.animate(withDuration: 0.25, animations: { [weak self, button = button] in
            guard let s = self else {
                return
            }
            
            s.line.center.x = button.center.x
            s.line.width = button.width
        }) { [weak self, isTriggeredByUserInteraction = isTriggeredByUserInteraction, button = button, completion = completion] _ in
            guard let s = self else {
                return
            }
            
            s.isAnimating = false
            
            if isTriggeredByUserInteraction {
                s.delegate?.tabBar?(tabBar: s, didSelect: button)
            }
            
            completion?(button)
        }
    }
}
