diff --git a/data/staking/src/main/java/com/tangem/data/staking/converters/YieldBalanceConverter.kt b/data/staking/src/main/java/com/tangem/data/staking/converters/YieldBalanceConverter.kt index 70577561ed..90b35354b7 100644 --- a/data/staking/src/main/java/com/tangem/data/staking/converters/YieldBalanceConverter.kt +++ b/data/staking/src/main/java/com/tangem/data/staking/converters/YieldBalanceConverter.kt @@ -3,6 +3,7 @@ package com.tangem.data.staking.converters import com.tangem.data.staking.converters.action.PendingActionConverter import com.tangem.datasource.api.stakekit.models.response.model.YieldBalanceWrapperDTO import com.tangem.domain.staking.model.stakekit.* +import com.tangem.domain.staking.model.stakekit.action.StakingActionType import com.tangem.utils.converter.Converter internal class YieldBalanceConverter : Converter { @@ -27,7 +28,10 @@ internal class YieldBalanceConverter : Converter { override fun transform(prevState: StakingUiState): StakingUiState { - val isTron = isTron(cryptoCurrencyStatus.currency.network.id.value) - val actionType = if (activeStake.pendingActions.isEmpty() || isTron) { + val isTronStakedBalance = isTronStakedBalance( + networkId = cryptoCurrencyStatus.currency.network.id, + activeStake = activeStake, + ) + val actionType = if (activeStake.pendingActions.isEmpty() || isTronStakedBalance) { StakingActionCommonType.EXIT } else { StakingActionCommonType.PENDING_OTHER @@ -21,4 +26,8 @@ internal class ActionTypeActiveStakeTransformer( return prevState.copy(actionType = actionType) } + + private fun isTronStakedBalance(networkId: Network.ID, activeStake: BalanceState): Boolean { + return isTron(networkId.value) && activeStake.type == BalanceType.STAKED + } } \ No newline at end of file