import { Fr } from '@aztec/foundation/curves/bn254';
import type { Logger } from '@aztec/foundation/log';
import { Timer } from '@aztec/foundation/timer';
import type { ContractArtifact } from '@aztec/stdlib/abi';
import { AztecAddress } from '@aztec/stdlib/aztec-address';

import { PublicTxSimulationTester } from './public_tx_simulation_tester.js';

export async function bulkTest(
  tester: PublicTxSimulationTester,
  logger: Logger,
  avmTestContractArtifact: ContractArtifact,
) {
  const timer = new Timer();

  const deployer = AztecAddress.fromNumberUnsafe(42);
  const avmTestContract = await tester.registerAndDeployContract(
    /*constructorArgs=*/ [],
    deployer,
    avmTestContractArtifact,
  );

  // Needed since we invoke the Fee Juice Contract in the bulk test.registerFeeJuiceContract
  await tester.registerFeeJuiceContract();
  // Register multiple different protocol contracts (to ensure we don't dedup bytecode hashing events):
  await tester.registerAuthContract();
  await tester.registerInstanceRegistryContract();
  await tester.registerClassRegistryContract();

  // Get a deployed contract instance to pass to the contract
  // for it to use as "expected" values when testing contract instance retrieval.
  const expectContractInstance = avmTestContract;
  const argsField = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  const argsU8 = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  // Pinned grumpkin-Poseidon2 Schnorr signature (mirrors the C++ `pinned_test_vector_large`
  // and noir-lang/schnorr v0.4.0's `pinned_vector_large`). Passing these in as calldata
  // (rather than baking them into Noir as constants) keeps MSM + Poseidon2 from being folded
  // by the Noir compiler.
  const schnorrInputs = [
    Fr.fromHexString('0x065812e335a97c2108ea8cf4ccfe2f9dd6b117a0714f5e18461575be93f61da6'), // pubkey.x
    Fr.fromHexString('0x1a915003e8ec534f9a15d926a7ded478e178468ccc4f28e236e67450a55ac622'), // pubkey.y
    Fr.fromHexString('0xf3bc3b7147acb9c621fd9f72dbf15ffa'), // sig_s.lo
    Fr.fromHexString('0x08599f379f0301dfefdbd0272554454d'), // sig_s.hi
    Fr.fromHexString('0x97065383ebbbd76620398792bd259bc2'), // sig_e.lo
    Fr.fromHexString('0x2ceaee87f45b7a417f0ffb05451a8c92'), // sig_e.hi
    Fr.fromHexString('0x0123456789abcdef0fedcba9876543210123456789abcdef0fedcba987654321'), // message
  ];
  const args = [
    argsField,
    argsU8,
    /*getInstanceForAddress=*/ expectContractInstance.address,
    /*expectedDeployer=*/ expectContractInstance.deployer,
    /*expectedClassId=*/ expectContractInstance.currentContractClassId,
    /*expectedInitializationHash=*/ expectContractInstance.initializationHash,
    /*expectedImmutablesHash=*/ expectContractInstance.immutablesHash,
    /*schnorrInputs=*/ schnorrInputs,
    /*skip_strictly_limited_side_effects=*/ false,
  ];

  const bulkResult = await tester.executeTxWithLabel(
    /*txLabel=*/ 'AvmTest/bulk_testing',
    /*sender=*/ deployer,
    /*setupCalls=*/ [],
    /*appCalls=*/ [
      {
        address: avmTestContract.address,
        fnName: 'bulk_testing',
        args,
      },
      // 3 calls creating calldata + asserting calldata copy:
      {
        address: avmTestContract.address,
        fnName: 'assert_calldata_copy_large',
        args: [Array.from({ length: 300 }, () => Fr.random()), /* with_selector: */ true],
      },
      {
        address: avmTestContract.address,
        fnName: 'assert_calldata_copy',
        args: [argsField.slice(0, 3), /* with_selector: */ true],
      },
      {
        address: avmTestContract.address,
        fnName: 'assert_calldata_copy_large',
        args: [Array.from({ length: 300 }, () => Fr.random()), /* with_selector: */ true],
      },
      // 3 calls to external contracts
      { address: avmTestContract.address, fnName: 'call_fee_juice', args: [] },
      { address: avmTestContract.address, fnName: 'call_auth_registry', args: [] },
      { address: avmTestContract.address, fnName: 'call_instance_registry', args: [] },
    ],
    /*teardownCall=*/ undefined,
    /*feePayer*/ undefined,
    /*privateInsertions=*/ {
      nonRevertible: {
        nullifiers: [new Fr(420000)],
        noteHashes: [new Fr(420001)],
      },
      revertible: {
        nullifiers: [new Fr(420002)],
        noteHashes: [new Fr(420003)],
      },
    },
  );

  logger.info(`Bulk test took ${timer.ms()}ms\n`);

  return bulkResult;
}

export async function megaBulkTest(
  tester: PublicTxSimulationTester,
  logger: Logger,
  avmTestContractArtifact: ContractArtifact,
) {
  const timer = new Timer();

  const deployer = AztecAddress.fromNumberUnsafe(42);
  const avmTestContract = await tester.registerAndDeployContract(
    /*constructorArgs=*/ [],
    deployer,
    avmTestContractArtifact,
  );

  // Needed since we invoke the Fee Juice Contract in the bulk test.registerFeeJuiceContract
  await tester.registerFeeJuiceContract();

  // Get a deployed contract instance to pass to the contract
  // for it to use as "expected" values when testing contract instance retrieval.
  const expectContractInstance = avmTestContract;
  const argsField0 = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  const argsField1 = [3, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  const argsField2 = [5, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  const argsField3 = [7, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  const argsField4 = [9, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  //const argsField5 = [11, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  //const argsField6 = [13, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  //const argsField7 = [15, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  //const argsField8 = [17, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  const argsU8 = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10].map(x => new Fr(x));
  const schnorrInputs = [
    Fr.fromHexString('0x065812e335a97c2108ea8cf4ccfe2f9dd6b117a0714f5e18461575be93f61da6'), // pubkey.x
    Fr.fromHexString('0x1a915003e8ec534f9a15d926a7ded478e178468ccc4f28e236e67450a55ac622'), // pubkey.y
    Fr.fromHexString('0xf3bc3b7147acb9c621fd9f72dbf15ffa'), // sig_s.lo
    Fr.fromHexString('0x08599f379f0301dfefdbd0272554454d'), // sig_s.hi
    Fr.fromHexString('0x97065383ebbbd76620398792bd259bc2'), // sig_e.lo
    Fr.fromHexString('0x2ceaee87f45b7a417f0ffb05451a8c92'), // sig_e.hi
    Fr.fromHexString('0x0123456789abcdef0fedcba9876543210123456789abcdef0fedcba987654321'), // message
  ];
  const genArgs = (argsField: Fr[]) => [
    argsField,
    argsU8,
    /*getInstanceForAddress=*/ expectContractInstance.address.toField(),
    /*expectedDeployer=*/ expectContractInstance.deployer.toField(),
    /*expectedClassId=*/ expectContractInstance.currentContractClassId.toField(),
    /*expectedInitializationHash=*/ expectContractInstance.initializationHash.toField(),
    /*expectedImmutablesHash=*/ expectContractInstance.immutablesHash.toField(),
    /*schnorrInputs=*/ schnorrInputs,
    // Must skip strictly limited side effects (logs, messages) so we can spam the bulk test several times.
    /*skip_strictly_limited_side_effects=*/ true,
  ];

  const bulkResult = await tester.executeTxWithLabel(
    /*txLabel=*/ 'AvmTest/mega_bulk_testing',
    /*sender=*/ deployer,
    /*setupCalls=*/ [],
    /*appCalls=*/ [
      { address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField0) },
      { address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField1) },
      { address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField2) },
      { address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField3) },
      { address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField4) },
      //{ address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField5) },
      //{ address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField6) },
      //{ address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField7) },
      //{ address: avmTestContract.address, fnName: 'bulk_testing', args: genArgs(argsField8) },
    ],
    /*teardownCall=*/ undefined,
    /*feePayer*/ undefined,
    /*privateInsertions=*/ {
      nonRevertible: {
        nullifiers: [new Fr(420000)],
        noteHashes: [new Fr(420001)],
      },
      revertible: {
        nullifiers: [new Fr(420002)],
        noteHashes: [new Fr(420003)],
      },
    },
  );

  logger.info(`Mega bulk test took ${timer.ms()}ms\n`);

  return bulkResult;
}