Merge pull request #6517 from BitGo/add-trx-consolidation-amount-script

feat: add script for fetching TRX balance that can be consolidated

Author: Vijay-Jagannathan

examples/ts/trx/fetch-addresses-with-trx-balance-for-consolidation.ts

@@ -0,0 +1,51 @@
+/**
+ * Copyright 2024, BitGo, Inc.  All Rights Reserved.
+ */
+import { BitGoAPI } from '@bitgo/sdk-api';
+import { Trx } from '@bitgo/sdk-coin-trx';
+import * as fs from 'fs';
+require('dotenv').config({ path: '../../.env' });
+
+const bitgo = new BitGoAPI({
+  accessToken: '',
+  env: 'prod',
+});
+
+const coin = 'trx';
+bitgo.register(coin, Trx.createInstance);
+
+const walletId = '';
+
+async function main() {
+  const wallet = await bitgo.coin(coin).wallets().get({ id: walletId });
+  let prevId = undefined;
+  let index = 1;
+
+  do {
+    const addresses = await wallet.addresses({ includeBalances: true, prevId });
+    prevId = addresses.nextBatchPrevId;
+
+    for (const { address, balance, needsConsolidation, tokenConsolidationState } of addresses.addresses) {
+      const { tokens, spendableBalanceString } = balance;
+      const usdtBalance = tokens['trx:usdt']?.spendableBalanceString;
+
+      console.log(`Address ${index}: ${address}`);
+      console.log('USDT token balance: ', usdtBalance || 'Does not have USDT balance');
+      console.log('TRX balance: ', spendableBalanceString);
+
+      if (usdtBalance > 0 && spendableBalanceString > 36000000) {
+        const data = `${address}, USDT balance: ${usdtBalance}, TRX balance: ${spendableBalanceString}, V1: ${needsConsolidation}, V2: ${tokenConsolidationState['trx']}\n`;
+        fs.appendFileSync('addresses-with-usdt-balance-and-trx-greater-than-36.txt', data);
+      }
+
+      if ((!usdtBalance || usdtBalance <= 0) && spendableBalanceString > 1000000) {
+        const data = `${address}, USDT balance: ${usdtBalance}, TRX balance: ${spendableBalanceString}, V1: ${needsConsolidation}, V2: ${tokenConsolidationState['trx']}\n`;
+        fs.appendFileSync('addresses-without-usdt-and-trx-greater-than-1.txt', data);
+      }
+
+      index += 1;
+    }
+  } while (prevId !== undefined);
+}
+
+main().catch((e) => console.error(e));