"bufio"
"bytes"
"fmt"
+ "io/ioutil"
"os"
"strconv"
+ "strings"
"github.com/zyedidia/tcell"
)
// We have to keep track of the cursor for prompting
cursorx int
+ // This map stores the history for all the different kinds of uses Prompt has
+ // It's a map of history type -> history array
+ history map[string][]string
+ historyNum int
+
// Is the current message a message from the gutter
gutterMessage bool
}
func (m *Messenger) YesNoPrompt(prompt string) (bool, bool) {
m.Message(prompt)
+ _, h := screen.Size()
for {
m.Clear()
m.Display()
+ screen.ShowCursor(Count(m.message), h-1)
screen.Show()
- event := screen.PollEvent()
+ event := <-events
switch e := event.(type) {
case *tcell.EventKey:
}
}
+type Completion int
+
+const (
+ NoCompletion Completion = iota
+ FileCompletion
+)
+
// Prompt sends the user a message and waits for a response to be typed in
// This function blocks the main loop while waiting for input
-func (m *Messenger) Prompt(prompt string) (string, bool) {
+func (m *Messenger) Prompt(prompt, historyType string, completionType Completion) (string, bool) {
m.hasPrompt = true
m.Message(prompt)
+ if _, ok := m.history[historyType]; !ok {
+ m.history[historyType] = []string{""}
+ } else {
+ m.history[historyType] = append(m.history[historyType], "")
+ }
+ m.historyNum = len(m.history[historyType]) - 1
response, canceled := "", true
m.Clear()
m.Display()
- event := screen.PollEvent()
+ event := <-events
switch e := event.(type) {
case *tcell.EventKey:
// User is done entering their response
m.hasPrompt = false
response, canceled = m.response, false
+ m.history[historyType][len(m.history[historyType])-1] = response
+ case tcell.KeyTab:
+ if completionType == FileCompletion {
+ dirs := strings.Split(m.response, "/")
+ var files []os.FileInfo
+ var err error
+ if len(dirs) > 1 {
+ files, err = ioutil.ReadDir(strings.Join(dirs[:len(dirs)-1], "/"))
+ } else {
+ files, err = ioutil.ReadDir(".")
+ }
+ if err != nil {
+ continue
+ }
+ var suggestions []string
+ for _, f := range files {
+ name := f.Name()
+ if f.IsDir() {
+ name += "/"
+ }
+ if strings.HasPrefix(name, dirs[len(dirs)-1]) {
+ suggestions = append(suggestions, name)
+ }
+ }
+ if len(suggestions) == 1 {
+ if len(dirs) > 1 {
+ m.response = strings.Join(dirs[:len(dirs)-1], "/") + "/" + suggestions[0]
+ } else {
+ m.response = suggestions[0]
+ }
+ m.cursorx = Count(m.response)
+ }
+ }
}
}
- m.HandleEvent(event)
+ m.HandleEvent(event, m.history[historyType])
if m.cursorx < 0 {
// Cancel
}
// HandleEvent handles an event for the prompter
-func (m *Messenger) HandleEvent(event tcell.Event) {
+func (m *Messenger) HandleEvent(event tcell.Event, history []string) {
switch e := event.(type) {
case *tcell.EventKey:
switch e.Key() {
+ case tcell.KeyUp:
+ if m.historyNum > 0 {
+ m.historyNum--
+ m.response = history[m.historyNum]
+ m.cursorx = Count(m.response)
+ }
+ case tcell.KeyDown:
+ if m.historyNum < len(history)-1 {
+ m.historyNum++
+ m.response = history[m.historyNum]
+ m.cursorx = Count(m.response)
+ }
case tcell.KeyLeft:
if m.cursorx > 0 {
m.cursorx--
m.response = Insert(m.response, m.cursorx, string(e.Rune()))
m.cursorx++
}
+ history[m.historyNum] = m.response
}
}