probedb

type AggregateAttestation struct {
	IPAddr          net.IP
	Source          string
	Method          string
	Slot            uint32
	CommitteeIndex  uint16
	AggregationBits []byte
	BeaconBlockRoot []byte
	SourceRoot      []byte
	TargetRoot      []byte
	DelayMS         uint32
}

type AggregateAttestationFilter struct {
	// IPAddr is the IP address from which to fetch results.
	// If empty then there is no IP address filter.
	IPAddr string

	// Sources are the beacon nodes from which to fetch results.
	// If empty then there is no source filter.
	Sources []string

	// Methods are the collection methods from which to fetch results.
	// If empty then there is no method filter.
	Methods []string

	// From is the slot of the earliest result to fetch.
	// If nil then there is no earliest slot.
	From *phase0.Slot

	// To is the slot of the latest result to fetch.
	// If nil then there is no latest slot.
	To *phase0.Slot

	// Order is either OrderEarliest, in which case the earliest results
	// that match the filter are returned, or OrderLatest, in which case the
	// latest results that match the filter are returned.
	// The default is OrderEarliest.
	Order Order

	// Limit is the maximum number of results to return.
	// If 0 then there is no limit.
	Limit uint32
}

type AggregateAttestationsProvider interface {
	// AggregateAttestations obtains the aggregate attestations for a filter.
	AggregateAttestations(ctx context.Context, filter *AggregateAttestationFilter) ([]*AggregateAttestation, error)
}

type AggregateAttestationsSetter interface {
	Service

	// SetAggregateAttestation sets an aggregate attestation.
	SetAggregateAttestation(ctx context.Context, aggregateAttestation *AggregateAttestation) error
}

type AttestationSummariesProvider interface {
	// AttestationSummaries obtains the attestation summaries for a filter.
	AttestationSummaries(ctx context.Context, filter *AttestationSummaryFilter) ([]*AttestationSummary, error)
}

type AttestationSummariesSetter interface {
	Service

	// SetAttestationSummary sets an attestation summary.
	SetAttestationSummary(ctx context.Context, summary *AttestationSummary) error
}

type AttestationSummary struct {
	IPAddr          net.IP
	Source          string
	Method          string
	Slot            uint32
	CommitteeIndex  uint16
	BeaconBlockRoot []byte
	SourceRoot      []byte
	TargetRoot      []byte
	// AttesterBuckets contains the information about when specific indices
	// were first seen.
	// This is a raw representation of a github.com/prysmaticlabs/go-bitfield.Bitlist
	AttesterBuckets [][]byte
}

type AttestationSummaryFilter struct {
	// IPAddr is the IP address from which to fetch data.
	// If empty then there is no IP address filter.
	IPAddr string

	// Sources are the beacon nodes from which to fetch results.
	// If empty then there is no source filter.
	Sources []string

	// Methods are the collection methods from which to fetch results.
	// If empty then there is no method filter.
	Methods []string

	// From is the slot of the earliest data to fetch.
	// If nil then there is no earliest slot.
	From *phase0.Slot

	// To is the slot of the latest data to fetch.
	// If nil then there is no latest slot.
	To *phase0.Slot

	// Order is either OrderEarliest, in which case the earliest results
	// that match the filter are returned, or OrderLatest, in which case the
	// latest results that match the filter are returned.
	// The default is OrderEarliest.
	Order Order

	// Limit is the maximum number of results to return.
	// If 0 then there is no limit.
	Limit uint32
}

type BlockDelaysProvider interface {
	// BlockDelays obtains the block delays for a range of slots.
	BlockDelays(ctx context.Context, filter *DelayFilter) ([]*Delay, error)
}

type BlockDelaysSetter interface {
	Service

	// SetBlockDelay sets a block delay.
	SetBlockDelay(ctx context.Context, delay *Delay) error
}

type Delay struct {
	IPAddr  net.IP
	Source  string
	Method  string
	Slot    uint32
	DelayMS uint32
}

type DelayFilter struct {
	// IPAddr is the IP address from which to fetch delays.
	// If empty then there is no IP address filter.
	IPAddr string

	// Sources are the beacon nodes from which to fetch results.
	// If empty then there is no source filter.
	Sources []string

	// Methods are the collection methods from which to fetch results.
	// If empty then there is no method filter.
	Methods []string

	// From is the slot of the earliest delay to fetch.
	// If nil then there is no earliest slot.
	From *phase0.Slot

	// To is the slot of the latest delay to fetch.
	// If nil then there is no latest slot.
	To *phase0.Slot

	// Order is either OrderEarliest, in which case the earliest results
	// that match the filter are returned, or OrderLatest, in which case the
	// latest results that match the filter are returned.
	// The default is OrderEarliest.
	Order Order

	// Selection is the selection of the delay(s).
	// The default is SelectionMinimum.
	Selection Selection
}

type HeadDelaysProvider interface {
	// HeadDelays obtains the minimum head delays for a range of slots.
	HeadDelays(ctx context.Context, filter *DelayFilter) ([]*Delay, error)
}

type HeadDelaysSetter interface {
	Service

	// SetHeadDelay sets a head delay.
	SetHeadDelay(ctx context.Context, delay *Delay) error
}

type Order uint8

type Selection uint8

type Service interface {
	// BeginTx begins a transaction.
	BeginTx(ctx context.Context) (context.Context, context.CancelFunc, error)

	// CommitTx commits a transaction.
	CommitTx(ctx context.Context) error

	// SetMetadata sets a metadata key to a JSON value.
	SetMetadata(ctx context.Context, key string, value []byte) error

	// Metadata obtains the JSON value from a metadata key.
	Metadata(ctx context.Context, key string) ([]byte, error)
}