import (
"bytes"
- "io/ioutil"
+ "fmt"
+ "io"
"os"
"os/exec"
"os/signal"
+ "path/filepath"
"regexp"
"strings"
"github.com/mitchellh/go-homedir"
)
-var commands map[string]func([]string)
-
-var commandActions = map[string]func([]string){
- "Set": Set,
- "Run": Run,
- "Bind": Bind,
- "Quit": Quit,
- "Save": Save,
- "Replace": Replace,
- "VSplit": VSplit,
- "HSplit": HSplit,
- "Tab": NewTab,
+type Command struct {
+ action func([]string)
+ completions []Completion
+}
+
+type StrCommand struct {
+ action string
+ completions []Completion
+}
+
+var commands map[string]Command
+
+var commandActions map[string]func([]string)
+
+func init() {
+ commandActions = map[string]func([]string){
+ "Set": Set,
+ "SetLocal": SetLocal,
+ "Show": Show,
+ "Run": Run,
+ "Bind": Bind,
+ "Quit": Quit,
+ "Save": Save,
+ "Replace": Replace,
+ "VSplit": VSplit,
+ "HSplit": HSplit,
+ "Tab": NewTab,
+ "Help": Help,
+ "Eval": Eval,
+ "ToggleLog": ToggleLog,
+ "Plugin": PluginCmd,
+ "Reload": Reload,
+ "Cd": Cd,
+ "Pwd": Pwd,
+ "Open": Open,
+ }
}
// InitCommands initializes the default commands
func InitCommands() {
- commands = make(map[string]func([]string))
+ commands = make(map[string]Command)
defaults := DefaultCommands()
parseCommands(defaults)
}
-func parseCommands(userCommands map[string]string) {
+func parseCommands(userCommands map[string]StrCommand) {
for k, v := range userCommands {
- MakeCommand(k, v)
+ MakeCommand(k, v.action, v.completions...)
}
}
// MakeCommand is a function to easily create new commands
// This can be called by plugins in Lua so that plugins can define their own commands
-func MakeCommand(name, function string) {
+func MakeCommand(name, function string, completions ...Completion) {
action := commandActions[function]
if _, ok := commandActions[function]; !ok {
// If the user seems to be binding a function that doesn't exist
action = LuaFunctionCommand(function)
}
- commands[name] = action
+ commands[name] = Command{action, completions}
}
// DefaultCommands returns a map containing micro's default commands
-func DefaultCommands() map[string]string {
- return map[string]string{
- "set": "Set",
- "bind": "Bind",
- "run": "Run",
- "quit": "Quit",
- "save": "Save",
- "replace": "Replace",
- "vsplit": "VSplit",
- "hsplit": "HSplit",
- "tab": "Tab",
+func DefaultCommands() map[string]StrCommand {
+ return map[string]StrCommand{
+ "set": {"Set", []Completion{OptionCompletion, NoCompletion}},
+ "setlocal": {"SetLocal", []Completion{OptionCompletion, NoCompletion}},
+ "show": {"Show", []Completion{OptionCompletion, NoCompletion}},
+ "bind": {"Bind", []Completion{NoCompletion}},
+ "run": {"Run", []Completion{NoCompletion}},
+ "quit": {"Quit", []Completion{NoCompletion}},
+ "save": {"Save", []Completion{NoCompletion}},
+ "replace": {"Replace", []Completion{NoCompletion}},
+ "vsplit": {"VSplit", []Completion{FileCompletion, NoCompletion}},
+ "hsplit": {"HSplit", []Completion{FileCompletion, NoCompletion}},
+ "tab": {"Tab", []Completion{FileCompletion, NoCompletion}},
+ "help": {"Help", []Completion{HelpCompletion, NoCompletion}},
+ "eval": {"Eval", []Completion{NoCompletion}},
+ "log": {"ToggleLog", []Completion{NoCompletion}},
+ "plugin": {"Plugin", []Completion{PluginCmdCompletion, PluginNameCompletion}},
+ "reload": {"Reload", []Completion{NoCompletion}},
+ "cd": {"Cd", []Completion{FileCompletion}},
+ "pwd": {"Pwd", []Completion{NoCompletion}},
+ "open": {"Open", []Completion{FileCompletion}},
+ }
+}
+
+// PluginCmd installs, removes, updates, lists, or searches for given plugins
+func PluginCmd(args []string) {
+ if len(args) >= 1 {
+ switch args[0] {
+ case "install":
+ installedVersions := GetInstalledVersions(false)
+ for _, plugin := range args[1:] {
+ pp := GetAllPluginPackages().Get(plugin)
+ if pp == nil {
+ messenger.Error("Unknown plugin \"" + plugin + "\"")
+ } else if err := pp.IsInstallable(); err != nil {
+ messenger.Error("Error installing ", plugin, ": ", err)
+ } else {
+ for _, installed := range installedVersions {
+ if pp.Name == installed.pack.Name {
+ if pp.Versions[0].Version.Compare(installed.Version) == 1 {
+ messenger.Error(pp.Name, " is already installed but out-of-date: use 'plugin update ", pp.Name, "' to update")
+ } else {
+ messenger.Error(pp.Name, " is already installed")
+ }
+ }
+ }
+ pp.Install()
+ }
+ }
+ case "remove":
+ removed := ""
+ for _, plugin := range args[1:] {
+ // check if the plugin exists.
+ for _, lp := range loadedPlugins {
+ if lp == plugin {
+ UninstallPlugin(plugin)
+ removed += plugin + " "
+ continue
+ }
+ }
+ }
+ if !IsSpaces(removed) {
+ messenger.Message("Removed ", removed)
+ } else {
+ messenger.Error("The requested plugins do not exist")
+ }
+ case "update":
+ UpdatePlugins(args[1:])
+ case "list":
+ plugins := GetInstalledVersions(false)
+ messenger.AddLog("----------------")
+ messenger.AddLog("The following plugins are currently installed:\n")
+ for _, p := range plugins {
+ messenger.AddLog(fmt.Sprintf("%s (%s)", p.pack.Name, p.Version))
+ }
+ messenger.AddLog("----------------")
+ if len(plugins) > 0 {
+ if CurView().Type != vtLog {
+ ToggleLog([]string{})
+ }
+ }
+ case "search":
+ plugins := SearchPlugin(args[1:])
+ messenger.Message(len(plugins), " plugins found")
+ for _, p := range plugins {
+ messenger.AddLog("----------------")
+ messenger.AddLog(p.String())
+ }
+ messenger.AddLog("----------------")
+ if len(plugins) > 0 {
+ if CurView().Type != vtLog {
+ ToggleLog([]string{})
+ }
+ }
+ case "available":
+ packages := GetAllPluginPackages()
+ messenger.AddLog("Available Plugins:")
+ for _, pkg := range packages {
+ messenger.AddLog(pkg.Name)
+ }
+ if CurView().Type != vtLog {
+ ToggleLog([]string{})
+ }
+ }
+ } else {
+ messenger.Error("Not enough arguments")
+ }
+}
+
+func Cd(args []string) {
+ if len(args) > 0 {
+ home, _ := homedir.Dir()
+ path := strings.Replace(args[0], "~", home, 1)
+ os.Chdir(path)
+ for _, tab := range tabs {
+ for _, view := range tab.views {
+ wd, _ := os.Getwd()
+ view.Buf.Path, _ = MakeRelative(view.Buf.AbsPath, wd)
+ if p, _ := filepath.Abs(view.Buf.Path); !strings.Contains(p, wd) {
+ view.Buf.Path = view.Buf.AbsPath
+ }
+ }
+ }
+ }
+}
+
+func Pwd(args []string) {
+ wd, err := os.Getwd()
+ if err != nil {
+ messenger.Message(err.Error())
+ } else {
+ messenger.Message(wd)
+ }
+}
+
+func Open(args []string) {
+ if len(args) > 0 {
+ filename := args[0]
+ // the filename might or might not be quoted, so unquote first then join the strings.
+ filename = strings.Join(SplitCommandArgs(filename), " ")
+
+ CurView().Open(filename)
+ } else {
+ messenger.Error("No filename")
+ }
+}
+
+func ToggleLog(args []string) {
+ buffer := messenger.getBuffer()
+ if CurView().Type != vtLog {
+ CurView().HSplit(buffer)
+ CurView().Type = vtLog
+ RedrawAll()
+ buffer.Cursor.Loc = buffer.Start()
+ CurView().Relocate()
+ buffer.Cursor.Loc = buffer.End()
+ CurView().Relocate()
+ } else {
+ CurView().Quit(true)
+ }
+}
+
+func Reload(args []string) {
+ LoadAll()
+}
+
+// Help tries to open the given help page in a horizontal split
+func Help(args []string) {
+ if len(args) < 1 {
+ // Open the default help if the user just typed "> help"
+ CurView().openHelp("help")
+ } else {
+ helpPage := args[0]
+ if FindRuntimeFile(RTHelp, helpPage) != nil {
+ CurView().openHelp(helpPage)
+ } else {
+ messenger.Error("Sorry, no help for ", helpPage)
+ }
}
}
// If no file is given, it opens an empty buffer in a new split
func VSplit(args []string) {
if len(args) == 0 {
- CurView().VSplit(NewBuffer([]byte{}, ""))
+ CurView().VSplit(NewBuffer(strings.NewReader(""), ""))
} else {
filename := args[0]
home, _ := homedir.Dir()
filename = strings.Replace(filename, "~", home, 1)
- file, err := ioutil.ReadFile(filename)
+ file, err := os.Open(filename)
+ defer file.Close()
var buf *Buffer
if err != nil {
// File does not exist -- create an empty buffer with that name
- buf = NewBuffer([]byte{}, filename)
+ buf = NewBuffer(strings.NewReader(""), filename)
} else {
buf = NewBuffer(file, filename)
}
// If no file is given, it opens an empty buffer in a new split
func HSplit(args []string) {
if len(args) == 0 {
- CurView().HSplit(NewBuffer([]byte{}, ""))
+ CurView().HSplit(NewBuffer(strings.NewReader(""), ""))
} else {
filename := args[0]
home, _ := homedir.Dir()
filename = strings.Replace(filename, "~", home, 1)
- file, err := ioutil.ReadFile(filename)
+ file, err := os.Open(filename)
+ defer file.Close()
var buf *Buffer
if err != nil {
// File does not exist -- create an empty buffer with that name
- buf = NewBuffer([]byte{}, filename)
+ buf = NewBuffer(strings.NewReader(""), filename)
} else {
buf = NewBuffer(file, filename)
}
}
}
-// Tab opens the given file in a new tab
+// Eval evaluates a lua expression
+func Eval(args []string) {
+ if len(args) >= 1 {
+ err := L.DoString(args[0])
+ if err != nil {
+ messenger.Error(err)
+ }
+ } else {
+ messenger.Error("Not enough arguments")
+ }
+}
+
+// NewTab opens the given file in a new tab
func NewTab(args []string) {
if len(args) == 0 {
- CurView().AddTab()
+ CurView().AddTab(true)
} else {
filename := args[0]
home, _ := homedir.Dir()
filename = strings.Replace(filename, "~", home, 1)
- file, _ := ioutil.ReadFile(filename)
+ file, _ := os.Open(filename)
+ defer file.Close()
tab := NewTabFromView(NewView(NewBuffer(file, filename)))
tab.SetNum(len(tabs))
tabs = append(tabs, tab)
- curTab++
- // if len(tabs) == 2 {
- // for _, t := range tabs {
- // for _, v := range t.views {
- // v.Resize(screen.Size())
- // }
- // }
- // }
+ curTab = len(tabs) - 1
+ if len(tabs) == 2 {
+ for _, t := range tabs {
+ for _, v := range t.views {
+ v.ToggleTabbar()
+ }
+ }
+ }
}
}
// Set sets an option
func Set(args []string) {
- // Set an option and we have to set it for every view
- for _, tab := range tabs {
- for _, view := range tab.views {
- SetOption(view, args)
- }
+ if len(args) < 2 {
+ messenger.Error("Not enough arguments")
+ return
+ }
+
+ option := strings.TrimSpace(args[0])
+ value := strings.TrimSpace(args[1])
+
+ SetOptionAndSettings(option, value)
+}
+
+// SetLocal sets an option local to the buffer
+func SetLocal(args []string) {
+ if len(args) < 2 {
+ messenger.Error("Not enough arguments")
+ return
+ }
+
+ option := strings.TrimSpace(args[0])
+ value := strings.TrimSpace(args[1])
+
+ err := SetLocalOption(option, value, CurView())
+ if err != nil {
+ messenger.Error(err.Error())
+ }
+}
+
+// Show shows the value of the given option
+func Show(args []string) {
+ if len(args) < 1 {
+ messenger.Error("Please provide an option to show")
+ return
}
+
+ option := GetOption(args[0])
+
+ if option == nil {
+ messenger.Error(args[0], " is not a valid option")
+ return
+ }
+
+ messenger.Message(option)
}
// Bind creates a new keybinding
func Bind(args []string) {
- if len(args) != 2 {
- messenger.Error("Incorrect number of arguments")
+ if len(args) < 2 {
+ messenger.Error("Not enough arguments")
return
}
BindKey(args[0], args[1])
// Run runs a shell command in the background
func Run(args []string) {
// Run a shell command in the background (openTerm is false)
- HandleShellCommand(strings.Join(args, " "), false)
+ HandleShellCommand(JoinCommandArgs(args...), false, true)
}
// Quit closes the main view
func Quit(args []string) {
// Close the main view
- CurView().Quit()
+ CurView().Quit(true)
}
// Save saves the buffer in the main view
func Save(args []string) {
- // Save the main view
- CurView().Save()
+ if len(args) == 0 {
+ // Save the main view
+ CurView().Save(true)
+ } else {
+ CurView().Buf.SaveAs(args[0])
+ }
}
// Replace runs search and replace
func Replace(args []string) {
- // This is a regex to parse the replace expression
- // We allow no quotes if there are no spaces, but if you want to search
- // for or replace an expression with spaces, you can add double quotes
- r := regexp.MustCompile(`"[^"\\]*(?:\\.[^"\\]*)*"|[^\s]*`)
- replaceCmd := r.FindAllString(strings.Join(args, " "), -1)
- if len(replaceCmd) < 2 {
+ if len(args) < 2 {
// We need to find both a search and replace expression
messenger.Error("Invalid replace statement: " + strings.Join(args, " "))
return
}
var flags string
- if len(replaceCmd) == 3 {
+ if len(args) == 3 {
// The user included some flags
- flags = replaceCmd[2]
- }
-
- search := string(replaceCmd[0])
- replace := string(replaceCmd[1])
-
- // If the search and replace expressions have quotes, we need to remove those
- if strings.HasPrefix(search, `"`) && strings.HasSuffix(search, `"`) {
- search = search[1 : len(search)-1]
- }
- if strings.HasPrefix(replace, `"`) && strings.HasSuffix(replace, `"`) {
- replace = replace[1 : len(replace)-1]
+ flags = args[2]
}
- // We replace all escaped double quotes to real double quotes
- search = strings.Replace(search, `\"`, `"`, -1)
- replace = strings.Replace(replace, `\"`, `"`, -1)
- // Replace some things so users can actually insert newlines and tabs in replacements
- replace = strings.Replace(replace, "\\n", "\n", -1)
- replace = strings.Replace(replace, "\\t", "\t", -1)
+ search := string(args[0])
+ replace := string(args[1])
- regex, err := regexp.Compile(search)
+ regex, err := regexp.Compile("(?m)" + search)
if err != nil {
// There was an error with the user's regex
messenger.Error(err.Error())
view := CurView()
found := 0
- for {
- match := regex.FindStringIndex(view.Buf.String())
- if match == nil {
- break
- }
- found++
- if strings.Contains(flags, "c") {
+ if strings.Contains(flags, "c") {
+ for {
// The 'check' flag was used
Search(search, view, true)
+ if !view.Cursor.HasSelection() {
+ break
+ }
view.Relocate()
- if settings["syntax"].(bool) {
+ if view.Buf.Settings["syntax"].(bool) {
view.matches = Match(view)
}
RedrawAll()
view.Cursor.ResetSelection()
}
messenger.Reset()
- return
+ break
}
if choice {
view.Cursor.DeleteSelection()
- view.Buf.Insert(FromCharPos(match[0], view.Buf), replace)
+ view.Buf.Insert(view.Cursor.Loc, replace)
view.Cursor.ResetSelection()
messenger.Reset()
+ found++
} else {
if view.Cursor.HasSelection() {
searchStart = ToCharPos(view.Cursor.CurSelection[1], view.Buf)
}
continue
}
- } else {
- view.Buf.Replace(FromCharPos(match[0], view.Buf), FromCharPos(match[1], view.Buf), replace)
+ }
+ } else {
+ bufStr := view.Buf.String()
+ matches := regex.FindAllStringIndex(bufStr, -1)
+ if matches != nil && len(matches) > 0 {
+ prevMatchCount := runePos(matches[0][0], bufStr)
+ searchCount := runePos(matches[0][1], bufStr) - prevMatchCount
+ from := FromCharPos(matches[0][0], view.Buf)
+ to := from.Move(searchCount, view.Buf)
+ adjust := Count(replace) - searchCount
+ view.Buf.Replace(from, to, replace)
+ if len(matches) > 1 {
+ for _, match := range matches[1:] {
+ found++
+ matchCount := runePos(match[0], bufStr)
+ searchCount = runePos(match[1], bufStr) - matchCount
+ from = from.Move(matchCount-prevMatchCount+adjust, view.Buf)
+ to = from.Move(searchCount, view.Buf)
+ view.Buf.Replace(from, to, replace)
+ prevMatchCount = matchCount
+ adjust = Count(replace) - searchCount
+ }
+ }
}
}
view.Cursor.Relocate()
if found > 1 {
- messenger.Message("Replaced ", found, " occurences of ", search)
+ messenger.Message("Replaced ", found, " occurrences of ", search)
} else if found == 1 {
- messenger.Message("Replaced ", found, " occurence of ", search)
+ messenger.Message("Replaced ", found, " occurrence of ", search)
} else {
messenger.Message("Nothing matched ", search)
}
// RunShellCommand executes a shell command and returns the output/error
func RunShellCommand(input string) (string, error) {
- inputCmd := strings.Split(input, " ")[0]
- args := strings.Split(input, " ")[1:]
+ inputCmd := SplitCommandArgs(input)[0]
+ args := SplitCommandArgs(input)[1:]
cmd := exec.Command(inputCmd, args...)
outputBytes := &bytes.Buffer{}
// HandleShellCommand runs the shell command
// The openTerm argument specifies whether a terminal should be opened (for viewing output
// or interacting with stdin)
-func HandleShellCommand(input string, openTerm bool) {
- inputCmd := strings.Split(input, " ")[0]
+func HandleShellCommand(input string, openTerm bool, waitToFinish bool) string {
+ inputCmd := SplitCommandArgs(input)[0]
if !openTerm {
// Simply run the command in the background and notify the user when it's done
messenger.Message("Running...")
screen.Fini()
screen = nil
- args := strings.Split(input, " ")[1:]
+ args := SplitCommandArgs(input)[1:]
// Set up everything for the command
+ var outputBuf bytes.Buffer
cmd := exec.Command(inputCmd, args...)
cmd.Stdin = os.Stdin
- cmd.Stdout = os.Stdout
+ cmd.Stdout = io.MultiWriter(os.Stdout, &outputBuf)
cmd.Stderr = os.Stderr
// This is a trap for Ctrl-C so that it doesn't kill micro
}
}()
- // Start the command
cmd.Start()
- cmd.Wait()
+ err := cmd.Wait()
+
+ output := outputBuf.String()
+ if err != nil {
+ output = err.Error()
+ }
- // This is just so we don't return right away and let the user press enter to return
- TermMessage("")
+ if waitToFinish {
+ // This is just so we don't return right away and let the user press enter to return
+ TermMessage("")
+ }
// Start the screen back up
InitScreen()
+
+ return output
}
+ return ""
}
// HandleCommand handles input from the user
func HandleCommand(input string) {
- inputCmd := strings.Split(input, " ")[0]
- args := strings.Split(input, " ")[1:]
+ args := SplitCommandArgs(input)
+ inputCmd := args[0]
if _, ok := commands[inputCmd]; !ok {
- messenger.Error("Unkown command ", inputCmd)
+ messenger.Error("Unknown command ", inputCmd)
} else {
- commands[inputCmd](args)
+ commands[inputCmd].action(args[1:])
}
}