go-ethereum/beacon/beaclient/client.go

// Copyright 2023 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

package beaclient

import (
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net/http"

	eth2client "github.com/attestantio/go-eth2-client"
	eth2api "github.com/attestantio/go-eth2-client/api"
	eth2http "github.com/attestantio/go-eth2-client/http"
	eth2spec "github.com/attestantio/go-eth2-client/spec"
	"github.com/ethereum/go-ethereum/beacon/types"
	"github.com/ethereum/go-ethereum/common"
	"github.com/rs/zerolog"
)

var (
	beaconLightClientBootstrap        = "eth/v1/beacon/light_client/bootstrap"
	beaconLightClientUpdate           = "eth/v1/beacon/light_client/updates"
	beaconLightClientOptimisticUpdate = "eth/v1/beacon/light_client/optimistic_update"
	beaconLightClientFinalityUpdate   = "eth/v1/beacon/light_client/finality_update"
)

type Client struct {
	ctx    context.Context
	url    string
	client eth2client.Service
}

func NewClient(ctx context.Context, server string) (*Client, error) {
	client, err := eth2http.New(
		ctx,
		eth2http.WithAddress(server),
		eth2http.WithLogLevel(zerolog.WarnLevel),
		eth2http.WithEnforceJSON(true),
	)
	if err != nil {
		return nil, err
	}
	return &Client{
		ctx:    ctx,
		url:    server,
		client: client,
	}, nil
}

func (c *Client) Bootstrap(root common.Hash) (*types.Bootstrap, error) {
	resp, err := http.Get(fmt.Sprintf("%s/%s/%s", c.url, beaconLightClientBootstrap, root.String()))
	if err != nil {
		return nil, fmt.Errorf("failed http request: %w", err)
	}
	b, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("failed to read response: %w", err)
	}
	var bs types.Bootstrap
	if err := json.Unmarshal(b, &bs); err != nil {
		return nil, fmt.Errorf("failed to unmarshal data: %v", err)
	}
	return &bs, nil
}

func (c *Client) GetRangeUpdate(start, count int) ([]*types.LightClientUpdate, error) {
	resp, err := http.Get(fmt.Sprintf("%s/%s?start_period=%d&count=%d", c.url, beaconLightClientUpdate, start, count))
	if err != nil {
		return nil, fmt.Errorf("failed http request: %w", err)
	}
	b, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("failed to read response: %w", err)
	}
	var u []*types.LightClientUpdate
	if err := json.Unmarshal(b, &u); err != nil {
		return nil, fmt.Errorf("failed to unmarshal data: %v", err)
	}
	return u, nil

}

func (c *Client) GetOptimisticUpdate() (*types.LightClientUpdate, error) {
	resp, err := http.Get(fmt.Sprintf("%s/%s", c.url, beaconLightClientOptimisticUpdate))
	if err != nil {
		return nil, fmt.Errorf("failed http request: %w", err)
	}
	b, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("failed to read response: %w", err)
	}
	var u types.LightClientUpdate
	if err := json.Unmarshal(b, &u); err != nil {
		return nil, fmt.Errorf("failed to unmarshal data: %v", err)
	}
	return &u, nil
}

func (c *Client) GetFinalityUpdate() (*types.LightClientUpdate, error) {
	resp, err := http.Get(fmt.Sprintf("%s/%s", c.url, beaconLightClientFinalityUpdate))
	if err != nil {
		return nil, fmt.Errorf("failed http request: %w", err)
	}
	b, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("failed to read response: %w", err)
	}
	var u types.LightClientUpdate
	if err := json.Unmarshal(b, &u); err != nil {
		return nil, fmt.Errorf("failed to unmarshal data: %v", err)
	}
	return &u, nil

}

func (c *Client) GetBlock(root common.Hash) (*eth2spec.VersionedSignedBeaconBlock, error) {
	provider, ok := c.client.(eth2client.SignedBeaconBlockProvider)
	if !ok {
		return nil, fmt.Errorf("beacon server does not support retrieving blocks")
	}
	resp, err := provider.SignedBeaconBlock(c.ctx, &eth2api.SignedBeaconBlockOpts{Block: root.String()})
	if err != nil {
		return nil, fmt.Errorf("failed http request: %w", err)
	}
	return resp.Data, nil
}