// File created from ScreenTemplate // $ createScreen.sh Secrets/Reset SecretsReset /* Copyright 2020 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 SecretsResetViewModel: SecretsResetViewModelType { // MARK: - Properties // MARK: Private private let session: MXSession private let recoveryService: MXRecoveryService private let crossSigningService: CrossSigningService // MARK: Public weak var viewDelegate: SecretsResetViewModelViewDelegate? weak var coordinatorDelegate: SecretsResetViewModelCoordinatorDelegate? // MARK: - Setup init(session: MXSession) { self.session = session self.recoveryService = session.crypto.recoveryService self.crossSigningService = CrossSigningService() } // MARK: - Public func process(viewAction: SecretsResetViewAction) { switch viewAction { case .loadData: break case .reset: self.askAuthentication() case .authenticationCancelled: self.authenticationCancelled() case .authenticationInfoEntered(let authParameters): self.resetSecrets(with: authParameters) case .cancel: self.coordinatorDelegate?.secretsResetViewModelDidCancel(self) } } // MARK: - Private func update(viewState: SecretsResetViewState) { self.viewDelegate?.secretsResetViewModel(self, didUpdateViewState: viewState) } private func resetSecrets(with authParameters: [String: Any]) { guard let crossSigning = self.session.crypto?.crossSigning else { return } MXLog.debug("[SecretsResetViewModel] resetSecrets") crossSigning.setup(withAuthParams: authParameters, success: { [weak self] in guard let self = self else { return } self.recoveryService.deleteRecovery(withDeleteServicesBackups: true, success: { [weak self] in guard let self = self else { return } self.update(viewState: .resetDone) self.coordinatorDelegate?.secretsResetViewModelDidResetSecrets(self) }, failure: { [weak self] error in guard let self = self else { return } self.update(viewState: .error(error)) }) }, failure: { [weak self] error in guard let self = self else { return } // bwi 7295 need to do authentication with session on an 401 error let nsError = error as NSError if let matrixData = nsError.userInfo["com.matrixsdk.httpclient.error.response.data"] as? [String: Any], let session = MXAuthenticationSession(fromJSON: matrixData) { self.coordinatorDelegate?.secretsResetViewModel(self, needsToAuthenticateWith: session) } else { self.update(viewState: .error(error)) } // bwi 7295 end }) } private func askAuthentication() { self.update(viewState: .resetting) let setupCrossSigningRequest = self.crossSigningService.setupCrossSigningRequest() self.coordinatorDelegate?.secretsResetViewModel(self, needsToAuthenticateWith: setupCrossSigningRequest) } private func authenticationCancelled() { self.update(viewState: .resetCancelled) } }