clmimicry

type CompiledPattern struct {
	Pattern *regexp.Regexp
	Config  *TopicShardingConfig
	// EventTypeConstraint specifies which event types this pattern applies to.
	// Empty string means it applies to all events (backward compatibility).
	// Can be an exact event name (e.g., "LIBP2P_TRACE_GOSSIPSUB_BEACON_ATTESTATION")
	// or a wildcard pattern (e.g., "LIBP2P_TRACE_RPC_META_*")
	EventTypeConstraint string
}

type Config struct {
	LoggingLevel string  `yaml:"logging" default:"info"`
	MetricsAddr  string  `yaml:"metricsAddr" default:":9090"`
	PProfAddr    *string `yaml:"pprofAddr"`
	ProbeAddr    *string `yaml:"probeAddr"`

	// The name of the mimicry
	Name string `yaml:"name"`

	// Ethereum configuration
	Ethereum ethereum.Config `yaml:"ethereum"`

	// Outputs configuration
	Outputs []output.Config `yaml:"outputs"`

	// Labels configures the mimicry with labels
	Labels map[string]string `yaml:"labels"`

	// NTP Server to use for clock drift correction
	NTPServer string `yaml:"ntpServer" default:"time.google.com"`

	// Events is the configuration for the events
	Events EventConfig `yaml:"events"`

	// Sharding is the configuration for event sharding
	Sharding ShardingConfig `yaml:"sharding"`
}

type DutiesProvider interface {
	GetValidatorIndex(epoch phase0.Epoch, slot phase0.Slot, committeeIndex phase0.CommitteeIndex, position uint64) (phase0.ValidatorIndex, error)
}

type EventCategorizer struct {
	// contains filtered or unexported fields
}

type EventConfig struct {
	RecvRPCEnabled                    bool `yaml:"recvRpcEnabled" default:"false"`
	SendRPCEnabled                    bool `yaml:"sendRpcEnabled" default:"false"`
	DropRPCEnabled                    bool `yaml:"dropRpcEnabled" default:"false"`
	RpcMetaControlIHaveEnabled        bool `yaml:"rpcMetaControlIHaveEnabled" default:"false"`
	RpcMetaControlIWantEnabled        bool `yaml:"rpcMetaControlIWantEnabled" default:"false"`
	RpcMetaControlIDontWantEnabled    bool `yaml:"rpcMetaControlIDontWantEnabled" default:"false"`
	RpcMetaControlGraftEnabled        bool `yaml:"rpcMetaControlGraftEnabled" default:"false"`
	RpcMetaControlPruneEnabled        bool `yaml:"rpcMetaControlPruneEnabled" default:"false"`
	RpcMetaSubscriptionEnabled        bool `yaml:"rpcMetaSubscriptionEnabled" default:"false"`
	RpcMetaMessageEnabled             bool `yaml:"rpcMetaMessageEnabled" default:"false"`
	AddPeerEnabled                    bool `yaml:"addPeerEnabled" default:"true"`
	RemovePeerEnabled                 bool `yaml:"removePeerEnabled" default:"true"`
	ConnectedEnabled                  bool `yaml:"connectedEnabled" default:"true"`
	DisconnectedEnabled               bool `yaml:"disconnectedEnabled" default:"true"`
	SyntheticHeartbeatEnabled         bool `yaml:"syntheticHeartbeatEnabled" default:"true"`
	JoinEnabled                       bool `yaml:"joinEnabled" default:"true"`
	LeaveEnabled                      bool `yaml:"leaveEnabled" default:"false"`
	GraftEnabled                      bool `yaml:"graftEnabled" default:"false"`
	PruneEnabled                      bool `yaml:"pruneEnabled" default:"false"`
	PublishMessageEnabled             bool `yaml:"publishMessageEnabled" default:"false"`
	RejectMessageEnabled              bool `yaml:"rejectMessageEnabled" default:"false"`
	DuplicateMessageEnabled           bool `yaml:"duplicateMessageEnabled" default:"false"`
	DeliverMessageEnabled             bool `yaml:"deliverMessageEnabled" default:"false"`
	HandleMetadataEnabled             bool `yaml:"handleMetadataEnabled" default:"true"`
	HandleStatusEnabled               bool `yaml:"handleStatusEnabled" default:"true"`
	CustodyProbeEnabled               bool `yaml:"custodyProbeEnabled" default:"true"`
	GossipSubBeaconBlockEnabled       bool `yaml:"gossipSubBeaconBlockEnabled" default:"true"`
	GossipSubAttestationEnabled       bool `yaml:"gossipSubAttestationEnabled" default:"true"`
	GossipSubAggregateAndProofEnabled bool `yaml:"gossipSubAggregateAndProofEnabled" default:"true"`
	GossipSubBlobSidecarEnabled       bool `yaml:"gossipSubBlobSidecarEnabled" default:"true"`
	GossipSubDataColumnSidecarEnabled bool `yaml:"gossipSubDataColumnSidecarEnabled" default:"true"`
	EngineAPINewPayloadEnabled        bool `yaml:"engineApiNewPayloadEnabled" default:"false"`
	EngineAPIGetBlobsEnabled          bool `yaml:"engineApiGetBlobsEnabled" default:"false"`
}

type EventInfo struct {
	Type          xatu.Event_Name
	ShardingGroup ShardingGroup
	HasTopic      bool
	HasMsgID      bool
	IsMeta        bool // True for RPC meta events
}

type FilteredMessageWithIndex struct {
	MessageID     *wrapperspb.StringValue
	OriginalIndex uint32
}

type MetaProvider interface {
	GetClientMeta(ctx context.Context) (*xatu.ClientMeta, error)
}

type MetadataProvider interface {
	Wallclock() *ethwallclock.EthereumBeaconChain
	ClockDrift() *time.Duration
	Network() *xatu.ClientMeta_Ethereum_Network
}

type Metrics struct {
	// contains filtered or unexported fields
}

type MetricsCollector interface {
	AddEvent(eventType, network string)
	AddProcessedMessage(eventType, network string)
	AddSkippedMessage(eventType, network string)
	AddShardingDecision(eventType, reason, network string)
	AddDecoratedEvent(count float64, eventType, network string)
}

type NoShardingKeyConfig struct {
	// Whether to record events without sharding keys (default: true)
	Enabled bool `yaml:"enabled" default:"true"`
}

type OutputHandler interface {
	HandleDecoratedEvent(ctx context.Context, event *xatu.DecoratedEvent) error
	HandleDecoratedEvents(ctx context.Context, events []*xatu.DecoratedEvent) error
}

type Override struct {
	MetricsAddr struct {
		Enabled bool
		Value   string
	}
}

type Processor struct {
	// contains filtered or unexported fields
}

type RPCMetaMessageInfo struct {
	MessageID *wrapperspb.StringValue
	Topic     *wrapperspb.StringValue // Optional: gossip topic for the message
}

type RPCMetaTopicInfo struct {
	Topic *wrapperspb.StringValue // Gossip topic for the event
}

type RpcControlGraft struct {
	TopicID string
}

type RpcControlIHave struct {
	TopicID string
	MsgIDs  []string
}

type RpcControlIWant struct {
	MsgIDs []string
}

type RpcControlIdontWant struct {
	MsgIDs []string
}

type RpcControlPrune struct {
	TopicID string
	PeerIDs []peer.ID
}

type RpcMeta struct {
	PeerID        peer.ID
	Subscriptions []RpcMetaSub    `json:"Subs,omitempty"`
	Messages      []RpcMetaMsg    `json:"Msgs,omitempty"`
	Control       *RpcMetaControl `json:"Control,omitempty"`
}

type RpcMetaControl struct {
	IHave     []RpcControlIHave     `json:"IHave,omitempty"`
	IWant     []RpcControlIWant     `json:"IWant,omitempty"`
	Graft     []RpcControlGraft     `json:"Graft,omitempty"`
	Prune     []RpcControlPrune     `json:"Prune,omitempty"`
	Idontwant []RpcControlIdontWant `json:"Idontwant,omitempty"`
}

type RpcMetaMsg struct {
	MsgID string `json:"MsgID,omitempty"`
	Topic string `json:"Topic,omitempty"`
}

type RpcMetaSub struct {
	Subscribe bool
	TopicID   string
}

type ShardableEvent struct {
	MsgID string
	Topic string
}

type ShardingConfig struct {
	// Topic-based patterns with sampling rates
	Topics map[string]*TopicShardingConfig `yaml:"topics"`

	// Events without sharding keys (Group D)
	NoShardingKeyEvents *NoShardingKeyConfig `yaml:"noShardingKeyEvents,omitempty"`
	// contains filtered or unexported fields
}

type ShardingGroup int

type TopicShardingConfig struct {
	// Total number of shards for this topic pattern
	TotalShards uint64 `yaml:"totalShards"`
	// Active shards for this topic pattern
	ActiveShards []uint64 `yaml:"activeShards"`
}

type TraceEvent struct {
	Type      string
	Topic     string
	PeerID    peer.ID
	Timestamp time.Time
	Payload   any `json:"Data"` // cannot use field "Data" because of gk.Record method
}

type TraceEventAltairBlock struct {
	TraceEventPayloadMetaData
	Block *ethtypes.SignedBeaconBlockAltair
}

type TraceEventAttestation struct {
	TraceEventPayloadMetaData
	Attestation *ethtypes.Attestation
}

type TraceEventAttestationElectra struct {
	TraceEventPayloadMetaData
	AttestationElectra *ethtypes.AttestationElectra
}

type TraceEventAttesterSlashing struct {
	TraceEventPayloadMetaData
	AttesterSlashing *ethtypes.AttesterSlashing
}

type TraceEventBLSToExecutionChange struct {
	TraceEventPayloadMetaData
	BLSToExecutionChange *ethtypes.BLSToExecutionChange
}

type TraceEventBellatrixBlock struct {
	TraceEventPayloadMetaData
	Block *ethtypes.SignedBeaconBlockBellatrix
}

type TraceEventBlobSidecar struct {
	TraceEventPayloadMetaData
	BlobSidecar *ethtypes.BlobSidecar
}

type TraceEventCapellaBlock struct {
	TraceEventPayloadMetaData
	Block *ethtypes.SignedBeaconBlockCapella
}

type TraceEventConsensusEngineAPIGetBlobs struct {
	TraceEventPayloadMetaData

	// Timing
	RequestedAt time.Time     `json:"requested_at"`
	Duration    time.Duration `json:"duration"`

	// Beacon context
	Slot            uint64 `json:"slot"`
	BlockRoot       string `json:"block_root"`
	ParentBlockRoot string `json:"parent_block_root"`

	// Request details
	RequestedCount  uint32   `json:"requested_count"`
	VersionedHashes []string `json:"versioned_hashes"`

	// Response
	ReturnedCount uint32 `json:"returned_count"`
	Status        string `json:"status"`
	ErrorMessage  string `json:"error_message"`

	// Meta
	MethodVersion string `json:"method_version"`

	// ExecutionClientVersion is the raw version string from web3_clientVersion RPC.
	// Parsed into components when converting to protobuf.
	ExecutionClientVersion string `json:"execution_client_version"`
}

type TraceEventConsensusEngineAPINewPayload struct {
	TraceEventPayloadMetaData

	// Timing
	RequestedAt time.Time     `json:"requested_at"`
	Duration    time.Duration `json:"duration"`

	// Beacon context
	Slot            uint64 `json:"slot"`
	BlockRoot       string `json:"block_root"`
	ParentBlockRoot string `json:"parent_block_root"`
	ProposerIndex   uint64 `json:"proposer_index"`

	// Execution payload
	BlockNumber uint64 `json:"block_number"`
	BlockHash   string `json:"block_hash"`
	ParentHash  string `json:"parent_hash"`
	GasUsed     uint64 `json:"gas_used"`
	GasLimit    uint64 `json:"gas_limit"`
	TxCount     uint32 `json:"tx_count"`
	BlobCount   uint32 `json:"blob_count"`

	// Response
	Status          string `json:"status"`
	LatestValidHash string `json:"latest_valid_hash"`
	ValidationError string `json:"validation_error"`

	// Meta
	MethodVersion string `json:"method_version"`

	// ExecutionClientVersion is the raw version string from web3_clientVersion RPC.
	// Parsed into components when converting to protobuf.
	ExecutionClientVersion string `json:"execution_client_version"`
}

type TraceEventCustodyProbe struct {
	TraceEventPayloadMetaData
	PeerID     *peer.ID      `json:"peer_id,omitempty"`
	Epoch      uint64        `json:"epoch"`
	Slot       uint64        `json:"slot"`
	BlockHash  string        `json:"block_hash"`
	Column     uint64        `json:"column_id"`
	Result     string        `json:"result,omitempty"`
	Duration   time.Duration `json:"duration,omitempty"`
	ColumnSize int           `json:"column_size,omitempty"`
	Error      string        `json:"error,omitempty"`
}

type TraceEventDataColumnSidecar struct {
	TraceEventPayloadMetaData
	DataColumnSidecar *ethtypes.DataColumnSidecar
}

type TraceEventDenebBlock struct {
	TraceEventPayloadMetaData
	Block *ethtypes.SignedBeaconBlockDeneb
}

type TraceEventElectraBlock struct {
	TraceEventPayloadMetaData
	Block *ethtypes.SignedBeaconBlockElectra
}

type TraceEventFuluBlock struct {
	TraceEventPayloadMetaData
	Block *ethtypes.SignedBeaconBlockFulu
}

type TraceEventPayloadMetaData struct {
	PeerID  string `json:"PeerID"`
	Topic   string `json:"Topic"`
	Seq     []byte `json:"Seq"`
	MsgID   string `json:"MsgID"`
	MsgSize int    `json:"MsgSize"`
}

type TraceEventPhase0Block struct {
	TraceEventPayloadMetaData
	Block *ethtypes.SignedBeaconBlock
}

type TraceEventProposerSlashing struct {
	TraceEventPayloadMetaData
	ProposerSlashing *ethtypes.ProposerSlashing
}

type TraceEventSignedAggregateAttestationAndProof struct {
	TraceEventPayloadMetaData
	SignedAggregateAttestationAndProof *ethtypes.SignedAggregateAttestationAndProof
}

type TraceEventSignedAggregateAttestationAndProofElectra struct {
	TraceEventPayloadMetaData
	SignedAggregateAttestationAndProofElectra *ethtypes.SignedAggregateAttestationAndProofElectra
}

type TraceEventSignedContributionAndProof struct {
	TraceEventPayloadMetaData
	SignedContributionAndProof *ethtypes.SignedContributionAndProof
}

type TraceEventSingleAttestation struct {
	TraceEventPayloadMetaData
	SingleAttestation *ethtypes.SingleAttestation
}

type TraceEventSyncCommitteeMessage struct {
	TraceEventPayloadMetaData
	SyncCommitteeMessage *ethtypes.SyncCommitteeMessage //nolint:staticcheck // gRPC API deprecated but still supported until v8 (2026)
}

type TraceEventVoluntaryExit struct {
	TraceEventPayloadMetaData
	VoluntaryExit *ethtypes.VoluntaryExit
}

type UnifiedSharder struct {
	// contains filtered or unexported fields
}