XLPagerTabStrip/Sources/TwitterPagerTabStripViewController.swift

//  TwitterPagerTabStripViewController.swift
//  XLPagerTabStrip ( https://github.com/xmartlabs/XLPagerTabStrip )
//
//  Copyright (c) 2016 Xmartlabs ( http://xmartlabs.com )
//
//
// 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

public struct TwitterPagerTabStripSettings {
    
    public struct Style {
        public var dotColor = UIColor(white: 1, alpha: 0.4)
        public var selectedDotColor = UIColor.whiteColor()
        public var portraitTitleFont = UIFont.systemFontOfSize(18)
        public var landscapeTitleFont = UIFont.systemFontOfSize(15)
        public var titleColor = UIColor.whiteColor()
    }
    
    public var style = Style()
}

public class TwitterPagerTabStripViewController: PagerTabStripViewController, PagerTabStripDataSource, PagerTabStripIsProgressiveDelegate {
    
    public var settings = TwitterPagerTabStripSettings()
    
    required public init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)
        delegate = self
        datasource = self
    }
    
    public override func viewDidLoad() {
        super.viewDidLoad()
        
        if titleView.superview == nil {
            navigationItem.titleView = titleView
        }
        
        // keep watching the frame of titleView
        titleView.addObserver(self, forKeyPath: "frame", options: [.New, .Old], context: nil)
        
        guard let navigationController = navigationController  else {
            fatalError("TwitterPagerTabStripViewController should be embedded in a UINavigationController")
        }
        titleView.frame = CGRectMake(0, 0, CGRectGetWidth(navigationController.navigationBar.frame), CGRectGetHeight(navigationController.navigationBar.frame))
        titleView.addSubview(titleScrollView)
        titleView.addSubview(pageControl)
        reloadNavigationViewItems()
    }
    
    public override func reloadPagerTabStripView() {
        super.reloadPagerTabStripView()
        guard isViewLoaded() else { return }
        
        reloadNavigationViewItems()
        setNavigationViewItemsPosition(updateAlpha: true)
    }
    
    // MARK: - PagerTabStripDelegate
    
    public func pagerTabStripViewController(pagerTabStripViewController: PagerTabStripViewController, updateIndicatorFromIndex fromIndex: Int, toIndex: Int) throws {
        fatalError()
    }
    
    public func pagerTabStripViewController(pagerTabStripViewController: PagerTabStripViewController, updateIndicatorFromIndex fromIndex: Int, toIndex: Int, withProgressPercentage progressPercentage: CGFloat, indexWasChanged: Bool) throws {
        
        // move indicator scroll view
        let distance = distanceValue
        var xOffset: CGFloat = 0
        if fromIndex < toIndex {
            xOffset = distance * CGFloat(fromIndex) + distance * progressPercentage
        }
        else if fromIndex > toIndex {
            xOffset = distance * CGFloat(fromIndex) - distance * progressPercentage
        }
        else {
            xOffset = distance * CGFloat(fromIndex)
        }
        
        titleScrollView.contentOffset = CGPointMake(xOffset, 0)
        
        // update alpha of titles
        setAlphaWithOffset(xOffset, andDistance: distance)
        
        // update page control page
        pageControl.currentPage = currentIndex
    }
    
    public override func observeValueForKeyPath(keyPath: String?, ofObject object: AnyObject?, change: [String : AnyObject]?, context: UnsafeMutablePointer<Void>) {
        guard object === titleView && keyPath == "frame" && change?[NSKeyValueChangeKindKey] as? UInt == NSKeyValueChange.Setting.rawValue else { return }
        let oldRect = change![NSKeyValueChangeOldKey]!.CGRectValue
        let newRect = change![NSKeyValueChangeOldKey]!.CGRectValue
        if CGRectEqualToRect(oldRect, newRect) {
            titleScrollView.frame = CGRectMake(0, 0, CGRectGetWidth(titleView.frame), CGRectGetHeight(titleScrollView.frame))
            setNavigationViewItemsPosition(updateAlpha: true)
        }
    }
    
    deinit {
        titleView.removeObserver(self, forKeyPath: "frame")
    }
    
    public override func viewDidLayoutSubviews() {
        super.viewDidLayoutSubviews()
        setNavigationViewItemsPosition(updateAlpha: false)
    }
    
    // MARK: - Helpers
    
    private lazy var titleView: UIView = { [unowned self] in
        let navigationView = UIView()
        navigationView.autoresizingMask = [.FlexibleWidth, .FlexibleHeight]
        return navigationView
        }()
    
    private lazy var titleScrollView: UIScrollView = { [unowned self] in
        let titleScrollView = UIScrollView(frame: CGRectMake(0, 0, CGRectGetWidth(self.view.bounds), 44))
        titleScrollView.autoresizingMask = [.FlexibleWidth, .FlexibleHeight]
        titleScrollView.bounces = true
        titleScrollView.scrollsToTop = false
        titleScrollView.delegate = self
        titleScrollView.showsVerticalScrollIndicator = false
        titleScrollView.showsHorizontalScrollIndicator = false
        titleScrollView.pagingEnabled = true
        titleScrollView.userInteractionEnabled = false
        titleScrollView.alwaysBounceHorizontal = true
        titleScrollView.alwaysBounceVertical = false
        return titleScrollView
    }()
    
    private lazy var pageControl: FXPageControl = { [unowned self] in
        let pageControl = FXPageControl()
        pageControl.backgroundColor = .clearColor()
        pageControl.dotSize = 3.8
        pageControl.dotSpacing = 4.0
        pageControl.dotColor = self.settings.style.dotColor
        pageControl.selectedDotColor = self.settings.style.selectedDotColor
        pageControl.userInteractionEnabled = false
        return pageControl
    }()
    
    private var childTitleLabels = [UILabel]()

    private func reloadNavigationViewItems() {
        // remove all child view controller header labels
        childTitleLabels.forEach { $0.removeFromSuperview() }
        childTitleLabels.removeAll()
        for (index, item) in viewControllers.enumerate() {
            let child = item as! IndicatorInfoProvider
            let indicatorInfo = child.indicatorInfoForPagerTabStrip(self)
            let navTitleLabel : UILabel = {
                let label = UILabel()
                label.text = indicatorInfo.title
                label.font = UIApplication.sharedApplication().statusBarOrientation.isPortrait ? settings.style.portraitTitleFont : settings.style.landscapeTitleFont
                label.textColor = settings.style.titleColor
                label.alpha = 0
                return label
            }()
            navTitleLabel.alpha = currentIndex == index ? 1 : 0
            navTitleLabel.textColor = settings.style.titleColor
            titleScrollView.addSubview(navTitleLabel)
            childTitleLabels.append(navTitleLabel)
        }
    }
    
    private func setNavigationViewItemsPosition(updateAlpha updateAlpha: Bool) {
        let distance = distanceValue
        let isPortrait = UIApplication.sharedApplication().statusBarOrientation.isPortrait
        let navBarHeight: CGFloat = navigationController!.navigationBar.frame.size.height
        for (index, label) in childTitleLabels.enumerate() {
            if updateAlpha {
                label.alpha = currentIndex == index ? 1 : 0
            }
            label.font = isPortrait ? settings.style.portraitTitleFont : settings.style.landscapeTitleFont
            let viewSize = label.intrinsicContentSize()
            let originX = distance - viewSize.width/2 + CGFloat(index) * distance
            let originY = (CGFloat(navBarHeight) - viewSize.height) / 2
            label.frame = CGRectMake(originX, originY - 2, viewSize.width, viewSize.height)
            label.tag = index
        }
        
        let xOffset = distance * CGFloat(currentIndex)
        titleScrollView.contentOffset = CGPointMake(xOffset, 0)
        
        pageControl.numberOfPages = childTitleLabels.count
        pageControl.currentPage = currentIndex
        let viewSize = pageControl.sizeForNumberOfPages(childTitleLabels.count)
        let originX = distance - viewSize.width / 2
        pageControl.frame = CGRectMake(originX, navBarHeight - 10, viewSize.width, viewSize.height)
    }
    
    private func setAlphaWithOffset(offset: CGFloat, andDistance distance: CGFloat) {
        for (index, label) in childTitleLabels.enumerate() {
            label.alpha = {
                if offset < distance * CGFloat(index) {
                    return (offset - distance * CGFloat(index - 1)) / distance
                }
                else {
                    return 1 - ((offset - distance * CGFloat(index)) / distance)
                }
            }()
        }
    }
    
    private var distanceValue: CGFloat {
        let middle = navigationController!.navigationBar.convertPoint(navigationController!.navigationBar.center, toView: titleView)
        return middle.x
    }
}