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/akash.go

@@ -0,0 +1,161 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+	"strconv"
+
+	"github.com/eve-network/eve/airdrop/config"
+	"github.com/joho/godotenv"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/metadata"
+
+	sdkmath "cosmossdk.io/math"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	grpctypes "github.com/cosmos/cosmos-sdk/types/grpc"
+	"github.com/cosmos/cosmos-sdk/types/query"
+	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 {
+		return nil, nil, 0, fmt.Errorf("failed to load env: %w", err)
+	}
+
+	blockHeight, err := getLatestHeightWithRetry(config.GetAkashConfig().RPC + "/status")
+	if err != nil {
+		return nil, nil, 0, fmt.Errorf("failed to get latest height for Akash: %w", err)
+	}
+
+	grpcAddr := config.GetAkashConfig().GRPCAddr
+	grpcConn, err := grpc.Dial(grpcAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		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 := getValidators(stakingClient, blockHeight)
+	if err != nil {
+		return nil, nil, 0, fmt.Errorf("failed to get Akash validators: %w", err)
+	}
+	fmt.Println("Validators: ", len(validators))
+	for validatorIndex, validator := range validators {
+		var header metadata.MD
+		delegationsResponse, err := stakingClient.ValidatorDelegations(
+			metadata.AppendToOutgoingContext(context.Background(), grpctypes.GRPCBlockHeightHeader, blockHeight), // Add metadata to request
+			&stakingtypes.QueryValidatorDelegationsRequest{
+				ValidatorAddr: validator.OperatorAddress,
+				Pagination: &query.PageRequest{
+					CountTotal: true,
+					Limit:      LimitPerPage,
+				},
+			},
+			grpc.Header(&header), // Retrieve header from response
+		)
+		if err != nil {
+			return nil, nil, 0, fmt.Errorf("failed to query delegate info for Akash validator: %w", err)
+		}
+		total := delegationsResponse.Pagination.Total
+		fmt.Println("Response ", len(delegationsResponse.DelegationResponses))
+		fmt.Println("Akash validator "+strconv.Itoa(validatorIndex)+" ", total)
+		delegators = append(delegators, delegationsResponse.DelegationResponses...)
+	}
+
+	usd := sdkmath.LegacyMustNewDecFromStr("20")
+
+	apiURL := APICoingecko + config.GetAkashConfig().CoinID + "&vs_currencies=usd"
+	fetchTokenPrice := fetchTokenPriceWithRetry(fetchAkashTokenPrice)
+	tokenInUsd, err := fetchTokenPrice(apiURL)
+	if err != nil {
+		return nil, nil, 0, fmt.Errorf("failed to fetch Akash token price: %w", err)
+	}
+	tokenIn20Usd := usd.QuoTruncate(tokenInUsd)
+
+	rewardInfo := []config.Reward{}
+	balanceInfo := []banktypes.Balance{}
+
+	totalTokenDelegate := sdkmath.LegacyMustNewDecFromStr("0")
+	for _, delegator := range delegators {
+		validatorIndex := findValidatorInfo(validators, delegator.Delegation.ValidatorAddress)
+		validatorInfo := validators[validatorIndex]
+		token := (delegator.Delegation.Shares.MulInt(validatorInfo.Tokens)).QuoTruncate(validatorInfo.DelegatorShares)
+		if token.LT(tokenIn20Usd) {
+			continue
+		}
+		totalTokenDelegate = totalTokenDelegate.Add(token)
+	}
+	eveAirdrop := sdkmath.LegacyMustNewDecFromStr(EveAirdrop)
+	testAmount, err := sdkmath.LegacyNewDecFromStr("0")
+	if err != nil {
+		return nil, nil, 0, fmt.Errorf("failed to convert string to dec: %w", err)
+	}
+	for _, delegator := range delegators {
+		validatorIndex := 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 := convertBech32Address(delegator.Delegation.DelegatorAddress)
+		if err != nil {
+			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())),
+		})
+	}
+	fmt.Println("Akash ", 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
+}
+
+func fetchAkashTokenPrice(apiURL string) (sdkmath.LegacyDec, error) {
+	// Make a GET request to the API
+	response, err := makeGetRequest(apiURL)
+	if err != nil {
+		return sdkmath.LegacyDec{}, fmt.Errorf("error making GET request to fetch Akash token price: %w", err)
+	}
+	defer response.Body.Close()
+
+	// Read the response body
+	responseBody, err := io.ReadAll(response.Body)
+	if err != nil {
+		return sdkmath.LegacyDec{}, fmt.Errorf("error reading response body for Akash token price: %w", err)
+	}
+
+	var data config.AkashPrice
+
+	// Unmarshal the JSON byte slice into the defined struct
+	err = json.Unmarshal(responseBody, &data)
+	if err != nil {
+		return sdkmath.LegacyDec{}, fmt.Errorf("error unmarshalling JSON for Akash token price: %w", err)
+	}
+
+	tokenInUsd := sdkmath.LegacyMustNewDecFromStr(data.Token.USD.String())
+	return tokenInUsd, nil
+}

airdrop/bostrom.go

@@ -0,0 +1,131 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"strconv"
+	"strings"
+
+	"github.com/eve-network/eve/airdrop/config"
+
+	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(LimitPerPage) + "&pagination.count_total=true"
+	validatorsResponse, err := fetchValidators(rpc)
+	if err != nil {
+		return nil, nil, 0, fmt.Errorf("failed to fetch validators for Bostrom: %w", err)
+	}
+	validators := validatorsResponse.Validators
+	fmt.Println("Validators: ", len(validators))
+	for validatorIndex, validator := range validators {
+		url := config.GetBostromConfig().API + "/cosmos/staking/v1beta1/validators/" + validator.OperatorAddress + "/delegations?pagination.limit=" + strconv.Itoa(LimitPerPage) + "&pagination.count_total=true"
+		delegations, total, err := fetchDelegationsWithRetry(url)
+		if err != nil {
+			return nil, nil, 0, fmt.Errorf("failed to fetch delegations for Bostrom: %w", err)
+		}
+		fmt.Println(validator.OperatorAddress)
+		fmt.Println("Response ", len(delegations))
+		fmt.Println("Bostrom validator "+strconv.Itoa(validatorIndex)+" ", total)
+		delegators = append(delegators, delegations...)
+	}
+
+	usd := sdkmath.LegacyMustNewDecFromStr("20")
+
+	apiURL := APICoingecko + config.GetBostromConfig().CoinID + "&vs_currencies=usd"
+	fetchTokenPrice := fetchTokenPriceWithRetry(fetchBostromTokenPrice)
+	tokenInUsd, err := fetchTokenPrice(apiURL)
+	if err != nil {
+		return nil, nil, 0, fmt.Errorf("failed to fetch Bostrom token price: %w", err)
+	}
+	tokenIn20Usd := usd.QuoTruncate(tokenInUsd)
+
+	rewardInfo := []config.Reward{}
+	balanceInfo := []banktypes.Balance{}
+
+	totalTokenDelegate := sdkmath.LegacyMustNewDecFromStr("0")
+	for _, delegator := range delegators {
+		validatorIndex := findValidatorInfoCustomType(validators, delegator.Delegation.ValidatorAddress)
+		validatorInfo := validators[validatorIndex]
+		token := (delegator.Delegation.Shares.MulInt(validatorInfo.Tokens)).QuoTruncate(validatorInfo.DelegatorShares)
+		if token.LT(tokenIn20Usd) {
+			continue
+		}
+		totalTokenDelegate = totalTokenDelegate.Add(token)
+	}
+	eveAirdrop := sdkmath.LegacyMustNewDecFromStr(EveAirdrop)
+	testAmount, _ := sdkmath.LegacyNewDecFromStr("0")
+	for _, delegator := range delegators {
+		validatorIndex := 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 := convertBech32Address(delegator.Delegation.DelegatorAddress)
+		if err != nil {
+			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())),
+		})
+	}
+	fmt.Println("Bostrom ", 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
+}
+
+func fetchBostromTokenPrice(apiURL string) (sdkmath.LegacyDec, error) {
+	// Make a GET request to the API
+	response, err := makeGetRequest(apiURL)
+	if err != nil {
+		return sdkmath.LegacyDec{}, fmt.Errorf("error making GET request to fetch Bostrom token price: %w", err)
+	}
+	defer response.Body.Close()
+
+	// Read the response body
+	responseBody, err := io.ReadAll(response.Body)
+	if err != nil {
+		return sdkmath.LegacyDec{}, fmt.Errorf("error reading response body for Bostrom token price: %w", err)
+	}
+
+	var data config.BostromPrice
+
+	// Unmarshal the JSON byte slice into the defined struct
+	err = json.Unmarshal(responseBody, &data)
+	if err != nil {
+		return sdkmath.LegacyDec{}, fmt.Errorf("error unmarshalling JSON for Bostrom token price: %w", err)
+	}
+	rawPrice := strings.Split(data.Token.USD.String(), "e-")
+	base := rawPrice[0]
+	power := rawPrice[1]
+	powerInt, _ := strconv.ParseUint(power, 10, 64)
+	baseDec, _ := sdkmath.LegacyNewDecFromStr(base)
+	tenDec, _ := sdkmath.LegacyNewDecFromStr("10")
+	tokenInUsd := baseDec.Quo(tenDec.Power(powerInt))
+	return tokenInUsd, nil
+}