import { AnchorProvider, Wallet } from '@project-serum/anchor';
import { Connection, Keypair, PublicKey } from '@solana/web3.js';
import fs from 'fs';
import { Group } from '../accounts/group';
import { HealthType, MangoAccount } from '../accounts/mangoAccount';
import { MangoClient } from '../client';
import { MANGO_V4_ID } from '../constants';
import { toUiDecimalsForQuote } from '../utils';

async function debugUser(
  client: MangoClient,
  group: Group,
  mangoAccount: MangoAccount,
) {
  console.log(mangoAccount.toString(group));

  await mangoAccount.reload(client, group);

  console.log(
    'buildFixedAccountRetrieverHealthAccounts ' +
      client
        .buildFixedAccountRetrieverHealthAccounts(group, mangoAccount, [
          group.banksMapByName.get('BTC')[0],
          group.banksMapByName.get('USDC')[0],
        ])
        .map((pk) => pk.toBase58())
        .join(', '),
  );
  console.log(
    'mangoAccount.getEquity() ' +
      toUiDecimalsForQuote(mangoAccount.getEquity().toNumber()),
  );
  console.log(
    'mangoAccount.getHealth(HealthType.init) ' +
      toUiDecimalsForQuote(mangoAccount.getHealth(HealthType.init).toNumber()),
  );
  console.log(
    'mangoAccount.getHealthRatio(HealthType.init) ' +
      mangoAccount.getHealthRatio(HealthType.init).toNumber(),
  );
  console.log(
    'mangoAccount.getHealthRatioUi(HealthType.init) ' +
      mangoAccount.getHealthRatioUi(HealthType.init),
  );
  console.log(
    'mangoAccount.getHealthRatio(HealthType.maint) ' +
      mangoAccount.getHealthRatio(HealthType.maint).toNumber(),
  );
  console.log(
    'mangoAccount.getHealthRatioUi(HealthType.maint) ' +
      mangoAccount.getHealthRatioUi(HealthType.maint),
  );
  console.log(
    'mangoAccount.getCollateralValue() ' +
      toUiDecimalsForQuote(mangoAccount.getCollateralValue().toNumber()),
  );
  console.log(
    'mangoAccount.getAssetsValue() ' +
      toUiDecimalsForQuote(
        mangoAccount.getAssetsValue(HealthType.init).toNumber(),
      ),
  );
  console.log(
    'mangoAccount.getLiabsValue() ' +
      toUiDecimalsForQuote(
        mangoAccount.getLiabsValue(HealthType.init).toNumber(),
      ),
  );

  async function getMaxWithdrawWithBorrowForTokenUiWrapper(token) {
    console.log(
      `mangoAccount.getMaxWithdrawWithBorrowForTokenUi(group, ${token}) ` +
        mangoAccount.getMaxWithdrawWithBorrowForTokenUi(
          group,
          group.banksMapByName.get(token)[0].mint,
        ),
    );
  }
  for (const srcToken of Array.from(group.banksMapByName.keys())) {
    await getMaxWithdrawWithBorrowForTokenUiWrapper(srcToken);
  }

  function simHealthRatioWithTokenPositionChangesWrapper(debug, change) {
    console.log(
      `mangoAccount.simHealthRatioWithTokenPositionChanges ${debug}` +
        mangoAccount.simHealthRatioWithTokenPositionUiChanges(group, [change]),
    );
  }
  for (const srcToken of Array.from(group.banksMapByName.keys())) {
    simHealthRatioWithTokenPositionChangesWrapper(`${srcToken} 1  `, {
      mintPk: group.banksMapByName.get(srcToken)[0].mint,
      uiTokenAmount: 1,
    });
    simHealthRatioWithTokenPositionChangesWrapper(`${srcToken} -1  `, {
      mintPk: group.banksMapByName.get(srcToken)[0].mint,
      uiTokenAmount: -1,
    });
  }

  function getMaxSourceForTokenSwapWrapper(src, tgt) {
    // console.log();
    console.log(
      `getMaxSourceForTokenSwap ${src.padEnd(4)} ${tgt.padEnd(4)} ` +
        mangoAccount
          .getMaxSourceForTokenSwap(
            group,
            group.banksMapByName.get(src)[0].mint,
            group.banksMapByName.get(tgt)[0].mint,
            1,
          )
          .div(
            I80F48.fromNumber(
              Math.pow(10, group.banksMapByName.get(src)[0].mintDecimals),
            ),
          )
          .toNumber(),
    );
  }
  for (const srcToken of Array.from(group.banksMapByName.keys())) {
    for (const tgtToken of Array.from(group.banksMapByName.keys())) {
      // if (srcToken === 'SOL')
      // if (tgtToken === 'MSOL')
      getMaxSourceForTokenSwapWrapper(srcToken, tgtToken);
    }
  }
}

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

  const admin = Keypair.fromSecretKey(
    Buffer.from(
      JSON.parse(fs.readFileSync(process.env.MB_PAYER_KEYPAIR!, 'utf-8')),
    ),
  );
  console.log(`Admin ${admin.publicKey.toBase58()}`);

  const adminWallet = new Wallet(admin);
  const adminProvider = new AnchorProvider(connection, adminWallet, options);
  const client = MangoClient.connect(
    adminProvider,
    'mainnet-beta',
    MANGO_V4_ID['mainnet-beta'],
  );

  const group = await client.getGroupForCreator(admin.publicKey, 2);

  for (const keypair of [
    process.env.MB_PAYER_KEYPAIR,
    process.env.MB_USER2_KEYPAIR,
  ]) {
    console.log();
    const user = Keypair.fromSecretKey(
      Buffer.from(JSON.parse(fs.readFileSync(keypair, 'utf-8'))),
    );
    const userWallet = new Wallet(user);
    console.log(`User ${userWallet.publicKey.toBase58()}`);
    // const mangoAccounts = await client.getMangoAccountsForOwner(
    //   group,
    //   user.publicKey,
    // );
    const mangoAccounts = await Promise.all([
      await client.getMangoAccount({
        publicKey: new PublicKey(
          '6mqHfpJqnXcu6RgDYZSVW9CQXQPFyRYhgvdzvWXN9mPW',
        ),
      } as any),
    ]);
    for (const mangoAccount of mangoAccounts) {
      console.log(`MangoAccount ${mangoAccount.publicKey}`);
      // if (
      // 'etVgrWPAQe3aUMiuZT32tEMnbtuD2yRxip6eEkSRrLV' ===
      // '2UWD1ZBiYdusXgXputE68MBL4Kz47nZptzMn6rLnzn1K' ===
      // '9B8uwqH8FJqLn9kvGPVb5GEksLvmyXb3B8UKCFtRs5cq' ===
      // mangoAccount.publicKey.toBase58()
      // )
      {
        await debugUser(client, group, mangoAccount);
      }
    }
  }

  process.exit();
}

try {
  main();
} catch (error) {
  console.log(error);
}