Fix ADA temporarily out-dated balance bug

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Unreleased
 
+- fixed: (ADA) Improved balance calculation by deriving from UTXO state.
 - fixed: (ADA) Race condition between network queries and local transaction processing causing incorrect UTXO state.
 
 ## 4.26.2 (2024-10-29)

src/cardano/CardanoEngine.ts

@@ -238,14 +238,14 @@ export class CardanoEngine extends CurrencyEngine<
       })
       const clean = asKoiosBalance(raw)
 
-      const mainnetBal = clean[0]?.balance ?? '0'
-      this.updateBalance(this.currencyInfo.currencyCode, mainnetBal)
-
       // Merge unseen utxos into wallet state:
       const networkUtxos = clean[0]?.utxo_set ?? []
       for (const utxo of networkUtxos) {
         this.addUnseenUtxo(utxo)
       }
+
+      // Network balance may be out of date, so we'll calculate it from utxos:
+      this.updateBalanceFromUtxos(this.currencyInfo.currencyCode)
     } catch (e) {
       this.log.warn('queryBalance error: ', e)
     }
@@ -534,6 +534,9 @@ export class CardanoEngine extends CurrencyEngine<
         })
       }
     })
+
+    // Update balance incase the UTXO set changed:
+    this.updateBalanceFromUtxos(this.currencyInfo.currencyCode)
   }
 
   async signTx(
@@ -635,6 +638,11 @@ export class CardanoEngine extends CurrencyEngine<
     )
   }
 
+  private updateBalanceFromUtxos(currencyCode: string): void {
+    const balance = this.utxos.reduce((acc, utxo) => add(acc, utxo.value), '0')
+    this.updateBalance(currencyCode, balance)
+  }
+
   getStakeAddress = async (): Promise<string> => {
     const { bech32Address } = asSafeCardanoWalletInfo(this.walletInfo).keys