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"`

	// Node is the configuration for the node
	Node NodeConfig `yaml:"node"`

	// 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"`
	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"`
}

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 Mimicry struct {
	Config *Config
	// contains filtered or unexported fields
}

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

type NodeConfig struct {
	// The private key for the libp2p host and local enode in hex format
	PrivateKeyStr string `yaml:"privateKeyStr" default:""`

	// General timeout when communicating with other network participants
	DialTimeout time.Duration `yaml:"dialTimeout" default:"5s"`

	// The address information of the local ethereuem [enode.Node].
	Devp2pHost string `yaml:"devp2pHost" default:"0.0.0.0"`
	Devp2pPort int    `yaml:"devp2pPort" default:"0"`

	// The address information of the local libp2p host
	Libp2pHost string `yaml:"libp2pHost" default:"0.0.0.0"`
	Libp2pPort int    `yaml:"libp2pPort" default:"0"`

	// The address information where the Beacon API or Prysm's custom API is accessible at
	PrysmHost     string `yaml:"prysmHost" default:"127.0.0.1"`
	PrysmPortHTTP int    `yaml:"prysmPortHttp" default:"3500"`
	PrysmPortGRPC int    `yaml:"prysmPortGrpc" default:"4000"`
	PrysmUseTLS   bool   `yaml:"prysmUseTls" default:"false"`

	// The maximum number of peers our libp2p host can be connected to.
	MaxPeers int `yaml:"maxPeers" default:"30"`

	// Limits the number of concurrent connection establishment routines. When
	// we discover peers over discv5 and are not at our MaxPeers limit we try
	// to establish a connection to a peer. However, we limit the concurrency to
	// this DialConcurrency value.
	DialConcurrency int `yaml:"dialConcurrency" default:"16"`

	// DataStreamType is the type of data stream to use for the node (e.g. kinesis, callback, etc).
	DataStreamType string `yaml:"dataStreamType" default:"callback"`

	// Subnets is the configuration for gossipsub subnets.
	Subnets map[string]*hermes.SubnetConfig `yaml:"subnets"`
}

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 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 UnifiedSharder struct {
	// contains filtered or unexported fields
}