Add airdrop tool

.gitignore

@@ -46,4 +46,10 @@ dependency-graph.png
 *.out
 *.synctex.gz
 contract_tests/*
-scripts/wasm/*.wasm
+
+#Airdrop tool
+balance.json
+rewards.json
+balancenft.json
+.env
+scripts/wasm/*.wasm

airdrop/backoff/backoff.go

@@ -0,0 +1,24 @@
+package backoff
+
+import (
+	"context"
+	"time"
+
+	"github.com/cenkalti/backoff/v4"
+)
+
+var globalBackoffOptions = []backoff.ExponentialBackOffOpts{
+	func(b *backoff.ExponentialBackOff) {
+		b.InitialInterval = 1 * time.Second
+	},
+	func(b *backoff.ExponentialBackOff) {
+		b.MaxInterval = 15 * time.Second
+	},
+	func(b *backoff.ExponentialBackOff) {
+		b.Multiplier = 2
+	},
+}
+
+func NewBackoff(ctx context.Context) *backoff.ExponentialBackOff {
+	return backoff.NewExponentialBackOff(globalBackoffOptions...)
+}

airdrop/chains/akash.go

@@ -0,0 +1,123 @@
+package chains
+
+import (
+	"fmt"
+	"log"
+	"strconv"
+
+	"github.com/eve-network/eve/airdrop/config"
+	"github.com/eve-network/eve/airdrop/utils"
+	"github.com/joho/godotenv"
+
+	sdkmath "cosmossdk.io/math"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
+)
+
+func Akash() ([]banktypes.Balance, []config.Reward, int, error) {
+	err := godotenv.Load()
+	if err != nil {
+		log.Printf("Error loading Akash environment variables: %v", err)
+		return nil, nil, 0, fmt.Errorf("failed to load env: %w", err)
+	}
+
+	blockHeight, err := utils.GetLatestHeight(config.GetAkashConfig().RPC + "/status")
+	if err != nil {
+		log.Printf("Failed to get latest height for Akash: %v", err)
+		return nil, nil, 0, fmt.Errorf("failed to get latest height for Akash: %w", err)
+	}
+
+	grpcAddr := config.GetAkashConfig().GRPCAddr
+	grpcConn, err := utils.SetupGRPCConnection(grpcAddr)
+	if err != nil {
+		log.Printf("Failed to connect to gRPC Akash: %v", err)
+		return nil, nil, 0, fmt.Errorf("failed to connect to gRPC Akash: %w", err)
+	}
+	defer grpcConn.Close()
+	stakingClient := stakingtypes.NewQueryClient(grpcConn)
+
+	delegators := []stakingtypes.DelegationResponse{}
+
+	validators, err := utils.GetValidators(stakingClient, blockHeight)
+	if err != nil {
+		log.Printf("Failed to get Akash validators: %v", err)
+		return nil, nil, 0, fmt.Errorf("failed to get Akash validators: %w", err)
+	}
+	log.Println("Validators: ", len(validators))
+	for validatorIndex, validator := range validators {
+		delegationsResponse, err := utils.GetValidatorDelegations(stakingClient, validator.OperatorAddress, blockHeight)
+		if err != nil {
+			log.Printf("Failed to query delegate info for Akash validator: %v", err)
+			return nil, nil, 0, fmt.Errorf("failed to query delegate info for Akash validator: %w", err)
+		}
+		total := delegationsResponse.Pagination.Total
+		log.Println("Response ", len(delegationsResponse.DelegationResponses))
+		log.Println("Akash validator "+strconv.Itoa(validatorIndex)+" ", total)
+		delegators = append(delegators, delegationsResponse.DelegationResponses...)
+	}
+
+	usd := sdkmath.LegacyMustNewDecFromStr("20")
+
+	apiURL := config.APICoingecko + config.GetAkashConfig().CoinID + "&vs_currencies=usd"
+	tokenInUsd, err := utils.FetchTokenPrice(apiURL, config.GetAkashConfig().CoinID)
+	if err != nil {
+		log.Println("Failed to fetch Akash token price: %w", err)
+		return nil, nil, 0, fmt.Errorf("failed to fetch Akash token price: %w", err)
+	}
+	tokenIn20Usd := usd.Quo(tokenInUsd)
+
+	rewardInfo := []config.Reward{}
+	balanceInfo := []banktypes.Balance{}
+
+	totalTokenDelegate := sdkmath.LegacyMustNewDecFromStr("0")
+	for _, delegator := range delegators {
+		validatorIndex := utils.FindValidatorInfo(validators, delegator.Delegation.ValidatorAddress)
+		validatorInfo := validators[validatorIndex]
+		token := (delegator.Delegation.Shares.MulInt(validatorInfo.Tokens)).QuoTruncate(validatorInfo.DelegatorShares)
+		totalTokenDelegate = totalTokenDelegate.Add(token)
+	}
+	eveAirdrop, err := sdkmath.LegacyNewDecFromStr(config.EveAirdrop)
+	if err != nil {
+		log.Println("Failed to convert EveAirdrop string to dec: %w", err)
+		return nil, nil, 0, fmt.Errorf("failed to convert EveAirdrop string to dec: %w", err)
+	}
+	testAmount := sdkmath.LegacyMustNewDecFromStr("0")
+
+	for _, delegator := range delegators {
+		validatorIndex := utils.FindValidatorInfo(validators, delegator.Delegation.ValidatorAddress)
+		validatorInfo := validators[validatorIndex]
+		token := (delegator.Delegation.Shares.MulInt(validatorInfo.Tokens)).QuoTruncate(validatorInfo.DelegatorShares)
+		if token.LT(tokenIn20Usd) {
+			continue
+		}
+		eveAirdrop := (eveAirdrop.MulInt64(int64(config.GetAkashConfig().Percent))).QuoInt64(100).Mul(token).QuoTruncate(totalTokenDelegate)
+		eveBech32Address, err := utils.ConvertBech32Address(delegator.Delegation.DelegatorAddress)
+		if err != nil {
+			log.Println("Failed to convert Akash bech32 address: %w", err)
+			return nil, nil, 0, fmt.Errorf("failed to convert Bech32Address: %w", err)
+		}
+		rewardInfo = append(rewardInfo, config.Reward{
+			Address:         delegator.Delegation.DelegatorAddress,
+			EveAddress:      eveBech32Address,
+			Shares:          delegator.Delegation.Shares,
+			Token:           token,
+			EveAirdropToken: eveAirdrop,
+			ChainID:         config.GetAkashConfig().ChainID,
+		})
+		testAmount = eveAirdrop.Add(testAmount)
+		balanceInfo = append(balanceInfo, banktypes.Balance{
+			Address: eveBech32Address,
+			Coins:   sdk.NewCoins(sdk.NewCoin("eve", eveAirdrop.TruncateInt())),
+		})
+	}
+	log.Println("Akash balance: ", testAmount)
+	// Write delegations to file
+	// fileForDebug, _ := json.MarshalIndent(rewardInfo, "", " ")
+	// _ = os.WriteFile("rewards.json", fileForDebug, 0644)
+
+	// fileBalance, _ := json.MarshalIndent(balanceInfo, "", " ")
+	// _ = os.WriteFile("balance.json", fileBalance, 0644)
+	return balanceInfo, rewardInfo, len(balanceInfo), nil
+}

airdrop/chains/bostrom.go

@@ -0,0 +1,104 @@
+package chains
+
+import (
+	"fmt"
+	"log"
+	"strconv"
+
+	"github.com/eve-network/eve/airdrop/config"
+	"github.com/eve-network/eve/airdrop/utils"
+
+	sdkmath "cosmossdk.io/math"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
+)
+
+func Bostrom() ([]banktypes.Balance, []config.Reward, int, error) {
+	delegators := []stakingtypes.DelegationResponse{}
+
+	rpc := config.GetBostromConfig().API + "/cosmos/staking/v1beta1/validators?pagination.limit=" + strconv.Itoa(config.LimitPerPage) + "&pagination.count_total=true"
+	validatorsResponse, err := utils.FetchValidators(rpc)
+	if err != nil {
+		log.Println("Failed to fetch Bostrom validators: %w", err)
+		return nil, nil, 0, fmt.Errorf("failed to fetch validators for Bostrom: %w", err)
+	}
+	validators := validatorsResponse.Validators
+	log.Println("Validators: ", len(validators))
+	for validatorIndex, validator := range validators {
+		url := config.GetBostromConfig().API + "/cosmos/staking/v1beta1/validators/" + validator.OperatorAddress + "/delegations?pagination.limit=" + strconv.Itoa(config.LimitPerPage) + "&pagination.count_total=true"
+		delegations, total, err := utils.FetchDelegations(url)
+		if err != nil {
+			log.Println("Failed to fetch delegations for Bostrom: %w", err)
+			return nil, nil, 0, fmt.Errorf("failed to fetch delegations for Bostrom: %w", err)
+		}
+		log.Println(validator.OperatorAddress)
+		log.Println("Response ", len(delegations))
+		log.Println("Bostrom validator "+strconv.Itoa(validatorIndex)+" ", total)
+		delegators = append(delegators, delegations...)
+	}
+
+	usd := sdkmath.LegacyMustNewDecFromStr("20")
+
+	apiURL := config.APICoingecko + config.GetBostromConfig().CoinID + "&vs_currencies=usd"
+	tokenInUsd, err := utils.FetchTokenPrice(apiURL, config.GetBostromConfig().CoinID)
+	if err != nil {
+		log.Println("Failed to fetch Bostrom token price: %w", err)
+		return nil, nil, 0, fmt.Errorf("failed to fetch Bostrom token price: %w", err)
+	}
+	tokenIn20Usd := usd.Quo(tokenInUsd)
+
+	rewardInfo := []config.Reward{}
+	balanceInfo := []banktypes.Balance{}
+
+	totalTokenDelegate := sdkmath.LegacyMustNewDecFromStr("0")
+	for _, delegator := range delegators {
+		validatorIndex := utils.FindValidatorInfoCustomType(validators, delegator.Delegation.ValidatorAddress)
+		validatorInfo := validators[validatorIndex]
+		token := (delegator.Delegation.Shares.MulInt(validatorInfo.Tokens)).QuoTruncate(validatorInfo.DelegatorShares)
+		totalTokenDelegate = totalTokenDelegate.Add(token)
+	}
+	eveAirdrop, err := sdkmath.LegacyNewDecFromStr(config.EveAirdrop)
+	if err != nil {
+		log.Println("Failed to convert EveAirdrop string to dec: %w", err)
+		return nil, nil, 0, fmt.Errorf("failed to convert EveAirdrop string to dec: %w", err)
+	}
+	testAmount := sdkmath.LegacyMustNewDecFromStr("0")
+
+	for _, delegator := range delegators {
+		validatorIndex := utils.FindValidatorInfoCustomType(validators, delegator.Delegation.ValidatorAddress)
+		validatorInfo := validators[validatorIndex]
+		token := (delegator.Delegation.Shares.MulInt(validatorInfo.Tokens)).QuoTruncate(validatorInfo.DelegatorShares)
+		if token.LT(tokenIn20Usd) {
+			continue
+		}
+		eveAirdrop := (eveAirdrop.MulInt64(int64(config.GetBostromConfig().Percent))).QuoInt64(100).Mul(token).QuoTruncate(totalTokenDelegate)
+		eveBech32Address, err := utils.ConvertBech32Address(delegator.Delegation.DelegatorAddress)
+		if err != nil {
+			log.Println("Failed to convert Bostrom bech32 address: %w", err)
+			return nil, nil, 0, fmt.Errorf("failed to convert Bech32Address: %w", err)
+		}
+		rewardInfo = append(rewardInfo, config.Reward{
+			Address:         delegator.Delegation.DelegatorAddress,
+			EveAddress:      eveBech32Address,
+			Shares:          delegator.Delegation.Shares,
+			Token:           token,
+			EveAirdropToken: eveAirdrop,
+			ChainID:         config.GetBostromConfig().ChainID,
+		})
+		testAmount = eveAirdrop.Add(testAmount)
+		balanceInfo = append(balanceInfo, banktypes.Balance{
+			Address: eveBech32Address,
+			Coins:   sdk.NewCoins(sdk.NewCoin("eve", eveAirdrop.TruncateInt())),
+		})
+	}
+	log.Println("Bostrom balance: ", testAmount)
+	// Write delegations to file
+	// fileForDebug, _ := json.MarshalIndent(rewardInfo, "", " ")
+	// _ = os.WriteFile("rewards.json", fileForDebug, 0644)
+
+	// fileBalance, _ := json.MarshalIndent(balanceInfo, "", " ")
+	// _ = os.WriteFile("balance.json", fileBalance, 0644)
+	return balanceInfo, rewardInfo, len(balanceInfo), nil
+}

airdrop/chains/celestia.go

@@ -0,0 +1,125 @@
+package chains
+
+// error max size response
+import (
+	"fmt"
+	"log"
+	"strconv"
+
+	"github.com/eve-network/eve/airdrop/config"
+	"github.com/eve-network/eve/airdrop/utils"
+	"github.com/joho/godotenv"
+
+	sdkmath "cosmossdk.io/math"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
+)
+
+func Celestia() ([]banktypes.Balance, []config.Reward, int, error) {
+	err := godotenv.Load()
+	if err != nil {
+		log.Printf("Error loading Celestial environment variables: %v", err)
+		return nil, nil, 0, fmt.Errorf("failed to load env: %w", err)
+	}
+
+	blockHeight, err := utils.GetLatestHeight(config.GetCelestiaConfig().RPC + "/status")
+	if err != nil {
+		log.Printf("Failed to get latest height for Celestial: %v", err)
+		return nil, nil, 0, fmt.Errorf("failed to get latest height for Celestia: %w", err)
+	}
+
+	grpcAddr := config.GetCelestiaConfig().GRPCAddr
+	grpcConn, err := utils.SetupGRPCConnection(grpcAddr)
+	if err != nil {
+		log.Printf("Failed to connect to gRPC Celestial: %v", err)
+		return nil, nil, 0, fmt.Errorf("failed to connect to gRPC Celestia: %w", err)
+	}
+	defer grpcConn.Close()
+	stakingClient := stakingtypes.NewQueryClient(grpcConn)
+
+	delegators := []stakingtypes.DelegationResponse{}
+
+	validators, err := utils.GetValidators(stakingClient, blockHeight)
+	if err != nil {
+		log.Printf("Failed to get Celestial validators: %v", err)
+		return nil, nil, 0, fmt.Errorf("failed to get Celestia validators: %w", err)
+	}
+
+	log.Println("Validators: ", len(validators))
+	for validatorIndex, validator := range validators {
+		url := config.GetCelestiaConfig().API + "/cosmos/staking/v1beta1/validators/" + validator.OperatorAddress + "/delegations?pagination.limit=" + strconv.Itoa(config.LimitPerPage) + "&pagination.count_total=true"
+		delegations, total, err := utils.FetchDelegations(url)
+		if err != nil {
+			log.Printf("Failed to query delegate info for Celestial validator: %v", err)
+			return nil, nil, 0, fmt.Errorf("failed to fetch delegations for Celestia: %w", err)
+		}
+		log.Println(validator.OperatorAddress)
+		log.Println("Response ", len(delegations))
+		log.Println("Celestia validator "+strconv.Itoa(validatorIndex)+" ", total)
+		delegators = append(delegators, delegations...)
+	}
+
+	usd := sdkmath.LegacyMustNewDecFromStr("20")
+
+	apiURL := config.APICoingecko + config.GetCelestiaConfig().CoinID + "&vs_currencies=usd"
+	tokenInUsd, err := utils.FetchTokenPrice(apiURL, config.GetCelestiaConfig().CoinID)
+	if err != nil {
+		log.Println("Failed to fetch Celestial token price: %w", err)
+		return nil, nil, 0, fmt.Errorf("failed to fetch Celestia token price: %w", err)
+	}
+	tokenIn20Usd := usd.Quo(tokenInUsd)
+
+	rewardInfo := []config.Reward{}
+	balanceInfo := []banktypes.Balance{}
+
+	totalTokenDelegate := sdkmath.LegacyMustNewDecFromStr("0")
+	for _, delegator := range delegators {
+		validatorIndex := utils.FindValidatorInfo(validators, delegator.Delegation.ValidatorAddress)
+		validatorInfo := validators[validatorIndex]
+		token := (delegator.Delegation.Shares.MulInt(validatorInfo.Tokens)).QuoTruncate(validatorInfo.DelegatorShares)
+		totalTokenDelegate = totalTokenDelegate.Add(token)
+	}
+	eveAirdrop, err := sdkmath.LegacyNewDecFromStr(config.EveAirdrop)
+	if err != nil {
+		log.Println("Failed to convert EveAirdrop string to dec: %w", err)
+		return nil, nil, 0, fmt.Errorf("failed to convert EveAirdrop string to dec: %w", err)
+	}
+	testAmount := sdkmath.LegacyMustNewDecFromStr("0")
+	for _, delegator := range delegators {
+		validatorIndex := utils.FindValidatorInfo(validators, delegator.Delegation.ValidatorAddress)
+		validatorInfo := validators[validatorIndex]
+		token := (delegator.Delegation.Shares.MulInt(validatorInfo.Tokens)).QuoTruncate(validatorInfo.DelegatorShares)
+		if token.LT(tokenIn20Usd) {
+			continue
+		}
+		eveAirdrop := (eveAirdrop.MulInt64(int64(config.GetCelestiaConfig().Percent))).QuoInt64(100).Mul(token).QuoTruncate(totalTokenDelegate)
+		eveBech32Address, err := utils.ConvertBech32Address(delegator.Delegation.DelegatorAddress)
+		if err != nil {
+			log.Println("Failed to convert Celestial bech32 address: %w", err)
+			return nil, nil, 0, fmt.Errorf("failed to convert Bech32Address: %w", err)
+		}
+		rewardInfo = append(rewardInfo, config.Reward{
+			Address:         delegator.Delegation.DelegatorAddress,
+			EveAddress:      eveBech32Address,
+			Shares:          delegator.Delegation.Shares,
+			Token:           token,
+			EveAirdropToken: eveAirdrop,
+			ChainID:         config.GetCelestiaConfig().ChainID,
+		})
+		testAmount = eveAirdrop.Add(testAmount)
+		balanceInfo = append(balanceInfo, banktypes.Balance{
+			Address: eveBech32Address,
+			Coins:   sdk.NewCoins(sdk.NewCoin("eve", eveAirdrop.TruncateInt())),
+		})
+	}
+	log.Println("Celestia balance: ", testAmount)
+	// Write delegations to file
+	// fileForDebug, _ := json.MarshalIndent(rewardInfo, "", " ")
+	// _ = os.WriteFile("rewards.json", fileForDebug, 0644)
+
+	// fileBalance, _ := json.MarshalIndent(balanceInfo, "", " ")
+	// _ = os.WriteFile("balance.json", fileBalance, 0644)
+	return balanceInfo, rewardInfo, len(balanceInfo), nil
+}