Fix: don’t allow to reset secrets if it is already in progress.

element-hq · May 2, 2023 · 0240e35 · 0240e35
1 parent 5cf28f4
commit 0240e35
Show file tree

Hide file tree

Showing 6 changed files with 18 additions and 2 deletions.
diff --git a/Riot/Modules/Secrets/Reset/SecretsResetCoordinator.swift b/Riot/Modules/Secrets/Reset/SecretsResetCoordinator.swift
@@ -94,11 +94,11 @@ extension SecretsResetCoordinator: SecretsResetViewModelCoordinatorDelegate {
 extension SecretsResetCoordinator: ReauthenticationCoordinatorDelegate {
 
     func reauthenticationCoordinatorDidComplete(_ coordinator: ReauthenticationCoordinatorType, withAuthenticationParameters authenticationParameters: [String: Any]?) {
-
         self.secretsResetViewModel.process(viewAction: .authenticationInfoEntered(authenticationParameters ?? [:]))
     }
 
     func reauthenticationCoordinatorDidCancel(_ coordinator: ReauthenticationCoordinatorType) {
+        self.secretsResetViewModel.process(viewAction: .authenticationCancelled)
         self.remove(childCoordinator: coordinator)
     }
 

diff --git a/Riot/Modules/Secrets/Reset/SecretsResetViewAction.swift b/Riot/Modules/Secrets/Reset/SecretsResetViewAction.swift
@@ -22,6 +22,7 @@ import Foundation
 enum SecretsResetViewAction {
     case loadData
     case reset
+    case authenticationCancelled
     case authenticationInfoEntered(_ authInfo: [String: Any])
     case cancel
 }
diff --git a/Riot/Modules/Secrets/Reset/SecretsResetViewController.swift b/Riot/Modules/Secrets/Reset/SecretsResetViewController.swift
@@ -132,6 +132,8 @@ final class SecretsResetViewController: UIViewController {
             self.renderLoading()
         case .resetDone:
             self.renderLoaded()
+        case .resetCancelled:
+            self.renderCancelled()
         case .error(let error):
             self.render(error: error)
         }
@@ -145,6 +147,10 @@ final class SecretsResetViewController: UIViewController {
         self.activityPresenter.removeCurrentActivityIndicator(animated: true)
     }
 
+    private func renderCancelled() {
+        self.activityPresenter.removeCurrentActivityIndicator(animated: true)
+    }
+
     private func render(error: Error) {
         self.activityPresenter.removeCurrentActivityIndicator(animated: true)
         self.errorPresenter.presentError(from: self, forError: error, animated: true, handler: nil)

diff --git a/Riot/Modules/Secrets/Reset/SecretsResetViewModel.swift b/Riot/Modules/Secrets/Reset/SecretsResetViewModel.swift
@@ -49,6 +49,8 @@ final class SecretsResetViewModel: SecretsResetViewModelType {
             break
         case .reset:
             self.askAuthentication()
+        case .authenticationCancelled:
+            self.authenticationCancelled()
         case .authenticationInfoEntered(let authParameters):
             self.resetSecrets(with: authParameters)
         case .cancel:
@@ -68,7 +70,6 @@ final class SecretsResetViewModel: SecretsResetViewModelType {
         }
         MXLog.debug("[SecretsResetViewModel] resetSecrets")
 
-        self.update(viewState: .resetting)
         crossSigning.setup(withAuthParams: authParameters, success: { [weak self] in
             guard let self = self else {
                 return
@@ -96,7 +97,13 @@ final class SecretsResetViewModel: SecretsResetViewModelType {
     }
 
     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)
+    }
 }
diff --git a/Riot/Modules/Secrets/Reset/SecretsResetViewState.swift b/Riot/Modules/Secrets/Reset/SecretsResetViewState.swift
@@ -22,5 +22,6 @@ import Foundation
 enum SecretsResetViewState {
     case resetting
     case resetDone
+    case resetCancelled
     case error(Error)
 }
diff --git a/changelog.d/pr-7404.bugfix b/changelog.d/pr-7404.bugfix
@@ -0,0 +1 @@
+Fix an issue where the Secrets Reset screen would open twice.
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Fix an issue where the Secrets Reset screen would open twice.