db-migrator.go

1
/**
2
 * This file is part of the raoptimus/db-migrator.go library
3
 *
4
 * @copyright Copyright (c) Evgeniy Urvantsev
5
 * @license https://github.com/raoptimus/db-migrator.go/blob/master/LICENSE.md
6
 * @link https://github.com/raoptimus/db-migrator.go
7
 */
8

9
package console
10

11
import (
12
	"bufio"
13
	"os"
14
	"strings"
15
)
16

17
func Confirm(s string) bool {
18
	reader := bufio.NewReader(os.Stdin)
19

20
	for {
21
		Infof("%s [y/n]: ", s)
22

23
		response, err := reader.ReadString('\n')
24
		if err != nil {
25
			Fatal(err)
26
		}
27

28
		response = strings.ToLower(strings.TrimSpace(response))
29

30
		if response == "y" || response == "yes" {
31
			return true
32
		} else if response == "n" || response == "no" {
33
			return false
34
		}
35
	}
36
}
37