import type { Fr } from '@aztec/foundation/curves/bn254';

import { z } from 'zod';

import type { AztecAddress } from '../../aztec-address/index.js';
import { schemas, zodFor } from '../../schemas/index.js';
import type { UInt64 } from '../../types/shared.js';

/**
 * An update to a contract instance, changing its contract class.
 */
export interface ContractInstanceUpdate {
  /** Identifier of the previous contract class for this instance */
  prevContractClassId: Fr;
  /** Identifier of the new contract class for this instance. */
  newContractClassId: Fr;
  /** The timestamp at which the contract class in use will be the new one */
  timestampOfChange: UInt64;
}

export type ContractInstanceUpdateWithAddress = ContractInstanceUpdate & { address: AztecAddress };

export const ContractInstanceUpdateSchema = zodFor<ContractInstanceUpdate>()(
  z.object({
    prevContractClassId: schemas.Fr,
    newContractClassId: schemas.Fr,
    timestampOfChange: schemas.BigInt,
  }),
);

export const ContractInstanceUpdateWithAddressSchema = zodFor<ContractInstanceUpdateWithAddress>()(
  ContractInstanceUpdateSchema.and(z.object({ address: schemas.AztecAddress })),
);