feat(ledger): Added changes for decoding block headers of different shapes

Signed-off-by: Akhil Repala <[email protected]>

Author: arepala-uml

ledger/allegra.go

@@ -37,6 +37,7 @@ const (
 // Allegra functions
 var (
 	NewAllegraBlockFromCbor           = allegra.NewAllegraBlockFromCbor
+	NewAllegraBlockHeaderFromCbor     = allegra.NewAllegraBlockHeaderFromCbor
 	NewAllegraTransactionFromCbor     = allegra.NewAllegraTransactionFromCbor
 	NewAllegraTransactionBodyFromCbor = allegra.NewAllegraTransactionBodyFromCbor
 )

ledger/allegra/allegra.go

@@ -329,6 +329,14 @@ func NewAllegraBlockFromCbor(data []byte) (*AllegraBlock, error) {
 	return &allegraBlock, nil
 }
 
+func NewAllegraBlockHeaderFromCbor(data []byte) (*AllegraBlockHeader, error) {
+	var allegraBlockHeader AllegraBlockHeader
+	if _, err := cbor.Decode(data, &allegraBlockHeader); err != nil {
+		return nil, fmt.Errorf("Allegra block header decode error: %w", err)
+	}
+	return &allegraBlockHeader, nil
+}
+
 func NewAllegraTransactionBodyFromCbor(
 	data []byte,
 ) (*AllegraTransactionBody, error) {

ledger/alonzo.go

@@ -40,6 +40,7 @@ const (
 // Alonzo functions
 var (
 	NewAlonzoBlockFromCbor             = alonzo.NewAlonzoBlockFromCbor
+	NewAlonzoBlockHeaderFromCbor       = alonzo.NewAlonzoBlockFromCbor
 	NewAlonzoTransactionFromCbor       = alonzo.NewAlonzoTransactionFromCbor
 	NewAlonzoTransactionBodyFromCbor   = alonzo.NewAlonzoTransactionBodyFromCbor
 	NewAlonzoTransactionOutputFromCbor = alonzo.NewAlonzoTransactionOutputFromCbor

ledger/alonzo/alonzo.go

@@ -532,6 +532,14 @@ func NewAlonzoBlockFromCbor(data []byte) (*AlonzoBlock, error) {
 	return &alonzoBlock, nil
 }
 
+func NewAlonzoBlockHeaderFromCbor(data []byte) (*AlonzoBlockHeader, error) {
+	var alonzoBlockHeader AlonzoBlockHeader
+	if _, err := cbor.Decode(data, &alonzoBlockHeader); err != nil {
+		return nil, fmt.Errorf("Alonzo block header decode error: %w", err)
+	}
+	return &alonzoBlockHeader, nil
+}
+
 func NewAlonzoTransactionBodyFromCbor(
 	data []byte,
 ) (*AlonzoTransactionBody, error) {

ledger/block.go

@@ -54,12 +54,21 @@ func NewBlockHeaderFromCbor(blockType uint, data []byte) (BlockHeader, error) {
 	case BlockTypeByronMain:
 		return NewByronMainBlockHeaderFromCbor(data)
 	// TODO: break into separate cases and parse as specific block header types (#844)
-	case BlockTypeShelley, BlockTypeAllegra, BlockTypeMary, BlockTypeAlonzo:
+	case BlockTypeShelley:
 		return NewShelleyBlockHeaderFromCbor(data)
-	case BlockTypeBabbage, BlockTypeConway:
+	case BlockTypeAllegra:
+		return NewAllegraBlockHeaderFromCbor(data)
+	case BlockTypeMary:
+		return NewMaryBlockHeaderFromCbor(data)
+	case BlockTypeAlonzo:
+		return NewAlonzoBlockHeaderFromCbor(data)
+	case BlockTypeBabbage:
 		return NewBabbageBlockHeaderFromCbor(data)
+	case BlockTypeConway:
+		return NewConwayBlockHeaderFromCbor(data)
+	default:
+		return nil, fmt.Errorf("unknown node-to-node block type: %d", blockType)
 	}
-	return nil, fmt.Errorf("unknown node-to-node block type: %d", blockType)
 }
 
 func DetermineBlockType(data []byte) (uint, error) {

ledger/mary.go

@@ -40,6 +40,7 @@ const (
 // Mary functions
 var (
 	NewMaryBlockFromCbor             = mary.NewMaryBlockFromCbor
+	NewMaryBlockHeaderFromCbor       = mary.NewMaryBlockHeaderFromCbor
 	NewMaryTransactionFromCbor       = mary.NewMaryTransactionFromCbor
 	NewMaryTransactionBodyFromCbor   = mary.NewMaryTransactionBodyFromCbor
 	NewMaryTransactionOutputFromCbor = mary.NewMaryTransactionOutputFromCbor

ledger/mary/mary.go

@@ -414,6 +414,14 @@ func NewMaryBlockFromCbor(data []byte) (*MaryBlock, error) {
 	return &maryBlock, nil
 }
 
+func NewMaryBlockHeaderFromCbor(data []byte) (*MaryBlockHeader, error) {
+	var maryBlockHeader MaryBlockHeader
+	if _, err := cbor.Decode(data, &maryBlockHeader); err != nil {
+		return nil, fmt.Errorf("Mary block header decode error: %w", err)
+	}
+	return &maryBlockHeader, nil
+}
+
 func NewMaryTransactionBodyFromCbor(data []byte) (*MaryTransactionBody, error) {
 	var maryTx MaryTransactionBody
 	if _, err := cbor.Decode(data, &maryTx); err != nil {