Backup support

[micro.git] / internal / screen / message.go

diff --git a/internal/screen/message.go b/internal/screen/message.go
index cd4abce9d4aed924751302ed0b58c7be5cf62364..1ca9141ce9328b424b00adb3d86718ec5b624b93 100644 (file)
--- a/internal/screen/message.go
+++ b/internal/screen/message.go
@@ -5,6 +5,7 @@ import (
        "fmt"
        "os"
        "strconv"
+       "strings"
 )
 
 // TermMessage sends a message to the user in the terminal. This usually occurs before
@@ -25,6 +26,38 @@ func TermMessage(msg ...interface{}) {
        TempStart(screenb)
 }
 
+// TermPrompt prints a prompt and requests the user for a response
+// The result is matched against a list of options and the index of
+// the match is returned
+// If wait is true, the prompt re-prompts until a valid option is
+// chosen, otherwise if wait is false, -1 is returned for no match
+func TermPrompt(prompt string, options []string, wait bool) int {
+       screenb := TempFini()
+
+       idx := -1
+       // same behavior as do { ... } while (wait && idx == -1)
+       for ok := true; ok; ok = wait && idx == -1 {
+               reader := bufio.NewReader(os.Stdin)
+               fmt.Print(prompt)
+               resp, _ := reader.ReadString('\n')
+               resp = strings.TrimSpace(resp)
+
+               for i, opt := range options {
+                       if resp == opt {
+                               idx = i
+                       }
+               }
+
+               if wait && idx == -1 {
+                       fmt.Println("\nInvalid choice.")
+               }
+       }
+
+       TempStart(screenb)
+
+       return idx
+}
+
 // TermError sends an error to the user in the terminal. Like TermMessage except formatted
 // as an error
 func TermError(filename string, lineNum int, err string) {