From da530d986c1b0049b66d7d3234295a2e72e34845 Mon Sep 17 00:00:00 2001 From: xavierror Date: Thu, 6 Aug 2020 10:44:31 +0800 Subject: [PATCH] refactor: optimized code --- utils/utils.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/utils/utils.go b/utils/utils.go index 2a29633..928f536 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -153,8 +153,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) {