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

//
// example script to close accounts - banks, markets, group etc. which require admin to be the signer
//

const GROUP_NUM = Number(process.env.GROUP_NUM || 0);

export const DEVNET_MINTS = new Map([
  ['USDC', '8FRFC6MoGGkMFQwngccyu69VnYbzykGeez7ignHVAFSN'], // use devnet usdc
]);

async function main() {
  const options = AnchorProvider.defaultOptions();
  const connection = new Connection(
    'https://mango.devnet.rpcpool.com',
    options,
  );

  const admin = Keypair.fromSecretKey(
    Buffer.from(
      JSON.parse(fs.readFileSync(process.env.ADMIN_KEYPAIR!, 'utf-8')),
    ),
  );
  const adminWallet = new Wallet(admin);
  console.log(`Admin ${adminWallet.publicKey.toBase58()}`);
  const adminProvider = new AnchorProvider(connection, adminWallet, options);
  const client = await MangoClient.connect(
    adminProvider,
    'devnet',
    MANGO_V4_ID['devnet'],
    {
      idsSource: 'get-program-accounts',
    },
  );

  const group = await client.getGroupForCreator(admin.publicKey, GROUP_NUM);
  console.log(`Group ${group.publicKey}`);

  let sig;

  // close stub oracle
  const usdcDevnetMint = new PublicKey(DEVNET_MINTS.get('USDC')!);

  const usdcDevnetOracle = (
    await client.getStubOracle(group, usdcDevnetMint)
  )[0];
  sig = await client.stubOracleClose(group, usdcDevnetOracle.publicKey);
  console.log(
    `Closed USDC stub oracle, sig https://explorer.solana.com/tx/${sig}?cluster=devnet`,
  );

  // close all bank
  group.consoleLogBanks();

  // deregister all serum markets
  for (const market of group.serum3MarketsMapByExternal.values()) {
    sig = await client.serum3deregisterMarket(
      group,
      market.serumMarketExternal,
    );
    console.log(
      `Deregistered serum market ${market.name}, sig https://explorer.solana.com/tx/${sig}?cluster=devnet`,
    );
  }

  // close all perp markets
  for (const market of group.perpMarketsMapByMarketIndex.values()) {
    sig = await client.perpCloseMarket(group, market.perpMarketIndex);
    console.log(
      `Closed perp market ${market.name}, sig https://explorer.solana.com/tx/${sig}?cluster=devnet`,
    );
  }

  // finally, close the group

  sig = await client.groupClose(group);
  console.log(
    `Closed group, sig https://explorer.solana.com/tx/${sig}?cluster=devnet`,
  );

  process.exit();
}

main();