github.com/KiraCore/sekai@v0.3.43/x/basket/client/cli/query.go

package cli

import (
	"context"
	"strconv"
	"strings"

	"github.com/spf13/cobra"

	"github.com/cosmos/cosmos-sdk/client"
	"github.com/cosmos/cosmos-sdk/client/flags"

	"github.com/KiraCore/sekai/x/basket/types"
)

// NewQueryCmd returns a root CLI command handler for all x/basket transaction commands.
func NewQueryCmd() *cobra.Command {
	queryCmd := &cobra.Command{
		Use:   types.RouterKey,
		Short: "query commands for the basket module",
	}
	queryCmd.AddCommand(
		GetCmdQueryTokenBasketById(),
		GetCmdQueryTokenBasketByDenom(),
		GetCmdQueryTokenBaskets(),
		GetCmdQueryBaksetHistoricalMints(),
		GetCmdQueryBaksetHistoricalBurns(),
		GetCmdQueryBaksetHistoricalSwaps(),
	)

	return queryCmd
}

func GetCmdQueryTokenBasketById() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "token-basket-by-id [id]",
		Short: "Queries a single basket by id",
		Args:  cobra.MinimumNArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			clientCtx := client.GetClientContextFromCmd(cmd)

			id, err := strconv.Atoi(args[0])
			if err != nil {
				return err
			}
			queryClient := types.NewQueryClient(clientCtx)
			res, err := queryClient.TokenBasketById(context.Background(), &types.QueryTokenBasketByIdRequest{
				Id: uint64(id),
			})
			if err != nil {
				return err
			}

			return clientCtx.PrintProto(res)
		},
	}

	flags.AddQueryFlagsToCmd(cmd)

	return cmd
}

func GetCmdQueryTokenBasketByDenom() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "token-basket-by-denom [denom]",
		Short: "Queries a single basket by denom",
		Args:  cobra.MinimumNArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			clientCtx := client.GetClientContextFromCmd(cmd)

			queryClient := types.NewQueryClient(clientCtx)
			res, err := queryClient.TokenBasketByDenom(context.Background(), &types.QueryTokenBasketByDenomRequest{
				Denom: args[0],
			})
			if err != nil {
				return err
			}

			return clientCtx.PrintProto(res)
		},
	}

	flags.AddQueryFlagsToCmd(cmd)

	return cmd
}

func GetCmdQueryTokenBaskets() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "token-baskets [tokens] [derivatives_only]",
		Short: "Queries token baskets by filter",
		Args:  cobra.MinimumNArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			clientCtx := client.GetClientContextFromCmd(cmd)

			tokens := []string{}
			if args[0] != "" {
				tokens = strings.Split(args[0], ",")
			}
			derivativesOnly, err := strconv.ParseBool(args[1])
			if err != nil {
				return err
			}

			queryClient := types.NewQueryClient(clientCtx)
			res, err := queryClient.TokenBaskets(context.Background(), &types.QueryTokenBasketsRequest{
				Tokens:          tokens,
				DerivativesOnly: derivativesOnly,
			})
			if err != nil {
				return err
			}

			return clientCtx.PrintProto(res)
		},
	}

	flags.AddQueryFlagsToCmd(cmd)

	return cmd
}

func GetCmdQueryBaksetHistoricalMints() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "historical-mints [basket_id]",
		Short: "Queries basket historical mints on limited period",
		Args:  cobra.MinimumNArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			clientCtx := client.GetClientContextFromCmd(cmd)

			basketId, err := strconv.Atoi(args[0])
			if err != nil {
				return err
			}

			queryClient := types.NewQueryClient(clientCtx)
			res, err := queryClient.BaksetHistoricalMints(context.Background(), &types.QueryBasketHistoricalMintsRequest{
				BasketId: uint64(basketId),
			})
			if err != nil {
				return err
			}

			return clientCtx.PrintProto(res)
		},
	}

	flags.AddQueryFlagsToCmd(cmd)

	return cmd
}

func GetCmdQueryBaksetHistoricalBurns() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "historical-burns [basket_id]",
		Short: "Queries basket historical burns on limited period",
		Args:  cobra.MinimumNArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			clientCtx := client.GetClientContextFromCmd(cmd)

			basketId, err := strconv.Atoi(args[0])
			if err != nil {
				return err
			}

			queryClient := types.NewQueryClient(clientCtx)
			res, err := queryClient.BaksetHistoricalBurns(context.Background(), &types.QueryBasketHistoricalBurnsRequest{
				BasketId: uint64(basketId),
			})
			if err != nil {
				return err
			}

			return clientCtx.PrintProto(res)
		},
	}

	flags.AddQueryFlagsToCmd(cmd)

	return cmd
}

func GetCmdQueryBaksetHistoricalSwaps() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "historical-swaps [basket_id]",
		Short: "Queries basket historical swaps on limited period",
		Args:  cobra.MinimumNArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			clientCtx := client.GetClientContextFromCmd(cmd)

			basketId, err := strconv.Atoi(args[0])
			if err != nil {
				return err
			}

			queryClient := types.NewQueryClient(clientCtx)
			res, err := queryClient.BaksetHistoricalSwaps(context.Background(), &types.QueryBasketHistoricalSwapsRequest{
				BasketId: uint64(basketId),
			})
			if err != nil {
				return err
			}

			return clientCtx.PrintProto(res)
		},
	}

	flags.AddQueryFlagsToCmd(cmd)

	return cmd
}