From 09aa1133a6851356cfa10733f46166a743cfeb9e Mon Sep 17 00:00:00 2001
From: Thebora Kompanioni <theborakompanioni@users.noreply.github.com>
Date: Tue, 6 Aug 2024 13:32:45 +0200
Subject: [PATCH] feat: unit and visibility toggle on main wallet view (#806)

---
 src/components/MainWalletView.tsx | 33 ++++++++++++++++++++-----------
 1 file changed, 22 insertions(+), 11 deletions(-)

diff --git a/src/components/MainWalletView.tsx b/src/components/MainWalletView.tsx
index 4b636c310..efadf1d18 100644
--- a/src/components/MainWalletView.tsx
+++ b/src/components/MainWalletView.tsx
@@ -148,18 +148,29 @@ export default function MainWalletView({ wallet }: MainWalletViewProps) {
         <rb.Row>
           <WalletHeaderRescanning walletName={wallet.displayName} isLoading={isLoading} />
         </rb.Row>
+      ) : !currentWalletInfo || isLoading ? (
+        <rb.Row>
+          <WalletHeaderPlaceholder />
+        </rb.Row>
       ) : (
-        <rb.Row onClick={() => settingsDispatch({ showBalance: !settings.showBalance })} className="cursor-pointer">
-          {!currentWalletInfo || isLoading ? (
-            <WalletHeaderPlaceholder />
-          ) : (
-            <WalletHeader
-              walletName={wallet.displayName}
-              balance={currentWalletInfo.balanceSummary.calculatedTotalBalanceInSats}
-              unit={settings.unit}
-              showBalance={settings.showBalance}
-            />
-          )}
+        <rb.Row
+          className="cursor-pointer"
+          onClick={() => {
+            if (!settings.showBalance) {
+              settingsDispatch({ unit: 'BTC', showBalance: true })
+            } else if (settings.unit === 'BTC') {
+              settingsDispatch({ unit: 'sats', showBalance: true })
+            } else {
+              settingsDispatch({ unit: 'BTC', showBalance: false })
+            }
+          }}
+        >
+          <WalletHeader
+            walletName={wallet.displayName}
+            balance={currentWalletInfo.balanceSummary.calculatedTotalBalanceInSats}
+            unit={settings.unit}
+            showBalance={settings.showBalance}
+          />
         </rb.Row>
       )}
       <div className={styles.walletBody}>