41 lines
665 B
Go
41 lines
665 B
Go
|
package prompt
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"golang.org/x/term"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TODO: use charm stuff for this
|
||
|
|
|
||
|
|
func YesNo(prompt string) bool {
|
||
|
|
return AskRune(prompt, "y/n") == 'y'
|
||
|
|
}
|
||
|
|
|
||
|
|
func AskRune(prompt, options string) byte {
|
||
|
|
// TODO: handle errors better
|
||
|
|
// switch stdin into 'raw' mode
|
||
|
|
oldState, err := term.MakeRaw(int(os.Stdin.Fd()))
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
defer func() {
|
||
|
|
if err := term.Restore(int(os.Stdin.Fd()), oldState); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
fmt.Printf("%s [%s]: ", prompt, options)
|
||
|
|
|
||
|
|
// read one byte from stdin
|
||
|
|
b := make([]byte, 1)
|
||
|
|
_, err = os.Stdin.Read(b)
|
||
|
|
if err != nil {
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
|
||
|
|
return bytes.ToLower(b)[0]
|
||
|
|
}
|