import { FUNCTION_TREE_HEIGHT, PUBLIC_DATA_TREE_HEIGHT, UPDATES_VALUE_SIZE } from '@aztec/constants';
import { Fr } from '@aztec/foundation/curves/bn254';
import { BufferReader, serializeToBuffer } from '@aztec/foundation/serialize';
import { MembershipWitness } from '@aztec/foundation/trees';
import type { FieldsOf } from '@aztec/foundation/types';

import { DelayedPublicMutableValues } from '../delayed_public_mutable/delayed_public_mutable_values.js';
import { PublicKeys } from '../keys/public_keys.js';
import { PublicDataTreeLeafPreimage } from '../trees/index.js';
import { VerificationKeyAsFields } from '../vks/verification_key.js';
import { PrivateCircuitPublicInputs } from './private_circuit_public_inputs.js';

/**
 * Private call data.
 */
export class PrivateCallData {
  constructor(
    /**
     * Public inputs of the private function circuit.
     */
    public publicInputs: PrivateCircuitPublicInputs,

    /**
     * The verification key for the function being invoked.
     */
    public vk: VerificationKeyAsFields,

    /**
     * Hints for the validation of the vk
     */
    public verificationKeyHints: PrivateVerificationKeyHints,
  ) {}

  /**
   * Serialize into a field array. Low-level utility.
   * @param fields - Object with fields.
   * @returns The array.
   */
  static getFields(fields: FieldsOf<PrivateCallData>) {
    return [fields.publicInputs, fields.vk, fields.verificationKeyHints] as const;
  }

  static from(fields: FieldsOf<PrivateCallData>): PrivateCallData {
    return new PrivateCallData(...PrivateCallData.getFields(fields));
  }

  /**
   * Serialize this as a buffer.
   * @returns The buffer.
   */
  toBuffer(): Buffer {
    return serializeToBuffer(...PrivateCallData.getFields(this));
  }

  /**
   * Deserializes from a buffer or reader.
   * @param buffer - Buffer or reader to read from.
   * @returns The deserialized instance.
   */
  static fromBuffer(buffer: Buffer | BufferReader): PrivateCallData {
    const reader = BufferReader.asReader(buffer);
    return new PrivateCallData(
      reader.readObject(PrivateCircuitPublicInputs),
      reader.readObject(VerificationKeyAsFields),
      reader.readObject(PrivateVerificationKeyHints),
    );
  }
}

export class PrivateVerificationKeyHints {
  constructor(
    /**
     * Artifact hash of the contract class for this private call.
     */
    public contractClassArtifactHash: Fr,
    /**
     * Public bytecode commitment for the contract class for this private call.
     */
    public contractClassPublicBytecodeCommitment: Fr,
    /**
     * Public keys hash of the contract instance.
     */
    public publicKeys: PublicKeys,
    /**
     * Salted initialization hash of the contract instance.
     */
    public saltedInitializationHash: Fr,
    /**
     * The membership witness for the function leaf corresponding to the function being invoked.
     */
    public functionLeafMembershipWitness: MembershipWitness<typeof FUNCTION_TREE_HEIGHT>,

    public updatedClassIdHints: UpdatedClassIdHints,
  ) {}

  /**
   * Serialize into a field array. Low-level utility.
   * @param fields - Object with fields.
   * @returns The array.
   */
  static getFields(fields: FieldsOf<PrivateVerificationKeyHints>) {
    return [
      fields.contractClassArtifactHash,
      fields.contractClassPublicBytecodeCommitment,
      fields.publicKeys,
      fields.saltedInitializationHash,
      fields.functionLeafMembershipWitness,
      fields.updatedClassIdHints,
    ] as const;
  }

  static from(fields: FieldsOf<PrivateVerificationKeyHints>): PrivateVerificationKeyHints {
    return new PrivateVerificationKeyHints(...PrivateVerificationKeyHints.getFields(fields));
  }

  /**
   * Serialize this as a buffer.
   * @returns The buffer.
   */
  toBuffer(): Buffer {
    return serializeToBuffer(...PrivateVerificationKeyHints.getFields(this));
  }

  /**
   * Deserializes from a buffer or reader.
   * @param buffer - Buffer or reader to read from.
   * @returns The deserialized instance.
   */
  static fromBuffer(buffer: Buffer | BufferReader): PrivateVerificationKeyHints {
    const reader = BufferReader.asReader(buffer);
    return new PrivateVerificationKeyHints(
      reader.readObject(Fr),
      reader.readObject(Fr),
      reader.readObject(PublicKeys),
      reader.readObject(Fr),
      reader.readObject(MembershipWitness.deserializer(FUNCTION_TREE_HEIGHT)),
      reader.readObject(UpdatedClassIdHints),
    );
  }
}

export class UpdatedClassIdHints {
  constructor(
    public updatedClassIdWitness: MembershipWitness<typeof PUBLIC_DATA_TREE_HEIGHT>,
    public updatedClassIdLeaf: PublicDataTreeLeafPreimage,
    public updatedClassIdValues: DelayedPublicMutableValues,
  ) {}

  static getFields(fields: FieldsOf<UpdatedClassIdHints>) {
    return [fields.updatedClassIdWitness, fields.updatedClassIdLeaf, fields.updatedClassIdValues] as const;
  }

  static from(fields: FieldsOf<UpdatedClassIdHints>): UpdatedClassIdHints {
    return new UpdatedClassIdHints(...UpdatedClassIdHints.getFields(fields));
  }

  /**
   * Serialize this as a buffer.
   * @returns The buffer.
   */
  toBuffer(): Buffer {
    return serializeToBuffer(...UpdatedClassIdHints.getFields(this));
  }

  /**
   * Deserializes from a buffer or reader.
   * @param buffer - Buffer or reader to read from.
   * @returns The deserialized instance.
   */
  static fromBuffer(buffer: Buffer | BufferReader): UpdatedClassIdHints {
    const reader = BufferReader.asReader(buffer);
    return new UpdatedClassIdHints(
      reader.readObject(MembershipWitness.deserializer(PUBLIC_DATA_TREE_HEIGHT)),
      reader.readObject(PublicDataTreeLeafPreimage),
      reader.readObject({
        fromBuffer(reader) {
          return DelayedPublicMutableValues.fromBuffer(reader, UPDATES_VALUE_SIZE);
        },
      }),
    );
  }
}