balances.ts

import { BatchCall } from "@airswap/libraries";
import { toDecimalString } from "@airswap/utils";
import { Command } from "@oclif/command";
import { getTable } from "console.table";
import { ethers } from "ethers";
import { cancelled } from "../lib/prompt";
import * as utils from "../lib/utils";
import { getWallet } from "../lib/wallet";

export default class Balances extends Command {
	public static description = "display token balances";
	public async run() {
		try {
			const wallet = await getWallet(this);
			const chainId = (await wallet.provider.getNetwork()).chainId;
			const metadata = await utils.getMetadata(this, chainId);
			utils.displayDescription(this, Balances.description, chainId);

			const startTime = Date.now();

			const addresses = Object.keys(metadata.byAddress);
			for (let i = addresses.length; i >= 0; i--) {
				if (!ethers.utils.isAddress(addresses[i])) {
					addresses.splice(i, 1);
				}
			}

			this.log(`Checking balances for ${addresses.length} tokens...\n`);

			const batchCallContract = BatchCall.getContract(wallet, chainId);
			const chunk = 750;
			const count = addresses.length;
			let balances = [];
			let index = 0;
			while (index < count) {
				balances = balances.concat(
					await batchCallContract.walletBalances(
						wallet.address,
						addresses.slice(index, index + chunk),
					),
				);
				index += chunk;
			}

			const result = [];
			for (let i = 0; i < addresses.length; i++) {
				const token = metadata.byAddress[addresses[i]];
				if (!balances[i].eq(0)) {
					const balanceDecimal = toDecimalString(
						balances[i],
						metadata.byAddress[token.address].decimals,
					);
					result.push({
						Token: token.symbol,
						Balance: balanceDecimal,
					});
				}
			}

			if (result.length) {
				this.log(getTable(result));
				this.log(
					`Balances displayed for ${result.length} of ${
						addresses.length
					} known tokens. (${Date.now() - startTime}ms)\n`,
				);
			} else {
				this.log(
					`The current account holds no balances in any of ${addresses.length} known tokens.\n`,
				);
			}
		} catch (e) {
			cancelled(e);
		}
	}
}