From a745850f9d73a90f95ce0d9a5c06edff260a3b84 Mon Sep 17 00:00:00 2001
From: gopherjk121 <strdev123strdev@gmail.com>
Date: Mon, 30 Jan 2023 22:39:46 +0800
Subject: [PATCH] batch send command for crew3

---
 x/airdrop/client/cli/batch_send.go | 99 ++++++++++++++++++++++++++++++
 x/airdrop/client/cli/tx.go         |  1 +
 2 files changed, 100 insertions(+)
 create mode 100644 x/airdrop/client/cli/batch_send.go

diff --git a/x/airdrop/client/cli/batch_send.go b/x/airdrop/client/cli/batch_send.go
new file mode 100644
index 0000000..dcf955a
--- /dev/null
+++ b/x/airdrop/client/cli/batch_send.go
@@ -0,0 +1,99 @@
+package cli
+
+import (
+	"encoding/csv"
+	"fmt"
+	"os"
+	"strconv"
+
+	"github.com/cosmos/cosmos-sdk/client"
+	"github.com/cosmos/cosmos-sdk/client/flags"
+	"github.com/cosmos/cosmos-sdk/client/tx"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+	"github.com/spf13/cobra"
+)
+
+func parseCSV(path string) [][]string {
+	f, err := os.Open(path)
+	if err != nil {
+		panic(err)
+	}
+	defer f.Close()
+
+	csvReader := csv.NewReader(f)
+	records, err := csvReader.ReadAll()
+	if err != nil {
+		panic(err)
+	}
+
+	return records
+}
+
+// NewBatchSendCmd returns a CLI command for multi-send
+func NewBatchSendCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "batch-send [csv_file] [denom] [startIndex] [threshold]",
+		Short:   "Execute batch-send based on csv file",
+		Example: `teritorid tx airdrop batch-send "./airdrop.csv" utori 0 100 --from=ACCOUNT --keyring-backend=test --gas=10000000 -y --fees=100000utori`,
+		Args:    cobra.ExactArgs(4),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			clientCtx, err := client.GetClientTxContext(cmd)
+			if err != nil {
+				return err
+			}
+
+			sendMsgs := []banktypes.MsgSend{}
+			amountRecords := parseCSV(args[0])
+
+			for _, line := range amountRecords[1:] {
+				addr, amountStr := line[0], line[1]
+				amountDec := sdk.MustNewDecFromStr(amountStr)
+				decimal := sdk.NewInt(1000_000) // 10^6
+				amount := amountDec.Mul(decimal.ToDec()).TruncateInt()
+
+				msg := banktypes.MsgSend{
+					FromAddress: clientCtx.FromAddress.String(),
+					ToAddress:   addr,
+					Amount:      sdk.Coins{sdk.NewCoin(args[1], amount)},
+				}
+				sendMsgs = append(sendMsgs, msg)
+			}
+
+			startIndex, err := strconv.Atoi(args[2])
+			if err != nil {
+				return err
+			}
+			threshold, err := strconv.Atoi(args[3])
+			if err != nil {
+				return err
+			}
+
+			msgs := []sdk.Msg{}
+			for index, msg := range sendMsgs {
+				if index < startIndex {
+					continue
+				}
+				msgs = append(msgs, &banktypes.MsgSend{
+					FromAddress: msg.FromAddress,
+					ToAddress:   msg.ToAddress,
+					Amount:      msg.Amount,
+				})
+				if len(msgs) >= threshold || index+1 == len(sendMsgs) {
+					err := tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msgs...)
+					if err != nil {
+						return err
+					}
+					fmt.Println("executed batch ", index)
+					msgs = []sdk.Msg{}
+				}
+			}
+			fmt.Println("finalized batch execution")
+
+			return nil
+		},
+	}
+
+	flags.AddTxFlagsToCmd(cmd)
+	return cmd
+}
diff --git a/x/airdrop/client/cli/tx.go b/x/airdrop/client/cli/tx.go
index cc7b57f..be2d1f1 100644
--- a/x/airdrop/client/cli/tx.go
+++ b/x/airdrop/client/cli/tx.go
@@ -28,6 +28,7 @@ func GetTxCmd() *cobra.Command {
 		AllocateFurtherAirdropCmd(),
 		FetchAndRemoveAirdropCmd(),
 		AllocateStarsAirdropCmd(),
+		NewBatchSendCmd(),
 	)
 
 	return txCmd