youtubebeat/vendor/github.com/elastic/beats/libbeat/common/terminal/terminal.go

70 lines
1.8 KiB
Go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package terminal
import (
"bufio"
"fmt"
"os"
"strings"
)
// ReadInput Capture user input for a question
func ReadInput() (string, error) {
reader := bufio.NewReader(os.Stdin)
input, err := reader.ReadString('\n')
if err != nil {
return "", err
}
return strings.TrimSuffix(input, "\n"), nil
}
// PromptYesNo Returns true if the user has entered Y or YES, capitalization is ignored, we are
// matching elasticsearch behavior
func PromptYesNo(prompt string, defaultAnswer bool) bool {
var defaultYNprompt string
if defaultAnswer == true {
defaultYNprompt = "[Y/n]"
} else {
defaultYNprompt = "[y/N]"
}
fmt.Printf("%s %s: ", prompt, defaultYNprompt)
for {
input, err := ReadInput()
if err != nil {
panic("could not read from input")
}
response := strings.TrimSpace(input)
response = strings.ToLower(response)
if response == "" {
return defaultAnswer
} else if response == "y" || response == "yes" {
return true
} else if response == "n" || response == "no" {
return false
}
fmt.Printf("Did not understand the answer '%s'\n", input)
}
}