Add SSZ support to the Proposer API (#685)

* Add SSZ support to registerValidator

* Fix merge mistake

* Run go mod tidy

* Add SSZ support for getHeader

* Add a couple getHeader SSZ tests

* Add SSZ support for getPayload

* Do some clean up

* Add some debug prints

* Add back fast registration parsing

---------

Co-authored-by: Chris Hager <[email protected]>

Author: jtraglia

.golangci.yaml

@@ -30,6 +30,7 @@ linters:
     - exhaustruct
     - tenv
     - gocognit
+    - dupl
 
     #
     # Disabled because of generics:

common/constants.go

@@ -0,0 +1,8 @@
+package common
+
+const (
+	EthConsensusVersionBellatrix = "bellatrix"
+	EthConsensusVersionCapella   = "capella"
+	EthConsensusVersionDeneb     = "deneb"
+	EthConsensusVersionElectra   = "electra"
+)

common/types_spec.go

@@ -1,6 +1,7 @@
 package common
 
 import (
+	"bytes"
 	"encoding/json"
 	"fmt"
 
@@ -11,6 +12,7 @@ import (
 	builderApiV1 "github.com/attestantio/go-builder-client/api/v1"
 	builderSpec "github.com/attestantio/go-builder-client/spec"
 	eth2Api "github.com/attestantio/go-eth2-client/api"
+	eth2ApiV1Bellatrix "github.com/attestantio/go-eth2-client/api/v1/bellatrix"
 	eth2ApiV1Capella "github.com/attestantio/go-eth2-client/api/v1/capella"
 	eth2ApiV1Deneb "github.com/attestantio/go-eth2-client/api/v1/deneb"
 	eth2ApiV1Electra "github.com/attestantio/go-eth2-client/api/v1/electra"
@@ -589,3 +591,36 @@ func (r *VersionedSignedBlindedBeaconBlock) UnmarshalJSON(input []byte) error {
 	}
 	return errors.Wrap(err, "failed to unmarshal SignedBlindedBeaconBlock")
 }
+
+func (r *VersionedSignedBlindedBeaconBlock) Unmarshal(input []byte, contentType, ethConsensusVersion string) error {
+	switch contentType {
+	case "application/octet-stream":
+		if ethConsensusVersion != "" {
+			switch ethConsensusVersion {
+			case EthConsensusVersionBellatrix:
+				r.Version = spec.DataVersionBellatrix
+				r.Bellatrix = new(eth2ApiV1Bellatrix.SignedBlindedBeaconBlock)
+				return r.Bellatrix.UnmarshalSSZ(input)
+			case EthConsensusVersionCapella:
+				r.Version = spec.DataVersionCapella
+				r.Capella = new(eth2ApiV1Capella.SignedBlindedBeaconBlock)
+				return r.Capella.UnmarshalSSZ(input)
+			case EthConsensusVersionDeneb:
+				r.Version = spec.DataVersionDeneb
+				r.Deneb = new(eth2ApiV1Deneb.SignedBlindedBeaconBlock)
+				return r.Deneb.UnmarshalSSZ(input)
+			case EthConsensusVersionElectra:
+				r.Version = spec.DataVersionElectra
+				r.Electra = new(eth2ApiV1Electra.SignedBlindedBeaconBlock)
+				return r.Electra.UnmarshalSSZ(input)
+			default:
+				return ErrInvalidForkVersion
+			}
+		} else {
+			return ErrMissingEthConsensusVersion
+		}
+	case "application/json":
+		return json.NewDecoder(bytes.NewReader(input)).Decode(r)
+	}
+	return ErrInvalidContentType
+}

common/utils.go

@@ -26,9 +26,11 @@ import (
 )
 
 var (
-	ErrInvalidForkVersion = errors.New("invalid fork version")
-	ErrHTTPErrorResponse  = errors.New("got an HTTP error response")
-	ErrIncorrectLength    = errors.New("incorrect length")
+	ErrInvalidForkVersion         = errors.New("invalid fork version")
+	ErrHTTPErrorResponse          = errors.New("got an HTTP error response")
+	ErrIncorrectLength            = errors.New("incorrect length")
+	ErrMissingEthConsensusVersion = errors.New("missing eth-consensus-version")
+	ErrInvalidContentType         = errors.New("invalid content type")
 )
 
 // SlotPos returns the slot's position in the epoch (1-based, i.e. 1..32)

datastore/redis.go

@@ -847,6 +847,6 @@ func (r *RedisCache) NewPipeline() redis.Pipeliner { //nolint:ireturn,nolintlint
 	return r.client.Pipeline()
 }
 
-func (r *RedisCache) NewTxPipeline() redis.Pipeliner { //nolint:ireturn
+func (r *RedisCache) NewTxPipeline() redis.Pipeliner { //nolint:ireturn,nolintlint
 	return r.client.TxPipeline()
 }