import "mocha";

import {
  AggregatorAccount,
  AggregatorPdaAccounts,
  CrankAccount,
  QueueAccount,
  SolanaClock,
  types,
} from "../src/index.js";

import { createFeed, createFeeds, setupTest, TestContext } from "./utils.js";

import { Keypair, LAMPORTS_PER_SOL, PublicKey } from "@solana/web3.js";
import { BN, promiseWithTimeout, sleep } from "@switchboard-xyz/common";
import assert from "assert";

describe("Crank Tests", () => {
  const CRANK_SIZE = 50;
  const QUEUE_REWARD = 15_000 / LAMPORTS_PER_SOL;
  // const QUEUE_REWARD = 0;

  let ctx: TestContext;

  const queueAuthority = Keypair.generate();

  let userTokenAddress: PublicKey;

  let queueAccount: QueueAccount;
  let queue: types.OracleQueueAccountData;

  let crankAccount: CrankAccount;

  before(async () => {
    ctx = await setupTest();

    [userTokenAddress] = await ctx.program.mint.getOrCreateWrappedUser(
      ctx.program.walletPubkey,
      { fundUpTo: 50 }
    );

    [queueAccount] = await QueueAccount.create(ctx.program, {
      name: "q1",
      metadata: "",
      queueSize: 2,
      reward: QUEUE_REWARD,
      minStake: 0,
      oracleTimeout: 600,
      slashingEnabled: false,
      unpermissionedFeeds: false,
      unpermissionedVrf: true,
      enableBufferRelayers: false,
      authority: queueAuthority.publicKey,
    });

    queue = await queueAccount.loadData();
    assert(
      queue.authority.equals(queueAuthority.publicKey),
      "Incorrect queue authority"
    );

    const [oracle1] = await queueAccount.createOracle({
      name: "Oracle 1",
      enable: true,
      queueAuthority: queueAuthority,
    });
    await oracle1.heartbeat();

    const [oracle2] = await queueAccount.createOracle({
      name: "Oracle 2",
      enable: true,
      queueAuthority: queueAuthority,
    });
    await oracle2.heartbeat();
  });

  it("Creates a Crank", async () => {
    [crankAccount] = await queueAccount.createCrank({
      name: "Crank #1",
      maxRows: CRANK_SIZE,
    });
    const crank = await crankAccount.loadData();
    const crankRows = await crankAccount.loadCrank();
    assert(
      crankRows.length === 0,
      `Crank should be empty but found ${crankRows.length} rows`
    );
  });

  it("Adds a set of feeds to the crank", async () => {
    const aggregatorAccounts: Array<AggregatorAccount> = [];

    // create in batches
    for (const i of Array.from(Array(10).keys())) {
      const newAggregatorAccounts = await createFeeds(
        queueAccount,
        Math.round(CRANK_SIZE / 10),
        {
          queueAuthority,
          // minUpdateDelaySeconds: 5 + Math.floor(Math.random() * 25), // 5 - 30 sec
          fundAmount: QUEUE_REWARD * 10,
          funderTokenWallet: userTokenAddress,
          slidingWindow: Boolean(i % 2),
        }
      );
      await Promise.all(
        newAggregatorAccounts.map(async (aggregatorAccount) => {
          await crankAccount.push({ aggregatorAccount });
        })
      );
      aggregatorAccounts.push(...newAggregatorAccounts);
    }

    assert(
      aggregatorAccounts.length === CRANK_SIZE,
      `Failed to create ${CRANK_SIZE} AggregatorAccount's`
    );

    const crankRows = await crankAccount.loadCrank();
    assert(
      crankRows.length === CRANK_SIZE,
      `Failed to push all ${CRANK_SIZE} aggregators onto the crank, crank size = ${crankRows.length}`
    );

    for (const row of crankRows) {
      const idx = aggregatorAccounts.findIndex((aggregator) =>
        aggregator.publicKey.equals(row.pubkey)
      );
      assert(
        idx !== -1,
        `Failed to find aggregatorAccount on crank, ${row.pubkey}`
      );
    }
  });

  it("Fails to push a non-permitted aggregator onto the crank", async () => {
    const newAggregatorAccount = await createFeed(queueAccount, {
      name: "No Crank Aggregator",
      // queueAuthority,
    });

    await assert.rejects(async () => {
      await crankAccount.push({ aggregatorAccount: newAggregatorAccount });
    }, new RegExp(/PermissionDenied|6035|0x1793/g));
  });

  it("Fails to push a new aggregator onto a full crank", async () => {
    const newAggregatorAccount = await createFeed(queueAccount, {
      name: "No Crank Aggregator",
      queueAuthority,
    });

    await assert.rejects(async () => {
      await crankAccount.push({ aggregatorAccount: newAggregatorAccount });
    }, new RegExp(/CrankMaxCapacityError|6022|0x1786/g));
  });

  it("Crank pop tests", async () => {
    const getTimestamp = async (): Promise<BN> => {
      return (await SolanaClock.fetch(ctx.program.connection)).unixTimestamp;
    };

    const crank = await crankAccount.loadData();

    let timestamp = await getTimestamp();

    const nextAvailable = (await crankAccount.loadCrank()).reduce(
      (nextTimestamp, row) => {
        return nextTimestamp < row.nextTimestamp.toNumber()
          ? row.nextTimestamp.toNumber()
          : nextTimestamp;
      },
      0
    );

    // sleep until crank pop is ready
    const delay = nextAvailable - timestamp.toNumber();
    if (delay > 0) {
      await sleep((delay + 1) * 1000);
    }

    const initialCrankRows = await crankAccount.loadCrank();

    const crankAccounts = crankAccount.getCrankAccounts(
      initialCrankRows,
      queueAccount,
      queue.authority
    );
    assert(
      initialCrankRows.length === crankAccounts.size,
      `Failed to load all crank accounts`
    );

    timestamp = await getTimestamp();

    const readyRows = initialCrankRows.filter((row) =>
      timestamp.gte(row.nextTimestamp)
    );
    assert(readyRows.length > 0, `No aggregators ready!`);

    const readyAggregators: Array<[AggregatorAccount, AggregatorPdaAccounts]> =
      readyRows.map((r) => {
        if (crankAccounts.has(r.pubkey.toBase58())) {
          return [
            new AggregatorAccount(ctx.program, r.pubkey),
            crankAccounts.get(r.pubkey.toBase58())!,
          ];
        } else {
          const aggregatorAccount = new AggregatorAccount(
            ctx.program,
            r.pubkey
          );
          return [
            aggregatorAccount,
            aggregatorAccount.getAccounts(queueAccount, queue.authority),
          ];
        }
      });

    const packedTxns = crankAccount.packAndPopInstructions(
      ctx.payer.publicKey,
      {
        payoutTokenWallet: userTokenAddress,

        queuePubkey: queueAccount.publicKey,
        queueAuthority: queue.authority,
        queueDataBuffer: queue.dataBuffer,
        crankDataBuffer: crank.dataBuffer,

        readyAggregators: readyAggregators,
        failOpenOnMismatch: true,
      }
    );
    assert(packedTxns.length > 0, `Failed to create packed transactions`);

    const startingTokenBalance = (await ctx.program.mint.fetchBalance(
      userTokenAddress
    ))!;

    const signatures = await ctx.program.signAndSendAll(
      packedTxns,
      {
        skipPreflight: true,
        skipConfrimation: true,
      },
      undefined,
      10
    );

    await promiseWithTimeout(
      5000,
      Promise.allSettled(
        signatures.map(
          async (s) => await ctx.program.connection.confirmTransaction(s)
        )
      )
    );

    const newCrankRows = await crankAccount.loadCrank();

    let numPopped = 0;
    for (const [i, row] of initialCrankRows.entries()) {
      const idx = newCrankRows.findIndex((newRow) =>
        newRow.pubkey.equals(row.pubkey)
      );
      assert(idx !== -1, `Failed to find aggregator #${i} - ${row.pubkey}`);
      if (!row.nextTimestamp.eq(newCrankRows[idx].nextTimestamp)) {
        numPopped = numPopped + 1;
      }
    }

    console.log(`Popped ${numPopped}/${readyRows.length} rows`);

    const finalTokenBalance = ctx.round(
      (await ctx.program.mint.fetchBalance(userTokenAddress)) ?? 0,
      3
    );

    const queueReward =
      (await queueAccount.loadData()).reward.toNumber() / LAMPORTS_PER_SOL;

    const expectedTokenBalance = ctx.round(
      startingTokenBalance + numPopped * queueReward,
      3
    );

    assert(
      finalTokenBalance === expectedTokenBalance,
      `Crank turner was not rewarded sufficiently, expected ${expectedTokenBalance}, received ${finalTokenBalance}`
    );

    assert(
      numPopped >= 0.5 * readyRows.length,
      `Failed to pop at least 50% of the crank, ${numPopped}/${readyRows.length}`
    );
  });
});