import { Connection, Keypair, PublicKey } from "@solana/web3.js";
import IDL from "../programs/openbook_v2.json";
import { Program, web3, BN } from "@project-serum/anchor";
import { createAccount } from "../general/solana_utils";
import { MintUtils } from "../general/mint_utils";
import { I80F48, I80F48Dto } from "@blockworks-foundation/mango-v4";
import { OpenbookV2 } from "./openbook_v2";
import { TestProvider } from "../anchor_utils";

export interface Market {
  name: string;
  admin: number[];
  market_pk: PublicKey;
  oracle: PublicKey;
  asks: PublicKey;
  bids: PublicKey;
  event_queue: PublicKey;
  base_vault: PublicKey;
  quote_vault: PublicKey;
  base_mint: PublicKey;
  quote_mint: PublicKey;
  market_index: number;
}

export async function createMarket(
  program: Program<OpenbookV2>,
  anchorProvider: TestProvider,
  mintUtils: MintUtils,
  adminKp: Keypair,
  openbookProgramId: PublicKey,
  baseMint: PublicKey,
  quoteMint: PublicKey,
  index: number
): Promise<Market> {
  let [oracleId, _tmp] = PublicKey.findProgramAddressSync(
    [Buffer.from("StubOracle"), baseMint.toBytes()],
    openbookProgramId
  );
  const admin: PublicKey = adminKp.publicKey;

  await program.methods
    .stubOracleCreate({ val: I80F48.fromNumber(1.0).getData() })
    .accounts({
      oracle: oracleId,
      admin,
      mint: baseMint,
      payer: anchorProvider.wallet.publicKey,
      systemProgram: web3.SystemProgram.programId,
    })
    .signers([adminKp])
    .rpc();

  // bookside size = 123720
  let asks = await createAccount(
    anchorProvider.connection,
    anchorProvider.keypair,
    123720,
    openbookProgramId
  );
  let bids = await createAccount(
    anchorProvider.connection,
    anchorProvider.keypair,
    123720,
    openbookProgramId
  );
  let eventQueue = await createAccount(
    anchorProvider.connection,
    anchorProvider.keypair,
    97688,
    openbookProgramId
  );
  let marketIndex: BN = new BN(index);

  let [marketPk, _tmp2] = PublicKey.findProgramAddressSync(
    [Buffer.from("Market"), admin.toBuffer(), marketIndex.toBuffer("le", 4)],
    openbookProgramId
  );

  let baseVault = await mintUtils.createTokenAccount(
    baseMint,
    anchorProvider.keypair,
    marketPk
  );
  let quoteVault = await mintUtils.createTokenAccount(
    quoteMint,
    anchorProvider.keypair,
    marketPk
  );
  let name = "index " + index.toString() + " wrt 0";

  await program.methods
    .createMarket(
      marketIndex,
      name,
      {
        confFilter: 0,
        maxStalenessSlots: 100,
      },
      new BN(1),
      new BN(1),
      0,
      0,
      0
    )
    .accounts({
      admin,
      market: marketPk,
      bids,
      asks,
      eventQueue,
      payer: anchorProvider.publicKey,
      baseVault,
      quoteVault,
      baseMint,
      quoteMint,
      systemProgram: web3.SystemProgram.programId,
      oracle: oracleId,
    })
    .signers([adminKp])
    .rpc();

  return {
    admin: Array.from(adminKp.secretKey),
    name,
    bids,
    asks,
    event_queue: eventQueue,
    base_mint: baseMint,
    base_vault: baseVault,
    market_index: index,
    market_pk: marketPk,
    oracle: oracleId,
    quote_mint: quoteMint,
    quote_vault: quoteVault,
  };
}