stdin.go


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
50
51
52
53
54
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.ToLower(s)
		switch s {
		case "y":
			log.Info("got y")
			return
		case "n":
			log.Info("got n")
			badExit(nil, err)
		case "":
			if b {
				return
			} else {
				badExit(nil, err)
			}
		default:
			badExit(nil, err)
		}
	}
}