Create key backup recover from recovery key scene with coordinator, view model, view controller

Riot/Modules/KeyBackup/Recover/RecoveryKey/KeyBackupRecoverFromRecoveryKeyViewController.swift

@@ -0,0 +1,274 @@
+/*
+ Copyright 2019 New Vector Ltd
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ */
+
+import UIKit
+import MobileCoreServices
+
+final class KeyBackupRecoverFromRecoveryKeyViewController: UIViewController {
+    
+    // MARK: - Properties
+    
+    // MARK: Outlets
+    
+    @IBOutlet private weak var scrollView: UIScrollView!
+    
+    @IBOutlet private weak var shieldImageView: UIImageView!
+    
+    @IBOutlet private weak var informationLabel: UILabel!
+    
+    @IBOutlet private weak var recoveryKeyTitleLabel: UILabel!
+    @IBOutlet private weak var recoveryKeyTextField: UITextField!
+    @IBOutlet private weak var recoveryKeyTextFieldBackgroundView: UIView!
+    
+    @IBOutlet private weak var importFileButton: UIButton!
+    
+    @IBOutlet private weak var unknownRecoveryKeyButton: UIButton!
+    
+    @IBOutlet private weak var recoverButtonBackgroundView: UIView!
+    @IBOutlet private weak var recoverButton: UIButton!
+    
+    // MARK: Private
+    
+    private var viewModel: KeyBackupRecoverFromRecoveryKeyViewModelType!
+    private var keyboardAvoider: KeyboardAvoider?
+    private var theme: Theme!
+    private var errorPresenter: MXKErrorPresentation!
+    private var activityPresenter: ActivityIndicatorPresenter!
+    private weak var skipAlertController: UIAlertController?
+    
+    // MARK: Public
+    
+    // MARK: - Setup
+    
+    class func instantiate(with viewModel: KeyBackupRecoverFromRecoveryKeyViewModelType) -> KeyBackupRecoverFromRecoveryKeyViewController {
+        let viewController = StoryboardScene.KeyBackupRecoverFromRecoveryKeyViewController.initialScene.instantiate()
+        viewController.viewModel = viewModel
+        viewController.theme = ThemeService.shared().theme
+        return viewController
+    }
+    
+    // MARK: - Life cycle
+    
+    override func viewDidLoad() {
+        super.viewDidLoad()
+        
+        // Do any additional setup after loading the view.
+        
+        self.title = VectorL10n.keyBackupRecoverTitle
+        self.vc_removeBackTitle()
+        
+        self.setupViews()
+        self.keyboardAvoider = KeyboardAvoider(scrollViewContainerView: self.view, scrollView: self.scrollView)
+        self.activityPresenter = ActivityIndicatorPresenter()
+        self.errorPresenter = MXKErrorAlertPresentation()
+        
+        self.registerThemeServiceDidChangeThemeNotification()
+        self.update(theme: self.theme)
+        
+        self.viewModel.viewDelegate = self
+    }
+    
+    override var preferredStatusBarStyle: UIStatusBarStyle {
+        return self.theme.statusBarStyle
+    }
+    
+    // MARK: - Private
+    
+    private func setupViews() {
+        let cancelBarButtonItem = MXKBarButtonItem(title: VectorL10n.cancel, style: .plain) { [weak self] in
+            self?.cancelButtonAction()
+        }
+        
+        self.navigationItem.rightBarButtonItem = cancelBarButtonItem
+        
+        self.scrollView.keyboardDismissMode = .interactive
+        
+        let shieldImage = Asset.Images.shield.image.withRenderingMode(.alwaysTemplate)
+        self.shieldImageView.image = shieldImage
+        
+        self.informationLabel.text = VectorL10n.keyBackupRecoverFromRecoveryKeyInfo
+        
+        self.recoveryKeyTitleLabel.text = VectorL10n.keyBackupRecoverFromRecoveryKeyRecoveryKeyTitle
+        self.recoveryKeyTextField.addTarget(self, action: #selector(recoveryKeyTextFieldDidChange(_:)), for: .editingChanged)
+        
+        let importFileImage = Asset.Images.importFilesButton.image.withRenderingMode(.alwaysTemplate)
+        self.importFileButton.setImage(importFileImage, for: .normal)
+        
+        self.unknownRecoveryKeyButton.vc_enableMultiLinesTitle()
+        self.unknownRecoveryKeyButton.setTitle(VectorL10n.keyBackupRecoverFromRecoveryKeyLostRecoveryKeyAction, for: .normal)
+        // Interaction is disabled for the moment
+        self.unknownRecoveryKeyButton.isUserInteractionEnabled = false
+        
+        self.recoverButton.vc_enableMultiLinesTitle()
+        self.recoverButton.setTitle(VectorL10n.keyBackupRecoverFromPassphraseRecoverAction, for: .normal)
+        
+        self.updateRecoverButton()
+    }
+    
+    private func update(theme: Theme) {
+        self.theme = theme
+        
+        self.view.backgroundColor = theme.headerBackgroundColor
+        
+        if let navigationBar = self.navigationController?.navigationBar {
+            theme.applyStyle(onNavigationBar: navigationBar)
+        }
+        
+        self.informationLabel.textColor = theme.textPrimaryColor
+        
+        self.shieldImageView.tintColor = theme.textPrimaryColor
+        
+        self.recoveryKeyTextFieldBackgroundView.backgroundColor = theme.backgroundColor
+        self.recoveryKeyTitleLabel.textColor = theme.textPrimaryColor
+        theme.applyStyle(onTextField: self.recoveryKeyTextField)
+        self.recoveryKeyTextField.attributedPlaceholder = NSAttributedString(string: VectorL10n.keyBackupRecoverFromRecoveryKeyRecoveryKeyPlaceholder,
+                                                                            attributes: [.foregroundColor : theme.placeholderTextColor])
+        
+        theme.applyStyle(onButton: self.importFileButton)        
+        
+        self.recoverButtonBackgroundView.backgroundColor = theme.backgroundColor
+        theme.applyStyle(onButton: self.recoverButton)
+        
+        self.unknownRecoveryKeyButton.setTitleColor(theme.textPrimaryColor, for: .normal)
+    }
+    
+    private func registerThemeServiceDidChangeThemeNotification() {
+        NotificationCenter.default.addObserver(self, selector: #selector(themeDidChange), name: .themeServiceDidChangeTheme, object: nil)
+    }
+    
+    @objc private func themeDidChange() {
+        self.update(theme: ThemeService.shared().theme)
+    }
+    
+    private func updateRecoverButton() {
+        self.recoverButton.isEnabled = self.viewModel.isFormValid
+    }
+    
+    private func render(viewState: KeyBackupRecoverFromRecoveryKeyViewState) {
+        switch viewState {
+        case .loading:
+            self.renderLoading()
+        case .loaded(totalKeys: let totalKeys):
+            self.renderLoaded(with: totalKeys)
+        case .error(let error):
+            self.render(error: error)
+        }
+    }
+    
+    private func renderLoading() {
+        self.view.endEditing(true)
+        self.activityPresenter.presentActivityIndicator(on: self.view, animated: true)
+    }
+    
+    private func renderLoaded(with totalKeys: UInt) {
+        self.activityPresenter.removeCurrentActivityIndicator(animated: true)
+        
+        if totalKeys == 0 {
+            self.errorPresenter.presentError(from: self,
+                                             title: VectorL10n.keyBackupRecoverEmptyBackupTitle,
+                                             message: VectorL10n.keyBackupRecoverEmptyBackupMessage,
+                                             animated: true,
+                                             handler: nil)
+        }
+    }
+    
+    private func render(error: Error) {
+        self.activityPresenter.removeCurrentActivityIndicator(animated: true)
+        self.errorPresenter.presentError(from: self, forError: error, animated: true, handler: nil)
+    }
+    
+    private func showFileSelection() {
+        // Show only text documents
+        let documentPicker = UIDocumentPickerViewController(documentTypes: [kUTTypeText as String], in: .import)
+        documentPicker.delegate = self
+        self.present(documentPicker, animated: true, completion: nil)
+    }
+    
+    private func importRecoveryKey(from url: URL) {
+        if let recoveryKey = self.getDocumentContent(from: url) {
+            self.recoveryKeyTextField.text = recoveryKey
+        } else {
+            self.errorPresenter.presentGenericError(from: self, animated: true, handler: nil)
+        }
+    }
+    
+    private func getDocumentContent(from documentURL: URL) -> String? {
+        let documentContent: String?
+        
+        do {
+            documentContent = try String(contentsOf: documentURL)
+        } catch {
+            print("Error: \(error)")
+            documentContent = nil
+        }
+        
+        return documentContent
+    }
+    
+    // MARK: - Actions
+    
+    private func cancelButtonAction() {
+        self.viewModel.process(viewAction: .cancel)
+    }
+    
+    @IBAction private func importFileButtonAction(_ sender: Any) {
+        self.showFileSelection()
+    }
+    
+    @objc private func recoveryKeyTextFieldDidChange(_ textField: UITextField) {
+        self.viewModel.recoveryKey = textField.text
+        self.updateRecoverButton()
+    }
+    
+    @IBAction private func usePassphraseButtonAction(_ sender: Any) {
+        self.viewModel.process(viewAction: .recover)
+    }
+    
+    @IBAction private func unknownPassphraseButtonAction(_ sender: Any) {
+        self.viewModel.process(viewAction: .unknownRecoveryKey)
+    }
+}
+
+// MARK: - UITextFieldDelegate
+extension KeyBackupRecoverFromRecoveryKeyViewController: UITextFieldDelegate {
+    
+    func textFieldShouldReturn(_ textField: UITextField) -> Bool {
+        textField.resignFirstResponder()
+        return true
+    }
+}
+
+// MARK: - KeyBackupRecoverFromRecoveryKeyViewModelViewDelegate
+extension KeyBackupRecoverFromRecoveryKeyViewController: KeyBackupRecoverFromRecoveryKeyViewModelViewDelegate {
+    func keyBackupRecoverFromPassphraseViewModel(_ viewModel: KeyBackupRecoverFromRecoveryKeyViewModelType, didUpdateViewState viewSate: KeyBackupRecoverFromRecoveryKeyViewState) {
+        self.render(viewState: viewSate)
+    }
+}
+
+// MARK: - UIDocumentPickerDelegate
+extension KeyBackupRecoverFromRecoveryKeyViewController: UIDocumentPickerDelegate {
+    
+    func documentPicker(_ controller: UIDocumentPickerViewController, didPickDocumentsAt urls: [URL]) {
+        guard let documentUrl = urls.first else {
+            return
+        }
+        self.importRecoveryKey(from: documentUrl)
+    }
+    
+    func documentPicker(_ controller: UIDocumentPickerViewController, didPickDocumentAt url: URL) {
+        self.importRecoveryKey(from: url)
+    }
+}