add API get balance rune, inscription

Author: phonglnDEV

packages/extension-base/src/constants/index.ts

@@ -10,6 +10,7 @@ export const CRON_AUTO_RECOVER_DOTSAMA_INTERVAL = 60000;
 export const CRON_AUTO_RECOVER_WEB3_INTERVAL = 90000;
 export const ACALA_REFRESH_CROWDLOAN_INTERVAL = 300000;
 export const ASTAR_REFRESH_BALANCE_INTERVAL = 60000;
+export const BITCOIN_REFRESH_BALANCE_INTERVAL = 600000;
 export const SUB_TOKEN_REFRESH_BALANCE_INTERVAL = 60000;
 export const CRON_REFRESH_NFT_INTERVAL = 7200000;
 export const CRON_REFRESH_MKT_CAMPAIGN_INTERVAL = 15 * BASE_MINUTE_INTERVAL;

packages/extension-base/src/services/balance-service/helpers/subscribe/bitcoin.ts

@@ -1,10 +1,9 @@
 // Copyright 2019-2022 @subwallet/extension-base
 // SPDX-License-Identifier: Apache-2.0
 
-import { _AssetType, _ChainAsset, _ChainInfo } from '@subwallet/chain-list/types';
-import { BitcoinBalanceMetadata } from '@subwallet/extension-base/background/KoniTypes';
+import { APIItemState, BitcoinBalanceMetadata } from '@subwallet/extension-base/background/KoniTypes';
+import { BITCOIN_REFRESH_BALANCE_INTERVAL } from '@subwallet/extension-base/constants';
 import { _BitcoinApi } from '@subwallet/extension-base/services/chain-service/types';
-import { _getChainNativeTokenSlug } from '@subwallet/extension-base/services/chain-service/utils';
 import { BalanceItem, UtxoResponseItem } from '@subwallet/extension-base/types';
 import { filteredOutTxsUtxos, getInscriptionUtxos, getRuneUtxos } from '@subwallet/extension-base/utils';
 import BigN from 'bignumber.js';
@@ -41,7 +40,7 @@ export const getTransferableBitcoinUtxos = async (bitcoinApi: _BitcoinApi, addre
   }
 };
 
-async function getBitcoinBalance (bitcoinApi: _BitcoinApi, addresses: string[]): Promise<BalanceItem[]> {
+async function getBitcoinBalance (bitcoinApi: _BitcoinApi, addresses: string[]) {
   return await Promise.all(addresses.map(async (address) => {
     try {
       const [filteredUtxos, addressSummaryInfo] = await Promise.all([
@@ -51,7 +50,9 @@ async function getBitcoinBalance (bitcoinApi: _BitcoinApi, addresses: string[]):
 
       console.log('addressSummaryInfo', addressSummaryInfo);
       const bitcoinBalanceMetadata = {
-        inscriptionCount: addressSummaryInfo.total_inscription
+        inscriptionCount: addressSummaryInfo.total_inscription,
+        runeBalance: addressSummaryInfo.balance_rune,
+        inscriptionBalance: addressSummaryInfo.balance_inscription
       } as BitcoinBalanceMetadata;
 
       let balanceValue = new BigN(0);
@@ -70,29 +71,57 @@ async function getBitcoinBalance (bitcoinApi: _BitcoinApi, addresses: string[]):
       return {
         balance: '0',
         bitcoinBalanceMetadata: {
-          inscriptionCount: 0
+          inscriptionCount: 0,
+          runeBalance: 0,
+          inscriptionBalance: 0
         }
       };
     }
   }));
 }
 
-export const subscribeBitcoinBalance = async (addresses: string[], bitcoinApi: _BitcoinApi, callback: (rs: BalanceItem[]) => void) => {
-  console.log('subscribeBitcoinBalanceBalance', addresses);
-
-  const getBalance = async (): Promise<BalanceItem[]> => {
-    try {
-      const balances = await getBitcoinBalance(bitcoinApi, addresses);
-
-      callback(balances);
-    } catch (e) {
-      console.error('Error while fetching cardano balance', e);
-    }
+export function subscribeBitcoinBalance (addresses: string[], bitcoinApi: _BitcoinApi, callback: (rs: BalanceItem[]) => void): () => void {
+  const getBalance = () => {
+    getBitcoinBalance(bitcoinApi, addresses)
+      .then((balances) => {
+        return balances.map(({ balance, bitcoinBalanceMetadata }, index): BalanceItem => {
+          return {
+            address: addresses[index],
+            tokenSlug: 'bitcoin-NATIVE-BTC',
+            state: APIItemState.READY,
+            free: balance,
+            locked: (
+              parseInt(bitcoinBalanceMetadata.runeBalance.toString()) +
+              parseInt(bitcoinBalanceMetadata.inscriptionBalance.toString())
+            ).toString(),
+            metadata: bitcoinBalanceMetadata
+          };
+        });
+      })
+      .catch((e) => {
+        console.error('Error on get Bitcoin balance with token bitcoin', e);
+
+        return addresses.map((address): BalanceItem => {
+          return {
+            address: address,
+            tokenSlug: 'bitcoin',
+            state: APIItemState.READY,
+            free: '0',
+            locked: '0'
+          };
+        });
+      })
+      .then((items) => {
+        callback(items);
+      })
+      .catch(console.error);
   };
 
-  await getBalance();
+  const interval = setInterval(getBalance, BITCOIN_REFRESH_BALANCE_INTERVAL);
+
+  getBalance();
 
   return () => {
-    console.log('unsub');
+    clearInterval(interval);
   };
-};
+}

packages/extension-base/src/services/chain-service/handler/bitcoin/strategy/BlockStream/types.ts

@@ -34,7 +34,9 @@ export interface BitcoinAddressSummaryInfo {
     tx_count: number
   },
   balance: number,
-  total_inscription: number
+  total_inscription: number,
+  balance_rune: string,
+  balance_inscription: string,
 }
 
 // todo: combine RunesByAddressResponse & RunesCollectionInfoResponse