import type { EthAddress } from '@aztec/foundation/eth-address';
import { schemas, zodFor } from '@aztec/foundation/schemas';

import { z } from 'zod';

export type SlasherClientType = 'empire' | 'tally';

export interface SlasherConfig {
  slashOverridePayload?: EthAddress;
  slashMinPenaltyPercentage: number;
  slashMaxPenaltyPercentage: number;
  slashSelfAllowed?: boolean; // Whether to allow slashes to own validators
  slashValidatorsAlways: EthAddress[]; // Array of validator addresses
  slashValidatorsNever: EthAddress[]; // Array of validator addresses
  slashInactivityTargetPercentage: number; // 0-1, 0.9 means 90%. Must be greater than 0
  slashInactivityConsecutiveEpochThreshold: number; // Number of consecutive epochs a validator must be inactive before slashing
  slashPrunePenalty: bigint;
  slashDataWithholdingPenalty: bigint;
  slashInactivityPenalty: bigint;
  slashBroadcastedInvalidBlockPenalty: bigint;
  slashDuplicateProposalPenalty: bigint;
  slashDuplicateAttestationPenalty: bigint;
  slashProposeInvalidAttestationsPenalty: bigint;
  slashAttestDescendantOfInvalidPenalty: bigint;
  slashUnknownPenalty: bigint;
  slashOffenseExpirationRounds: number; // Number of rounds after which pending offenses expire
  slashMaxPayloadSize: number; // Maximum number of offenses to include in a single slash payload
  slashGracePeriodL2Slots: number; // Number of L2 slots to wait after genesis before slashing for most offenses
  slashExecuteRoundsLookBack: number; // How many rounds to look back when searching for a round to execute
}

export const SlasherConfigSchema = zodFor<SlasherConfig>()(
  z.object({
    slashOverridePayload: schemas.EthAddress.optional(),
    slashMinPenaltyPercentage: z.number(),
    slashMaxPenaltyPercentage: z.number(),
    slashValidatorsAlways: z.array(schemas.EthAddress),
    slashValidatorsNever: z.array(schemas.EthAddress),
    slashPrunePenalty: schemas.BigInt,
    slashDataWithholdingPenalty: schemas.BigInt,
    slashInactivityTargetPercentage: z.number(),
    slashInactivityConsecutiveEpochThreshold: z.number(),
    slashInactivityPenalty: schemas.BigInt,
    slashProposeInvalidAttestationsPenalty: schemas.BigInt,
    slashDuplicateProposalPenalty: schemas.BigInt,
    slashDuplicateAttestationPenalty: schemas.BigInt,
    slashAttestDescendantOfInvalidPenalty: schemas.BigInt,
    slashUnknownPenalty: schemas.BigInt,
    slashOffenseExpirationRounds: z.number(),
    slashMaxPayloadSize: z.number(),
    slashGracePeriodL2Slots: z.number(),
    slashBroadcastedInvalidBlockPenalty: schemas.BigInt,
    slashExecuteRoundsLookBack: z.number(),
    slashSelfAllowed: z.boolean().optional(),
  }),
);