feat: claim all claimable tokens if token addresses is empty

Layr-Labs · Nov 25, 2024 · ab89bb0 · ab89bb0
1 parent 74bcf53
commit ab89bb0
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/pkg/rewards/claim.go b/pkg/rewards/claim.go
@@ -118,8 +118,15 @@ func BatchClaim(
 		earnerAddr := gethcommon.HexToAddress(claimConfig.EarnerAddress)
 
 		var tokenAddrs []gethcommon.Address
-		for _, addr := range claimConfig.TokenAddresses {
-			tokenAddrs = append(tokenAddrs, gethcommon.HexToAddress(addr))
+
+		// Empty token addresses list will create a claim for all tokens claimable
+		// by the earner address.
+		if len(claimConfig.TokenAddresses) == 0 {
+			tokenAddrs = nil
+		} else {
+			for _, addr := range claimConfig.TokenAddresses {
+				tokenAddrs = append(tokenAddrs, gethcommon.HexToAddress(addr))
+			}
 		}
 
 		elClaim, _, _, err := generateClaimPayload(