IOS-8906 [Balance caching] Repository (#4470)

tangem · Jan 23, 2025 · da69139 · da69139
1 parent df9812f
commit da69139
Show file tree

Hide file tree

Showing 53 changed files with 477 additions and 155 deletions.
diff --git a/Tangem/App/Factories/WalletModels/CommonWalletModelsFactory.swift b/Tangem/App/Factories/WalletModels/CommonWalletModelsFactory.swift
@@ -13,9 +13,11 @@ import TangemStaking
 
 struct CommonWalletModelsFactory {
     private let config: UserWalletConfig
+    private let userWalletId: UserWalletId
 
-    init(config: UserWalletConfig) {
+    init(config: UserWalletConfig, userWalletId: UserWalletId) {
         self.config = config
+        self.userWalletId = userWalletId
     }
 
     private func isDerivationDefault(blockchain: Blockchain, derivationPath: DerivationPath?) -> Bool {
@@ -92,6 +94,7 @@ extension CommonWalletModelsFactory: WalletModelsFactory {
         let isMainCoinCustom = !isDerivationDefault(blockchain: currentBlockchain, derivationPath: currentDerivation)
         let blockchainNetwork = BlockchainNetwork(currentBlockchain, derivationPath: currentDerivation)
         let sendAvailabilityProvider = TransactionSendAvailabilityProvider(isSendingSupportedByCard: config.hasFeature(.send))
+        let tokenBalancesRepository = CommonTokenBalancesRepository(userWalletId: userWalletId)
 
         if types.contains(.coin) {
             let tokenItem: TokenItem = .blockchain(blockchainNetwork)
@@ -111,7 +114,8 @@ extension CommonWalletModelsFactory: WalletModelsFactory {
                 amountType: .coin,
                 shouldPerformHealthCheck: shouldPerformHealthCheck,
                 isCustom: isMainCoinCustom,
-                sendAvailabilityProvider: sendAvailabilityProvider
+                sendAvailabilityProvider: sendAvailabilityProvider,
+                tokenBalancesRepository: tokenBalancesRepository
             )
             models.append(mainCoinModel)
         }
@@ -137,7 +141,8 @@ extension CommonWalletModelsFactory: WalletModelsFactory {
                     amountType: amountType,
                     shouldPerformHealthCheck: shouldPerformHealthCheck,
                     isCustom: isTokenCustom,
-                    sendAvailabilityProvider: sendAvailabilityProvider
+                    sendAvailabilityProvider: sendAvailabilityProvider,
+                    tokenBalancesRepository: tokenBalancesRepository
                 )
                 models.append(tokenModel)
             }

diff --git a/Tangem/App/Factories/WalletModels/DemoWalletModelsFactory.swift b/Tangem/App/Factories/WalletModels/DemoWalletModelsFactory.swift
@@ -12,8 +12,8 @@ import BlockchainSdk
 struct DemoWalletModelsFactory {
     private let factory: CommonWalletModelsFactory
 
-    init(config: UserWalletConfig) {
-        factory = CommonWalletModelsFactory(config: config)
+    init(config: UserWalletConfig, userWalletId: UserWalletId) {
+        factory = CommonWalletModelsFactory(config: config, userWalletId: userWalletId)
     }
 }
 

diff --git a/Tangem/App/Models/UserWallet/Implementations/GenericConfig.swift b/Tangem/App/Models/UserWallet/Implementations/GenericConfig.swift
@@ -203,8 +203,8 @@ extension GenericConfig: UserWalletConfig {
         }
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
-        return CommonWalletModelsFactory(config: self)
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
+        return CommonWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/Implementations/GenericDemoConfig.swift b/Tangem/App/Models/UserWallet/Implementations/GenericDemoConfig.swift
@@ -196,8 +196,8 @@ extension GenericDemoConfig: UserWalletConfig {
         }
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
-        return DemoWalletModelsFactory(config: self)
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
+        return DemoWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/Implementations/LegacyConfig.swift b/Tangem/App/Models/UserWallet/Implementations/LegacyConfig.swift
@@ -224,8 +224,8 @@ extension LegacyConfig: UserWalletConfig {
         }
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
-        return CommonWalletModelsFactory(config: self)
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
+        return CommonWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/Implementations/NoteConfig.swift b/Tangem/App/Models/UserWallet/Implementations/NoteConfig.swift
@@ -148,8 +148,8 @@ extension NoteConfig: UserWalletConfig {
         }
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
-        return CommonWalletModelsFactory(config: self)
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
+        return CommonWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/Implementations/NoteDemoConfig.swift b/Tangem/App/Models/UserWallet/Implementations/NoteDemoConfig.swift
@@ -150,8 +150,8 @@ extension NoteDemoConfig: UserWalletConfig {
         }
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
-        return DemoWalletModelsFactory(config: self)
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
+        return DemoWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/Implementations/Start2CoinConfig.swift b/Tangem/App/Models/UserWallet/Implementations/Start2CoinConfig.swift
@@ -155,8 +155,8 @@ extension Start2CoinConfig: UserWalletConfig {
         )
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
-        return CommonWalletModelsFactory(config: self)
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
+        return CommonWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/Implementations/TwinConfig.swift b/Tangem/App/Models/UserWallet/Implementations/TwinConfig.swift
@@ -171,8 +171,8 @@ extension TwinConfig: UserWalletConfig {
         }
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
-        return CommonWalletModelsFactory(config: self)
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
+        return CommonWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/Implementations/VisaConfig.swift b/Tangem/App/Models/UserWallet/Implementations/VisaConfig.swift
@@ -141,8 +141,8 @@ extension VisaConfig: UserWalletConfig {
         }
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
-        return CommonWalletModelsFactory(config: self)
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
+        return CommonWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/Implementations/Wallet2Config.swift b/Tangem/App/Models/UserWallet/Implementations/Wallet2Config.swift
@@ -365,12 +365,12 @@ extension Wallet2Config: UserWalletConfig {
         }
     }
 
-    func makeWalletModelsFactory() -> WalletModelsFactory {
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory {
         if isDemo {
-            return DemoWalletModelsFactory(config: self)
+            return DemoWalletModelsFactory(config: self, userWalletId: userWalletId)
         }
 
-        return CommonWalletModelsFactory(config: self)
+        return CommonWalletModelsFactory(config: self, userWalletId: userWalletId)
     }
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory {

diff --git a/Tangem/App/Models/UserWallet/UserWalletConfig.swift b/Tangem/App/Models/UserWallet/UserWalletConfig.swift
@@ -62,7 +62,7 @@ protocol UserWalletConfig: OnboardingStepsBuilderFactory, BackupServiceFactory,
 
     func getFeatureAvailability(_ feature: UserWalletFeature) -> UserWalletFeature.Availability
 
-    func makeWalletModelsFactory() -> WalletModelsFactory
+    func makeWalletModelsFactory(userWalletId: UserWalletId) -> WalletModelsFactory
 
     func makeAnyWalletManagerFactory() throws -> AnyWalletManagerFactory
 

diff --git a/Tangem/App/Services/BalanceProvider/FormattedTokenBalanceType.swift b/Tangem/App/Services/BalanceProvider/FormattedTokenBalanceType.swift
@@ -53,7 +53,7 @@ extension FormattedTokenBalanceType: CustomStringConvertible {
         switch self {
         case .loading(let cached): "Loading cached balance: \(String(describing: cached))"
         case .failure(let cached): "Failure cached balance: \(String(describing: cached))"
-        case .loaded(let balance): "Loaded balance: \(balance)"
+        case .loaded: "Loaded balance"
         }
     }
 }
@@ -73,8 +73,12 @@ extension FormattedTokenBalanceType {
         }
     }
 
-    struct Cached: Hashable {
+    struct Cached: Hashable, CustomStringConvertible {
         let balance: String
         let date: Date
+
+        var description: String {
+            "Cached balance on date: \(date.formatted())"
+        }
     }
 }
diff --git a/Tangem/App/Services/BalanceProvider/Implementation/FiatTokenBalanceProvider.swift b/Tangem/App/Services/BalanceProvider/Implementation/FiatTokenBalanceProvider.swift
@@ -30,7 +30,6 @@ extension FiatTokenBalanceProvider: TokenBalanceProvider {
 
     var balanceTypePublisher: AnyPublisher<TokenBalanceType, Never> {
         Publishers.CombineLatest(
-            // Listen if rate was loaded after main balance
             walletModel.ratePublisher.removeDuplicates(),
             cryptoBalanceProvider.balanceTypePublisher.removeDuplicates()
         )
@@ -52,52 +51,70 @@ extension FiatTokenBalanceProvider: TokenBalanceProvider {
 // MARK: - Private
 
 extension FiatTokenBalanceProvider {
-    func mapToTokenBalance(rate: LoadingResult<WalletModel.Rate?, Never>, balanceType: TokenBalanceType) -> TokenBalanceType {
+    func mapToTokenBalance(rate: WalletModel.Rate, balanceType: TokenBalanceType) -> TokenBalanceType {
         switch (rate, balanceType) {
-        // There is no rate because it's custom token
-        case (.success(.none), _) where walletModel.isCustom:
+        // There is no rate because token is custom
+        case (.custom, _):
             return .empty(.custom)
 
-        // There is no crypto value to convert
+        // There is no crypto
         case (_, .empty(let reason)):
             return .empty(reason)
 
-        // There is no rate but main balance is still loading
+        // Main balance is still loading
         // Then show loading to avoid showing empty fiat when crypto is loading
-        case (.success(.none), .loading):
+        // Or rate is loading
+        case (_, .loading(.none)), (.loading(.none), _):
             return .loading(.none)
 
-        // There is no rate
-        case (.success(.none), _):
-            return .empty(.noData)
-
-        // There is no crypto value because there was an error
-        case (_, .failure(.none)):
+        // There is no crypto or no rate value without cache
+        case (.failure(.none), _), (_, .failure(.none)):
             return .failure(.none)
 
-        // There is rate is loading and we can't convert it
-        case (.loading, _), (_, .loading(.none)):
-            return .loading(.none)
+        // Both is loading and has cached value
+        case (.loading(.some(let rate)), .loading(.some(let cached))):
+            let fiat = cached.balance * rate.price
+            return .loading(.init(balance: fiat, date: cached.date))
+
+        // Both is failure and has cached value
+        case (.failure(.some(let rate)), .failure(.some(let cached))):
+            let fiat = cached.balance * rate.price
+            return .failure(.init(balance: fiat, date: cached.date))
 
-        // Has some rate and cached value
-        case (.success(.some(let rate)), .loading(.some(let cached))):
-            let fiat = cached.balance * rate.value
+        // Has some rate and loading cached value
+        case (.loaded(let rate), .loading(.some(let cached))):
+            let fiat = cached.balance * rate.price
             return .loading(.init(balance: fiat, date: cached.date))
 
-        // Has some rate and cached value
-        case (.success(.some(let rate)), .failure(.some(let cached))):
-            let fiat = cached.balance * rate.value
+        // Has some rate and failure cached value
+        case (.loaded(let rate), .failure(.some(let cached))):
+            let fiat = cached.balance * rate.price
             return .failure(.init(balance: fiat, date: cached.date))
 
-        // Has cached rate and cached value
-        case (.success(.some(.cached(let rate))), .loaded(let value)):
-            let fiat = value * rate.balance
+        // Has some rate and loaded value
+        case (.loaded(let rate), .loaded(let value)):
+            let fiat = value * rate.price
+            return .loaded(fiat)
+
+        // Has some cached rate and loading cached value
+        case (.failure(.some(let rate)), .loading(.some(let cached))):
+            let fiat = cached.balance * rate.price
+            return .loading(.init(balance: fiat, date: rate.date))
+
+        // Has some cached rate and loaded crypto value
+        case (.failure(.some(let rate)), .loaded(let value)):
+            let fiat = value * rate.price
             return .failure(.init(balance: fiat, date: rate.date))
 
-        // Has some rate and cached value
-        case (.success(.some(.actual(let rate))), .loaded(let value)):
-            let fiat = value * rate
-            return .loaded(fiat)
+        // Rate is loading with cached value and crypto cached value
+        case (.loading(.some(let rate)), .failure(.some(let cached))):
+            let fiat = cached.balance * rate.price
+            return .loading(.init(balance: fiat, date: rate.date))
+
+        // Rate is loading with cached value and crypto loaded value
+        case (.loading(.some(let rate)), .loaded(let value)):
+            let fiat = value * rate.price
+            return .loading(.init(balance: fiat, date: rate.date))
         }
     }
 

diff --git a/Tangem/App/Services/BalanceProvider/TokenBalanceType.swift b/Tangem/App/Services/BalanceProvider/TokenBalanceType.swift
@@ -39,7 +39,7 @@ extension TokenBalanceType {
         case .empty: nil
         case .loading(let cached): cached
         case .failure(let cached): cached
-        case .loaded(let value): nil
+        case .loaded: nil
         }
     }
 
@@ -64,9 +64,9 @@ extension TokenBalanceType: CustomStringConvertible {
     var description: String {
         switch self {
         case .empty(let reason): "Empty \(reason)"
-        case .loading(let cached): "Loading cached: \(String(describing: cached))"
+        case .loading(let cached): "Loading with cached: \(String(describing: cached))"
         case .failure(let cached): "Failure cached: \(String(describing: cached))"
-        case .loaded(let balance): "Loaded: \(balance)"
+        case .loaded: "Loaded"
         }
     }
 }
@@ -85,8 +85,12 @@ extension TokenBalanceType {
         case noAccount(message: String)
     }
 
-    struct Cached: Hashable {
+    struct Cached: Hashable, CustomStringConvertible {
         let balance: Decimal
         let date: Date
+
+        var description: String {
+            "Cached balance on date: \(date.formatted())"
+        }
     }
 }
diff --git a/Tangem/App/Services/PersistentStorage/PersistentStorageKey.swift b/Tangem/App/Services/PersistentStorage/PersistentStorageKey.swift
@@ -17,6 +17,8 @@ enum PersistentStorageKey {
     case pendingOnrampTransactions
     case pendingStakingTransactions
     case onrampPreference
+    case cachedBalances
+    case cachedQuotes
 
     var path: String {
         switch self {
@@ -36,6 +38,10 @@ enum PersistentStorageKey {
             return "staking_pending_transactions"
         case .onrampPreference:
             return "onramp_preference"
+        case .cachedBalances:
+            return "cached_balances"
+        case .cachedQuotes:
+            return "cached_quotes"
         }
     }
 }
diff --git a/Tangem/App/Services/TokenBalancesRepository/CommonTokenBalancesRepository.swift b/Tangem/App/Services/TokenBalancesRepository/CommonTokenBalancesRepository.swift
@@ -0,0 +1,33 @@
+//
+//  CommonTokenBalancesRepository.swift
+//  TangemApp
+//
+//  Created by Sergey Balashov on 26.12.2024.
+//  Copyright © 2024 Tangem AG. All rights reserved.
+//
+
+import Foundation
+import Combine
+import TangemFoundation
+
+struct CommonTokenBalancesRepository {
+    @Injected(\.tokenBalancesStorage)
+    private var storage: TokenBalancesStorage
+    private let userWalletId: UserWalletId
+
+    init(userWalletId: UserWalletId) {
+        self.userWalletId = userWalletId
+    }
+}
+
+// MARK: - TokenBalancesRepository
+
+extension CommonTokenBalancesRepository: TokenBalancesRepository {
+    func balance(walletModel: WalletModel, type: CachedBalanceType) -> CachedBalance? {
+        storage.balance(for: walletModel.id, userWalletId: userWalletId, type: type)
+    }
+
+    func store(balance: CachedBalance, for walletModel: WalletModel, type: CachedBalanceType) {
+        storage.store(balance: balance, type: type, id: walletModel.id, userWalletId: userWalletId)
+    }
+}