add timeouts to ledger sign process

kukai-wallet · Aug 21, 2024 · 6bf4f57 · 6bf4f57
1 parent 6512793
commit 6bf4f57
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Sources/KukaiCoreSwift/Models/LedgerWallet.swift b/Sources/KukaiCoreSwift/Models/LedgerWallet.swift
@@ -78,8 +78,14 @@ public class LedgerWallet: Wallet {
  let isWatermarkedOperation = (String(hex.prefix(2)) == "03") && hex.count != 32
 
  LedgerService.shared.connectTo(uuid: ledgerUUID)
- .flatMap { _ -> AnyPublisher<String, KukaiError> in
+ .timeout(.seconds(10), scheduler: RunLoop.main, customError: {
+ return KukaiError.knownErrorMessage("Timed out waiting for device to connect. Check device/bluetooth is turned on and try again")
+ })
+ .flatMap { _ -> Publishers.Timeout<AnyPublisher<String, KukaiError>, RunLoop> in
  return LedgerService.shared.sign(hex: hex, parse: isWatermarkedOperation)
+ .timeout(.seconds(10), scheduler: RunLoop.main, customError: {
+ return KukaiError.knownErrorMessage("Timed out waiting for device to connect. Check device/bluetooth is turned on and try again")
+ })
  }
  .sink(onError: { error in
  completion(Result.failure(error))