feat: implement rewards v2.2 - Operator Set Rewards with Unique & Total Stake

internal/config/config.go

@@ -47,6 +47,7 @@ const (
 	RewardsFork_Colorado    ForkName = "colorado"
 	RewardsFork_Red         ForkName = "red"
 	RewardsFork_Pecos       ForkName = "pecos"
+	RewardsFork_Sabine      ForkName = "sabine"
 )
 
 func normalizeFlagName(name string) string {
@@ -104,6 +105,7 @@ type RewardsConfig struct {
 	ValidateRewardsRoot          bool
 	GenerateStakerOperatorsTable bool
 	CalculateRewardsDaily        bool
+	RewardsV2_2Enabled           bool
 }
 
 type StatsdConfig struct {
@@ -212,6 +214,7 @@ var (
 	RewardsValidateRewardsRoot          = "rewards.validate_rewards_root"
 	RewardsGenerateStakerOperatorsTable = "rewards.generate_staker_operators_table"
 	RewardsCalculateRewardsDaily        = "rewards.calculate_rewards_daily"
+	RewardsV2_2Enabled                  = "rewards.v2_2_enabled"
 
 	EthereumRpcBaseUrl               = "ethereum.rpc_url"
 	EthereumRpcContractCallBatchSize = "ethereum.contract_call_batch_size"
@@ -246,6 +249,13 @@ var (
 	CoingeckoApiKey = "coingecko.api-key"
 )
 
+func FloatWithDefault(value, defaultValue float64) float64 {
+	if value == 0.0 {
+		return defaultValue
+	}
+	return value
+}
+
 func NewConfig() *Config {
 	return &Config{
 		Debug: viper.GetBool(normalizeFlagName("debug")),
@@ -297,6 +307,7 @@ func NewConfig() *Config {
 			ValidateRewardsRoot:          viper.GetBool(normalizeFlagName(RewardsValidateRewardsRoot)),
 			GenerateStakerOperatorsTable: viper.GetBool(normalizeFlagName(RewardsGenerateStakerOperatorsTable)),
 			CalculateRewardsDaily:        viper.GetBool(normalizeFlagName(RewardsCalculateRewardsDaily)),
+			RewardsV2_2Enabled:           viper.GetBool(normalizeFlagName(RewardsV2_2Enabled)),
 		},
 
 		DataDogConfig: DataDogConfig{
@@ -419,13 +430,19 @@ func (c *Config) GetContractsMapForChain() *ContractAddresses {
 		}
 	} else if c.Chain == Chain_Hoodi {
 		return &ContractAddresses{
-			AllocationManager:  "0x95a7431400f362f3647a69535c5666ca0133caa0",
-			AvsDirectory:       "0xd58f6844f79eb1fbd9f7091d05f7cb30d3363926",
-			DelegationManager:  "0x867837a9722c512e0862d8c2e15b8be220e8b87d",
-			EigenpodManager:    "0xcd1442415fc5c29aa848a49d2e232720be07976c",
-			RewardsCoordinator: "0x29e8572678e0c272350aa0b4b8f304e47ebcd5e7",
-			StrategyManager:    "0xee45e76ddbedda2918b8c7e3035cd37eab3b5d41",
-			KeyRegistrar:       "0x5737e38a260545d8feccb4cae2cfd984da4130ed",
+			AllocationManager:        "0x95a7431400f362f3647a69535c5666ca0133caa0",
+			AvsDirectory:             "0xd58f6844f79eb1fbd9f7091d05f7cb30d3363926",
+			DelegationManager:        "0x867837a9722c512e0862d8c2e15b8be220e8b87d",
+			EigenpodManager:          "0xcd1442415fc5c29aa848a49d2e232720be07976c",
+			RewardsCoordinator:       "0x29e8572678e0c272350aa0b4b8f304e47ebcd5e7",
+			StrategyManager:          "0xee45e76ddbedda2918b8c7e3035cd37eab3b5d41",
+			KeyRegistrar:             "0x5737e38a260545d8feccb4cae2cfd984da4130ed",
+			ReleaseManager:           "0xe863060013cb95473b96f7c3e1444e3e3df65671",
+			CrossChainRegistry:       "0x9269432451965996be7796582c062cb0795d3e8b",
+			OperatorTableUpdater:     "0xb02a15c6bd0882b35e9936a9579f35fb26e11476",
+			ECDSACertificateVerifier: "0xb3cd1a457dea9a9a6f6406c6419b1c326670a96f",
+			BN254CertificateVerifier: "0xff58a373c18268f483c1f5ca03cf885c0c43373a",
+			TaskMailbox:              "0xb99cc53e8db7018f557606c2a5b066527bf96b26",
 		}
 	} else if c.Chain == Chain_PreprodHoodi {
 		return &ContractAddresses{
@@ -552,6 +569,10 @@ func (c *Config) GetRewardsSqlForkDates() (ForkMap, error) {
 				Date:        "2025-05-14",
 				BlockNumber: 3840004,
 			},
+			RewardsFork_Sabine: Fork{
+				Date:        "1970-01-01",
+				BlockNumber: 0,
+			},
 		}, nil
 	case Chain_Holesky:
 		return ForkMap{
@@ -590,6 +611,10 @@ func (c *Config) GetRewardsSqlForkDates() (ForkMap, error) {
 				Date:        "2025-05-14",
 				BlockNumber: 3840004,
 			},
+			RewardsFork_Sabine: Fork{
+				Date:        "1970-01-01",
+				BlockNumber: 0,
+			},
 		}, nil
 	case Chain_Sepolia:
 		return ForkMap{
@@ -618,6 +643,10 @@ func (c *Config) GetRewardsSqlForkDates() (ForkMap, error) {
 				Date:        "2025-05-14",
 				BlockNumber: 8327038,
 			},
+			RewardsFork_Sabine: Fork{
+				Date:        "2026-02-09",
+				BlockNumber: 10225000,
+			},
 		}, nil
 	case Chain_Hoodi:
 		return ForkMap{
@@ -646,6 +675,10 @@ func (c *Config) GetRewardsSqlForkDates() (ForkMap, error) {
 				Date:        "1970-01-01",
 				BlockNumber: 0,
 			},
+			RewardsFork_Sabine: Fork{
+				Date:        "2026-02-09",
+				BlockNumber: 1990000,
+			},
 		}, nil
 	case Chain_PreprodHoodi:
 		return ForkMap{
@@ -674,6 +707,10 @@ func (c *Config) GetRewardsSqlForkDates() (ForkMap, error) {
 				Date:        "1970-01-01",
 				BlockNumber: 0,
 			},
+			RewardsFork_Sabine: Fork{
+				Date:        "1970-01-01",
+				BlockNumber: 0,
+			},
 		}, nil
 	case Chain_Mainnet:
 		return ForkMap{
@@ -716,6 +753,10 @@ func (c *Config) GetRewardsSqlForkDates() (ForkMap, error) {
 				Date:        "2025-05-14",
 				BlockNumber: 22483225,
 			},
+			RewardsFork_Sabine: Fork{
+				Date:        "2026-02-16",
+				BlockNumber: 24274311,
+			},
 		}, nil
 	}
 	return nil, errors.New("unsupported chain")
@@ -841,6 +882,23 @@ func (c *Config) IsRewardsV2_1EnabledForCutoffDate(cutoffDate string) (bool, err
 	return cutoffDateTime.Compare(mississippiForkDateTime) >= 0, nil
 }
 
+func (c *Config) IsRewardsV2_2EnabledForCutoffDate(cutoffDate string) (bool, error) {
+	forks, err := c.GetRewardsSqlForkDates()
+	if err != nil {
+		return false, err
+	}
+	cutoffDateTime, err := time.Parse(time.DateOnly, cutoffDate)
+	if err != nil {
+		return false, errors.Join(fmt.Errorf("failed to parse cutoff date %s", cutoffDate), err)
+	}
+	sabineForkDateTime, err := time.Parse(time.DateOnly, forks[RewardsFork_Sabine].Date)
+	if err != nil {
+		return false, errors.Join(fmt.Errorf("failed to parse Sabine fork date %s", forks[RewardsFork_Sabine].Date), err)
+	}
+
+	return cutoffDateTime.Compare(sabineForkDateTime) >= 0, nil
+}
+
 // CanIgnoreIncorrectRewardsRoot returns true if the rewards root can be ignored for the given block number
 //
 // Due to inconsistencies in the rewards root calculation on testnet, we know that some roots