feat: add support for TX attribute: protocol parameters update (#599)

Signed-off-by: Ales Verbic <verbotenj@blinklabs.io>
blinklabs-io · Apr 29, 2024 · cdfeb4b · cdfeb4b
1 parent 77a9387
commit cdfeb4b
Show file tree

Hide file tree

Showing 8 changed files with 58 additions and 0 deletions.
diff --git a/ledger/allegra.go b/ledger/allegra.go
@@ -170,6 +170,14 @@ func (t AllegraTransaction) IsValid() bool {
 	return true
 }
 
+func (t *AllegraTransaction) ProtocolParametersUpdate() map[Blake2b224]any {
+	updateMap := make(map[Blake2b224]any)
+	for k, v := range t.Body.Update.ProtocolParamUpdates {
+		updateMap[k] = v
+	}
+	return updateMap
+}
+
 func (t *AllegraTransaction) Cbor() []byte {
 	// Return stored CBOR if we have any
 	cborData := t.DecodeStoreCbor.Cbor()

diff --git a/ledger/alonzo.go b/ledger/alonzo.go
@@ -268,6 +268,14 @@ func (t AlonzoTransaction) IsValid() bool {
 	return t.IsTxValid
 }
 
+func (t *AlonzoTransaction) ProtocolParametersUpdate() map[Blake2b224]any {
+	updateMap := make(map[Blake2b224]any)
+	for k, v := range t.Body.Update.ProtocolParamUpdates {
+		updateMap[k] = v
+	}
+	return updateMap
+}
+
 func (t *AlonzoTransaction) Cbor() []byte {
 	// Return stored CBOR if we have any
 	cborData := t.DecodeStoreCbor.Cbor()

diff --git a/ledger/babbage.go b/ledger/babbage.go
@@ -438,6 +438,14 @@ func (t BabbageTransaction) IsValid() bool {
 	return t.IsTxValid
 }
 
+func (t *BabbageTransaction) ProtocolParametersUpdate() map[Blake2b224]any {
+	updateMap := make(map[Blake2b224]any)
+	for k, v := range t.Body.Update.ProtocolParamUpdates {
+		updateMap[k] = v
+	}
+	return updateMap
+}
+
 func (t *BabbageTransaction) Cbor() []byte {
 	// Return stored CBOR if we have any
 	cborData := t.DecodeStoreCbor.Cbor()

diff --git a/ledger/byron.go b/ledger/byron.go
@@ -187,6 +187,15 @@ func (t *ByronTransaction) Utxorpc() *utxorpc.Tx {
 	return &utxorpc.Tx{}
 }
 
+func (t *ByronTransaction) ProtocolParametersUpdate() map[Blake2b224]any {
+	// TODO: Implement this add missing Body TransactionBody
+	updateMap := make(map[Blake2b224]any)
+	// for k, v := range t.Body.Update.ProtocolParamUpdates {
+	// 	updateMap[k] = v
+	// }
+	return updateMap
+}
+
 type ByronTransactionInput struct {
 	cbor.StructAsArray
 	TxId        Blake2b256

diff --git a/ledger/conway.go b/ledger/conway.go
@@ -172,6 +172,14 @@ func (t ConwayTransaction) IsValid() bool {
 	return t.IsTxValid
 }
 
+func (t *ConwayTransaction) ProtocolParametersUpdate() map[Blake2b224]any {
+	updateMap := make(map[Blake2b224]any)
+	for k, v := range t.Body.Update.ProtocolParamUpdates {
+		updateMap[k] = v
+	}
+	return updateMap
+}
+
 func (t *ConwayTransaction) Cbor() []byte {
 	// Return stored CBOR if we have any
 	cborData := t.DecodeStoreCbor.Cbor()

diff --git a/ledger/mary.go b/ledger/mary.go
@@ -183,6 +183,14 @@ func (t MaryTransaction) IsValid() bool {
 	return true
 }
 
+func (t *MaryTransaction) ProtocolParametersUpdate() map[Blake2b224]any {
+	updateMap := make(map[Blake2b224]any)
+	for k, v := range t.Body.Update.ProtocolParamUpdates {
+		updateMap[k] = v
+	}
+	return updateMap
+}
+
 func (t *MaryTransaction) Cbor() []byte {
 	// Return stored CBOR if we have any
 	cborData := t.DecodeStoreCbor.Cbor()

diff --git a/ledger/shelley.go b/ledger/shelley.go
@@ -371,6 +371,14 @@ func (t ShelleyTransaction) Utxorpc() *utxorpc.Tx {
 	return t.Body.Utxorpc()
 }
 
+func (t *ShelleyTransaction) ProtocolParametersUpdate() map[Blake2b224]any {
+	updateMap := make(map[Blake2b224]any)
+	for k, v := range t.Body.Update.ProtocolParamUpdates {
+		updateMap[k] = v
+	}
+	return updateMap
+}
+
 func (t *ShelleyTransaction) Cbor() []byte {
 	// Return stored CBOR if we have any
 	cborData := t.DecodeStoreCbor.Cbor()

diff --git a/ledger/tx.go b/ledger/tx.go
@@ -28,6 +28,7 @@ type Transaction interface {
 	TransactionBody
 	Metadata() *cbor.Value
 	IsValid() bool
+	ProtocolParametersUpdate() map[Blake2b224]any
 }
 
 type TransactionBody interface {