mirror of https://github.com/cheat/cheat.git
38 lines
650 B
Go
38 lines
650 B
Go
|
package installer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Prompt prompts the user for a answer
|
||
|
|
func Prompt(prompt string, def bool) (bool, error) {
|
||
|
|
|
||
|
|
// initialize a line reader
|
||
|
|
reader := bufio.NewReader(os.Stdin)
|
||
|
|
|
||
|
|
// display the prompt
|
||
|
|
fmt.Print(fmt.Sprintf("%s: ", prompt))
|
||
|
|
|
||
|
|
// read the answer
|
||
|
|
ans, err := reader.ReadString('\n')
|
||
|
|
if err != nil {
|
||
|
|
return false, fmt.Errorf("failed to parse input: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// normalize the answer
|
||
|
|
ans = strings.ToLower(strings.TrimRight(ans, "\n"))
|
||
|
|
|
||
|
|
// return the appropriate response
|
||
|
|
switch ans {
|
||
|
|
case "y":
|
||
|
|
return true, nil
|
||
|
|
case "":
|
||
|
|
return def, nil
|
||
|
|
default:
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
}
|