IOS-2464 Keep explicit reference

tangem · Nov 2, 2022 · de192dd · de192dd
1 parent ecf1205
commit de192dd
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/TangemSdk/TangemSdk/Operations/Backup/BackupService.swift b/TangemSdk/TangemSdk/Operations/Backup/BackupService.swift
@@ -271,11 +271,15 @@ public class BackupService: ObservableObject {
                 Message(header: nil,
                         body: "backup_finalize_primary_card_message_format".localized($0))
             }
+
+            currentCommand = task
 
             sdk.startSession(with: task,
                              cardId: primaryCard.cardId,
-                             initialMessage: initialMessage,
-                             completion: completion)
+                             initialMessage: initialMessage) {[weak self] result in
+                completion(result)
+                self?.currentCommand = nil
+            }
 
         } catch {
             completion(.failure(error.toTangemSdkError()))
@@ -332,17 +336,23 @@ public class BackupService: ObservableObject {
                 Message(header: nil,
                         body: "backup_finalize_backup_card_message_format".localized($0))
             }
+
+            currentCommand = command
 
             sdk.startSession(with: command,
                              cardId: backupCard.cardId,
-                             initialMessage: initialMessage) { result in
+                             initialMessage: initialMessage) {[weak self] result in
+                guard let self = self else { return }
+
                 switch result {
                 case .success(let card):
                     self.repo.data.finalizedBackupCardsCount += 1
                     completion(.success(card))
                 case .failure(let error):
                     completion(.failure(error))
                 }
+
+                self.currentCommand = nil
             }
 
         } catch {