Implement key backup setup recovery key scene with view controller, view model and coordinator

2026-04-20 08:32:53 +02:00 · 2019-01-22 11:04:06 +01:00
parent 4511d06743
commit e947ed8452
9 changed files with 748 additions and 3 deletions
@@ -0,0 +1,102 @@
+/*
+ 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 Foundation
+
+final class KeyBackupSetupRecoveryKeyViewModel: KeyBackupSetupRecoveryKeyViewModelType {
+    
+    // MARK: - Properties
+    
+    // MARK: Private
+    
+    private let megolmBackupCreationInfo: MXMegolmBackupCreationInfo
+    private let keyBackup: MXKeyBackup
+    private let coordinatorDelegateQueue: OperationQueue
+    private var createKeyBackupOperation: MXHTTPOperation?
+    
+    // MARK: Public
+    
+    let recoveryKey: String
+    
+    weak var viewDelegate: KeyBackupSetupRecoveryKeyViewModelViewDelegate?
+    weak var coordinatorDelegate: KeyBackupSetupRecoveryKeyViewModelCoordinatorDelegate?
+    
+    // MARK: - Setup
+    
+    init(keyBackup: MXKeyBackup, megolmBackupCreationInfo: MXMegolmBackupCreationInfo) {
+        self.megolmBackupCreationInfo = megolmBackupCreationInfo
+        self.recoveryKey = megolmBackupCreationInfo.recoveryKey
+        self.keyBackup = keyBackup
+        
+        let coordinatorDelegateQueue = OperationQueue()
+        coordinatorDelegateQueue.name = "KeyBackupSetupRecoveryKeyViewModel.coordinatorDelegateQueue"
+        coordinatorDelegateQueue.maxConcurrentOperationCount = 1
+        self.coordinatorDelegateQueue = coordinatorDelegateQueue
+    }
+    
+    deinit {
+        self.createKeyBackupOperation?.cancel()
+    }
+    
+    // MARK: - Public
+    
+    func process(viewAction: KeyBackupSetupRecoveryKeyViewAction) {
+        switch viewAction {
+        case .madeCopy:
+            self.createBackup()
+        case .skip:
+            self.pauseCoordinatorOperations()
+            self.viewDelegate?.keyBackupSetupPassphraseViewModelShowSkipAlert(self)
+        case.skipAlertContinue:
+            self.resumeCoordinatorOperations()
+        case.skipAlertSkip:
+            self.createKeyBackupOperation?.cancel()
+            self.cancelCoordinatorOperations()
+            self.coordinatorDelegate?.keyBackupSetupRecoveryKeyViewModelDidCancel(self)
+        }
+    }
+    
+    // MARK: - Private
+    
+    func createBackup() {
+        self.viewDelegate?.keyBackupSetupRecoveryKeyViewModel(self, didUpdateViewState: .loading)
+        
+        self.keyBackup.createKeyBackupVersion(self.megolmBackupCreationInfo, success: { [weak self] (keyBackupVersion) in
+            guard let sself = self else {
+                return
+            }
+            sself.viewDelegate?.keyBackupSetupRecoveryKeyViewModel(sself, didUpdateViewState: .loaded)
+            sself.coordinatorDelegate?.keyBackupSetupRecoveryKeyViewModelDidCreateBackup(sself)
+        }, failure: { [weak self] error in
+            guard let sself = self else {
+                return
+            }
+            sself.viewDelegate?.keyBackupSetupRecoveryKeyViewModel(sself, didUpdateViewState: .error(error))
+        })    
+    }
+    
+    private func pauseCoordinatorOperations() {
+        self.coordinatorDelegateQueue.isSuspended = true
+    }
+    
+    private func resumeCoordinatorOperations() {
+        self.coordinatorDelegateQueue.isSuspended = false
+    }
+    
+    private func cancelCoordinatorOperations() {
+        self.coordinatorDelegateQueue.cancelAllOperations()
+    }
+}