import { DomainSeparator, PRIVATE_TO_PUBLIC_KERNEL_CIRCUIT_PUBLIC_INPUTS_LENGTH } from '@aztec/constants';
import { poseidon2HashWithSeparator } from '@aztec/foundation/crypto/poseidon';
import type { Fr } from '@aztec/foundation/curves/bn254';
import { bufferSchemaFor } from '@aztec/foundation/schemas';
import { BufferReader, bigintToUInt64BE, serializeToBuffer, serializeToFields } from '@aztec/foundation/serialize';
import { bufferToHex, hexToBuffer } from '@aztec/foundation/string';
import type { FieldsOf } from '@aztec/foundation/types';

import { AztecAddress } from '../aztec-address/index.js';
import { Gas } from '../gas/gas.js';
import { TxConstantData } from '../tx/tx_constant_data.js';
import type { UInt64 } from '../types/index.js';
import { PrivateToPublicAccumulatedData } from './private_to_public_accumulated_data.js';
import { PublicCallRequest } from './public_call_request.js';

export class PrivateToPublicKernelCircuitPublicInputs {
  constructor(
    public constants: TxConstantData,
    public nonRevertibleAccumulatedData: PrivateToPublicAccumulatedData,
    public revertibleAccumulatedData: PrivateToPublicAccumulatedData,
    public publicTeardownCallRequest: PublicCallRequest,
    public gasUsed: Gas,
    public feePayer: AztecAddress,
    public expirationTimestamp: UInt64,
  ) {}

  toBuffer() {
    return serializeToBuffer(
      this.constants,
      this.nonRevertibleAccumulatedData,
      this.revertibleAccumulatedData,
      this.publicTeardownCallRequest,
      this.gasUsed,
      this.feePayer,
      bigintToUInt64BE(this.expirationTimestamp),
    );
  }

  static getFields(fields: FieldsOf<PrivateToPublicKernelCircuitPublicInputs>) {
    return [
      fields.constants,
      fields.nonRevertibleAccumulatedData,
      fields.revertibleAccumulatedData,
      fields.publicTeardownCallRequest,
      fields.gasUsed,
      fields.feePayer,
      fields.expirationTimestamp,
    ] as const;
  }

  static fromBuffer(buffer: Buffer | BufferReader) {
    const reader = BufferReader.asReader(buffer);
    return new PrivateToPublicKernelCircuitPublicInputs(
      reader.readObject(TxConstantData),
      reader.readObject(PrivateToPublicAccumulatedData),
      reader.readObject(PrivateToPublicAccumulatedData),
      reader.readObject(PublicCallRequest),
      reader.readObject(Gas),
      reader.readObject(AztecAddress),
      reader.readUInt64(),
    );
  }

  static empty() {
    return new PrivateToPublicKernelCircuitPublicInputs(
      TxConstantData.empty(),
      PrivateToPublicAccumulatedData.empty(),
      PrivateToPublicAccumulatedData.empty(),
      PublicCallRequest.empty(),
      Gas.empty(),
      AztecAddress.ZERO,
      0n,
    );
  }

  static fromString(str: string) {
    return PrivateToPublicKernelCircuitPublicInputs.fromBuffer(hexToBuffer(str));
  }

  toString() {
    return bufferToHex(this.toBuffer());
  }

  toFields(): Fr[] {
    const fields = serializeToFields(...PrivateToPublicKernelCircuitPublicInputs.getFields(this));
    if (fields.length !== PRIVATE_TO_PUBLIC_KERNEL_CIRCUIT_PUBLIC_INPUTS_LENGTH) {
      throw new Error(
        `Invalid number of fields for PrivateToPublicKernelCircuitPublicInputs. Expected ${PRIVATE_TO_PUBLIC_KERNEL_CIRCUIT_PUBLIC_INPUTS_LENGTH}, got ${fields.length}`,
      );
    }
    return fields;
  }

  hash() {
    return poseidon2HashWithSeparator(this.toFields(), DomainSeparator.PUBLIC_TX_HASH);
  }

  toJSON() {
    return this.toBuffer();
  }

  static get schema() {
    return bufferSchemaFor(PrivateToPublicKernelCircuitPublicInputs);
  }
}