liuxiaolong
2022-06-28 37714b1093c04061e636e5b1d27179652e671c0a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package speakeasy
 
import (
    "fmt"
    "io"
    "os"
    "strings"
)
 
// Ask the user to enter a password with input hidden. prompt is a string to
// display before the user's input. Returns the provided password, or an error
// if the command failed.
func Ask(prompt string) (password string, err error) {
    return FAsk(os.Stdout, prompt)
}
 
// FAsk is the same as Ask, except it is possible to specify the file to write
// the prompt to. If 'nil' is passed as the writer, no prompt will be written.
func FAsk(wr io.Writer, prompt string) (password string, err error) {
    if wr != nil && prompt != "" {
        fmt.Fprint(wr, prompt) // Display the prompt.
    }
    password, err = getPassword()
 
    // Carriage return after the user input.
    if wr != nil {
        fmt.Fprintln(wr, "")
    }
    return
}
 
func readline() (value string, err error) {
    var valb []byte
    var n int
    b := make([]byte, 1)
    for {
        // read one byte at a time so we don't accidentally read extra bytes
        n, err = os.Stdin.Read(b)
        if err != nil && err != io.EOF {
            return "", err
        }
        if n == 0 || b[0] == '\n' {
            break
        }
        valb = append(valb, b[0])
    }
 
    return strings.TrimSuffix(string(valb), "\r"), nil
}