github.com/fibonacci-chain/fbc@v0.0.0-20231124064014-c7636198c1e9/x/ammswap/client/cli/tx.go

package cli

import (
	"bufio"
	"fmt"
	"strings"
	"time"

	"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/client/context"
	client "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/client/flags"
	"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/codec"
	sdk "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types"
	"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/x/auth"
	"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/x/auth/client/utils"
	"github.com/fibonacci-chain/fbc/x/ammswap/types"
	"github.com/spf13/cobra"
)

// flags
const (
	flagMinLiquidity     = "min-liquidity"
	flagMaxBaseAmount    = "max-base-amount"
	flagQuoteAmount      = "quote-amount"
	flagDeadlineDuration = "deadline-duration"
	flagLiquidity        = "liquidity"
	flagMinBaseAmount    = "min-base-amount"
	flagMinQuoteAmount   = "min-quote-amount"
	flagSellAmount       = "sell-amount"
	flagMinBuyAmount     = "min-buy-amount"
	flagRecipient        = "recipient"
	flagToken0           = "token0"
	flagToken1           = "token1"
)

// GetTxCmd returns the transaction commands for this module
func GetTxCmd(cdc *codec.Codec) *cobra.Command {
	txCmd := &cobra.Command{
		Use:   "swap",
		Short: "Swap transactions subcommands",
	}

	txCmd.AddCommand(client.PostCommands(
		getCmdAddLiquidity(cdc),
		getCmdRemoveLiquidity(cdc),
		getCmdCreateExchange(cdc),
		getCmdTokenSwap(cdc),
	)...)

	return txCmd
}

func getCmdAddLiquidity(cdc *codec.Codec) *cobra.Command {
	// flags
	var minLiquidity string
	var maxBaseAmount string
	var quoteAmount string
	var deadlineDuration string
	cmd := &cobra.Command{
		Use:   "add-liquidity",
		Short: "add liquidity",
		Long: strings.TrimSpace(
			fmt.Sprintf(`add liquidity.

Example:
$ fbchaincli tx swap add-liquidity --max-base-amount 10eth-355 --quote-amount 100btc-366 --min-liquidity 0.001

`),
		),
		RunE: func(cmd *cobra.Command, args []string) error {
			cliCtx := context.NewCLIContext().WithCodec(cdc)
			inBuf := bufio.NewReader(cmd.InOrStdin())
			txBldr := auth.NewTxBuilderFromCLI(inBuf).WithTxEncoder(utils.GetTxEncoder(cdc))

			minLiquidityDec, sdkErr := sdk.NewDecFromStr(minLiquidity)
			if sdkErr != nil {
				return sdkErr
			}
			maxBaseAmountDecCoin, err := sdk.ParseDecCoin(maxBaseAmount)
			if err != nil {
				return err
			}
			quoteAmountDecCoin, err := sdk.ParseDecCoin(quoteAmount)
			if err != nil {
				return err
			}
			duration, err := time.ParseDuration(deadlineDuration)
			if err != nil {
				return err
			}
			deadline := time.Now().Add(duration).Unix()
			msg := types.NewMsgAddLiquidity(minLiquidityDec, maxBaseAmountDecCoin, quoteAmountDecCoin, deadline, cliCtx.FromAddress)

			return utils.CompleteAndBroadcastTxCLI(txBldr, cliCtx, []sdk.Msg{msg})
		},
	}

	cmd.Flags().StringVarP(&minLiquidity, flagMinLiquidity, "l", "", "Minimum number of sender will mint if total pool token supply is greater than 0")
	cmd.Flags().StringVarP(&maxBaseAmount, flagMaxBaseAmount, "", "", "Maximum number of base amount deposited. Deposits max amount if total pool token supply is 0. For example \"100xxb\"")
	cmd.Flags().StringVarP(&quoteAmount, flagQuoteAmount, "q", "", "The number of quote amount. For example \"100okb\"")
	cmd.Flags().StringVarP(&deadlineDuration, flagDeadlineDuration, "d", "30s", "Duration after which this transaction can no longer be executed. such as \"300ms\", \"1.5h\" or \"2h45m\". Valid time units are \"ns\", \"us\" (or \"µs\"), \"ms\", \"s\", \"m\", \"h\".")
	cmd.MarkFlagRequired(flagMinLiquidity)
	cmd.MarkFlagRequired(flagMaxBaseAmount)
	cmd.MarkFlagRequired(flagQuoteAmount)
	return cmd
}

func getCmdRemoveLiquidity(cdc *codec.Codec) *cobra.Command {
	// flags
	var liquidity string
	var minBaseAmount string
	var minQuoteAmount string
	var deadlineDuration string
	cmd := &cobra.Command{
		Use:   "remove-liquidity",
		Short: "remove liquidity",
		Long: strings.TrimSpace(
			fmt.Sprintf(`remove liquidity.

Example:
$ fbchaincli tx swap remove-liquidity --liquidity 1 --min-base-amount 10eth-355 --min-quote-amount 1btc-366

`),
		),
		RunE: func(cmd *cobra.Command, args []string) error {
			cliCtx := context.NewCLIContext().WithCodec(cdc)
			inBuf := bufio.NewReader(cmd.InOrStdin())
			txBldr := auth.NewTxBuilderFromCLI(inBuf).WithTxEncoder(utils.GetTxEncoder(cdc))

			liquidityDec, sdkErr := sdk.NewDecFromStr(liquidity)
			if sdkErr != nil {
				return sdkErr
			}
			minBaseAmountDecCoin, err := sdk.ParseDecCoin(minBaseAmount)
			if err != nil {
				return err
			}
			minQuoteAmountDecCoin, err := sdk.ParseDecCoin(minQuoteAmount)
			if err != nil {
				return err
			}
			duration, err := time.ParseDuration(deadlineDuration)
			if err != nil {
				return err
			}
			deadline := time.Now().Add(duration).Unix()
			msg := types.NewMsgRemoveLiquidity(liquidityDec, minBaseAmountDecCoin, minQuoteAmountDecCoin, deadline, cliCtx.FromAddress)

			return utils.CompleteAndBroadcastTxCLI(txBldr, cliCtx, []sdk.Msg{msg})
		},
	}

	cmd.Flags().StringVarP(&liquidity, flagLiquidity, "l", "", "Liquidity amount of sender will burn")
	cmd.Flags().StringVarP(&minBaseAmount, flagMinBaseAmount, "", "", "Minimum number of base amount withdrawn")
	cmd.Flags().StringVarP(&minQuoteAmount, flagMinQuoteAmount, "q", "", "Minimum number of quote amount withdrawn")
	cmd.Flags().StringVarP(&deadlineDuration, flagDeadlineDuration, "d", "30s", "Duration after which this transaction can no longer be executed. such as \"300ms\", \"1.5h\" or \"2h45m\". Valid time units are \"ns\", \"us\" (or \"µs\"), \"ms\", \"s\", \"m\", \"h\".")
	cmd.MarkFlagRequired(flagLiquidity)
	cmd.MarkFlagRequired(flagMinBaseAmount)
	cmd.MarkFlagRequired(flagMinQuoteAmount)
	return cmd
}

func getCmdCreateExchange(cdc *codec.Codec) *cobra.Command {
	// flags
	var token0 string
	var token1 string
	cmd := &cobra.Command{
		Use:   "create-pair",
		Short: "create token pair",
		Long: strings.TrimSpace(
			fmt.Sprintf(`create token pair.

Example:
$ fbchaincli tx swap create-pair --token0 eth-355 --token1 btc-366 --fees 0.01fibo 

`),
		),

		RunE: func(cmd *cobra.Command, args []string) error {
			cliCtx := context.NewCLIContext().WithCodec(cdc)
			inBuf := bufio.NewReader(cmd.InOrStdin())
			txBldr := auth.NewTxBuilderFromCLI(inBuf).WithTxEncoder(utils.GetTxEncoder(cdc))
			msg := types.NewMsgCreateExchange(token0, token1, cliCtx.FromAddress)

			return utils.CompleteAndBroadcastTxCLI(txBldr, cliCtx, []sdk.Msg{msg})
		},
	}

	cmd.Flags().StringVar(&token0, flagToken0, "", "the base token name is required to create an AMM swap pair")
	cmd.Flags().StringVar(&token1, flagToken1, "", "the quote token name is required to create an AMM swap pair")
	cmd.MarkFlagRequired(flagToken0)
	cmd.MarkFlagRequired(flagToken1)
	return cmd
}

func getCmdTokenSwap(cdc *codec.Codec) *cobra.Command {
	// flags
	var soldTokenAmount string
	var minBoughtTokenAmount string
	var deadline string
	var recipient string
	cmd := &cobra.Command{
		Use:   "token",
		Short: "swap token",
		Long: strings.TrimSpace(
			fmt.Sprintf(`swap token.

Example:
$ fbchaincli tx swap token --sell-amount 1eth-355 --min-buy-amount 60btc-366

`),
		),
		RunE: func(cmd *cobra.Command, args []string) error {
			cliCtx := context.NewCLIContext().WithCodec(cdc)
			inBuf := bufio.NewReader(cmd.InOrStdin())
			txBldr := auth.NewTxBuilderFromCLI(inBuf).WithTxEncoder(utils.GetTxEncoder(cdc))

			soldTokenAmount, err := sdk.ParseDecCoin(soldTokenAmount)
			if err != nil {
				return err
			}
			minBoughtTokenAmount, err := sdk.ParseDecCoin(minBoughtTokenAmount)
			if err != nil {
				return err
			}
			dur, err := time.ParseDuration(deadline)
			if err != nil {
				return err
			}
			deadline := time.Now().Add(dur).Unix()
			var recip sdk.AccAddress
			if recipient == "" {
				recip = cliCtx.FromAddress
			} else {
				recip, err = sdk.AccAddressFromBech32(recipient)
				if err != nil {
					return err
				}
			}

			msg := types.NewMsgTokenToToken(soldTokenAmount, minBoughtTokenAmount,
				deadline, recip, cliCtx.FromAddress)

			return utils.CompleteAndBroadcastTxCLI(txBldr, cliCtx, []sdk.Msg{msg})
		},
	}

	cmd.Flags().StringVarP(&soldTokenAmount, flagSellAmount, "", "",
		"Amount expected to sell")
	cmd.Flags().StringVarP(&minBoughtTokenAmount, flagMinBuyAmount, "", "",
		"Minimum amount expected to buy")
	cmd.Flags().StringVarP(&recipient, flagRecipient, "", "",
		"The address to receive the amount bought")
	cmd.Flags().StringVarP(&deadline, flagDeadlineDuration, "", "100s",
		"Duration after which this transaction can no longer be executed. such as \"300ms\", \"1.5h\" or \"2h45m\". Valid time units are \"ns\", \"us\" (or \"µs\"), \"ms\", \"s\", \"m\", \"h\".")
	cmd.MarkFlagRequired(flagSellAmount)
	cmd.MarkFlagRequired(flagMinBuyAmount)

	return cmd
}