hide passwd when entered

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

72
io.go

@ -6,6 +6,8 @@ import (
"os" "os"
"strconv" "strconv"
"strings" "strings"
"golang.org/x/crypto/ssh/terminal"
) )
type InputMsg struct { type InputMsg struct {
@ -13,6 +15,41 @@ type InputMsg struct {
err error 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 { func MessageBox(hint string, defaultVal string) InputMsg {
if hint != "" { if hint != "" {
fmt.Print(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) pressLen := len(trigger)
if trigger == "" { if trigger == "" {
pressLen = 1 pressLen = 1
} else { } else {
pressLen = len(trigger) pressLen = len(trigger)
} }
ioBuf := make([]byte, pressLen) state, err := terminal.MakeRaw(int(os.Stdin.Fd()))
if err != nil {
return err
}
defer terminal.Restore(0, state)
inputReader := bufio.NewReader(os.Stdin)
//ioBuf := make([]byte, pressLen)
if hint != "" && !repeat {
fmt.Print(hint)
}
i := 0
for { for {
if hint != "" { b, err := inputReader.ReadByte()
fmt.Print(hint)
}
inputReader := bufio.NewReader(os.Stdin)
_, err := inputReader.Read(ioBuf)
if err != nil { if err != nil {
return err return err
} }
if string(ioBuf) == trigger || trigger == "" { if trigger == "" {
break break
} }
fmt.Print("\n") if b == trigger[i] {
i++
if i == pressLen {
break
}
continue
}
i = 0
if hint != "" && repeat {
fmt.Print("\n")
fmt.Print(hint)
}
} }
return nil return nil
} }

Loading…
Cancel
Save