hide passwd when entered

master v0.0.2
兔子 4 years ago
parent 195b967331
commit b3ad857712

68
io.go

@ -6,6 +6,8 @@ import (
"os"
"strconv"
"strings"
"golang.org/x/crypto/ssh/terminal"
)
type InputMsg struct {
@ -13,6 +15,41 @@ type InputMsg struct {
err error
}
func Passwd(hint string, defaultVal string) InputMsg {
return passwd(hint, defaultVal, "")
}
func PasswdWithMask(hint string, defaultVal string, mask string) InputMsg {
return passwd(hint, defaultVal, mask)
}
func passwd(hint string, defaultVal string, mask string) InputMsg {
var ioBuf []byte
if hint != "" {
fmt.Print(hint)
}
state, err := terminal.MakeRaw(int(os.Stdin.Fd()))
if err != nil {
return InputMsg{"", err}
}
defer terminal.Restore(0, state)
inputReader := bufio.NewReader(os.Stdin)
for {
b, err := inputReader.ReadByte()
if err != nil {
return InputMsg{"", err}
}
if b == 0x0d {
fmt.Println()
return InputMsg{strings.TrimSpace(string(ioBuf)), nil}
}
if mask != "" {
fmt.Print(mask)
}
ioBuf = append(ioBuf, b)
}
}
func MessageBox(hint string, defaultVal string) InputMsg {
if hint != "" {
fmt.Print(hint)
@ -125,27 +162,44 @@ func YesNo(hint string, defaults bool) bool {
}
}
func StopUntil(hint string, trigger string) error {
func StopUntil(hint string, trigger string, repeat bool) error {
pressLen := len(trigger)
if trigger == "" {
pressLen = 1
} else {
pressLen = len(trigger)
}
ioBuf := make([]byte, pressLen)
for {
if hint != "" {
fmt.Print(hint)
state, err := terminal.MakeRaw(int(os.Stdin.Fd()))
if err != nil {
return err
}
defer terminal.Restore(0, state)
inputReader := bufio.NewReader(os.Stdin)
_, err := inputReader.Read(ioBuf)
//ioBuf := make([]byte, pressLen)
if hint != "" && !repeat {
fmt.Print(hint)
}
i := 0
for {
b, err := inputReader.ReadByte()
if err != nil {
return err
}
if string(ioBuf) == trigger || trigger == "" {
if trigger == "" {
break
}
if b == trigger[i] {
i++
if i == pressLen {
break
}
continue
}
i = 0
if hint != "" && repeat {
fmt.Print("\n")
fmt.Print(hint)
}
}
return nil
}

Loading…
Cancel
Save