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

2026-04-19 16:13:42 +02:00 · 2020-03-18 17:39:26 +01:00
parent a8f28e0ec0
commit d6296b3150
91 changed files with 819 additions and 803 deletions
@@ -0,0 +1,123 @@
+// File created from ScreenTemplate
+// $ createScreen.sh DeviceVerification/Incoming DeviceVerificationIncoming
+/*
+ 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 DeviceVerificationIncomingViewModel: DeviceVerificationIncomingViewModelType {
+    
+    // MARK: - Properties
+    
+    // MARK: Private
+
+    private let session: MXSession
+    private let transaction: MXIncomingSASTransaction
+    
+    // MARK: Public
+
+    let userId: String
+    let userDisplayName: String?
+    let avatarUrl: String?
+    let deviceId: String
+
+    let mediaManager: MXMediaManager
+
+    weak var viewDelegate: DeviceVerificationIncomingViewModelViewDelegate?
+    weak var coordinatorDelegate: DeviceVerificationIncomingViewModelCoordinatorDelegate?
+    
+    // MARK: - Setup
+    
+    init(session: MXSession, otherUser: MXUser, transaction: MXIncomingSASTransaction) {
+        self.session = session
+        self.transaction = transaction
+        self.userId = otherUser.userId
+        self.userDisplayName = otherUser.displayname
+        self.avatarUrl = otherUser.avatarUrl
+        self.deviceId = transaction.otherDeviceId
+
+        self.mediaManager = session.mediaManager
+    }
+    
+    deinit {
+    }
+    
+    // MARK: - Public
+    
+    func process(viewAction: DeviceVerificationIncomingViewAction) {
+        switch viewAction {
+        case .loadData:
+            self.registerTransactionDidStateChangeNotification(transaction: transaction)
+        case .accept:
+            self.acceptIncomingDeviceVerification()
+        case .cancel:
+            self.rejectIncomingDeviceVerification()
+            self.coordinatorDelegate?.deviceVerificationIncomingViewModelDidCancel(self)
+        }
+    }
+    
+    // MARK: - Private
+    
+    private func acceptIncomingDeviceVerification() {
+        self.update(viewState: .loading)
+        self.transaction.accept()
+    }
+
+    private func rejectIncomingDeviceVerification() {
+        self.transaction.cancel(with: MXTransactionCancelCode.user())
+    }
+    
+    private func update(viewState: DeviceVerificationIncomingViewState) {
+        self.viewDelegate?.deviceVerificationIncomingViewModel(self, didUpdateViewState: viewState)
+    }
+
+    // MARK: - MXKeyVerificationTransactionDidChange
+
+    private func registerTransactionDidStateChangeNotification(transaction: MXIncomingSASTransaction) {
+        NotificationCenter.default.addObserver(self, selector: #selector(transactionDidStateChange(notification:)), name: NSNotification.Name.MXKeyVerificationTransactionDidChange, object: transaction)
+    }
+    
+    private func unregisterTransactionDidStateChangeNotification() {
+        NotificationCenter.default.removeObserver(self, name: .MXKeyVerificationTransactionDidChange, object: nil)
+    }
+
+    @objc private func transactionDidStateChange(notification: Notification) {
+        guard let transaction = notification.object as? MXIncomingSASTransaction else {
+            return
+        }
+
+        switch transaction.state {
+        case MXSASTransactionStateShowSAS:
+            self.unregisterTransactionDidStateChangeNotification()
+            self.update(viewState: .loaded)
+            self.coordinatorDelegate?.deviceVerificationIncomingViewModel(self, didAcceptTransaction: self.transaction)
+        case MXSASTransactionStateCancelled:
+            guard let reason = transaction.reasonCancelCode else {
+                return
+            }
+            self.unregisterTransactionDidStateChangeNotification()
+            self.update(viewState: .cancelled(reason))
+        case MXSASTransactionStateCancelledByMe:
+            guard let reason = transaction.reasonCancelCode else {
+                return
+            }
+            self.unregisterTransactionDidStateChangeNotification()
+            self.update(viewState: .cancelledByMe(reason))
+        default:
+            break
+        }
+    }
+}