feat: add external wallet balance option

bin/index.ts

@@ -9,10 +9,11 @@ import chalk from "chalk";
 import { deployCommand } from "../src/commands/deploy.js";
 import { verifyCommand } from "../src/commands/verify.js";
 import { ReadContract } from "../src/commands/contract.js";
+import { Address } from "viem";
 
 interface CommandOptions {
   testnet?: boolean;
-  address?: string;
+  address?: Address;
   value?: string;
   txid?: string;
   abi?: string;
@@ -55,8 +56,9 @@ program
   .command("balance")
   .description("Check the balance of the saved wallet")
   .option("-t, --testnet", "Check the balance on the testnet")
+  .option("-a, --address <address>", "Check the balance of a specific address")
   .action(async (options: CommandOptions) => {
-    await balanceCommand(!!options.testnet);
+    await balanceCommand(!!options.testnet, options.address);
   });
 
 program

package.json

@@ -11,6 +11,7 @@
     "build": "tsc",
     "wallet": "pnpm run build && node dist/bin/index.js wallet",
     "balance": "pnpm run build && node dist/bin/index.js balance",
+    "balance-of": "pnpm run build && node dist/bin/index.js balance -t --address 0x02C8345B2DF9Ff6122b0bE7B79cB219d956bd701",
     "transfer": "pnpm run build && node dist/bin/index.js transfer --testnet --address 0xa5f45f5bddefC810C48aCC1D5CdA5e5a4c6BC59E --value 0.001",
     "tx-status": "pnpm run build && node dist/bin/index.js tx --testnet --txid 0x876a0a9b167889350c41930a4204e5d9acf5704a7f201447a337094189af961c4"
   },

src/commands/balance.ts

@@ -2,34 +2,42 @@ import ViemProvider from "../utils/viemProvider.js";
 import fs from "fs";
 import path from "path";
 import chalk from "chalk";
-
+import { Address, isAddress } from "viem";
 const walletFilePath = path.join(process.cwd(), "rootstock-wallet.json");
 
-export async function balanceCommand(testnet: boolean) {
+export async function balanceCommand(testnet: boolean, address?: Address) {
   try {
-    if (!fs.existsSync(walletFilePath)) {
-      console.log(
-        chalk.red("🚫 No saved wallet found. Please create a wallet first.")
-      );
-      return;
-    }
+    let targetAddress: Address;
 
-    const walletData = JSON.parse(fs.readFileSync(walletFilePath, "utf8"));
-    const { address } = walletData;
-
-    if (!address) {
-      console.log(chalk.red("⚠️ No valid address found in the saved wallet."));
-      return;
+    if (address) {
+      if (!isAddress(address)) {
+        console.log(chalk.red("🚫 Invalid address provided"));
+        return;
+      }
+      targetAddress = address;
+    } else {
+      if (!fs.existsSync(walletFilePath)) {
+        console.log(chalk.red("🚫 No saved wallet found"));
+        return;
+      }
+      const { address: savedAddress } = JSON.parse(
+        fs.readFileSync(walletFilePath, "utf8")
+      );
+      if (!savedAddress) {
+        console.log(chalk.red("⚠️ Invalid wallet data"));
+        return;
+      }
+      targetAddress = savedAddress;
     }
 
     const provider = new ViemProvider(testnet);
     const client = await provider.getPublicClient();
 
-    const balance = await client.getBalance({ address });
+    const balance = await client.getBalance({ address: targetAddress });
 
     const rbtcBalance = Number(balance) / 10 ** 18;
 
-    console.log(chalk.white(`📄 Wallet Address:`), chalk.green(address));
+    console.log(chalk.white(`📄 Wallet Address:`), chalk.green(targetAddress));
     console.log(
       chalk.white(`🌐 Network:`),
       chalk.green(testnet ? "Rootstock Testnet" : "Rootstock Mainnet")