import { PublicKey, SYSVAR_RENT_PUBKEY, SystemProgram, TransactionInstruction, Connection } from '@solana/web3.js';
import { KaminoMarket } from '../classes';
import { PublicKeySet, referrerStatePda, referrerTokenStatePda, shortUrlPda, userMetadataPda } from '../utils';
import {
  PROGRAM_ID,
  ReferrerState,
  deleteReferrerStateAndShortUrl,
  initReferrerStateAndShortUrl,
  initReferrerTokenState,
} from '../lib';

export const getInitAllReferrerTokenStateIxs = async ({
  referrer,
  kaminoMarket,
  payer = referrer,
}: {
  referrer: PublicKey;
  kaminoMarket: KaminoMarket;
  payer?: PublicKey;
}) => {
  if (referrer.equals(PublicKey.default)) {
    throw new Error('Referrer not set');
  }

  await kaminoMarket.loadReserves();

  const initReferrerTokenStateIxs: TransactionInstruction[] = [];

  const tokenStatesToCreate: [PublicKey, PublicKey][] = [];
  const reserves = kaminoMarket.getReserves();
  const referrerTokenStates = reserves.map((reserve) => {
    return referrerTokenStatePda(referrer, reserve.address, kaminoMarket.programId)[0];
  });
  const uniqueReferrerTokenStates = new PublicKeySet<PublicKey>(referrerTokenStates).toArray();
  const accounts = await kaminoMarket.getConnection().getMultipleAccountsInfo(uniqueReferrerTokenStates);
  for (let i = 0; i < uniqueReferrerTokenStates.length; i++) {
    if (!accounts[i]) {
      tokenStatesToCreate.push([uniqueReferrerTokenStates[i], reserves[i].address]);
    }
  }

  tokenStatesToCreate.forEach(([referrerTokenStateAddress, reserveAddress]) => {
    const initReferrerTokenStateIx = initReferrerTokenState(
      {
        lendingMarket: kaminoMarket.getAddress(),
        payer,
        reserve: reserveAddress,
        referrer,
        referrerTokenState: referrerTokenStateAddress,
        rent: SYSVAR_RENT_PUBKEY,
        systemProgram: SystemProgram.programId,
      },
      kaminoMarket.programId
    );

    initReferrerTokenStateIxs.push(initReferrerTokenStateIx);
  });

  return initReferrerTokenStateIxs;
};

export const getInitReferrerStateAndShortUrlIxs = ({
  referrer,
  shortUrl,
  programId = PROGRAM_ID,
}: {
  referrer: PublicKey;
  shortUrl: string;
  programId: PublicKey;
}) => {
  const [referrerStateAddress] = referrerStatePda(referrer, programId);
  const [shortUrlAddress] = shortUrlPda(shortUrl, programId);

  const referrerUserMetadataAddress = userMetadataPda(referrer, programId)[0];

  const initReferrerStateAndShortUrlIx = initReferrerStateAndShortUrl(
    {
      shortUrl: shortUrl,
    },
    {
      referrer: referrer,
      referrerState: referrerStateAddress,
      referrerShortUrl: shortUrlAddress,
      referrerUserMetadata: referrerUserMetadataAddress,
      rent: SYSVAR_RENT_PUBKEY,
      systemProgram: SystemProgram.programId,
    },
    programId
  );

  return initReferrerStateAndShortUrlIx;
};

// TODO: 1 thing left before adding program id
export const getDeleteReferrerStateAndShortUrlIxs = async ({
  referrer,
  connection,
  programId = PROGRAM_ID,
}: {
  referrer: PublicKey;
  connection: Connection;
  programId: PublicKey;
}) => {
  const [referrerStateAddress] = referrerStatePda(referrer, programId);
  const referrerState = await ReferrerState.fetch(connection, referrerStateAddress, programId);

  const initReferrerStateAndShortUrlIx = deleteReferrerStateAndShortUrl(
    {
      referrer: referrer,
      referrerState: referrerStateAddress,
      shortUrl: referrerState!.shortUrl,
      rent: SYSVAR_RENT_PUBKEY,
      systemProgram: SystemProgram.programId,
    },
    programId
  );

  return initReferrerStateAndShortUrlIx;
};