50 lines
824 B
Go
50 lines
824 B
Go
package util
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
type Prompter struct {
|
|
scanner *bufio.Scanner
|
|
}
|
|
|
|
func NewPrompter(scanner *bufio.Scanner) *Prompter {
|
|
return &Prompter{scanner}
|
|
}
|
|
|
|
func (p Prompter) Prompt(prompt string) (string, error) {
|
|
printPrompt(prompt)
|
|
|
|
if p.scanner.Scan() {
|
|
return strings.TrimSpace(p.scanner.Text()), nil
|
|
}
|
|
|
|
if err := p.scanner.Err(); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return "", fmt.Errorf("%s must not be empty", prompt)
|
|
}
|
|
|
|
func PromptForPassword(prompt string) (string, error) {
|
|
printPrompt(prompt)
|
|
|
|
password, err := term.ReadPassword(int(syscall.Stdin))
|
|
fmt.Fprintln(os.Stderr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return strings.TrimSpace(string(password)), nil
|
|
}
|
|
|
|
func printPrompt(prompt string) {
|
|
fmt.Fprint(os.Stderr, prompt, ": ")
|
|
}
|