mango-v4/ts/client/src/scripts/mb-liqtest-settle-and-close-all.ts

import { AnchorProvider, Wallet } from '@project-serum/anchor';
import { Connection, Keypair, PublicKey } from '@solana/web3.js';
import fs from 'fs';
import { MangoClient } from '../client';
import { MANGO_V4_ID } from '../constants';

//
// This script tries to withdraw all positive balances for all accounts
// by MANGO_MAINNET_PAYER_KEYPAIR in the group.
//

const GROUP_NUM = Number(process.env.GROUP_NUM || 200);
const CLUSTER_URL =
  process.env.CLUSTER_URL ||
  'https://mango.rpcpool.com/946ef7337da3f5b8d3e4a34e7f88';
const MANGO_MAINNET_PAYER_KEYPAIR =
  process.env.MANGO_MAINNET_PAYER_KEYPAIR || '';

async function main() {
  const options = AnchorProvider.defaultOptions();
  const connection = new Connection(CLUSTER_URL!, options);

  const admin = Keypair.fromSecretKey(
    Buffer.from(
      JSON.parse(fs.readFileSync(MANGO_MAINNET_PAYER_KEYPAIR, 'utf-8')),
    ),
  );
  const userWallet = new Wallet(admin);
  const userProvider = new AnchorProvider(connection, userWallet, options);
  const client = await MangoClient.connect(
    userProvider,
    'mainnet-beta',
    MANGO_V4_ID['mainnet-beta'],
    {},
    'get-program-accounts'
  );
  console.log(`User ${userWallet.publicKey.toBase58()}`);

  // const groups = await client.getGroupsForCreator(admin.publicKey);
  // console.log(
  //   'groups: ',
  //   groups.map((g) => g.publicKey.toString() + '  ' + g.groupNum.toString()),
  // );

  const group = await client.getGroupForCreator(admin.publicKey, GROUP_NUM);
  console.log(group.toString());

  let accounts = await client.getMangoAccountsForOwner(group, admin.publicKey);
  for (let account of accounts) {
    console.log(`settling borrows on account: ${account}`);

    // first, settle all borrows
    for (let token of account.tokensActive()) {
      const bank = group.getFirstBankByTokenIndex(token.tokenIndex);
      const amount = token.balance(bank).toNumber();
      if (amount < 0) {
        try {
          await client.tokenDepositNative(
            group,
            account,
            bank.mint,
            Math.ceil(-amount),
          );
          await account.reload(client, group);
        } catch (error) {
          console.log(
            `failed to deposit ${bank.name} into ${account.publicKey}: ${error}`,
          );
        }
      }
    }
  }

  accounts = await client.getMangoAccountsForOwner(group, admin.publicKey);
  for (let account of accounts) {
    console.log(`withdrawing deposits of account: ${account}`);

    // withdraw all funds
    for (let token of account.tokensActive()) {
      const bank = group.getFirstBankByTokenIndex(token.tokenIndex);
      const amount = token.balance(bank).toNumber();
      if (amount > 0) {
        try {
          const allowBorrow = false;
          await client.tokenWithdrawNative(
            group,
            account,
            bank.mint,
            amount,
            allowBorrow,
          );
          await account.reload(client, group);
        } catch (error) {
          console.log(
            `failed to withdraw ${bank.name} from ${account.publicKey}: ${error}`,
          );
        }
      }
    }

    // close account
    try {
      console.log(`closing account: ${account}`);
      await client.closeMangoAccount(group, account);
    } catch (error) {
      console.log(`failed to close ${account.publicKey}: ${error}`);
    }
  }

  process.exit();
}

main();