Swiftgram/Display/ViewController.swift

import Foundation
import UIKit
import AsyncDisplayKit
import SwiftSignalKit

public func ==(lhs: ViewControllerLayout, rhs: ViewControllerLayout) -> Bool {
    return lhs.size == rhs.size && lhs.insets == rhs.insets && lhs.inputViewHeight == rhs.inputViewHeight && lhs.statusBarHeight == rhs.statusBarHeight
}

@objc public class ViewController: UIViewController, WindowContentController {
    private var _displayNode: ASDisplayNode?
    public final var displayNode: ASDisplayNode {
        get {
            if let value = self._displayNode {
                return value
            }
            else {
                self.loadDisplayNode()
                if self._displayNode == nil {
                    fatalError("displayNode should be initialized after loadDisplayNode()")
                }
                return self._displayNode!
            }
        }
        set(value) {
            self._displayNode = value
        }
    }
    
    public final var isNodeLoaded: Bool {
        return self._displayNode != nil
    }
    
    public let statusBar: StatusBar
    public let navigationBar: NavigationBar
    
    private let _ready = Promise<Bool>(true)
    public var ready: Promise<Bool> {
        return self._ready
    }
    
    private var updateLayoutOnLayout: (ViewControllerLayout, NSTimeInterval, UInt)?
    public private(set) var layout: ViewControllerLayout?
    
    var keyboardFrameObserver: AnyObject?
    
    private var scrollToTopView: ScrollToTopView?
    public var scrollToTop: (() -> Void)? {
        didSet {
            if self.isViewLoaded() {
                self.updateScrollToTopView()
            }
        }
    }
    
    private func updateScrollToTopView() {
        if let scrollToTop = self.scrollToTop {
            if let displayNode = self._displayNode where self.scrollToTopView == nil {
                let scrollToTopView = ScrollToTopView(frame: CGRect(x: 0.0, y: -1.0, width: displayNode.frame.size.width, height: 1.0))
                scrollToTopView.action = { [weak self] in
                    if let scrollToTop = self?.scrollToTop {
                        scrollToTop()
                    }
                }
                self.scrollToTopView = scrollToTopView
                self.view.addSubview(scrollToTopView)
            }
        } else if let scrollToTopView = self.scrollToTopView {
            scrollToTopView.removeFromSuperview()
            self.scrollToTopView = nil
        }
    }
    
    public init() {
        self.statusBar = StatusBar()
        self.navigationBar = NavigationBar()
        
        super.init(nibName: nil, bundle: nil)
        
        self.navigationBar.item = self.navigationItem
        self.automaticallyAdjustsScrollViewInsets = false
        
        self.keyboardFrameObserver = NSNotificationCenter.defaultCenter().addObserverForName(UIKeyboardWillChangeFrameNotification, object: nil, queue: nil, usingBlock: { [weak self] notification in
            if let strongSelf = self, _ = strongSelf._displayNode {
                let keyboardFrame: CGRect = (notification.userInfo?[UIKeyboardFrameEndUserInfoKey] as? NSValue)?.CGRectValue() ?? CGRect()
                let keyboardHeight = max(0.0, UIScreen.mainScreen().bounds.size.height - keyboardFrame.minY)
                var duration: Double = (notification.userInfo?[UIKeyboardAnimationDurationUserInfoKey] as? NSNumber)?.doubleValue ?? 0.0
                if duration > DBL_EPSILON {
                    duration = 0.5
                }
                var curve: UInt = (notification.userInfo?[UIKeyboardAnimationCurveUserInfoKey] as? NSNumber)?.unsignedIntegerValue ?? UInt(7 << 16)
                
                let previousLayout: ViewControllerLayout?
                var previousDurationAndCurve: (NSTimeInterval, UInt)?
                if let updateLayoutOnLayout = strongSelf.updateLayoutOnLayout {
                    previousLayout = updateLayoutOnLayout.0
                    previousDurationAndCurve = (updateLayoutOnLayout.1, updateLayoutOnLayout.2)
                } else{
                    previousLayout = strongSelf.layout
                }
                let layout = ViewControllerLayout(size: previousLayout?.size ?? CGSize(), insets: previousLayout?.insets ?? UIEdgeInsets(), inputViewHeight: keyboardHeight, statusBarHeight: previousLayout?.statusBarHeight ?? 20.0)
                let updated: Bool
                if let previousLayout = previousLayout {
                    updated = previousLayout != layout
                    if duration < DBL_EPSILON && abs(min(previousLayout.inputViewHeight, layout.inputViewHeight) - 225.0) < CGFloat(FLT_EPSILON) && abs(max(previousLayout.inputViewHeight, layout.inputViewHeight) - 225.0 - 33.0) < CGFloat(FLT_EPSILON) {
                        duration = 0.1
                        curve = 0
                    }
                } else {
                    updated = true
                }
                if updated {
                    //print("keyboard layout change: \(layout) rotating: \(strongSelf.view.window?.isRotating())")
                    
                    let durationAndCurve: (NSTimeInterval, UInt) = previousDurationAndCurve ?? (duration, curve)
                    strongSelf.updateLayoutOnLayout = (layout, durationAndCurve.0, durationAndCurve.1)
                    strongSelf.view.setNeedsLayout()
                }
            }
        })
    }
    
    required public init(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    deinit {
        if let keyboardFrameObserver = keyboardFrameObserver {
            NSNotificationCenter.defaultCenter().removeObserver(keyboardFrameObserver)
        }
    }
    
    public override func loadView() {
        self.view = self.displayNode.view
        self.displayNode.addSubnode(self.navigationBar)
        self.view.addSubview(self.statusBar.view)
    }
    
    public func loadDisplayNode() {
        self.displayNode = ASDisplayNode()
        self.displayNodeDidLoad()
    }
    
    public func displayNodeDidLoad() {
        self.updateScrollToTopView()
    }
    
    public func setParentLayout(layout: ViewControllerLayout, duration: NSTimeInterval, curve: UInt) {
        if self._displayNode == nil {
            self.loadDisplayNode()
        }
        
        let previousLayout: ViewControllerLayout?
        if let updateLayoutOnLayout = self.updateLayoutOnLayout {
            previousLayout = updateLayoutOnLayout.0
        } else {
            previousLayout = self.layout
        }
        
        var insets = layout.insets
        insets.top += 22.0
        
        let layout = ViewControllerLayout(size: layout.size, insets: insets, inputViewHeight: previousLayout?.inputViewHeight ?? 0.0, statusBarHeight: layout.statusBarHeight)
        let updated: Bool
        if let previousLayout = previousLayout {
            updated = previousLayout != layout
        } else {
            updated = true
        }
        if updated {
            if previousLayout == nil {
                self.layout = layout
                self.updateLayout(layout, previousLayout: previousLayout, duration: duration, curve: 0)
            } else {
                self.updateLayoutOnLayout = (layout, duration, 0)
                self.view.setNeedsLayout()
            }
        }
    }
    
    public func updateLayout(layout: ViewControllerLayout, previousLayout: ViewControllerLayout?, duration: Double, curve: UInt) {
        self.statusBar.frame = CGRect(origin: CGPoint(), size: CGSize(width: layout.size.width, height: 40.0))
        self.navigationBar.frame = CGRect(origin: CGPoint(x: 0.0, y: 0.0), size: CGSize(width: layout.size.width, height: 44.0 + 20.0))
        if let scrollToTopView = self.scrollToTopView {
            scrollToTopView.frame = CGRect(x: 0.0, y: 0.0, width: layout.size.width, height: 10.0)
        }
    }
    
    public func setNeedsLayoutWithDuration(duration: Double, curve: UInt) {
        let previousLayout: ViewControllerLayout?
        var previousDurationAndCurve: (NSTimeInterval, UInt)?
        if let updateLayoutOnLayout = self.updateLayoutOnLayout {
            previousLayout = updateLayoutOnLayout.0
            previousDurationAndCurve = (updateLayoutOnLayout.1, updateLayoutOnLayout.2)
        } else{
            previousLayout = self.layout
        }
        if let previousLayout = previousLayout {
            let durationAndCurve: (NSTimeInterval, UInt) = previousDurationAndCurve ?? (duration, curve)
            self.updateLayoutOnLayout = (previousLayout, durationAndCurve.0, durationAndCurve.1)
            self.view.setNeedsLayout()
        }
    }
    
    override public func viewDidLayoutSubviews() {
        if let updateLayoutOnLayout = self.updateLayoutOnLayout {
            if !Window.isDeviceRotating() {
                if !((self.view.window as? Window)?.isUpdatingOrientationLayout ?? false) {
                    //print("\(self) apply inputHeight: \(updateLayoutOnLayout.0.inputViewHeight)")
                    let previousLayout = self.layout
                    self.layout = updateLayoutOnLayout.0
                    self.updateLayout(updateLayoutOnLayout.0, previousLayout: previousLayout, duration: updateLayoutOnLayout.1, curve: updateLayoutOnLayout.2)
                    self.view.frame = CGRect(origin: self.view.frame.origin, size: updateLayoutOnLayout.0.size)
                    
                    self.updateLayoutOnLayout = nil
                } else {
                    (self.view.window as? Window)?.addPostUpdateToInterfaceOrientationBlock({ [weak self] in
                        if let strongSelf = self {
                            strongSelf.view.setNeedsLayout()
                        }
                    })
                }
            } else {
                Window.addPostDeviceOrientationDidChangeBlock({ [weak self] in
                    if let strongSelf = self {
                        strongSelf.view.setNeedsLayout()
                    }
                })
            }
        }
    }
    
    override public func presentViewController(viewControllerToPresent: UIViewController, animated flag: Bool, completion: (() -> Void)?) {
        if let navigationController = self.navigationController as? NavigationController {
            navigationController.presentViewController(viewControllerToPresent, animated: flag, completion: completion)
        } else {
            super.presentViewController(viewControllerToPresent, animated: flag, completion: completion)
        }
    }
}