IOS-8268 [Onramp] Repository for user preference (#4110)

tangem · Oct 29, 2024 · 9836302 · 9836302
1 parent 09989d1
commit 9836302
Show file tree

Hide file tree

Showing 38 changed files with 616 additions and 188 deletions.
diff --git a/Tangem/App/Services/Onramp/CommonOnrampStorage.swift b/Tangem/App/Services/Onramp/CommonOnrampStorage.swift
@@ -0,0 +1,31 @@
+//
+//  CommonOnrampStorage.swift
+//  TangemApp
+//
+//  Created by Sergey Balashov on 21.10.2024.
+//  Copyright © 2024 Tangem AG. All rights reserved.
+//
+
+import TangemExpress
+import TangemFoundation
+
+struct CommonOnrampStorage: OnrampStorage {
+    @Injected(\.persistentStorage) private var storage: PersistentStorageProtocol
+
+    func save(preference: OnrampUserPreference) {
+        do {
+            try storage.store(value: preference, for: .onrampPreference)
+        } catch {
+            AppLog.shared.debug("[CommonOnrampStorage] Failed to save changes in storage. Reason: \(error)")
+        }
+    }
+
+    func preference() -> OnrampUserPreference? {
+        do {
+            return try storage.value(for: .onrampPreference)
+        } catch {
+            AppLog.shared.debug("[CommonOnrampStorage] Couldn't get the staking transactions list from the storage with error \(error)")
+            return nil
+        }
+    }
+}
diff --git a/Tangem/App/Services/PersistentStorage/PersistentStorageKey.swift b/Tangem/App/Services/PersistentStorage/PersistentStorageKey.swift
@@ -15,6 +15,7 @@ enum PersistentStorageKey {
     case walletConnectSessions(userWalletId: String)
     case pendingExpressTransactions
     case pendingStakingTransactions
+    case onrampPreference
 
     var path: String {
         switch self {
@@ -30,6 +31,8 @@ enum PersistentStorageKey {
             return "express_pending_transactions"
         case .pendingStakingTransactions:
             return "staking_pending_transactions"
+        case .onrampPreference:
+            return "onramp_preference"
         }
     }
 }
diff --git a/Tangem/Common/UI/SendDecimalNumberTextField/SendDecimalNumberTextField.swift b/Tangem/Common/UI/SendDecimalNumberTextField/SendDecimalNumberTextField.swift
@@ -231,7 +231,7 @@ extension SendDecimalNumberTextField: Setupable {
         map { $0.toolbarType = toolbarType }
     }
 
-    func prefixSuffixOptions(_ prefixSuffixOptions: PrefixSuffixOptions) -> Self {
+    func prefixSuffixOptions(_ prefixSuffixOptions: PrefixSuffixOptions?) -> Self {
         map { $0.prefixSuffixOptions = prefixSuffixOptions }
     }
 

diff --git a/.../Common/UI/SendDecimalNumberTextField/SendDecimalNumberTextFieldPrefixSuffixOptions.swift b/.../Common/UI/SendDecimalNumberTextField/SendDecimalNumberTextFieldPrefixSuffixOptions.swift
@@ -17,17 +17,13 @@ extension SendDecimalNumberTextField {
 
 extension SendDecimalNumberTextField {
     struct PrefixSuffixOptionsFactory {
-        private let cryptoCurrencyCode: String
-        private let fiatCurrencyCode: String
         private let locale: Locale
 
-        init(cryptoCurrencyCode: String, fiatCurrencyCode: String, locale: Locale = .current) {
-            self.cryptoCurrencyCode = cryptoCurrencyCode
-            self.fiatCurrencyCode = fiatCurrencyCode
+        init(locale: Locale = .current) {
             self.locale = locale
         }
 
-        func makeCryptoOptions() -> PrefixSuffixOptions {
+        func makeCryptoOptions(cryptoCurrencyCode: String) -> PrefixSuffixOptions {
             let numberFormatter = NumberFormatter()
             numberFormatter.numberStyle = .currency
             numberFormatter.currencySymbol = cryptoCurrencyCode
@@ -36,7 +32,7 @@ extension SendDecimalNumberTextField {
             return options(from: numberFormatter, currency: cryptoCurrencyCode, forceSpace: true)
         }
 
-        func makeFiatOptions() -> PrefixSuffixOptions {
+        func makeFiatOptions(fiatCurrencyCode: String) -> PrefixSuffixOptions {
             let numberFormatter = NumberFormatter()
             numberFormatter.numberStyle = .currency
             numberFormatter.currencyCode = fiatCurrencyCode

diff --git a/Tangem/Modules/Main/MainCoordinator.swift b/Tangem/Modules/Main/MainCoordinator.swift
@@ -396,8 +396,6 @@ extension MainCoordinator: SingleTokenBaseRoutable {
     }
 
     func openOnramp(walletModel: WalletModel, userWalletModel: UserWalletModel) {
-        mainBottomSheetUIManager.hide()
-
         let dismissAction: Action<(walletModel: WalletModel, userWalletModel: UserWalletModel)?> = { [weak self] navigationInfo in
             self?.sendCoordinator = nil
 

diff --git a/Tangem/Modules/OnrampCountry/OnrampCountryViewModel.swift b/Tangem/Modules/OnrampCountry/OnrampCountryViewModel.swift
@@ -51,6 +51,7 @@ final class OnrampCountryViewModel: ObservableObject, Identifiable {
     func didTapMainButton() {
         switch style {
         case .info:
+            repository.saveChanges()
             coordinator?.dismissConfirmCountryView()
         case .notSupport:
             coordinator?.dismiss()

diff --git a/Tangem/Modules/Send/Factories/FlowBuilders/OnrampFlowBaseBuilder.swift b/Tangem/Modules/Send/Factories/FlowBuilders/OnrampFlowBaseBuilder.swift
@@ -17,13 +17,22 @@ struct OnrampFlowBaseBuilder {
     let sendFinishStepBuilder: SendFinishStepBuilder
     let builder: SendDependenciesBuilder
 
-    func makeSendViewModel(onrampManager: some OnrampManager, router: SendRoutable) -> SendViewModel {
+    func makeSendViewModel(router: SendRoutable) -> SendViewModel {
+        let expressAPIProvider = ExpressAPIProviderFactory().makeExpressAPIProvider(userId: userWalletModel.userWalletId.stringValue, logger: AppLog.shared)
+        let onrampRepository = builder.makeOnrampRepository()
+
+        let onrampManager = TangemExpressFactory().makeOnrampManager(
+            expressAPIProvider: expressAPIProvider,
+            onrampRepository: onrampRepository,
+            logger: AppLog.shared
+        )
+
         let onrampModel = builder.makeOnrampModel(onrampManager: onrampManager)
 
         let onrampAmountViewModel = sendAmountStepBuilder.makeOnrampAmountViewModel(
             io: (input: onrampModel, output: onrampModel),
-            sendAmountValidator: builder.makeOnrampAmountValidator(),
-            amountModifier: .none
+            repository: onrampRepository,
+            sendAmountValidator: builder.makeOnrampAmountValidator()
         )
 
         let sendAmountCompactViewModel = sendAmountStepBuilder.makeSendAmountCompactViewModel(
@@ -47,24 +56,29 @@ struct OnrampFlowBaseBuilder {
 
         let stepsManager = CommonOnrampStepsManager(
             onrampStep: onramp.step,
-            finishStep: finish
+            finishStep: finish,
+            // If user already has saved country in the repository then the bottom sheet will not show
+            // And we can show keyboard safely
+            shouldActivateKeyboard: onrampRepository.savedCountry != nil
         )
 
         let interactor = CommonSendBaseInteractor(input: onrampModel, output: onrampModel)
+        let dataBuilder = builder.makeOnrampBaseDataBuilder(input: onrampModel, onrampRepository: onrampRepository)
 
         let viewModel = SendViewModel(
             interactor: interactor,
             stepsManager: stepsManager,
             userWalletModel: userWalletModel,
             alertBuilder: builder.makeSendAlertBuilder(),
-            dataBuilder: builder.makeSendBaseDataBuilder(input: onrampModel),
+            dataBuilder: dataBuilder,
             tokenItem: walletModel.tokenItem,
             feeTokenItem: walletModel.feeTokenItem,
             coordinator: router
         )
 
         stepsManager.set(output: viewModel)
         onrampModel.router = viewModel
+        onrampModel.alertPresenter = viewModel
 
         return viewModel
     }

diff --git a/Tangem/Modules/Send/Factories/FlowBuilders/RestakingFlowBaseBuilder.swift b/Tangem/Modules/Send/Factories/FlowBuilders/RestakingFlowBaseBuilder.swift
@@ -77,7 +77,7 @@ struct RestakingFlowBaseBuilder {
             stepsManager: stepsManager,
             userWalletModel: userWalletModel,
             alertBuilder: builder.makeStakingAlertBuilder(),
-            dataBuilder: builder.makeSendBaseDataBuilder(input: restakingModel),
+            dataBuilder: builder.makeStakingBaseDataBuilder(input: restakingModel),
             tokenItem: walletModel.tokenItem,
             feeTokenItem: walletModel.feeTokenItem,
             coordinator: router

diff --git a/Tangem/Modules/Send/Factories/FlowBuilders/StakingFlowBaseBuilder.swift b/Tangem/Modules/Send/Factories/FlowBuilders/StakingFlowBaseBuilder.swift
@@ -83,7 +83,7 @@ struct StakingFlowBaseBuilder {
             stepsManager: stepsManager,
             userWalletModel: userWalletModel,
             alertBuilder: builder.makeStakingAlertBuilder(),
-            dataBuilder: builder.makeSendBaseDataBuilder(input: stakingModel),
+            dataBuilder: builder.makeStakingBaseDataBuilder(input: stakingModel),
             tokenItem: walletModel.tokenItem,
             feeTokenItem: walletModel.feeTokenItem,
             coordinator: router

diff --git a/Tangem/Modules/Send/Factories/FlowBuilders/StakingSingleActionFlowBaseBuilder.swift b/Tangem/Modules/Send/Factories/FlowBuilders/StakingSingleActionFlowBaseBuilder.swift
@@ -65,7 +65,7 @@ struct StakingSingleActionFlowBaseBuilder {
             stepsManager: stepsManager,
             userWalletModel: userWalletModel,
             alertBuilder: builder.makeStakingAlertBuilder(),
-            dataBuilder: builder.makeSendBaseDataBuilder(input: actionModel),
+            dataBuilder: builder.makeStakingBaseDataBuilder(input: actionModel),
             tokenItem: walletModel.tokenItem,
             feeTokenItem: walletModel.feeTokenItem,
             coordinator: router

diff --git a/Tangem/Modules/Send/Factories/FlowBuilders/UnstakingFlowBaseBuilder.swift b/Tangem/Modules/Send/Factories/FlowBuilders/UnstakingFlowBaseBuilder.swift
@@ -83,7 +83,7 @@ struct UnstakingFlowBaseBuilder {
             stepsManager: stepsManager,
             userWalletModel: userWalletModel,
             alertBuilder: builder.makeStakingAlertBuilder(),
-            dataBuilder: builder.makeSendBaseDataBuilder(input: unstakingModel),
+            dataBuilder: builder.makeStakingBaseDataBuilder(input: unstakingModel),
             tokenItem: walletModel.tokenItem,
             feeTokenItem: walletModel.feeTokenItem,
             coordinator: router

diff --git a/Tangem/Modules/Send/Factories/SendDependenciesBuilder.swift b/Tangem/Modules/Send/Factories/SendDependenciesBuilder.swift
@@ -145,10 +145,6 @@ struct SendDependenciesBuilder {
         CommonFeeIncludedCalculator(validator: walletModel.transactionValidator)
     }
 
-    func makeSendBaseDataBuilder(input: SendBaseDataBuilderInput) -> SendBaseDataBuilder {
-        SendBaseDataBuilder(input: input, walletModel: walletModel, emailDataProvider: userWalletModel)
-    }
-
     // MARK: - Send, Sell
 
     func makeSendModel(
@@ -224,6 +220,10 @@ struct SendDependenciesBuilder {
         CommonSendAlertBuilder()
     }
 
+    func makeSendBaseDataBuilder(input: SendBaseDataBuilderInput) -> SendBaseDataBuilder {
+        CommonSendBaseDataBuilder(input: input, walletModel: walletModel, emailDataProvider: userWalletModel)
+    }
+
     // MARK: - Staking
 
     func makeStakingModel(stakingManager: some StakingManager) -> StakingModel {
@@ -323,18 +323,36 @@ struct SendDependenciesBuilder {
         StakingAmountModifier(tokenItem: walletModel.tokenItem)
     }
 
+    func makeStakingBaseDataBuilder(input: StakingBaseDataBuilderInput) -> StakingBaseDataBuilder {
+        CommonStakingBaseDataBuilder(input: input, walletModel: walletModel, emailDataProvider: userWalletModel)
+    }
+
     // MARK: - Onramp
 
     func makeOnrampModel(onrampManager: some OnrampManager) -> OnrampModel {
         OnrampModel(onrampManager: onrampManager)
     }
 
-    func makeOnrampManager(userWalletId: String) -> OnrampManager {
-        let expressAPIProvider = ExpressAPIProviderFactory().makeExpressAPIProvider(userId: userWalletId, logger: AppLog.shared)
-        return TangemExpressFactory().makeOnrampManager(expressAPIProvider: expressAPIProvider, logger: AppLog.shared)
+    func makeOnrampManager(userWalletId: String, onrampRepository: OnrampRepository) -> OnrampManager {
+        let expressAPIProvider = ExpressAPIProviderFactory()
+            .makeExpressAPIProvider(userId: userWalletId, logger: AppLog.shared)
+
+        return TangemExpressFactory().makeOnrampManager(
+            expressAPIProvider: expressAPIProvider,
+            onrampRepository: onrampRepository,
+            logger: AppLog.shared
+        )
     }
 
     func makeOnrampAmountValidator() -> SendAmountValidator {
         OnrampAmountValidator()
     }
+
+    func makeOnrampBaseDataBuilder(input: OnrampBaseDataBuilderInput, onrampRepository: OnrampRepository) -> OnrampBaseDataBuilder {
+        CommonOnrampBaseDataBuilder(input: input, walletModel: walletModel, onrampRepository: onrampRepository)
+    }
+
+    func makeOnrampRepository() -> OnrampRepository {
+        TangemExpressFactory().makeOnrampRepository(storage: CommonOnrampStorage())
+    }
 }
diff --git a/Tangem/Modules/Send/Factories/SendFlowFactory.swift b/Tangem/Modules/Send/Factories/SendFlowFactory.swift
@@ -151,7 +151,6 @@ struct SendFlowFactory {
         let sendAmountStepBuilder = SendAmountStepBuilder(walletModel: walletModel, builder: builder)
         let onrampStepBuilder = OnrampStepBuilder()
         let sendFinishStepBuilder = SendFinishStepBuilder(walletModel: walletModel)
-        let onrampManager = builder.makeOnrampManager(userWalletId: userWalletModel.userWalletId.stringValue)
 
         let baseBuilder = OnrampFlowBaseBuilder(
             userWalletModel: userWalletModel,
@@ -162,6 +161,6 @@ struct SendFlowFactory {
             builder: builder
         )
 
-        return baseBuilder.makeSendViewModel(onrampManager: onrampManager, router: router)
+        return baseBuilder.makeSendViewModel(router: router)
     }
 }
diff --git a/Tangem/Modules/Send/Factories/StepBuilders/SendAmountStepBuilder.swift b/Tangem/Modules/Send/Factories/StepBuilders/SendAmountStepBuilder.swift
@@ -7,6 +7,7 @@
 //
 
 import Foundation
+import TangemExpress
 
 struct SendAmountStepBuilder {
     typealias IO = (input: SendAmountInput, output: SendAmountOutput)
@@ -57,20 +58,17 @@ struct SendAmountStepBuilder {
 
     func makeOnrampAmountViewModel(
         io: IO,
-        sendAmountValidator: SendAmountValidator,
-        amountModifier: SendAmountModifier?
+        repository: OnrampRepository,
+        sendAmountValidator: SendAmountValidator
     ) -> OnrampAmountViewModel {
         let interactor = makeSendAmountInteractor(
             io: io,
             sendAmountValidator: sendAmountValidator,
-            amountModifier: amountModifier,
+            amountModifier: nil,
             type: .fiat
         )
 
-        return OnrampAmountViewModel(
-            settings: .init(tokenIconInfo: builder.makeTokenIconInfo(), tokenItem: walletModel.tokenItem),
-            interactor: interactor
-        )
+        return OnrampAmountViewModel(tokenItem: walletModel.tokenItem, repository: repository, interactor: interactor)
     }
 }
 

diff --git a/Tangem/Modules/Send/Services/ManagmentModels/OnrampModel.swift b/Tangem/Modules/Send/Services/ManagmentModels/OnrampModel.swift
@@ -11,6 +11,7 @@ import TangemExpress
 import Combine
 
 protocol OnrampModelRoutable: AnyObject {
+    func openOnrampCountriesSelector()
     func openOnrampCountryBottomSheet(country: OnrampCountry)
 }
 
@@ -24,24 +25,42 @@ class OnrampModel {
     // MARK: - Dependencies
 
     weak var router: OnrampModelRoutable?
+    weak var alertPresenter: SendViewAlertPresenter?
 
     // MARK: - Private injections
 
     private let onrampManager: OnrampManager
 
+    private var task: Task<Void, Never>?
     private var bag: Set<AnyCancellable> = []
 
     init(onrampManager: OnrampManager) {
         self.onrampManager = onrampManager
 
         bind()
+        updateCountry()
     }
 }
 
 // MARK: - Bind
 
 private extension OnrampModel {
     func bind() {}
+
+    func updateCountry() {
+        task = runTask(in: self) { model in
+            do {
+                let country = try await model.onrampManager.getCountry()
+                await runOnMain {
+                    model.router?.openOnrampCountryBottomSheet(country: country)
+                }
+            } catch {
+                await runOnMain {
+                    model.alertPresenter?.showAlert(error.alertBinder)
+                }
+            }
+        }
+    }
 }
 
 // MARK: - Buy
@@ -132,12 +151,6 @@ extension OnrampModel: SendBaseOutput {
     }
 }
 
-// MARK: - SendBaseDataBuilderInput
+// MARK: - OnrampBaseDataBuilderInput
 
-extension OnrampModel: SendBaseDataBuilderInput {
-    var bsdkAmount: BSDKAmount? { nil }
-
-    var bsdkFee: BSDKFee? { nil }
-
-    var isFeeIncluded: Bool { false }
-}
+extension OnrampModel: OnrampBaseDataBuilderInput {}
diff --git a/Tangem/Modules/Send/Services/ManagmentModels/RestakingModel.swift b/Tangem/Modules/Send/Services/ManagmentModels/RestakingModel.swift
@@ -368,15 +368,21 @@ extension RestakingModel: NotificationTapDelegate {
     }
 }
 
-// MARK: - SendBaseDataBuilderInput
+// MARK: - StakingBaseDataBuilderInput
 
-extension RestakingModel: SendBaseDataBuilderInput {
+extension RestakingModel: StakingBaseDataBuilderInput {
     var bsdkAmount: BSDKAmount? { makeAmount(value: action.amount) }
 
     var bsdkFee: BlockchainSdk.Fee? { selectedFee.value.value }
 
     var isFeeIncluded: Bool { false }
 
+    var stakingActionType: TangemStaking.StakingAction.ActionType? { stakingAction.type }
+
+    var selectedPolicy: ApprovePolicy? { nil }
+
+    var approveViewModelInput: (any ApproveViewModelInput)? { nil }
+
     var validator: ValidatorInfo? { action.validatorInfo }
 }