import Foundation import Display import AsyncDisplayKit import TelegramCore final class AuthorizationSequenceCodeEntryController: ViewController { private var controllerNode: AuthorizationSequenceCodeEntryControllerNode { return self.displayNode as! AuthorizationSequenceCodeEntryControllerNode } var loginWithCode: ((String) -> Void)? var requestNextOption: (() -> Void)? var data: (String, SentAuthorizationCodeType, AuthorizationCodeNextType?, Int32?)? private let hapticFeedback = HapticFeedback() var inProgress: Bool = false { didSet { if self.inProgress { let item = UIBarButtonItem(customDisplayNode: ProgressNavigationButtonNode()) self.navigationItem.rightBarButtonItem = item } else { self.navigationItem.rightBarButtonItem = UIBarButtonItem(title: "Next", style: .done, target: self, action: #selector(self.nextPressed)) } self.controllerNode.inProgress = self.inProgress } } override init(navigationBar: NavigationBar = NavigationBar()) { super.init(navigationBar: navigationBar) self.navigationBar.backgroundColor = nil self.navigationBar.isOpaque = false self.navigationBar.stripeColor = UIColor.clear self.navigationItem.rightBarButtonItem = UIBarButtonItem(title: "Next", style: .done, target: self, action: #selector(self.nextPressed)) } required init(coder aDecoder: NSCoder) { fatalError("init(coder:) has not been implemented") } override public func loadDisplayNode() { self.displayNode = AuthorizationSequenceCodeEntryControllerNode() self.displayNodeDidLoad() self.controllerNode.loginWithCode = { [weak self] code in self?.loginWithCode?(code) } self.controllerNode.requestNextOption = { [weak self] in self?.requestNextOption?() } if let (number, codeType, nextType, timeout) = self.data { self.controllerNode.updateData(number: number, codeType: codeType, nextType: nextType, timeout: timeout) } } override func viewDidAppear(_ animated: Bool) { super.viewDidAppear(animated) self.controllerNode.activateInput() } func updateData(number: String, codeType: SentAuthorizationCodeType, nextType: AuthorizationCodeNextType?, timeout: Int32?) { if self.data?.0 != number || self.data?.1 != codeType || self.data?.2 != nextType || self.data?.3 != timeout { self.data = (number, codeType, nextType, timeout) if self.isNodeLoaded { self.controllerNode.updateData(number: number, codeType: codeType, nextType: nextType, timeout: timeout) self.requestLayout(transition: .immediate) } } } override func containerLayoutUpdated(_ layout: ContainerViewLayout, transition: ContainedViewLayoutTransition) { super.containerLayoutUpdated(layout, transition: transition) self.controllerNode.containerLayoutUpdated(layout, navigationBarHeight: 0.0, transition: transition) } @objc func nextPressed() { if self.controllerNode.currentCode.isEmpty { hapticFeedback.error() self.controllerNode.animateError() } else { self.loginWithCode?(self.controllerNode.currentCode) } } }