From 555ee2535c20648ad9d14284d34b931918187904 Mon Sep 17 00:00:00 2001
From: Tangem <git@tangem.com>
Date: Thu, 28 Dec 2023 13:46:27 +0300
Subject: [PATCH] Updated on 2026-08-14

---
 .../feature/swap/domain/SwapInteractorImpl.kt | 43 +++++++++++--------
 1 file changed, 25 insertions(+), 18 deletions(-)

diff --git a/features/swap/domain/src/main/java/com/tangem/feature/swap/domain/SwapInteractorImpl.kt b/features/swap/domain/src/main/java/com/tangem/feature/swap/domain/SwapInteractorImpl.kt
index a1420cc27f..38f55b3561 100644
--- a/features/swap/domain/src/main/java/com/tangem/feature/swap/domain/SwapInteractorImpl.kt
+++ b/features/swap/domain/src/main/java/com/tangem/feature/swap/domain/SwapInteractorImpl.kt
@@ -852,19 +852,12 @@ internal class SwapInteractorImpl @Inject constructor(
         amount: SwapAmount,
         fromToken: CryptoCurrency,
     ): IncludeFeeInAmount {
-        if (fromToken is CryptoCurrency.Token) {
-            return IncludeFeeInAmount.Excluded
-        }
-
         val tokenForFeeBalance =
             userWalletManager.getNativeTokenBalance(
                 networkId,
                 fromToken.network.derivationPath.value,
             ) ?: ProxyAmount.empty()
 
-        if (amount.value > tokenForFeeBalance.value) {
-            return IncludeFeeInAmount.BalanceNotEnough
-        }
         val feeValue = when (txFee) {
             TxFeeState.Empty -> BigDecimal.ZERO
             is TxFeeState.MultipleFeeState -> txFee.priorityFee.feeValue
@@ -872,19 +865,33 @@ internal class SwapInteractorImpl @Inject constructor(
         }
 
         val amountWithFee = amount.value + feeValue
-        return if (amountWithFee < tokenForFeeBalance.value) {
-            IncludeFeeInAmount.Excluded
-        } else {
-            if (feeValue < amount.value) {
-                IncludeFeeInAmount.Included(
-                    SwapAmount(
-                        tokenForFeeBalance.value - feeValue,
-                        transactionManager.getNativeTokenDecimals(networkId),
-                    ),
-                )
-            } else {
+
+        return when {
+            fromToken is CryptoCurrency.Token -> {
+                if (feeValue > tokenForFeeBalance.value) {
+                    IncludeFeeInAmount.BalanceNotEnough
+                } else {
+                    IncludeFeeInAmount.Excluded
+                }
+            }
+            amount.value > tokenForFeeBalance.value -> {
                 IncludeFeeInAmount.BalanceNotEnough
             }
+            amountWithFee < tokenForFeeBalance.value -> {
+                IncludeFeeInAmount.Excluded
+            }
+            else -> {
+                if (feeValue < amount.value) {
+                    IncludeFeeInAmount.Included(
+                        SwapAmount(
+                            tokenForFeeBalance.value - feeValue,
+                            transactionManager.getNativeTokenDecimals(networkId),
+                        ),
+                    )
+                } else {
+                    IncludeFeeInAmount.BalanceNotEnough
+                }
+            }
         }
     }