feat: changes to support cost models while updating Alonzon protocols

Signed-off-by: Jenita <[email protected]>

Author: Jenita

ledger/alonzo/genesis.go

@@ -16,6 +16,7 @@ package alonzo
 
 import (
 	"encoding/json"
+	"fmt"
 	"io"
 	"math/big"
 	"os"
@@ -29,7 +30,7 @@ type AlonzoGenesis struct {
 	ExecutionPrices      AlonzoGenesisExecutionPrices `json:"executionPrices"`
 	MaxTxExUnits         AlonzoGenesisExUnits         `json:"maxTxExUnits"`
 	MaxBlockExUnits      AlonzoGenesisExUnits         `json:"maxBlockExUnits"`
-	CostModels           map[string]map[string]int    `json:"costModels"`
+	CostModels           map[string]interface{}       `json:"costModels"`
 }
 
 func NewAlonzoGenesisFromReader(r io.Reader) (AlonzoGenesis, error) {
@@ -39,6 +40,9 @@ func NewAlonzoGenesisFromReader(r io.Reader) (AlonzoGenesis, error) {
 	if err := dec.Decode(&ret); err != nil {
 		return ret, err
 	}
+	if err := ret.NormalizeCostModels(); err != nil {
+		return ret, err
+	}
 	return ret, nil
 }
 
@@ -76,3 +80,42 @@ func (r *AlonzoGenesisExecutionPricesRat) UnmarshalJSON(data []byte) error {
 	r.Rat = big.NewRat(tmpData.Numerator, tmpData.Denominator)
 	return nil
 }
+
+func (a *AlonzoGenesis) NormalizeCostModels() error {
+	if a.CostModels == nil {
+		return nil
+	}
+
+	normalized := make(map[string]map[string]int)
+	for version, model := range a.CostModels {
+		if modelMap, ok := model.(map[string]interface{}); ok {
+			versionMap := make(map[string]int)
+			for k, v := range modelMap {
+				switch val := v.(type) {
+				case float64:
+					versionMap[k] = int(val)
+				case int:
+					versionMap[k] = val
+				case json.Number:
+					intVal, err := val.Int64()
+					if err != nil {
+						floatVal, err := val.Float64()
+						if err != nil {
+							return fmt.Errorf("invalid number in cost model: %v", val)
+						}
+						intVal = int64(floatVal)
+					}
+					versionMap[k] = int(intVal)
+				default:
+					return fmt.Errorf("invalid cost model value type: %T", v)
+				}
+			}
+			normalized[version] = versionMap
+		}
+	}
+	a.CostModels = make(map[string]interface{})
+	for k, v := range normalized {
+		a.CostModels[k] = v
+	}
+	return nil
+}

ledger/alonzo/genesis_test.go

@@ -243,8 +243,8 @@ var expectedGenesisObj = alonzo.AlonzoGenesis{
 		Mem:   50000000,
 		Steps: 40000000000,
 	},
-	CostModels: map[string]map[string]int{
-		"PlutusV1": {
+	CostModels: map[string]interface{}{
+		"PlutusV1": map[string]int{
 			"addInteger-cpu-arguments-intercept":                       197209,
 			"addInteger-cpu-arguments-slope":                           0,
 			"addInteger-memory-arguments-intercept":                    1,
@@ -484,8 +484,8 @@ func TestNewAlonzoGenesisFromReader(t *testing.T) {
 		t.Logf("prMem is correct: %v", result.ExecutionPrices.Mem.Rat)
 	}
 
-	expectedCostModels := map[string]map[string]int{
-		"PlutusV1": {
+	expectedCostModels := map[string]interface{}{
+		"PlutusV1": map[string]int{
 			"addInteger-cpu-arguments-intercept": 205665,
 			"addInteger-cpu-arguments-slope":     812,
 		},