Refactor DeviceVerification prefix classes to KeyVerification where needed. And move key verification related classes into the same module KeyVerification.

Riot/Modules/KeyVerification/Common/Loading/KeyVerificationDataLoadingViewModel.swift

@@ -0,0 +1,150 @@
+// File created from ScreenTemplate
+// $ createScreen.sh DeviceVerification/Loading DeviceVerificationDataLoading
+/*
+ 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
+
+enum KeyVerificationDataLoadingViewModelError: Error {
+    case unknown
+    case transactionCancelled
+    case transactionCancelledByMe(reason: MXTransactionCancelCode)
+}
+
+final class KeyVerificationDataLoadingViewModel: KeyVerificationDataLoadingViewModelType {
+    
+    // MARK: - Properties
+    
+    // MARK: Private
+
+    private let session: MXSession
+    private let otherUserId: String?
+    private let otherDeviceId: String?
+    private let keyVerificationService = KeyVerificationService()
+    
+    private let keyVerificationRequest: MXKeyVerificationRequest?
+    
+    private var currentOperation: MXHTTPOperation?
+    
+    // MARK: Public
+
+    weak var viewDelegate: KeyVerificationDataLoadingViewModelViewDelegate?
+    weak var coordinatorDelegate: KeyVerificationDataLoadingViewModelCoordinatorDelegate?
+    
+    // MARK: - Setup
+    
+    init(session: MXSession, otherUserId: String, otherDeviceId: String) {
+        self.session = session
+        self.otherUserId = otherUserId
+        self.otherDeviceId = otherDeviceId
+        self.keyVerificationRequest = nil
+    }
+    
+    init(session: MXSession, keyVerificationRequest: MXKeyVerificationRequest) {
+        self.session = session
+        self.otherUserId = nil
+        self.otherDeviceId = nil
+        self.keyVerificationRequest = keyVerificationRequest
+    }
+    
+    deinit {
+        self.currentOperation?.cancel()
+    }
+    
+    // MARK: - Public
+    
+    func process(viewAction: KeyVerificationDataLoadingViewAction) {
+        switch viewAction {
+        case .loadData:
+            self.loadData()
+        case .cancel:
+            self.coordinatorDelegate?.keyVerificationDataLoadingViewModelDidCancel(self)
+        }
+    }
+    
+    // MARK: - Private
+    
+    private func loadData() {
+        if let keyVerificationRequest = self.keyVerificationRequest {
+            self.acceptKeyVerificationRequest(keyVerificationRequest)
+        } else {
+            self.downloadOtherDeviceKeys()
+        }
+    }
+    
+    private func acceptKeyVerificationRequest(_ keyVerificationRequest: MXKeyVerificationRequest) {
+        
+        self.update(viewState: .loading)
+        
+        keyVerificationRequest.accept(withMethods: self.keyVerificationService.supportedKeyVerificationMethods(), success: { [weak self] in
+            guard let self = self else {
+                return
+            }
+            
+            self.coordinatorDelegate?.keyVerificationDataLoadingViewModel(self, didAcceptKeyVerificationRequest: keyVerificationRequest)
+            
+        }, failure: { [weak self] (error) in
+            guard let self = self else {
+                return
+            }
+            self.update(viewState: .error(error))
+        })
+    }
+    
+    private func downloadOtherDeviceKeys() {
+        guard let crypto = session.crypto,
+            let otherUserId = self.otherUserId,
+            let otherDeviceId = self.otherDeviceId else {
+            self.update(viewState: .errorMessage(VectorL10n.deviceVerificationErrorCannotLoadDevice))
+            NSLog("[KeyVerificationDataLoadingViewModel] Error session.crypto is nil")
+            return
+        }
+        
+        if let otherUser = session.user(withUserId: otherUserId) {
+            
+            self.update(viewState: .loading)
+           
+            self.currentOperation = crypto.downloadKeys([otherUserId], forceDownload: false, success: { [weak self] (usersDevicesMap, crossSigningKeysMap) in
+                guard let sself = self else {
+                    return
+                }
+                
+                if let otherDevice = usersDevicesMap?.object(forDevice: otherDeviceId, forUser: otherUserId) {
+                    sself.update(viewState: .loaded)
+                    sself.coordinatorDelegate?.keyVerificationDataLoadingViewModel(sself, didLoadUser: otherUser, device: otherDevice)
+                } else {
+                    sself.update(viewState: .errorMessage(VectorL10n.deviceVerificationErrorCannotLoadDevice))
+                }
+                
+                }, failure: { [weak self] (error) in
+                    guard let sself = self else {
+                        return
+                    }
+                    
+                    let finalError = error ?? KeyVerificationDataLoadingViewModelError.unknown
+                    
+                    sself.update(viewState: .error(finalError))
+            })
+            
+        } else {
+            self.update(viewState: .errorMessage(VectorL10n.deviceVerificationErrorCannotLoadDevice))
+        }
+    }
+    
+    private func update(viewState: KeyVerificationDataLoadingViewState) {
+        self.viewDelegate?.keyVerificationDataLoadingViewModel(self, didUpdateViewState: viewState)
+    }
+}