diff --git a/lib/android/src/main/kotlin/im/molly/monero/Balance.kt b/lib/android/src/main/kotlin/im/molly/monero/Balance.kt index e83bed6..83ff62f 100644 --- a/lib/android/src/main/kotlin/im/molly/monero/Balance.kt +++ b/lib/android/src/main/kotlin/im/molly/monero/Balance.kt @@ -23,12 +23,14 @@ data class Balance( } } -fun Iterable>.calculateBalance(): Balance { +fun Iterable>.calculateBalance( + accountFilter: (owner: AccountAddress) -> Boolean = { true }, +): Balance { val lockedAmounts = mutableListOf>() var pendingAmount = MoneroAmount.ZERO - for (timeLocked in filter { !it.value.spent }) { + for (timeLocked in filter { !it.value.spent && accountFilter(it.value.owner) }) { if (timeLocked.value.age == 0) { pendingAmount += timeLocked.value.amount } else {