summaryrefslogtreecommitdiff
path: root/stdin.go
diff options
context:
space:
mode:
Diffstat (limited to 'stdin.go')
-rw-r--r--stdin.go55
1 files changed, 55 insertions, 0 deletions
diff --git a/stdin.go b/stdin.go
new file mode 100644
index 0000000..9093246
--- /dev/null
+++ b/stdin.go
@@ -0,0 +1,55 @@
+package main
+
+import (
+ "bufio"
+ "errors"
+ "fmt"
+ "os"
+ "strings"
+
+ "go.wit.com/log"
+)
+
+func showOptions(b bool, s []string) {
+ fmt.Println("")
+ for _, line := range s {
+ fmt.Println(line)
+ }
+ fmt.Println("")
+ if b {
+ fmt.Println("Enter (Y/n)")
+ } else {
+ fmt.Println("Enter (y/N)")
+ }
+}
+
+// if b == true, default is to continue with 'Y'
+func simpleStdin(b bool, s []string) {
+ if argv.Auto {
+ return
+ }
+ err := errors.New("user cancelled via stdin")
+ showOptions(b, s)
+ scanner := bufio.NewScanner(os.Stdin)
+ for scanner.Scan() {
+ s := scanner.Text()
+ s = strings.TrimSpace(s)
+ // s = strings.Lower(s)
+ switch s {
+ case "y":
+ log.Info("got y")
+ return
+ case "n":
+ log.Info("got n")
+ badExit(err)
+ case "":
+ if b {
+ return
+ } else {
+ badExit(err)
+ }
+ default:
+ badExit(err)
+ }
+ }
+}