Merge pull request #12 from formysister/update-class-structure

bitcoin- getBalance() added, testcase- getBalance() added

src/wallet_class/bitcoin/index.ts

@@ -3,6 +3,8 @@ import * as bip32 from 'bip32'
 import * as bitcoin from 'bitcoinjs-lib'
 import { PrivateKey } from 'bitcore-lib';
 
+import axios from 'axios';
+
 import { BtcNetwork, BtcWallet, BtcAccount } from "../../type/type"
 import { BITCOIN_DEFAULT, BTC_MAINNET, BTC_REGTEST, BTC_TESTNET } from "../../constant";
 
@@ -167,6 +169,21 @@ class BitCoinWallet {
             privateKey: privateKey,
         }
     }
+
+    /**
+     * 
+     * @param address 
+     * @returns {Number}
+     */
+    getBalance = async (address: string): Promise<number> => {
+        try {
+            const response = await axios.get(`https://blockchain.info/q/addressbalance/${address}/`)
+
+            return response.data
+        } catch (error) {
+            throw error
+        }
+    }
 }
 
 export default BitCoinWallet

test/sample_data.ts

@@ -8,6 +8,9 @@ const SAMPLE_DATA = {
         SAMPLE_TOKEN_ADDRESS: "0xa8De46D005919aa6a8851C85E60f09884869c50a",
         SAMPLE_721_NFT_ADDRESS: "0x7EC3f3CE817F5A5CA3e285022573Df1b1B4DeF7d",
         SAMPLE_1155_NFT_ADDRESS: "0xA8bf99a3aDA507bCD0a8ce8982d3d94a4896440B"
+    },
+    BITCOIN: {
+        SAMPLE_ADDRESS: 'bc1pt4da3t0n347v7y95cpnqfw7el7u6uk47d9uvsxvntnkr7nsjwj7qyx8avw'
     }
 }
 

test/wallet.test.ts

@@ -136,5 +136,11 @@ describe("Wallet Test", () => {
             expect(typeof wallet.privateKey).toBe('string')
             expect(typeof wallet.address).toBe('string')
         })
+
+        it("getBalance", async () => {
+            const balance = await bitcoinWallet.getBalance(SAMPLE_DATA.BITCOIN.SAMPLE_ADDRESS)
+
+            console.log(typeof balance)
+        })
     })
 })