From b4c656b3050dac15b83933a6f6fcbbaf4d81db1d Mon Sep 17 00:00:00 2001
From: Tangem <git@tangem.com>
Date: Thu, 5 Sep 2024 15:57:48 +0500
Subject: [PATCH] Updated on 2026-08-14

---
 .../AddStakingNotificationsTransformer.kt     | 19 +++++++++++-------
 .../viewmodel/StakingViewModel.kt             | 20 +++++++++++++++++--
 2 files changed, 30 insertions(+), 9 deletions(-)

diff --git a/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/state/transformers/AddStakingNotificationsTransformer.kt b/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/state/transformers/AddStakingNotificationsTransformer.kt
index 3635796fe7..f6c6eaac4c 100644
--- a/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/state/transformers/AddStakingNotificationsTransformer.kt
+++ b/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/state/transformers/AddStakingNotificationsTransformer.kt
@@ -67,13 +67,18 @@ internal class AddStakingNotificationsTransformer(
             isSubtractAvailable = isSubtractAvailable,
             reduceAmountBy = reduceAmountBy,
         )
-        val sendingAmount = checkAndCalculateSubtractedAmount(
-            isAmountSubtractAvailable = isSubtractAvailable,
-            cryptoCurrencyStatus = cryptoCurrencyStatus,
-            amountValue = amountValue,
-            feeValue = feeValue,
-            reduceAmountBy = reduceAmountBy,
-        )
+        val sendingAmount = if (isEnterAction) {
+            checkAndCalculateSubtractedAmount(
+                isAmountSubtractAvailable = isSubtractAvailable,
+                cryptoCurrencyStatus = cryptoCurrencyStatus,
+                amountValue = amountValue,
+                feeValue = feeValue,
+                reduceAmountBy = reduceAmountBy,
+            )
+        } else {
+            // No amount is taken from account balance on exit or pending actions
+            BigDecimal.ZERO
+        }
 
         val notifications = buildList {
             // errors
diff --git a/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/viewmodel/StakingViewModel.kt b/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/viewmodel/StakingViewModel.kt
index 1f1afe7e27..43da4818a4 100644
--- a/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/viewmodel/StakingViewModel.kt
+++ b/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/viewmodel/StakingViewModel.kt
@@ -56,9 +56,11 @@ import com.tangem.features.staking.impl.presentation.state.transformers.approval
 import com.tangem.features.staking.impl.presentation.state.transformers.approval.SetConfirmationStateAssentApprovalTransformer
 import com.tangem.features.staking.impl.presentation.state.transformers.approval.ShowApprovalBottomSheetTransformer
 import com.tangem.features.staking.impl.presentation.state.transformers.validator.ValidatorSelectChangeTransformer
+import com.tangem.features.staking.impl.presentation.state.utils.checkAndCalculateSubtractedAmount
 import com.tangem.utils.Provider
 import com.tangem.utils.coroutines.*
 import com.tangem.utils.extensions.isSingleItem
+import com.tangem.utils.extensions.orZero
 import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.*
 import kotlinx.coroutines.flow.*
@@ -236,7 +238,7 @@ internal class StakingViewModel @Inject constructor(
                 val validatorState = confirmationState.validatorState as? ValidatorState.Content
                     ?: error("No validator provided")
                 val amountState = value.amountState as? AmountState.Data ?: error("No amount provided")
-                val amountValue = amountState.amountTextField.cryptoAmount.value ?: error("No amount value")
+
                 val fee = (confirmationState.feeState as? FeeState.Content)?.fee ?: error("No fee provided")
                 val defaultAddress = cryptoCurrencyStatus.value.networkAddress?.defaultAddress?.value
                     ?: error("No available address")
@@ -248,7 +250,7 @@ internal class StakingViewModel @Inject constructor(
                     params = ActionParams(
                         actionCommonType = value.actionType,
                         integrationId = yield.id,
-                        amount = amountValue,
+                        amount = getAmount(amountState, fee, confirmationState.reduceAmountBy),
                         address = defaultAddress,
                         validatorAddress = validatorState.chosenValidator.address,
                         token = yield.token,
@@ -893,6 +895,20 @@ internal class StakingViewModel @Inject constructor(
             .getOrElse { false }
     }
 
+    private fun getAmount(amountState: AmountState.Data, fee: Fee, reduceAmountBy: BigDecimal?): BigDecimal {
+        val amountValue = amountState.amountTextField.cryptoAmount.value ?: error("No amount value")
+        val feeValue = fee.amount.value ?: error("No fee value")
+        val isEnterAction = stateController.value.actionType == StakingActionCommonType.ENTER
+
+        return checkAndCalculateSubtractedAmount(
+            isAmountSubtractAvailable = isAmountSubtractAvailable && isEnterAction,
+            cryptoCurrencyStatus = cryptoCurrencyStatus,
+            amountValue = amountValue,
+            feeValue = feeValue,
+            reduceAmountBy = reduceAmountBy.orZero(),
+        )
+    }
+
     private data class FullTransactionData(
         val stakeKitTransaction: StakingTransaction,
         val tangemTransaction: TransactionData.Compiled,