import type {
  MAX_KEY_VALIDATION_REQUESTS_PER_TX,
  MAX_NOTE_HASH_READ_REQUESTS_PER_TX,
  MAX_NULLIFIERS_PER_TX,
  MAX_NULLIFIER_READ_REQUESTS_PER_TX,
} from '@aztec/constants';
import { serializeToBuffer } from '@aztec/foundation/serialize';

import type { PrivateKernelResetHints } from './hints/private_kernel_reset_hints.js';
import type { PaddedSideEffects } from './padded_side_effects.js';
import type { PrivateKernelData } from './private_kernel_data.js';
import type { PrivateKernelResetDimensions } from './private_kernel_reset_dimensions.js';

export class PrivateKernelResetCircuitPrivateInputsVariants<
  NH_RR_PENDING extends number,
  NH_RR_SETTLED extends number,
  NLL_RR_PENDING extends number,
  NLL_RR_SETTLED extends number,
  KEY_VALIDATION_HINTS_LEN extends number,
  TRANSIENT_DATA_HINTS_LEN extends number,
> {
  constructor(
    public previousKernel: PrivateKernelData,
    public paddedSideEffects: PaddedSideEffects,
    public hints: PrivateKernelResetHints<
      NH_RR_PENDING,
      NH_RR_SETTLED,
      NLL_RR_PENDING,
      NLL_RR_SETTLED,
      KEY_VALIDATION_HINTS_LEN,
      TRANSIENT_DATA_HINTS_LEN
    >,
  ) {}

  toBuffer() {
    return serializeToBuffer(this.previousKernel, this.paddedSideEffects, this.hints);
  }
}

/**
 * Input to the private kernel circuit - reset call.
 */
export class PrivateKernelResetCircuitPrivateInputs {
  constructor(
    /**
     * The previous kernel data
     */
    public previousKernel: PrivateKernelData,
    public paddedSideEffects: PaddedSideEffects,
    public hints: PrivateKernelResetHints<
      typeof MAX_NOTE_HASH_READ_REQUESTS_PER_TX,
      typeof MAX_NOTE_HASH_READ_REQUESTS_PER_TX,
      typeof MAX_NULLIFIER_READ_REQUESTS_PER_TX,
      typeof MAX_NULLIFIER_READ_REQUESTS_PER_TX,
      typeof MAX_KEY_VALIDATION_REQUESTS_PER_TX,
      typeof MAX_NULLIFIERS_PER_TX
    >,
    public dimensions: PrivateKernelResetDimensions,
  ) {}

  /**
   * Serialize this as a buffer.
   * @returns The buffer.
   */
  toBuffer() {
    return serializeToBuffer(this.previousKernel, this.paddedSideEffects, this.hints, this.dimensions);
  }

  trimToSizes() {
    const hints = this.hints.trimToSizes(
      this.dimensions.NOTE_HASH_PENDING_READ,
      this.dimensions.NOTE_HASH_SETTLED_READ,
      this.dimensions.NULLIFIER_PENDING_READ,
      this.dimensions.NULLIFIER_SETTLED_READ,
      this.dimensions.KEY_VALIDATION,
      this.dimensions.TRANSIENT_DATA_SQUASHING,
    );
    return new PrivateKernelResetCircuitPrivateInputsVariants(this.previousKernel, this.paddedSideEffects, hints);
  }
}