Normalize addresses

Layr-Labs · Sep 9, 2024 · b232ad4 · b232ad4
1 parent 817befb
commit b232ad4
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git a/internal/eigenState/avsOperators/avsOperators.go b/internal/eigenState/avsOperators/avsOperators.go
@@ -123,8 +123,8 @@ func (a *AvsOperatorsModel) GetStateTransitions() (types.StateTransitions[Accumu
 			return nil, xerrors.Errorf("No state accumulator found for block %d", log.BlockNumber)
 		}
 
-		avs := arguments[0].Value.(string)
-		operator := arguments[1].Value.(string)
+		avs := strings.ToLower(arguments[0].Value.(string))
+		operator := strings.ToLower(arguments[1].Value.(string))
 
 		registered := false
 		if val, ok := outputData["status"]; ok {

diff --git a/internal/eigenState/operatorShares/operatorShares.go b/internal/eigenState/operatorShares/operatorShares.go
@@ -109,6 +109,10 @@ func parseLogOutputForOperatorShares(outputDataStr string) (*operatorSharesOutpu
 	decoder.UseNumber()
 
 	err := decoder.Decode(&outputData)
+	if err != nil {
+		return nil, err
+	}
+	outputData.Strategy = strings.ToLower(outputData.Strategy)
 	return outputData, err
 }
 
@@ -129,7 +133,7 @@ func (osm *OperatorSharesModel) GetStateTransitions() (types.StateTransitions[Ac
 		if _, ok := osm.stateAccumulator[log.BlockNumber]; !ok {
 			return nil, xerrors.Errorf("No state accumulator found for block %d", log.BlockNumber)
 		}
-		operator := arguments[0].Value.(string)
+		operator := strings.ToLower(arguments[0].Value.(string))
 
 		sharesStr := outputData.Shares.String()
 		shares, err := uint256.FromDecimal(sharesStr)

diff --git a/internal/eigenState/stakerDelegations/stakerDelegations.go b/internal/eigenState/stakerDelegations/stakerDelegations.go
@@ -107,8 +107,8 @@ func (s *StakerDelegationsModel) GetStateTransitions() (types.StateTransitions[A
 			return nil, xerrors.Errorf("No state accumulator found for block %d", log.BlockNumber)
 		}
 
-		staker := arguments[0].Value.(string)
-		operator := arguments[1].Value.(string)
+		staker := strings.ToLower(arguments[0].Value.(string))
+		operator := strings.ToLower(arguments[1].Value.(string))
 
 		slotId := NewSlotId(staker, operator)
 		record, ok := s.stateAccumulator[log.BlockNumber][slotId]

diff --git a/internal/eigenState/stakerShares/stakerShares.go b/internal/eigenState/stakerShares/stakerShares.go
@@ -114,6 +114,7 @@ func parseLogOutputForDepositEvent(outputDataStr string) (*depositOutputData, er
 		return nil, err
 	}
 	outputData.Staker = strings.ToLower(outputData.Staker)
+	outputData.Depositor = strings.ToLower(outputData.Depositor)
 	outputData.Strategy = strings.ToLower(outputData.Strategy)
 	return outputData, err
 }