diff --git a/utils/utils.go b/utils/utils.go index 398aa1b..8128e3c 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -88,7 +88,7 @@ func IsInGOPATH(thePath string) bool { // IsBeegoProject checks whether the current path is a Beego application or not func IsBeegoProject(thePath string) bool { mainFiles := []string{} - hasBeegoRegex := regexp.MustCompile(`(?s)package main.*?import.*?\(.*?github.com/astaxie/beego".*?\).*func main()`) + hasBeegoRegex := regexp.MustCompile(`(?s)package main.*?import.*?\(.*?github.com/astaxie/beego".*?\).*func main\(\)`) c := make(chan error) // Walk the application path tree to look for main files. // Main files must satisfy the 'hasBeegoRegex' regular expression. @@ -162,8 +162,10 @@ func AskForConfirmation() bool { if err != nil { beeLogger.Log.Fatalf("%s", err) } - okayResponses := []string{"y", "Y", "yes", "Yes", "YES"} - nokayResponses := []string{"n", "N", "no", "No", "NO"} + response = strings.ToLower(strings.TrimSpace(response)) + + okayResponses := []string{"y", "yes"} + nokayResponses := []string{"n", "no"} if containsString(okayResponses, response) { return true } else if containsString(nokayResponses, response) {