update with latest methods

chainio/clients/elcontracts/reader.go

@@ -485,23 +485,18 @@ func (r *ChainReader) GetStrategiesForOperatorSet(
 	}
 }
 
-type OperatorSetStakes struct {
-	OperatorSet     allocationmanager.OperatorSet
-	Strategies      []gethcommon.Address
-	Operators       []gethcommon.Address
-	DelegatedStakes [][]*big.Int
-	SlashableStakes [][]*big.Int
-}
-
-// GetDelegatedAndSlashableSharesForOperatorSets returns the strategies the operatorSets take into account, their
-// operators,
-// and the minimum amount of shares that multiple operators delegated to them and slashable by the operatorSets.
+// GetSlashableSharesForOperatorSets returns the strategies the operatorSets take into account, their
+// operators, and the minimum amount of shares that are slashable by the operatorSets.
 // Not supported for M2 AVSs
-func (r *ChainReader) GetDelegatedAndSlashableSharesForOperatorSets(
+func (r *ChainReader) GetSlashableSharesForOperatorSets(
 	opts *bind.CallOpts,
 	operatorSets []allocationmanager.OperatorSet,
 ) ([]OperatorSetStakes, error) {
 	operatorSetStakes := make([]OperatorSetStakes, len(operatorSets))
+	currentBlock, err := r.ethClient.BlockNumber(opts.Context)
+	if err != nil {
+		return nil, err
+	}
 	for i, operatorSet := range operatorSets {
 		operators, err := r.GetOperatorsForOperatorSet(opts, operatorSet)
 		if err != nil {
@@ -513,14 +508,15 @@ func (r *ChainReader) GetDelegatedAndSlashableSharesForOperatorSets(
 			return nil, err
 		}
 
-		delegatedShares, slashableShares, err := r.allocationManager.GetCurrentDelegatedAndSlashableOperatorShares(
+		slashableShares, err := r.allocationManager.GetMinimumSlashableStake(
 			opts,
 			allocationmanager.OperatorSet{
 				Id:  operatorSet.Id,
 				Avs: operatorSet.Avs,
 			},
 			operators,
 			strategies,
+			uint32(currentBlock),
 		)
 		if err != nil {
 			return nil, err
@@ -530,7 +526,6 @@ func (r *ChainReader) GetDelegatedAndSlashableSharesForOperatorSets(
 			OperatorSet:     operatorSet,
 			Strategies:      strategies,
 			Operators:       operators,
-			DelegatedStakes: delegatedShares,
 			SlashableStakes: slashableShares,
 		}
 	}
@@ -546,7 +541,7 @@ func (r *ChainReader) GetDelegatedAndSlashableSharesForOperatorSets(
 func (r *ChainReader) GetDelegatedAndSlashableSharesForOperatorSetsBefore(
 	opts *bind.CallOpts,
 	operatorSets []allocationmanager.OperatorSet,
-	beforeTimestamp uint32,
+	futureBlock uint32,
 ) ([]OperatorSetStakes, error) {
 	operatorSetStakes := make([]OperatorSetStakes, len(operatorSets))
 	for i, operatorSet := range operatorSets {
@@ -560,15 +555,15 @@ func (r *ChainReader) GetDelegatedAndSlashableSharesForOperatorSetsBefore(
 			return nil, err
 		}
 
-		delegatedShares, slashableShares, err := r.allocationManager.GetMinDelegatedAndSlashableOperatorSharesBefore(
+		slashableShares, err := r.allocationManager.GetMinimumSlashableStake(
 			opts,
 			allocationmanager.OperatorSet{
 				Id:  operatorSet.Id,
 				Avs: operatorSet.Avs,
 			},
 			operators,
 			strategies,
-			beforeTimestamp,
+			futureBlock,
 		)
 		if err != nil {
 			return nil, err
@@ -578,7 +573,6 @@ func (r *ChainReader) GetDelegatedAndSlashableSharesForOperatorSetsBefore(
 			OperatorSet:     operatorSet,
 			Strategies:      strategies,
 			Operators:       operators,
-			DelegatedStakes: delegatedShares,
 			SlashableStakes: slashableShares,
 		}
 	}

chainio/clients/elcontracts/types.go

@@ -3,6 +3,8 @@ package elcontracts
 import (
 	"math/big"
 
+	allocationmanager "github.com/Layr-Labs/eigensdk-go/contracts/bindings/AllocationManager"
+
 	"github.com/ethereum/go-ethereum/common"
 )
 
@@ -18,3 +20,10 @@ type AllocationInfo struct {
 	OperatorSetId    uint32
 	AvsAddress       common.Address
 }
+
+type OperatorSetStakes struct {
+	OperatorSet     allocationmanager.OperatorSet
+	Strategies      []common.Address
+	Operators       []common.Address
+	SlashableStakes [][]*big.Int
+}