package cli

import (
	"fmt"

	"github.com/pkg/errors"
	"github.com/spf13/cobra"

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

	"github.com/cosmos/ethermint/x/evm/types"
)

// GetQueryCmd defines evm module queries through the cli
func GetQueryCmd(moduleName string, cdc *codec.Codec) *cobra.Command {
	evmQueryCmd := &cobra.Command{
		Use:                        types.ModuleName,
		Short:                      "Querying commands for the evm module",
		DisableFlagParsing:         true,
		SuggestionsMinimumDistance: 2,
		RunE:                       client.ValidateCmd,
	}
	evmQueryCmd.AddCommand(flags.GetCommands(
		GetCmdGetStorageAt(moduleName, cdc),
		GetCmdGetCode(moduleName, cdc),
	)...)
	return evmQueryCmd
}

// GetCmdGetStorageAt queries a key in an accounts storage
func GetCmdGetStorageAt(queryRoute string, cdc *codec.Codec) *cobra.Command {
	return &cobra.Command{
		Use:   "storage [account] [key]",
		Short: "Gets storage for an account at a given key",
		Args:  cobra.ExactArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			clientCtx := context.NewCLIContext().WithCodec(cdc)

			account, err := accountToHex(args[0])
			if err != nil {
				return errors.Wrap(err, "could not parse account address")
			}

			key := formatKeyToHash(args[1])

			res, _, err := clientCtx.Query(
				fmt.Sprintf("custom/%s/storage/%s/%s", queryRoute, account, key))

			if err != nil {
				return fmt.Errorf("could not resolve: %s", err)
			}
			var out types.QueryResStorage
			cdc.MustUnmarshalJSON(res, &out)
			return clientCtx.PrintOutput(out)
		},
	}
}

// GetCmdGetCode queries the code field of a given address
func GetCmdGetCode(queryRoute string, cdc *codec.Codec) *cobra.Command {
	return &cobra.Command{
		Use:   "code [account]",
		Short: "Gets code from an account",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			clientCtx := context.NewCLIContext().WithCodec(cdc)

			account, err := accountToHex(args[0])
			if err != nil {
				return errors.Wrap(err, "could not parse account address")
			}

			res, _, err := clientCtx.Query(
				fmt.Sprintf("custom/%s/code/%s", queryRoute, account))

			if err != nil {
				return fmt.Errorf("could not resolve: %s", err)
			}

			var out types.QueryResCode
			cdc.MustUnmarshalJSON(res, &out)
			return clientCtx.PrintOutput(out)
		},
	}
}