[NayNay] Balance --all

CHANGELOG.md

@@ -22,6 +22,9 @@ Version header format: `[version] Name - year-month-day (entropy-core compatibil
 - TUI
   - animation on tui load (while entropy loads) [#288](https://github.com/entropyxyz/cli/pull/288)
 
+- Programmatic CLI
+  - added option for users to view all balances of admin accounts in the config [#310](https://github.com/entropyxyz/cli/pull/310)
+
 ### Changes
 
 - Shared

src/balance/command.ts

@@ -5,25 +5,40 @@ import { EntropyBalance } from "./main";
 import { endpointOption, cliWrite, loadEntropy } from "../common/utils-cli";
 import { findAccountByAddressOrName } from "../common/utils";
 import * as config from "../config";
+import { EntropyAccountConfig } from "src/config/types";
+import { formattedBalances } from "./utils";
 
 export function entropyBalanceCommand () {
   const balanceCommand = new Command('balance')
+  // account can no longer be a required argument if a user wishes to
+  // view the balances of all accounts
   balanceCommand
     .description('Command to retrieive the balance of an account on the Entropy Network')
-    .argument('<account>', [
+    .argument('[account]', [
       'The address an account address whose balance you want to query.',
       'Can also be the human-readable name of one of your accounts'
     ].join(' '))
     .addOption(endpointOption())
+    .option('--all', 'Get balances for all admin accounts in the config')
     .action(async (account, opts) => {
-      const entropy: Entropy = await loadEntropy(account, opts.endpoint)
-      const BalanceService = new EntropyBalance(entropy, opts.endpoint)
-
       const { accounts } = await config.get()
-      const address = findAccountByAddressOrName(accounts, account)?.address
+      // when trying to get the balance of all accounts, need to use a temporary address
+      // to initialize entropy in order to use substrate
+      const tempAddress = account || accounts[0].address
+      const entropy: Entropy = await loadEntropy(tempAddress, opts.endpoint)
+      const BalanceService = new EntropyBalance(entropy, opts.endpoint)
 
-      const balance = await BalanceService.getBalance(address)
-      cliWrite(`${balance.toLocaleString('en-US')} BITS`)
+      if (opts.all) {
+        // Balances for all admin accounts
+        const addresses: string[] = accounts.map((acct: EntropyAccountConfig) => acct.address)
+        const balances = formattedBalances(await BalanceService.getBalances(addresses))
+        cliWrite(balances)
+      } else {
+        // Balance for singular account
+        const address = findAccountByAddressOrName(accounts, account)?.address
+        const balance = await BalanceService.getBalance(address)
+        cliWrite(`${balance.toLocaleString('en-US')} BITS`)
+      }
       process.exit(0)
     })
 

src/balance/utils.ts

@@ -1 +1,9 @@
-export const hexToBigInt = (hexString: string) => BigInt(hexString)
+import { BalanceInfo } from "./types"
+
+export const hexToBigInt = (hexString: string) => BigInt(hexString)
+
+export function formattedBalances (balanceInfo: BalanceInfo) {
+  const balances = {}
+  Object.keys(balanceInfo).forEach(key => balances[key] = !balanceInfo[key].error ? `${balanceInfo[key].balance.toLocaleString('en-us')} BITS` : null)
+  return balances
+}