fix: bug with signer

leather-io · Apr 9, 2024 · 2cc50d2 · 2cc50d2
1 parent cbb7159
commit 2cc50d2
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/app/components/balance-btc.tsx b/src/app/components/balance-btc.tsx
@@ -1,9 +1,17 @@
 import { formatMoney } from '@app/common/money/format-money';
-import { useCurrentNativeSegwitAddressBalance } from '@app/query/bitcoin/balance/btc-native-segwit-balance.hooks';
 import { Caption } from '@app/ui/components/typography/caption';
 
-export function BtcBalance() {
-  const { balance } = useCurrentNativeSegwitAddressBalance();
+import { BitcoinNativeSegwitAccountLoader } from './account/bitcoin-account-loader';
+import { BitcoinBalanceLoader } from './balance/bitcoin-balance-loader';
 
-  return <Caption>{formatMoney(balance)}</Caption>;
+export function BtcBalance() {
+  return (
+    <BitcoinNativeSegwitAccountLoader current>
+      {signer => (
+        <BitcoinBalanceLoader address={signer.address}>
+          {balance => <Caption>{formatMoney(balance.balance)}</Caption>}
+        </BitcoinBalanceLoader>
+      )}
+    </BitcoinNativeSegwitAccountLoader>
+  );
 }