From c28b40899d95c0f8b6f9ff7ba14adf206d54dc48 Mon Sep 17 00:00:00 2001
From: Oscar Mira <valldrac@molly.im>
Date: Wed, 28 Feb 2024 01:25:54 +0100
Subject: [PATCH] lib: add account filtering for balance calculation

---
 lib/android/src/main/kotlin/im/molly/monero/Balance.kt | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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<TimeLocked<Enote>>.calculateBalance(): Balance {
+fun Iterable<TimeLocked<Enote>>.calculateBalance(
+    accountFilter: (owner: AccountAddress) -> Boolean = { true },
+): Balance {
     val lockedAmounts = mutableListOf<TimeLocked<MoneroAmount>>()
 
     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 {