Merge pull request #441 from blinklabs-io/feat/ledger-block-type-map

feat: move block header to block type map to ledger

Author: agaffney

ledger/era.go

@@ -57,3 +57,25 @@ func GetEraById(eraId uint8) *Era {
 	}
 	return &era
 }
+
+// BlockHeaderToBlockTypeMap is a mapping of NtC chainsync block header types
+// to NtN block types (era ID)
+var BlockHeaderToBlockTypeMap = map[uint]uint{
+	BlockHeaderTypeShelley: BlockTypeShelley,
+	BlockHeaderTypeAllegra: BlockTypeAllegra,
+	BlockHeaderTypeMary:    BlockTypeMary,
+	BlockHeaderTypeAlonzo:  BlockTypeAlonzo,
+	BlockHeaderTypeBabbage: BlockTypeBabbage,
+	BlockHeaderTypeConway:  BlockTypeConway,
+}
+
+// BlockToBlockHeaderTypeMap is a mapping of NtN chainsync block types (era ID)
+// to NtC block header types
+var BlockToBlockHeaderTypeMap = map[uint]uint{
+	BlockTypeShelley: BlockHeaderTypeShelley,
+	BlockTypeAllegra: BlockHeaderTypeAllegra,
+	BlockTypeMary:    BlockHeaderTypeMary,
+	BlockTypeAlonzo:  BlockHeaderTypeAlonzo,
+	BlockTypeBabbage: BlockHeaderTypeBabbage,
+	BlockTypeConway:  BlockHeaderTypeConway,
+}

protocol/chainsync/client.go

@@ -275,16 +275,8 @@ func (c *Client) handleRollForward(msgGeneric protocol.Message) error {
 				return err
 			}
 		default:
-			// Map block header types to block types
-			blockTypeMap := map[uint]uint{
-				ledger.BlockHeaderTypeShelley: ledger.BlockTypeShelley,
-				ledger.BlockHeaderTypeAllegra: ledger.BlockTypeAllegra,
-				ledger.BlockHeaderTypeMary:    ledger.BlockTypeMary,
-				ledger.BlockHeaderTypeAlonzo:  ledger.BlockTypeAlonzo,
-				ledger.BlockHeaderTypeBabbage: ledger.BlockTypeBabbage,
-				ledger.BlockHeaderTypeConway:  ledger.BlockTypeConway,
-			}
-			blockType = blockTypeMap[blockEra]
+			// Map block header type to block type
+			blockType = ledger.BlockHeaderToBlockTypeMap[blockEra]
 			var err error
 			blockHeader, err = ledger.NewBlockHeaderFromCbor(
 				blockType,