Fix key listener

element-hq · Nov 28, 2023 · 51ddc74 · 51ddc74
1 parent 92efd5f
commit 51ddc74
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ElementX/Sources/Services/SecureBackup/SecureBackupController.swift b/ElementX/Sources/Services/SecureBackup/SecureBackupController.swift
@@ -96,6 +96,7 @@ class SecureBackupController: SecureBackupControllerProtocol {
                 return .success(key)
             }
 
+            var keyUploadErrored = false
             let recoveryKey = try await encryption.enableRecovery(waitForBackupsToUpload: false, progressListener: SecureBackupEnableRecoveryProgressListener { [weak self] state in
                 guard let self else { return }
 
@@ -105,12 +106,11 @@ class SecureBackupController: SecureBackupControllerProtocol {
                 case .done:
                     recoveryKeyStateSubject.send(.enabled)
                 case .roomKeyUploadError:
-                    #warning("AG: fix me")
-                    return
+                    keyUploadErrored = true
                 }
             })
 
-            return .success(recoveryKey)
+            return keyUploadErrored ? .failure(.failedGeneratingRecoveryKey) : .success(recoveryKey)
         } catch {
             return .failure(.failedGeneratingRecoveryKey)
         }