2020-01-05 04:32:09 +00:00
|
|
|
package cmd
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bufio"
|
|
|
|
"fmt"
|
|
|
|
"os"
|
|
|
|
"strings"
|
|
|
|
|
2020-05-06 23:50:19 +00:00
|
|
|
"github.com/pkg/errors"
|
2020-01-05 04:32:09 +00:00
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
)
|
|
|
|
|
|
|
|
var log = logrus.WithField("prefix", "node")
|
|
|
|
|
|
|
|
// ConfirmAction uses the passed in actionText as the confirmation text displayed in the terminal.
|
|
|
|
// The user must enter Y or N to indicate whether they confirm the action detailed in the warning text.
|
|
|
|
// Returns a boolean representing the user's answer.
|
|
|
|
func ConfirmAction(actionText string, deniedText string) (bool, error) {
|
|
|
|
var confirmed bool
|
|
|
|
reader := bufio.NewReader(os.Stdin)
|
|
|
|
log.Warn(actionText)
|
|
|
|
|
|
|
|
for {
|
|
|
|
fmt.Print(">> ")
|
|
|
|
|
|
|
|
line, _, err := reader.ReadLine()
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
trimmedLine := strings.TrimSpace(string(line))
|
|
|
|
lineInput := strings.ToUpper(trimmedLine)
|
|
|
|
if lineInput != "Y" && lineInput != "N" {
|
|
|
|
log.Errorf("Invalid option of %s chosen, please only enter Y/N", line)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
if lineInput == "Y" {
|
|
|
|
confirmed = true
|
|
|
|
break
|
|
|
|
}
|
|
|
|
log.Warn(deniedText)
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
return confirmed, nil
|
|
|
|
}
|
2020-05-05 20:13:18 +00:00
|
|
|
|
|
|
|
// EnterPassword queries the user for their password through the terminal, in order to make sure it is
|
|
|
|
// not passed in a visible way to the terminal.
|
2020-05-12 15:01:02 +00:00
|
|
|
func EnterPassword(confirmPassword bool, pr PasswordReader) (string, error) {
|
2020-05-05 20:13:18 +00:00
|
|
|
var passphrase string
|
|
|
|
log.Info("Enter a password:")
|
2020-05-12 15:01:02 +00:00
|
|
|
bytePassword, err := pr.ReadPassword()
|
2020-05-05 20:13:18 +00:00
|
|
|
if err != nil {
|
2020-05-06 23:50:19 +00:00
|
|
|
return "", errors.Wrap(err, "could not read account password")
|
2020-05-05 20:13:18 +00:00
|
|
|
}
|
|
|
|
text := string(bytePassword)
|
|
|
|
passphrase = strings.Replace(text, "\n", "", -1)
|
2020-05-06 23:50:19 +00:00
|
|
|
if confirmPassword {
|
|
|
|
log.Info("Please re-enter your password:")
|
2020-05-12 15:01:02 +00:00
|
|
|
bytePassword, err := pr.ReadPassword()
|
2020-05-06 23:50:19 +00:00
|
|
|
if err != nil {
|
|
|
|
return "", errors.Wrap(err, "could not read account password")
|
|
|
|
}
|
|
|
|
text := string(bytePassword)
|
|
|
|
confirmedPass := strings.Replace(text, "\n", "", -1)
|
|
|
|
if passphrase != confirmedPass {
|
|
|
|
log.Info("Passwords did not match, please try again")
|
2020-05-12 15:01:02 +00:00
|
|
|
return EnterPassword(true, pr)
|
2020-05-06 23:50:19 +00:00
|
|
|
}
|
|
|
|
}
|
2020-05-05 20:13:18 +00:00
|
|
|
return passphrase, nil
|
|
|
|
}
|