Skip to content

Commit

Permalink
refactor: inline flags.GetKeyringDir and remove it
Browse files Browse the repository at this point in the history
  • Loading branch information
michaelfig committed Oct 7, 2020
1 parent e756cab commit bc15c59
Show file tree
Hide file tree
Showing 11 changed files with 43 additions and 25 deletions.
5 changes: 4 additions & 1 deletion client/cmd.go
Original file line number Diff line number Diff line change
Expand Up @@ -103,8 +103,11 @@ func ReadPersistentCommandFlags(clientCtx Context, flagSet *pflag.FlagSet) (Cont
}

if clientCtx.Keyring == nil || flagSet.Changed(flags.FlagKeyringBackend) {
keyringDir, _ := flagSet.GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir, _ = flagSet.GetString(flags.FlagHome)
}
keyringBackend, _ := flagSet.GetString(flags.FlagKeyringBackend)
keyringDir, _ := flags.GetKeyringDir(flagSet, clientCtx.HomeDir)

if keyringBackend != "" {
kr, err := newKeyringFromFlags(clientCtx, keyringBackend, keyringDir)
Expand Down
15 changes: 0 additions & 15 deletions client/flags/flags.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@ import (
"strconv"

"github.com/spf13/cobra"
"github.com/spf13/pflag"
tmcli "github.com/tendermint/tendermint/libs/cli"

"github.com/cosmos/cosmos-sdk/crypto/keyring"
Expand Down Expand Up @@ -114,20 +113,6 @@ func AddTxFlagsToCmd(cmd *cobra.Command) {
cmd.SetOut(cmd.OutOrStdout())
}

// GetKeyringDir finds the directory for the keyring storage
func GetKeyringDir(pflags *pflag.FlagSet, homeDir string) (string, error) {
keyringDir, err := pflags.GetString(FlagKeyringDir)
if keyringDir == "" {
// Use the specified home directory.
keyringDir = homeDir
}
if keyringDir == "" {
// Default to the node's home directory.
keyringDir, err = pflags.GetString(FlagHome)
}
return keyringDir, err
}

// AddPaginationFlagsToCmd adds common pagination flags to cmd
func AddPaginationFlagsToCmd(cmd *cobra.Command, query string) {
cmd.Flags().Uint64(FlagPage, 1, fmt.Sprintf("pagination page of %s to query for. This sets offset to a multiple of limit", query))
Expand Down
5 changes: 4 additions & 1 deletion client/keys/add.go
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,10 @@ the flag --nosort is set.
func runAddCmd(cmd *cobra.Command, args []string) error {
buf := bufio.NewReader(cmd.InOrStdin())

keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), "")
keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir, _ = cmd.Flags().GetString(flags.FlagHome)
}
dryRun, _ := cmd.Flags().GetBool(flags.FlagDryRun)

var (
Expand Down
5 changes: 4 additions & 1 deletion client/keys/delete.go
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,11 @@ private keys stored in a ledger device cannot be deleted with the CLI.
RunE: func(cmd *cobra.Command, args []string) error {
buf := bufio.NewReader(cmd.InOrStdin())

keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir, _ = cmd.Flags().GetString(flags.FlagHome)
}
backend, _ := cmd.Flags().GetString(flags.FlagKeyringBackend)
keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), "")
kb, err := keyring.New(sdk.KeyringServiceName(), backend, keyringDir, buf)
if err != nil {
return err
Expand Down
5 changes: 4 additions & 1 deletion client/keys/export.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,11 @@ func ExportKeyCommand() *cobra.Command {
RunE: func(cmd *cobra.Command, args []string) error {
buf := bufio.NewReader(cmd.InOrStdin())

keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir, _ = cmd.Flags().GetString(flags.FlagHome)
}
backend, _ := cmd.Flags().GetString(flags.FlagKeyringBackend)
keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), "")
kb, err := keyring.New(sdk.KeyringServiceName(), backend, keyringDir, buf)
if err != nil {
return err
Expand Down
6 changes: 5 additions & 1 deletion client/keys/import.go
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,12 @@ func ImportKeyCommand() *cobra.Command {
RunE: func(cmd *cobra.Command, args []string) error {
buf := bufio.NewReader(cmd.InOrStdin())

keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir, _ = cmd.Flags().GetString(flags.FlagHome)
}
backend, _ := cmd.Flags().GetString(flags.FlagKeyringBackend)
keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), "")

kb, err := keyring.New(sdk.KeyringServiceName(), backend, keyringDir, buf)
if err != nil {
return err
Expand Down
6 changes: 5 additions & 1 deletion client/keys/list.go
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,12 @@ along with their associated name and address.`,
}

func runListCmd(cmd *cobra.Command, _ []string) error {
keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir, _ = cmd.Flags().GetString(flags.FlagHome)
}
backend, _ := cmd.Flags().GetString(flags.FlagKeyringBackend)
keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), "")

kb, err := keyring.New(sdk.KeyringServiceName(), backend, keyringDir, cmd.InOrStdin())
if err != nil {
return err
Expand Down
5 changes: 4 additions & 1 deletion client/keys/migrate.go
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,10 @@ It is recommended to run in 'dry-run' mode first to verify all key migration mat
}

func runMigrateCmd(cmd *cobra.Command, args []string) error {
keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), "")
keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir, _ = cmd.Flags().GetString(flags.FlagHome)
}

// instantiate legacy keybase
var legacyKb keyring.LegacyKeybase
Expand Down
5 changes: 4 additions & 1 deletion client/keys/show.go
Original file line number Diff line number Diff line change
Expand Up @@ -54,8 +54,11 @@ consisting of all the keys provided by name and multisig threshold.`,
func runShowCmd(cmd *cobra.Command, args []string) (err error) {
var info keyring.Info

keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir, _ = cmd.Flags().GetString(flags.FlagHome)
}
backend, _ := cmd.Flags().GetString(flags.FlagKeyringBackend)
keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), "")
kb, err := keyring.New(sdk.KeyringServiceName(), backend, keyringDir, cmd.InOrStdin())
if err != nil {
return err
Expand Down
6 changes: 5 additions & 1 deletion simapp/simd/cmd/genaccounts.go
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,11 @@ contain valid denominations. Accounts may optionally be supplied with vesting pa
addr, err := sdk.AccAddressFromBech32(args[0])
if err != nil {
inBuf := bufio.NewReader(cmd.InOrStdin())
keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), clientCtx.HomeDir)

keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir = clientCtx.HomeDir
}
keyringBackend, _ := cmd.Flags().GetString(flags.FlagKeyringBackend)

// attempt to lookup address from Keybase if no address was provided
Expand Down
5 changes: 4 additions & 1 deletion x/auth/client/cli/tx_multisign.go
Original file line number Diff line number Diff line change
Expand Up @@ -82,8 +82,11 @@ func makeMultiSignCmd() func(cmd *cobra.Command, args []string) error {
return err
}

keyringDir, _ := cmd.Flags().GetString(flags.FlagKeyringDir)
if keyringDir == "" {
keyringDir = clientCtx.HomeDir
}
backend, _ := cmd.Flags().GetString(flags.FlagKeyringBackend)
keyringDir, _ := flags.GetKeyringDir(cmd.Flags(), clientCtx.HomeDir)

inBuf := bufio.NewReader(cmd.InOrStdin())
kb, err := keyring.New(sdk.KeyringServiceName(), backend, keyringDir, inBuf)
Expand Down

0 comments on commit bc15c59

Please sign in to comment.