import {
  Keypair,
  LAMPORTS_PER_SOL,
  PublicKey,
  SystemProgram,
  Transaction,
} from '@solana/web3.js';
import { createTransferInstruction, InstructionType } from './instruction';
import { start } from 'solana-bankrun';
import { describe, test } from 'node:test';

describe('transfer-sol', async () => {
  const PROGRAM_ID = PublicKey.unique();
  const context = await start([{ name: 'transfer_sol_program', programId: PROGRAM_ID }],[]);
  const client = context.banksClient;
  const payer = context.payer;

  const transferAmount = 1 * LAMPORTS_PER_SOL;
  const test1Recipient = Keypair.generate();
  const test2Recipient1 = Keypair.generate();
  const test2Recipient2 = Keypair.generate();

  test('Transfer between accounts using the system program', async () => {
    await getBalances(payer.publicKey, test1Recipient.publicKey, 'Beginning');

    let ix = createTransferInstruction(
      payer.publicKey,
      test1Recipient.publicKey,
      PROGRAM_ID,
      InstructionType.CpiTransfer,
      transferAmount
    );

    const tx = new Transaction();
    const [blockhash,_] = await client.getLatestBlockhash();
    tx.recentBlockhash = blockhash;
    tx.add(ix).sign(payer);

    await client.processTransaction(tx);

    await getBalances(payer.publicKey, test1Recipient.publicKey, 'Resulting');
  });

  test('Create two accounts for the following test', async () => {
    const ix = (pubkey: PublicKey) => {
      return SystemProgram.createAccount({
        fromPubkey: payer.publicKey,
        newAccountPubkey: pubkey,
        space: 0,
        lamports: 2 * LAMPORTS_PER_SOL,
        programId: PROGRAM_ID,
      });
    };

    const tx = new Transaction();
    const [blockhash, _] = await client.getLatestBlockhash();
    tx.recentBlockhash = blockhash;
    tx.add(ix(test2Recipient1.publicKey))
      .add(ix(test2Recipient2.publicKey))
      .sign(payer, test2Recipient1, test2Recipient2);

    await client.processTransaction(tx);
  });

  test('Transfer between accounts using our program', async () => {
    await getBalances(
      test2Recipient1.publicKey,
      test2Recipient2.publicKey,
      'Beginning'
    );

    let ix = createTransferInstruction(
      test2Recipient1.publicKey,
      test2Recipient2.publicKey,
      PROGRAM_ID,
      InstructionType.ProgramTransfer,
      transferAmount
    );

    const tx = new Transaction();
    const [blockhash, _] = await client.getLatestBlockhash();
    tx.recentBlockhash = blockhash;
    tx.add(ix).sign(payer, test2Recipient1);

    await client.processTransaction(tx);

    await getBalances(
      test2Recipient1.publicKey,
      test2Recipient2.publicKey,
      'Resulting'
    );
  });

  async function getBalances(
    payerPubkey: PublicKey,
    recipientPubkey: PublicKey,
    timeframe: string
  ) {
    let payerBalance = await client.getBalance(payerPubkey);
    let recipientBalance = await client.getBalance(recipientPubkey);

    console.log(`${timeframe} balances:`);
    console.log(`   Payer: ${payerBalance}`);
    console.log(`   Recipient: ${recipientBalance}`);
  }
});