hodl-wallet 1.8.0 → 1.8.2

package/network/lib/BaseNetwork.js

@@ -47,7 +47,26 @@ export default class BaseNetwork {
         throw new Error("Method 'validateMnemonic' must be implemented.");
     }
 
-    // Add after line 51
+    async getTokenBalance(address, tokenSymbol) {
+        throw new Error("Method 'getTokenBalance' must be implemented.");
+    }
+
+    async getTokenBalances(address) {
+        const balances = [];
+        
+        // Get native token balance
+        const nativeBalance = await this.getBalance(address);
+        balances.push([this.config.nativeToken, parseFloat(nativeBalance)]);
+
+        // Get balances for all configured tokens
+        for (const symbol of Object.keys(this.config.tokens)) {
+            const tokenBalance = await this.getTokenBalance(address, symbol);
+            balances.push([symbol, tokenBalance]);
+        }
+
+        return balances;
+    }
+
     async sendSignedTransaction(signedTx) {
         throw new Error("Method 'sendSignedTransaction' must be implemented.");
     }

package/network/lib/BitcoinNetwork.js

@@ -30,14 +30,11 @@ export default class BitcoinNetwork extends BaseNetwork {
         }
     }
 
-
-    async getTokenBalances(address) {
-        const balances = [];
-
-        const nativeBalance = await this.getBalance(address);
-        balances.push([this.config.nativeToken, nativeBalance]);
-
-        return balances;
+    async getTokenBalance(address, tokenSymbol) {
+        if (tokenSymbol === this.config.nativeToken) {
+            return await this.getBalance(address);
+        }
+        throw new Error(`Token ${tokenSymbol} not supported on Bitcoin network`);
     }
 
     async transfer(from, to, amount, options = {}) {

package/network/lib/TONNetwork.js

@@ -137,32 +137,6 @@ export default class TONNetwork extends BaseNetwork {
         }));
     }
 
-    async getTokenBalances(address) {
-        const balances = [];
-        
-        // Add native TON balance
-        try {
-            const nativeBalance = await this.getBalance(address);
-            balances.push([this.config.nativeToken, nativeBalance]);
-        } catch (error) {
-            console.error('Failed to get TON balance:', error);
-            balances.push([this.config.nativeToken, 0]);
-        }
-
-        // Add token balances (not implemented yet)
-        for (const tokenSymbol in this.config.tokens) {
-            try {
-                const tokenBalance = await this.getTokenBalance(address, tokenSymbol);
-                balances.push([tokenSymbol, tokenBalance]);
-            } catch (error) {
-                console.error(`Failed to get ${tokenSymbol} balance:`, error);
-                balances.push([tokenSymbol, 0]);
-            }
-        }
-        
-        return balances;
-    }
-
     async sendSignedTransaction(signedTx) {
         // For TON, the transaction is already sent and signedTx contains the result
         let transactionHash;

package/network/lib/Web3Network.js

@@ -98,30 +98,21 @@ export default class Web3Network extends BaseNetwork {
         return this.transfer(account, recipient, amount, { gasLimit, gasPrice });
     }
 
-    async getTokenBalances(address) {
-        const balances = [];
-        
-        // Get native token balance
-        const nativeBalance = await this.getBalance(address);
-        balances.push([this.config.nativeToken, parseFloat(nativeBalance)]);
-
-        // Get balances for all configured tokens
-        for (const [symbol, tokenConfig] of Object.entries(this.config.tokens)) {
-            const contract = new this.web3.eth.Contract(
-                ERC20_ABI, 
-                tokenConfig.address
-            );
-            const balance = await contract.methods.balanceOf(address).call();
-            const decimals = await contract.methods.decimals().call();
-
-            const formattedBalance = Number(
-                (BigInt(balance) * 100n) / (10n ** BigInt(decimals))
-            ) / 100;
-
-            balances.push([symbol, formattedBalance]);
+    async getTokenBalance(address, tokenSymbol) {
+        if (tokenSymbol === this.config.nativeToken) {
+            return await this.getBalance(address);
         }
 
-        return balances;
+        const tokenConfig = this.config.tokens[tokenSymbol];
+        if (!tokenConfig) throw new Error(`Token ${tokenSymbol} not supported`);
+
+        const contract = new this.web3.eth.Contract(ERC20_ABI, tokenConfig.address);
+        const balance = await contract.methods.balanceOf(address).call();
+        const decimals = await contract.methods.decimals().call();
+
+        return Number(
+            (BigInt(balance) * 100n) / (10n ** BigInt(decimals))
+        ) / 100;
     }
 
     async privateKeyToAccount(privateKey) {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "hodl-wallet",
-    "version": "1.8.0",
+    "version": "1.8.2",
     "description": "🧊 HODL Wallet - Fast CLI crypto wallet!",
     "author": "Martin Clasen",
     "repository": {
@@ -29,6 +29,7 @@
         "bep20",
         "bnb",
         "eth",
+        "ton",
         "binance",
         "trust",
         "blockchain",