Add total balance calculation with private

Feature/add firo network

Author: reubenyap

packages/extension/src/providers/bitcoin/types/bitcoin-network.ts

@@ -20,6 +20,34 @@ import type HaskoinAPI from '@/providers/bitcoin/libs/api';
 import type SSAPI from '@/providers/bitcoin/libs/api-ss';
 import type FiroAPI from '@/providers/bitcoin/libs/api-firo';
 import { NFTCollection } from '@/types/nft';
+import { calculateCurrentSparkBalance } from '@/libs/utils/updateAndSync/calculateCurrentSparkBalance';
+
+type PrivateBalanceResolver = () => Promise<string>;
+
+const privateBalanceResolvers: Record<string, PrivateBalanceResolver> = {
+  FIRO: calculateCurrentSparkBalance,
+};
+
+const getFormattedBalanceTotal = async ({
+  networkName,
+  publicBalance,
+  decimals,
+}: {
+  networkName: string;
+  publicBalance: string;
+  decimals: number;
+}) => {
+  const resolvePrivateBalance = privateBalanceResolvers[networkName.toUpperCase()];
+  if (!resolvePrivateBalance) return '';
+
+  const privateBalance = await resolvePrivateBalance();
+  const privateBalanceFormatted = fromBase(privateBalance, decimals);
+  const totalBalance = new BigNumber(publicBalance)
+    .plus(privateBalanceFormatted)
+    .toString();
+
+  return formatFloatingPointValue(totalBalance).value;
+};
 
 export enum PaymentType {
   P2PKH = 'p2pkh',
@@ -127,6 +155,12 @@ export class BitcoinNetwork extends BaseNetwork {
     const usdBalance = new BigNumber(userBalance).times(
       marketData[0]?.current_price ?? 0,
     );
+    const balanceTotal = await getFormattedBalanceTotal({
+      networkName: this.currencyName,
+      publicBalance: userBalance,
+      decimals: this.decimals,
+    });
+
     const nativeAsset: AssetsType = {
       balance: balance,
       balancef: formatFloatingPointValue(userBalance).value,
@@ -137,6 +171,7 @@ export class BitcoinNetwork extends BaseNetwork {
       symbol: this.currencyName,
       value: marketData[0]?.current_price?.toString() ?? '0',
       valuef: marketData[0]?.current_price?.toString() ?? '0',
+      balanceTotal,
       contract: '',
       decimals: this.decimals,
       sparkline: marketData.length

packages/extension/src/types/provider.ts

@@ -219,6 +219,7 @@ export interface AssetsType {
   icon: string;
   balance: string;
   balancef: string;
+  balanceTotal?: string;
   balanceUSD: number;
   balanceUSDf: string;
   value: string;

packages/extension/src/ui/action/views/asset-detail-view/index.vue

@@ -56,14 +56,21 @@
         </div>
         <div class="asset-detail-view__balance-amount">
           <span class="asset-detail-view__balance-value">{{
-            token.balancef
+            token.balanceTotal || token.balancef
           }}</span>
           <span class="asset-detail-view__balance-symbol">{{
             token.symbol
           }}</span>
         </div>
         <p class="asset-detail-view__balance-usd">
-          ≈ {{ $filters.parseCurrency(token.balanceUSDf) }}
+          ≈
+          {{
+            $filters.parseCurrency(
+              token.balanceTotal
+                ? Number(token.balanceTotal) * Number(token.valuef)
+                : token.balanceUSDf,
+            )
+          }}
         </p>
       </div>
 

packages/extension/src/ui/action/views/network-assets/components/network-assets-item.vue

@@ -16,7 +16,7 @@
             </h4>
           </tooltip>
           <p class="asset-item__balance">
-            {{ token.balancef }}
+            {{ token.balanceTotal || token.balancef }}
             <span v-if="token.symbol.length <= 6" class="asset-item__symbol">
               {{ token.symbol }}
             </span>
@@ -48,7 +48,7 @@
       <!-- Value -->
       <div class="asset-item__value">
         <h4 class="asset-item__usd">
-          {{ $filters.parseCurrency(token.balanceUSD) }}
+          {{ $filters.parseCurrency(displayBalanceUSD) }}
         </h4>
         <p class="asset-item__price">
           {{ $filters.parseCurrency(token.value) }}
@@ -169,6 +169,19 @@ const isCustomToken = computed(() => {
   );
 });
 
+const displayBalanceUSD = computed(() => {
+  const hasExtraBalance = !!props.token.balanceTotal;
+  if (!hasExtraBalance) return props.token.balanceUSD;
+
+  const totalBalance = Number(props.token.balanceTotal);
+  const tokenPrice = Number(props.token.value);
+  if (!Number.isFinite(totalBalance) || !Number.isFinite(tokenPrice)) {
+    return props.token.balanceUSD;
+  }
+
+  return totalBalance * tokenPrice;
+});
+
 const emit = defineEmits<{
   (e: 'update:tokens'): void;
   (e: 'open:buy-action', token: AssetsType): void;