GiftedHacker/tangem-app-android-audited
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Updated on 2026-08-14

Browse source
This commit is contained in:
Tangem 2024-11-11 11:40:19 +02:00
parent cc81fb6d52
commit 1df2f17b79
1 changed files with 43 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

45
domain/staking/src/main/java/com/tangem/domain/staking/InvalidatePendingTransactionsUseCase.kt
View file

@ -8,6 +8,7 @@ import com.tangem.domain.staking.model.stakekit.action.StakingAction
import com.tangem.domain.staking.model.stakekit.action.StakingActionType
import com.tangem.domain.staking.repositories.StakingErrorResolver
import com.tangem.utils.extensions.isEqualTo
import java.math.BigDecimal
import java.util.UUID
class InvalidatePendingTransactionsUseCase(
@ -50,9 +51,15 @@ class InvalidatePendingTransactionsUseCase(
StakingActionType.UNLOCK_LOCKED -> {
modifyBalancesByStatus(balances, action, BalanceType.LOCKED)
}
StakingActionType.UNSTAKE, StakingActionType.RESTAKE -> {
StakingActionType.RESTAKE -> {
modifyBalancesByStatus(balances, action, BalanceType.STAKED)
}
StakingActionType.UNSTAKE -> {
val isFullUnstake = modifyBalancesByStatus(balances, action, BalanceType.STAKED)
if (!isFullUnstake) {
processPartialUnstake(balances, action)
}
}
else -> {
// intentionally do nothing
}
@ -86,12 +93,19 @@ class InvalidatePendingTransactionsUseCase(
)
}
private fun modifyBalancesByStatus(balances: MutableList<BalanceItem>, action: StakingAction, type: BalanceType) {
private fun modifyBalancesByStatus(
balances: MutableList<BalanceItem>,
action: StakingAction,
type: BalanceType,
): Boolean {
val index = findBalanceIndex(balances, action, type)
if (index != -1) {
balances[index] = balances[index].copy(isPending = true)
return true
}
return false
}
private fun findBalanceIndex(balances: MutableList<BalanceItem>, action: StakingAction, type: BalanceType): Int {
@ -99,4 +113,31 @@ class InvalidatePendingTransactionsUseCase(
!it.isPending && it.amount isEqualTo action.amount && it.type == type
}
}
private fun processPartialUnstake(balances: MutableList<BalanceItem>, action: StakingAction) {
val (index, pendingActionAmount) = findPartialUnstake(balances, action)
if (index != -1) {
val amount = balances[index].amount
balances[index] = balances[index].copy(
amount = amount - pendingActionAmount,
) // remnants of real one
balances.add(
balances[index].copy(
amount = pendingActionAmount,
isPending = true,
),
) // pending with amount from action
}
}
private fun findPartialUnstake(balances: MutableList<BalanceItem>, action: StakingAction): Pair<Int, BigDecimal> {
val index = balances.indexOfFirst {
!it.isPending && action.amount < it.amount &&
it.type == BalanceType.STAKED &&
it.validatorAddress == action.validatorAddress
}
return index to action.amount
}
}