import { Provider, Wallet } from '@project-serum/anchor';
import { Connection, Keypair, PublicKey } from '@solana/web3.js';
import fs from 'fs';
import {
  Serum3OrderType,
  Serum3SelfTradeBehavior,
  Serum3Side,
} from './accounts/serum3';
import { MangoClient } from './client';
import { DEVNET_GROUP, DEVNET_SERUM3_PROGRAM_ID } from './constants';

//
// An example for users based on high level api i.e. the client
//
async function main() {
  const options = Provider.defaultOptions();
  const connection = new Connection(
    'https://mango.devnet.rpcpool.com',
    options,
  );

  const user = Keypair.fromSecretKey(
    Buffer.from(
      JSON.parse(fs.readFileSync(process.env.USER_KEYPAIR!, 'utf-8')),
    ),
  );
  const userWallet = new Wallet(user);
  const userProvider = new Provider(connection, userWallet, options);
  const client = await MangoClient.connect(userProvider, true);
  console.log(`User ${userWallet.publicKey.toBase58()}`);

  // fetch group
  const group = await client.getGroup(new PublicKey(DEVNET_GROUP));
  console.log(`Group ${group.publicKey.toBase58()}`);

  for (const bank of group.banksMap.values()) {
    console.log(bank.publicKey.toBase58());
  }

  // create + fetch account
  const mangoAccount = await client.getOrCreateMangoAccount(
    group,
    user.publicKey,
    0,
    'my_mango_account',
  );
  console.log(`MangoAccount ${mangoAccount.publicKey}`);

  // deposit and withdraw
  console.log(`Depositing...5000000`);
  await client.deposit(group, mangoAccount, 'USDC', 50_000000);
  await client.deposit(group, mangoAccount, 'BTC', 5000000);
  console.log(`Withdrawing...1000000`);
  await client.withdraw(group, mangoAccount, 'USDC', 1_000000, false);

  // serum3
  console.log(
    `Placing serum3 bid which would not be settled since its relatively low then midprice`,
  );
  await client.serum3PlaceOrder(
    group,
    mangoAccount,
    DEVNET_SERUM3_PROGRAM_ID,
    'BTC/USDC',
    Serum3Side.bid,
    20,
    0.0001,
    Serum3SelfTradeBehavior.decrementTake,
    Serum3OrderType.limit,
    Date.now(),
    10,
  );
  console.log(`Placing serum3 bid way above midprice`);
  await client.serum3PlaceOrder(
    group,
    mangoAccount,
    DEVNET_SERUM3_PROGRAM_ID,
    'BTC/USDC',
    Serum3Side.bid,
    90000,
    0.0001,
    Serum3SelfTradeBehavior.decrementTake,
    Serum3OrderType.limit,
    Date.now(),
    10,
  );
  console.log(`Placing serum3 ask way below midprice`);
  await client.serum3PlaceOrder(
    group,
    mangoAccount,
    DEVNET_SERUM3_PROGRAM_ID,
    'BTC/USDC',
    Serum3Side.ask,
    30000,
    0.0001,
    Serum3SelfTradeBehavior.decrementTake,
    Serum3OrderType.limit,
    Date.now(),
    10,
  );

  console.log(`Current own orders on OB...`);
  let orders = await client.getSerum3Orders(
    group,
    DEVNET_SERUM3_PROGRAM_ID,
    'BTC/USDC',
  );
  for (const order of orders) {
    console.log(
      `Order orderId ${order.orderId}, ${order.side}, ${order.price}, ${order.size}`,
    );
    console.log(`Cancelling order with ${order.orderId}`);
    await client.serum3CancelOrder(
      group,
      mangoAccount,
      DEVNET_SERUM3_PROGRAM_ID,
      'BTC/USDC',
      order.side === 'buy' ? Serum3Side.bid : Serum3Side.ask,
      order.orderId,
    );
  }

  console.log(`Current own orders on OB...`);
  orders = await client.getSerum3Orders(
    group,
    DEVNET_SERUM3_PROGRAM_ID,
    'BTC/USDC',
  );
  for (const order of orders) {
    console.log(order);
  }

  // console.log(`Close mango account...`);
  // await client.closeMangoAccount(mangoAccount);

  console.log(`Settling funds...`);
  await client.serum3SettleFunds(
    group,
    mangoAccount,
    DEVNET_SERUM3_PROGRAM_ID,
    'BTC/USDC',
  );

  process.exit();
}

main();