Merge pull request #4 from Layr-Labs/sm-stakerDelegations

Add support for staker delegations

internal/eigenState/avsOperators/avsOperators.go

@@ -152,13 +152,7 @@ func (a *AvsOperators) getContractAddressesForEnvironment() map[string][]string
 // Given a log, determine if it is interesting to the state model
 func (a *AvsOperators) IsInterestingLog(log *storage.TransactionLog) bool {
 	addresses := a.getContractAddressesForEnvironment()
-	logAddress := strings.ToLower(log.Address)
-	if eventNames, ok := addresses[logAddress]; ok {
-		if slices.Contains(eventNames, log.EventName) {
-			return true
-		}
-	}
-	return false
+	return a.BaseEigenState.IsInterestingLog(addresses, log)
 }
 
 // Handle the state change for the given log

internal/eigenState/base/baseEigenState.go

@@ -6,6 +6,8 @@ import (
 	"github.com/Layr-Labs/sidecar/internal/parser"
 	"github.com/Layr-Labs/sidecar/internal/storage"
 	"go.uber.org/zap"
+	"slices"
+	"strings"
 )
 
 type BaseEigenState struct {
@@ -49,3 +51,13 @@ func (b *BaseEigenState) InitializeMerkleTreeBaseStateWithBlock(blockNumber uint
 		[]byte(fmt.Sprintf("%d", blockNumber)),
 	}
 }
+
+func (b *BaseEigenState) IsInterestingLog(contractsEvents map[string][]string, log *storage.TransactionLog) bool {
+	logAddress := strings.ToLower(log.Address)
+	if eventNames, ok := contractsEvents[logAddress]; ok {
+		if slices.Contains(eventNames, log.EventName) {
+			return true
+		}
+	}
+	return false
+}

internal/eigenState/operatorShares/operatorShares.go

@@ -152,13 +152,7 @@ func (osm *OperatorSharesModel) getContractAddressesForEnvironment() map[string]
 
 func (osm *OperatorSharesModel) IsInterestingLog(log *storage.TransactionLog) bool {
 	addresses := osm.getContractAddressesForEnvironment()
-	logAddress := strings.ToLower(log.Address)
-	if eventNames, ok := addresses[logAddress]; ok {
-		if slices.Contains(eventNames, log.EventName) {
-			return true
-		}
-	}
-	return false
+	return osm.BaseEigenState.IsInterestingLog(addresses, log)
 }
 
 func (osm *OperatorSharesModel) HandleStateChange(log *storage.TransactionLog) (interface{}, error) {