Add airdrop tool

.gitignore

@@ -46,3 +46,9 @@ dependency-graph.png
 *.out
 *.synctex.gz
 contract_tests/*
+
+#Airdrop tool
+balance.json
+rewards.json
+balancenft.json
+.env

airdrop/akash.go

@@ -0,0 +1,145 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"strconv"
+
+	"github.com/eve-network/eve/airdrop/config"
+	"github.com/joho/godotenv"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/metadata"
+
+	"cosmossdk.io/math"
+
+	"github.com/cosmos/cosmos-sdk/codec"
+	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) {
+	block_height := getLatestHeight(config.GetAkashConfig().RPC + "/status")
+	err := godotenv.Load()
+	if err != nil {
+		fmt.Println("Error loading env:", err)
+		panic("")
+	}
+	grpcAddr := config.GetAkashConfig().GRPCAddr
+	grpcConn, err := grpc.Dial(grpcAddr, grpc.WithDefaultCallOptions(grpc.ForceCodec(codec.NewProtoCodec(nil).GRPCCodec())))
+	if err != nil {
+		panic(err)
+	}
+	defer grpcConn.Close()
+	stakingClient := stakingtypes.NewQueryClient(grpcConn)
+
+	delegators := []stakingtypes.DelegationResponse{}
+
+	validators := getValidators(stakingClient, block_height)
+	fmt.Println("Validators: ", len(validators))
+	for validatorIndex, validator := range validators {
+		var header metadata.MD
+		delegationsResponse, _ := stakingClient.ValidatorDelegations(
+			metadata.AppendToOutgoingContext(context.Background(), grpctypes.GRPCBlockHeightHeader, block_height), // Add metadata to request
+			&stakingtypes.QueryValidatorDelegationsRequest{
+				ValidatorAddr: validator.OperatorAddress,
+				Pagination: &query.PageRequest{
+					CountTotal: true,
+					Limit:      LIMIT_PER_PAGE,
+				},
+			},
+			grpc.Header(&header), // Retrieve header from response
+		)
+		total := delegationsResponse.Pagination.Total
+		fmt.Println("Response ", len(delegationsResponse.DelegationResponses))
+		fmt.Println("Akash validator "+strconv.Itoa(validatorIndex)+" ", total)
+		delegators = append(delegators, delegationsResponse.DelegationResponses...)
+	}
+
+	usd := math.LegacyMustNewDecFromStr("20")
+
+	apiUrl := API_COINGECKO + config.GetAkashConfig().CoinId + "&vs_currencies=usd"
+	tokenInUsd := fetchAkashTokenPrice(apiUrl)
+	tokenIn20Usd := usd.QuoTruncate(tokenInUsd)
+
+	rewardInfo := []config.Reward{}
+	balanceInfo := []banktypes.Balance{}
+
+	totalTokenDelegate := math.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 := math.LegacyMustNewDecFromStr(EVE_AIRDROP)
+	testAmount, _ := math.LegacyNewDecFromStr("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
+		}
+		eveAirdrop := (eveAirdrop.MulInt64(int64(config.GetAkashConfig().Percent))).QuoInt64(100).Mul(token).QuoTruncate(totalTokenDelegate)
+		eveBech32Address := convertBech32Address(delegator.Delegation.DelegatorAddress)
+		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
+}
+
+func fetchAkashTokenPrice(apiUrl string) math.LegacyDec {
+	// Make a GET request to the API
+	response, err := http.Get(apiUrl) //nolint
+	if err != nil {
+		fmt.Println("Error making GET request:", err)
+		panic("")
+	}
+	defer response.Body.Close()
+
+	// Read the response body
+	responseBody, err := io.ReadAll(response.Body)
+	if err != nil {
+		fmt.Println("Error reading response body:", err)
+		panic("")
+	}
+
+	var data config.AkashPrice
+
+	// Unmarshal the JSON byte slice into the defined struct
+	err = json.Unmarshal(responseBody, &data)
+	if err != nil {
+		fmt.Println("Error unmarshalling JSON:", err)
+		panic("")
+	}
+
+	tokenInUsd := math.LegacyMustNewDecFromStr(data.Token.USD.String())
+	return tokenInUsd
+}

airdrop/bostrom.go

@@ -0,0 +1,122 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"strconv"
+	"strings"
+
+	"github.com/eve-network/eve/airdrop/config"
+
+	"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) {
+	delegators := []stakingtypes.DelegationResponse{}
+
+	rpc := config.GetBostromConfig().API + "/cosmos/staking/v1beta1/validators?pagination.limit=" + strconv.Itoa(LIMIT_PER_PAGE) + "&pagination.count_total=true"
+	validatorsResponse := fetchValidators(rpc)
+	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(LIMIT_PER_PAGE) + "&pagination.count_total=true"
+		delegations, total := fetchDelegations(url)
+		fmt.Println(validator.OperatorAddress)
+		fmt.Println("Response ", len(delegations))
+		fmt.Println("Bostrom validator "+strconv.Itoa(validatorIndex)+" ", total)
+		delegators = append(delegators, delegations...)
+	}
+
+	usd := math.LegacyMustNewDecFromStr("20")
+
+	apiUrl := API_COINGECKO + config.GetBostromConfig().CoinId + "&vs_currencies=usd"
+	tokenInUsd := fetchBostromTokenPrice(apiUrl)
+	tokenIn20Usd := usd.QuoTruncate(tokenInUsd)
+
+	rewardInfo := []config.Reward{}
+	balanceInfo := []banktypes.Balance{}
+
+	totalTokenDelegate := math.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 := math.LegacyMustNewDecFromStr(EVE_AIRDROP)
+	testAmount, _ := math.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 := convertBech32Address(delegator.Delegation.DelegatorAddress)
+		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
+}
+
+func fetchBostromTokenPrice(apiUrl string) math.LegacyDec {
+	// Make a GET request to the API
+	response, err := http.Get(apiUrl) //nolint
+	if err != nil {
+		fmt.Println("Error making GET request:", err)
+		panic("")
+	}
+	defer response.Body.Close()
+
+	// Read the response body
+	responseBody, err := io.ReadAll(response.Body)
+	if err != nil {
+		fmt.Println("Error reading response body:", err)
+		panic("")
+	}
+
+	var data config.BostromPrice
+
+	// Unmarshal the JSON byte slice into the defined struct
+	err = json.Unmarshal(responseBody, &data)
+	if err != nil {
+		fmt.Println("Error unmarshalling JSON:", err)
+		panic("")
+	}
+	rawPrice := strings.Split(data.Token.USD.String(), "e-")
+	base := rawPrice[0]
+	power := rawPrice[1]
+	powerInt, _ := strconv.ParseUint(power, 10, 64)
+	baseDec, _ := math.LegacyNewDecFromStr(base)
+	tenDec, _ := math.LegacyNewDecFromStr("10")
+	tokenInUsd := baseDec.Quo(tenDec.Power(powerInt))
+	return tokenInUsd
+}