]> git.lizzy.rs Git - micro.git/blobdiff - cmd/micro/messenger.go
Add savehistory option
[micro.git] / cmd / micro / messenger.go
index b53687b1f199e09af5205eb39a811436881c27eb..1212725b21a0746e873c636735e43416a6916f14 100644 (file)
@@ -3,11 +3,12 @@ package main
 import (
        "bufio"
        "bytes"
+       "encoding/gob"
        "fmt"
        "os"
        "strconv"
-       "strings"
 
+       "github.com/mattn/go-runewidth"
        "github.com/zyedidia/clipboard"
        "github.com/zyedidia/tcell"
 )
@@ -22,6 +23,7 @@ func TermMessage(msg ...interface{}) {
        screenWasNil := screen == nil
        if !screenWasNil {
                screen.Fini()
+               screen = nil
        }
 
        fmt.Println(msg...)
@@ -44,6 +46,7 @@ func TermError(filename string, lineNum int, err string) {
 // Messenger is an object that makes it easy to send messages to the user
 // and get input from the user
 type Messenger struct {
+       log *Buffer
        // Are we currently prompting the user?
        hasPrompt bool
        // Is there a message to print
@@ -68,38 +71,87 @@ type Messenger struct {
        gutterMessage bool
 }
 
+// AddLog sends a message to the log view
+func (m *Messenger) AddLog(msg ...interface{}) {
+       logMessage := fmt.Sprint(msg...)
+       buffer := m.getBuffer()
+       buffer.insert(buffer.End(), []byte(logMessage+"\n"))
+       buffer.Cursor.Loc = buffer.End()
+       buffer.Cursor.Relocate()
+}
+
+func (m *Messenger) getBuffer() *Buffer {
+       if m.log == nil {
+               m.log = NewBufferFromString("", "")
+               m.log.name = "Log"
+       }
+       return m.log
+}
+
 // Message sends a message to the user
 func (m *Messenger) Message(msg ...interface{}) {
-       buf := new(bytes.Buffer)
-       fmt.Fprint(buf, msg...)
-       m.message = buf.String()
-       m.style = defStyle
+       displayMessage := fmt.Sprint(msg...)
+       // only display a new message if there isn't an active prompt
+       // this is to prevent overwriting an existing prompt to the user
+       if m.hasPrompt == false {
+               // if there is no active prompt then style and display the message as normal
+               m.message = displayMessage
 
-       if _, ok := colorscheme["message"]; ok {
-               m.style = colorscheme["message"]
+               m.style = defStyle
+
+               if _, ok := colorscheme["message"]; ok {
+                       m.style = colorscheme["message"]
+               }
+
+               m.hasMessage = true
        }
-       m.hasMessage = true
+       // add the message to the log regardless of active prompts
+       m.AddLog(displayMessage)
 }
 
 // Error sends an error message to the user
 func (m *Messenger) Error(msg ...interface{}) {
        buf := new(bytes.Buffer)
        fmt.Fprint(buf, msg...)
-       m.message = buf.String()
-       m.style = defStyle.
-               Foreground(tcell.ColorBlack).
-               Background(tcell.ColorMaroon)
 
-       if _, ok := colorscheme["error-message"]; ok {
-               m.style = colorscheme["error-message"]
+       // only display a new message if there isn't an active prompt
+       // this is to prevent overwriting an existing prompt to the user
+       if m.hasPrompt == false {
+               // if there is no active prompt then style and display the message as normal
+               m.message = buf.String()
+               m.style = defStyle.
+                       Foreground(tcell.ColorBlack).
+                       Background(tcell.ColorMaroon)
+
+               if _, ok := colorscheme["error-message"]; ok {
+                       m.style = colorscheme["error-message"]
+               }
+               m.hasMessage = true
        }
+       // add the message to the log regardless of active prompts
+       m.AddLog(buf.String())
+}
+
+func (m *Messenger) PromptText(msg ...interface{}) {
+       displayMessage := fmt.Sprint(msg...)
+       // if there is no active prompt then style and display the message as normal
+       m.message = displayMessage
+
+       m.style = defStyle
+
+       if _, ok := colorscheme["message"]; ok {
+               m.style = colorscheme["message"]
+       }
+
        m.hasMessage = true
+       // add the message to the log regardless of active prompts
+       m.AddLog(displayMessage)
 }
 
 // YesNoPrompt asks the user a yes or no question (waits for y or n) and returns the result
 func (m *Messenger) YesNoPrompt(prompt string) (bool, bool) {
        m.hasPrompt = true
-       m.Message(prompt)
+       m.PromptText(prompt)
 
        _, h := screen.Size()
        for {
@@ -113,12 +165,20 @@ func (m *Messenger) YesNoPrompt(prompt string) (bool, bool) {
                case *tcell.EventKey:
                        switch e.Key() {
                        case tcell.KeyRune:
-                               if e.Rune() == 'y' {
+                               if e.Rune() == 'y' || e.Rune() == 'Y' {
+                                       m.AddLog("\t--> y")
+                                       m.hasPrompt = false
                                        return true, false
-                               } else if e.Rune() == 'n' {
+                               } else if e.Rune() == 'n' || e.Rune() == 'N' {
+                                       m.AddLog("\t--> n")
+                                       m.hasPrompt = false
                                        return false, false
                                }
                        case tcell.KeyCtrlC, tcell.KeyCtrlQ, tcell.KeyEscape:
+                               m.AddLog("\t--> (cancel)")
+                               m.Clear()
+                               m.Reset()
+                               m.hasPrompt = false
                                return false, true
                        }
                }
@@ -128,7 +188,7 @@ func (m *Messenger) YesNoPrompt(prompt string) (bool, bool) {
 // LetterPrompt gives the user a prompt and waits for a one letter response
 func (m *Messenger) LetterPrompt(prompt string, responses ...rune) (rune, bool) {
        m.hasPrompt = true
-       m.Message(prompt)
+       m.PromptText(prompt)
 
        _, h := screen.Size()
        for {
@@ -144,11 +204,18 @@ func (m *Messenger) LetterPrompt(prompt string, responses ...rune) (rune, bool)
                        case tcell.KeyRune:
                                for _, r := range responses {
                                        if e.Rune() == r {
+                                               m.AddLog("\t--> " + string(r))
+                                               m.Clear()
                                                m.Reset()
+                                               m.hasPrompt = false
                                                return r, false
                                        }
                                }
                        case tcell.KeyCtrlC, tcell.KeyCtrlQ, tcell.KeyEscape:
+                               m.AddLog("\t--> (cancel)")
+                               m.Clear()
+                               m.Reset()
+                               m.hasPrompt = false
                                return ' ', true
                        }
                }
@@ -163,13 +230,16 @@ const (
        CommandCompletion
        HelpCompletion
        OptionCompletion
+       PluginCmdCompletion
+       PluginNameCompletion
+       OptionValueCompletion
 )
 
 // 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, historyType string, completionTypes ...Completion) (string, bool) {
+func (m *Messenger) Prompt(prompt, placeholder, historyType string, completionTypes ...Completion) (string, bool) {
        m.hasPrompt = true
-       m.Message(prompt)
+       m.PromptText(prompt)
        if _, ok := m.history[historyType]; !ok {
                m.history[historyType] = []string{""}
        } else {
@@ -177,7 +247,9 @@ func (m *Messenger) Prompt(prompt, historyType string, completionTypes ...Comple
        }
        m.historyNum = len(m.history[historyType]) - 1
 
-       response, canceled := "", true
+       response, canceled := placeholder, true
+       m.response = response
+       m.cursorx = Count(placeholder)
 
        RedrawAll()
        for m.hasPrompt {
@@ -191,14 +263,16 @@ func (m *Messenger) Prompt(prompt, historyType string, completionTypes ...Comple
                        switch e.Key() {
                        case tcell.KeyCtrlQ, tcell.KeyCtrlC, tcell.KeyEscape:
                                // Cancel
+                               m.AddLog("\t--> (cancel)")
                                m.hasPrompt = false
                        case tcell.KeyEnter:
                                // User is done entering their response
+                               m.AddLog("\t--> " + m.response)
                                m.hasPrompt = false
                                response, canceled = m.response, false
                                m.history[historyType][len(m.history[historyType])-1] = response
                        case tcell.KeyTab:
-                               args := strings.Split(m.response, " ")
+                               args := SplitCommandArgs(m.response)
                                currentArgNum := len(args) - 1
                                currentArg := args[currentArgNum]
                                var completionType Completion
@@ -224,6 +298,16 @@ func (m *Messenger) Prompt(prompt, historyType string, completionTypes ...Comple
                                        chosen, suggestions = HelpComplete(currentArg)
                                } else if completionType == OptionCompletion {
                                        chosen, suggestions = OptionComplete(currentArg)
+                               } else if completionType == OptionValueCompletion {
+                                       if currentArgNum-1 > 0 {
+                                               chosen, suggestions = OptionValueComplete(args[currentArgNum-1], currentArg)
+                                       }
+                               } else if completionType == PluginCmdCompletion {
+                                       chosen, suggestions = PluginCmdComplete(currentArg)
+                               } else if completionType == PluginNameCompletion {
+                                       chosen, suggestions = PluginNameComplete(currentArg)
+                               } else if completionType < NoCompletion {
+                                       chosen, suggestions = PluginComplete(completionType, currentArg)
                                }
 
                                if len(suggestions) > 1 {
@@ -231,10 +315,7 @@ func (m *Messenger) Prompt(prompt, historyType string, completionTypes ...Comple
                                }
 
                                if chosen != "" {
-                                       if len(args) > 1 {
-                                               chosen = " " + chosen
-                                       }
-                                       m.response = strings.Join(args[:len(args)-1], " ") + chosen
+                                       m.response = JoinCommandArgs(append(args[:len(args)-1], chosen)...)
                                        m.cursorx = Count(m.response)
                                }
                        }
@@ -242,18 +323,19 @@ func (m *Messenger) Prompt(prompt, historyType string, completionTypes ...Comple
 
                m.HandleEvent(event, m.history[historyType])
 
-               messenger.Clear()
+               m.Clear()
                for _, v := range tabs[curTab].views {
                        v.Display()
                }
                DisplayTabs()
-               messenger.Display()
+               m.Display()
                if len(suggestions) > 1 {
                        m.DisplaySuggestions(suggestions)
                }
                screen.Show()
        }
 
+       m.Clear()
        m.Reset()
        return response, canceled
 }
@@ -262,36 +344,58 @@ func (m *Messenger) Prompt(prompt, historyType string, completionTypes ...Comple
 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--
-                       }
-               case tcell.KeyRight:
-                       if m.cursorx < Count(m.response) {
-                               m.cursorx++
-                       }
-               case tcell.KeyBackspace2, tcell.KeyBackspace:
-                       if m.cursorx > 0 {
-                               m.response = string([]rune(m.response)[:m.cursorx-1]) + string([]rune(m.response)[m.cursorx:])
-                               m.cursorx--
+               if e.Key() != tcell.KeyRune || e.Modifiers() != 0 {
+                       for key, actions := range bindings {
+                               if e.Key() == key.keyCode {
+                                       if e.Key() == tcell.KeyRune {
+                                               if e.Rune() != key.r {
+                                                       continue
+                                               }
+                                       }
+                                       if e.Modifiers() == key.modifiers {
+                                               for _, action := range actions {
+                                                       funcName := FuncName(action)
+                                                       switch funcName {
+                                                       case "main.(*View).CursorUp":
+                                                               if m.historyNum > 0 {
+                                                                       m.historyNum--
+                                                                       m.response = history[m.historyNum]
+                                                                       m.cursorx = Count(m.response)
+                                                               }
+                                                       case "main.(*View).CursorDown":
+                                                               if m.historyNum < len(history)-1 {
+                                                                       m.historyNum++
+                                                                       m.response = history[m.historyNum]
+                                                                       m.cursorx = Count(m.response)
+                                                               }
+                                                       case "main.(*View).CursorLeft":
+                                                               if m.cursorx > 0 {
+                                                                       m.cursorx--
+                                                               }
+                                                       case "main.(*View).CursorRight":
+                                                               if m.cursorx < Count(m.response) {
+                                                                       m.cursorx++
+                                                               }
+                                                       case "main.(*View).CursorStart", "main.(*View).StartOfLine":
+                                                               m.cursorx = 0
+                                                       case "main.(*View).CursorEnd", "main.(*View).EndOfLine":
+                                                               m.cursorx = Count(m.response)
+                                                       case "main.(*View).Backspace":
+                                                               if m.cursorx > 0 {
+                                                                       m.response = string([]rune(m.response)[:m.cursorx-1]) + string([]rune(m.response)[m.cursorx:])
+                                                                       m.cursorx--
+                                                               }
+                                                       case "main.(*View).Paste":
+                                                               clip, _ := clipboard.ReadAll("clipboard")
+                                                               m.response = Insert(m.response, m.cursorx, clip)
+                                                               m.cursorx += Count(clip)
+                                                       }
+                                               }
+                                       }
+                               }
                        }
-               case tcell.KeyCtrlV:
-                       clip, _ := clipboard.ReadAll()
-                       m.response = Insert(m.response, m.cursorx, clip)
-                       m.cursorx += Count(clip)
+               }
+               switch e.Key() {
                case tcell.KeyRune:
                        m.response = Insert(m.response, m.cursorx, string(e.Rune()))
                        m.cursorx++
@@ -302,6 +406,23 @@ func (m *Messenger) HandleEvent(event tcell.Event, history []string) {
                clip := e.Text()
                m.response = Insert(m.response, m.cursorx, clip)
                m.cursorx += Count(clip)
+       case *tcell.EventMouse:
+               x, y := e.Position()
+               x -= Count(m.message)
+               button := e.Buttons()
+               _, screenH := screen.Size()
+
+               if y == screenH-1 {
+                       switch button {
+                       case tcell.Button1:
+                               m.cursorx = x
+                               if m.cursorx < 0 {
+                                       m.cursorx = 0
+                               } else if m.cursorx > Count(m.response) {
+                                       m.cursorx = Count(m.response)
+                               }
+                       }
+               }
        }
 }
 
@@ -351,8 +472,10 @@ func (m *Messenger) Display() {
        if m.hasMessage {
                if m.hasPrompt || globalSettings["infobar"].(bool) {
                        runes := []rune(m.message + m.response)
+                       posx := 0
                        for x := 0; x < len(runes); x++ {
-                               screen.SetContent(x, h-1, runes[x], nil, m.style)
+                               screen.SetContent(posx, h-1, runes[x], nil, m.style)
+                               posx += runewidth.RuneWidth(runes[x])
                        }
                }
        }
@@ -363,6 +486,55 @@ func (m *Messenger) Display() {
        }
 }
 
+// LoadHistory attempts to load user history from configDir/buffers/history
+// into the history map
+// The savehistory option must be on
+func (m *Messenger) LoadHistory() {
+       if GetGlobalOption("savehistory").(bool) {
+               file, err := os.Open(configDir + "/buffers/history")
+               var decodedMap map[string][]string
+               if err == nil {
+                       decoder := gob.NewDecoder(file)
+                       err = decoder.Decode(&decodedMap)
+                       file.Close()
+               }
+
+               if err != nil {
+                       m.Error("Error loading history:", err)
+                       return
+               }
+
+               m.history = decodedMap
+       } else {
+               m.history = make(map[string][]string)
+       }
+}
+
+// SaveHistory saves the user's command history to configDir/buffers/history
+// only if the savehistory option is on
+func (m *Messenger) SaveHistory() {
+       if GetGlobalOption("savehistory").(bool) {
+               // Don't save history past 100
+               for k, v := range m.history {
+                       if len(v) > 100 {
+                               m.history[k] = v[0:100]
+                       }
+               }
+
+               file, err := os.Create(configDir + "/buffers/history")
+               if err == nil {
+                       encoder := gob.NewEncoder(file)
+
+                       err = encoder.Encode(m.history)
+                       if err != nil {
+                               m.Error("Error saving history:", err)
+                               return
+                       }
+                       file.Close()
+               }
+       }
+}
+
 // A GutterMessage is a message displayed on the side of the editor
 type GutterMessage struct {
        lineNum int