feat: Add FetchBeaconBlockHeader method

ethpandaops · Sep 19, 2024 · 2fc9d92 · 2fc9d92
1 parent d196706
commit 2fc9d92
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/pkg/beacon/beacon.go b/pkg/beacon/beacon.go
@@ -97,6 +97,8 @@ type Node interface {
 	FetchAttestationData(ctx context.Context, slot phase0.Slot, committeeIndex phase0.CommitteeIndex) (*phase0.AttestationData, error)
 	// FetchBeaconBlockBlobs fetches blob sidecars for the given block id.
 	FetchBeaconBlockBlobs(ctx context.Context, blockID string) ([]*deneb.BlobSidecar, error)
+	// FetchBeaconBlockHeader fetches beacon block headers.
+	FetchBeaconBlockHeader(ctx context.Context, opts *eapi.BeaconBlockHeaderOpts) (*v1.BeaconBlockHeader, error)
 
 	// Subscriptions
 	// - Proxied Beacon events

diff --git a/pkg/beacon/fetch.go b/pkg/beacon/fetch.go
@@ -272,3 +272,17 @@ func (n *node) FetchAttestationData(ctx context.Context, slot phase0.Slot, commi
 
 	return rsp.Data, nil
 }
+
+func (n *node) FetchBeaconBlockHeader(ctx context.Context, opts *api.BeaconBlockHeaderOpts) (*v1.BeaconBlockHeader, error) {
+	provider, isProvider := n.client.(eth2client.BeaconBlockHeadersProvider)
+	if !isProvider {
+		return nil, errors.New("client does not implement eth2client.BeaconBlockHeadersProvider")
+	}
+
+	rsp, err := provider.BeaconBlockHeader(ctx, opts)
+	if err != nil {
+		return nil, err
+	}
+
+	return rsp.Data, nil
+}