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

const CLUSTER: Cluster =
  (process.env.CLUSTER_OVERRIDE as Cluster) || 'mainnet-beta';
const CLUSTER_URL =
  process.env.CLUSTER_URL_OVERRIDE || process.env.MB_CLUSTER_URL;
const USER_KEYPAIR =
  process.env.USER_KEYPAIR_OVERRIDE || process.env.MB_PAYER_KEYPAIR;
const GROUP_PK =
  process.env.GROUP_PK || '78b8f4cGCwmZ9ysPFMWLaLTkkaYnUjwMJYStWe5RTSSX';
const PERP_MARKET_INDEX = Number(
  process.env.PERP_MARKET_INDEX,
) as PerpMarketIndex;

async function forceClosePerpPositions(): Promise<void> {
  const options = AnchorProvider.defaultOptions();
  const connection = new Connection(CLUSTER_URL!, options);
  const user = Keypair.fromSecretKey(
    Buffer.from(
      JSON.parse(
        process.env.KEYPAIR || fs.readFileSync(USER_KEYPAIR!, 'utf-8'),
      ),
    ),
  );
  const userWallet = new Wallet(user);
  const userProvider = new AnchorProvider(connection, userWallet, options);
  const client = await MangoClient.connect(
    userProvider,
    CLUSTER,
    MANGO_V4_ID[CLUSTER],
    {
      idsSource: 'get-program-accounts',
    },
  );

  const group = await client.getGroup(new PublicKey(GROUP_PK));
  const pm = group.getPerpMarketByMarketIndex(PERP_MARKET_INDEX);
  if (!pm.reduceOnly) {
    throw new Error(`Unexpected reduce only state ${pm.reduceOnly}`);
  }
  if (!pm.forceClose) {
    throw new Error(`Unexpected force close state ${pm.forceClose}`);
  }

  // Get all mango accounts who have a position in the given market
  const mangoAccounts = (await client.getAllMangoAccounts(group)).filter(
    (a) =>
      a.getPerpPosition(PERP_MARKET_INDEX) !== undefined &&
      a.getPerpPositionUi(group, PERP_MARKET_INDEX) !== 0,
  );
  // Sort descending
  mangoAccounts.sort(
    (a, b) =>
      b.getPerpPositionUi(group, PERP_MARKET_INDEX) -
      a.getPerpPositionUi(group, PERP_MARKET_INDEX),
  );

  let a: MangoAccount;
  let b: MangoAccount;
  let i = 0,
    j = mangoAccounts.length - 1;

  // i iterates forward to 2nd last account, and b iterates backward till 2nd account
  while (i < mangoAccounts.length - 1 && j > 0) {
    if (i === j) {
      break;
    }
    a = mangoAccounts[i];
    b = mangoAccounts[j];
    // PerpForceClosePosition ix expects a to be long, and b to short
    const sig = await client.perpForceClosePosition(
      group,
      PERP_MARKET_INDEX,
      a,
      b,
    );
    console.log(
      `PerpForceClosePosition ${a.publicKey} and ${
        b.publicKey
      } , sig https://explorer.solana.com/tx/${sig}?cluster=${
        CLUSTER == 'devnet' ? 'devnet' : ''
      }`,
    );
    a = await a.reload(client);
    b = await b.reload(client);
    // Move to previous account once b's position is completely reduced
    if (b.getPerpPositionUi(group, PERP_MARKET_INDEX) === 0) {
      console.log(`Fully reduced position for ${b.publicKey}`);
      j--;
    }
    // Move to next account once a's position is completely reduced
    if (a.getPerpPositionUi(group, PERP_MARKET_INDEX) === 0) {
      console.log(`Fully reduced position for ${a.publicKey}`);
      i++;
    }
  }
}

forceClosePerpPositions();