laconicd/keys/show.go

package keys

import (
	"errors"
	"fmt"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"

	"github.com/tendermint/tendermint/libs/cli"

	"github.com/cosmos/cosmos-sdk/client/flags"
	"github.com/cosmos/cosmos-sdk/crypto"
	cosmosKeys "github.com/cosmos/cosmos-sdk/crypto/keys"
	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/cosmos/ethermint/crypto/keys"
)

const (
	// FlagAddress is the flag for the user's address on the command line.
	FlagAddress = "address"
	// FlagAddress is the flag for the user's address on the command line.
	FlagETHAddress = "ethwallet"
	// FlagPublicKey represents the user's public key on the command line.
	FlagPublicKey = "pubkey"
	// FlagBechPrefix defines a desired Bech32 prefix encoding for a key.
	FlagBechPrefix = "bech"
	// FlagDevice indicates that the information should be shown in the device
	FlagDevice = "device"
)

func showKeysCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "show <name>",
		Short: "Show key info for the given name",
		Long:  `Return public details of a single local key.`,
		Args:  cobra.MinimumNArgs(1),
		RunE:  runShowCmd,
	}

	cmd.Flags().String(FlagBechPrefix, sdk.PrefixAccount, "The Bech32 prefix encoding for a key (acc|val|cons)")
	cmd.Flags().BoolP(FlagAddress, "a", false, "Output the address only (overrides --output)")
	cmd.Flags().BoolP(FlagETHAddress, "w", false, "Output the Ethereum address only (overrides --output)")
	cmd.Flags().BoolP(FlagPublicKey, "p", false, "Output the public key only (overrides --output)")
	cmd.Flags().BoolP(FlagDevice, "d", false, "Output the address in a ledger device")
	cmd.Flags().Bool(flags.FlagIndentResponse, false, "Add indent to JSON response")

	return cmd
}

func runShowCmd(cmd *cobra.Command, args []string) (err error) {
	var info cosmosKeys.Info

	if len(args) == 1 {
		info, err = GetKeyInfo(args[0])
		if err != nil {
			return err
		}
	} else {
		return errors.New("Must provide only one name")
	}

	isShowAddr := viper.GetBool(FlagAddress)
	isShowEthAddr := viper.GetBool(FlagETHAddress)
	isShowPubKey := viper.GetBool(FlagPublicKey)
	isShowDevice := viper.GetBool(FlagDevice)

	isOutputSet := false
	tmp := cmd.Flag(cli.OutputFlag)
	if tmp != nil {
		isOutputSet = tmp.Changed
	}

	isShowEitherAddr := isShowAddr || isShowEthAddr

	if isShowEitherAddr && isShowPubKey {
		return errors.New("cannot get address, with --address or --ethwallet, and --pubkey at once")
	}

	if isOutputSet && (isShowEitherAddr || isShowPubKey) {
		return errors.New("cannot use --output with --address or --pubkey")
	}

	keyOutputFunction := keys.Bech32KeyOutput

	switch {
	case isShowAddr:
		printKeyAddress(info, keyOutputFunction)
	case isShowEthAddr:
		printKeyEthAddress(info, keyOutputFunction)
	case isShowPubKey:
		printPubKey(info, keyOutputFunction)
	default:
		printKeyInfo(info, keyOutputFunction)
	}

	if isShowDevice {
		if isShowPubKey {
			return fmt.Errorf("the device flag (-d) can only be used for addresses not pubkeys")
		}
		if viper.GetString(FlagBechPrefix) != "acc" {
			return fmt.Errorf("the device flag (-d) can only be used for accounts")
		}
		// Override and show in the device
		if info.GetType() != cosmosKeys.TypeLedger {
			return fmt.Errorf("the device flag (-d) can only be used for accounts stored in devices")
		}

		hdpath, err := info.GetPath()
		if err != nil {
			return nil
		}

		return crypto.LedgerShowAddress(*hdpath, info.GetPubKey())
	}

	return nil
}

// TODO: Determine if the different prefixes are necessary for ethermint
// func getBechKeyOut(bechPrefix string) (bechKeyOutFn, error) {
// 	switch bechPrefix {
// 	case sdk.PrefixAccount:
// 		return keys.Bech32KeyOutput, nil
// 	case sdk.PrefixValidator:
// 		return keys.Bech32ValKeyOutput, nil
// 	case sdk.PrefixConsensus:
// 		return keys.Bech32ConsKeyOutput, nil
// 	}

// 	return nil, fmt.Errorf("invalid Bech32 prefix encoding provided: %s", bechPrefix)
// }