TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-11-26 09:42:22 +00:00
Code Issues Projects Releases Packages Wiki Activity

cli: unify password-related prompts

Browse source 
Close #2419.
This commit is contained in:
Anna Shaleva 2022-04-29 18:22:33 +03:00
parent 3ae1647940
commit 19e4c679d3
2 changed files with 20 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cli/wallet/validator.go
View file

@ -204,7 +204,7 @@ func getDecryptedAccount(ctx *cli.Context, wall *wallet.Wallet, addr util.Uint16
return nil, fmt.Errorf("can't find account for the address: %s", address.Uint160ToString(addr)) return nil, fmt.Errorf("can't find account for the address: %s", address.Uint160ToString(addr))
} }
if pass, err := input.ReadPassword("Password > "); err != nil { if pass, err := input.ReadPassword(EnterPasswordPrompt); err != nil {
fmt.Println("Error reading password", err) fmt.Println("Error reading password", err)
return nil, err return nil, err
} else if err := acc.Decrypt(pass, wall.Scrypt); err != nil { } else if err := acc.Decrypt(pass, wall.Scrypt); err != nil {

26
cli/wallet/wallet.go
View file

@ -22,6 +22,18 @@ import (
"github.com/urfave/cli" "github.com/urfave/cli"
) )
const (
// EnterPasswordPrompt is a prompt used to ask the user for a password.
EnterPasswordPrompt = "Enter password > "
// EnterNewPasswordPrompt is a prompt used to ask the user for a password on
// account creation.
EnterNewPasswordPrompt = "Enter new password > "
// EnterOldPasswordPrompt is a prompt used to ask the user for an old password.
EnterOldPasswordPrompt = "Enter old password > "
// ConfirmPasswordPrompt is a prompt used to confirm the password.
ConfirmPasswordPrompt = "Confirm password > "
)
var ( var (
errNoPath = errors.New("wallet path is mandatory and should be passed using (--wallet, -w) flags") errNoPath = errors.New("wallet path is mandatory and should be passed using (--wallet, -w) flags")
errPhraseMismatch = errors.New("the entered pass-phrases do not match. Maybe you have misspelled them") errPhraseMismatch = errors.New("the entered pass-phrases do not match. Maybe you have misspelled them")
@ -315,7 +327,7 @@ func changePassword(ctx *cli.Context) error {
} }
} }
oldPass, err := input.ReadPassword("Enter password > ") oldPass, err := input.ReadPassword(EnterOldPasswordPrompt)
if err != nil { if err != nil {
return cli.NewExitError(fmt.Errorf("Error reading old password: %w", err), 1) return cli.NewExitError(fmt.Errorf("Error reading old password: %w", err), 1)
} }
@ -368,7 +380,7 @@ func convertWallet(ctx *cli.Context) error {
newWallet.Scrypt = wall.Scrypt newWallet.Scrypt = wall.Scrypt
for _, acc := range wall.Accounts { for _, acc := range wall.Accounts {
pass, err := input.ReadPassword(fmt.Sprintf("Enter passphrase for account %s (label '%s') > ", acc.Address, acc.Label)) pass, err := input.ReadPassword(fmt.Sprintf("Enter password for account %s (label '%s') > ", acc.Address, acc.Label))
if err != nil { if err != nil {
return cli.NewExitError(fmt.Errorf("Error reading password: %w", err), 1) return cli.NewExitError(fmt.Errorf("Error reading password: %w", err), 1)
} }
@ -438,7 +450,7 @@ loop:
for _, wif := range wifs { for _, wif := range wifs {
if decrypt { if decrypt {
pass, err := input.ReadPassword("Enter password > ") pass, err := input.ReadPassword(EnterPasswordPrompt)
if err != nil { if err != nil {
return cli.NewExitError(fmt.Errorf("Error reading password: %w", err), 1) return cli.NewExitError(fmt.Errorf("Error reading password: %w", err), 1)
} }
@ -634,7 +646,7 @@ func dumpWallet(ctx *cli.Context) error {
return cli.NewExitError(err, 1) return cli.NewExitError(err, 1)
} }
if ctx.Bool("decrypt") { if ctx.Bool("decrypt") {
pass, err := input.ReadPassword("Enter wallet password > ") pass, err := input.ReadPassword(EnterPasswordPrompt)
if err != nil { if err != nil {
return cli.NewExitError(fmt.Errorf("Error reading password: %w", err), 1) return cli.NewExitError(fmt.Errorf("Error reading password: %w", err), 1)
} }
@ -734,11 +746,11 @@ func readAccountInfo() (string, string, error) {
} }
func readNewPassword() (string, error) { func readNewPassword() (string, error) {
phrase, err := input.ReadPassword("Enter passphrase > ") phrase, err := input.ReadPassword(EnterNewPasswordPrompt)
if err != nil { if err != nil {
return "", fmt.Errorf("Error reading password: %w", err) return "", fmt.Errorf("Error reading password: %w", err)
} }
phraseCheck, err := input.ReadPassword("Confirm passphrase > ") phraseCheck, err := input.ReadPassword(ConfirmPasswordPrompt)
if err != nil { if err != nil {
return "", fmt.Errorf("Error reading password: %w", err) return "", fmt.Errorf("Error reading password: %w", err)
} }
@ -785,7 +797,7 @@ func newAccountFromWIF(w io.Writer, wif string, scrypt keys.ScryptParams) (*wall
// note: NEP2 strings always have length of 58 even though // note: NEP2 strings always have length of 58 even though
// base58 strings can have different lengths even if slice lengths are equal // base58 strings can have different lengths even if slice lengths are equal
if len(wif) == 58 { if len(wif) == 58 {
pass, err := input.ReadPassword("Enter password > ") pass, err := input.ReadPassword(EnterPasswordPrompt)
if err != nil { if err != nil {
return nil, fmt.Errorf("Error reading password: %w", err) return nil, fmt.Errorf("Error reading password: %w", err)
} }