X-Git-Url: https://git.lizzy.rs/?a=blobdiff_plain;f=cmd%2Fmicro%2Fcommand.go;h=ebb7936551d82353829d5c5fdbed19fe892d9510;hb=015fcf5fec1f69d802438acef098e6d41a8be85a;hp=3c5baa264e48cdc074c9228ea026e45ba75682c1;hpb=12a4dd58f32f809641cf6faf25a2146fe8da83df;p=micro.git diff --git a/cmd/micro/command.go b/cmd/micro/command.go index 3c5baa26..ebb79365 100644 --- a/cmd/micro/command.go +++ b/cmd/micro/command.go @@ -1,12 +1,8 @@ package main import ( - "bytes" "fmt" - "io" "os" - "os/exec" - "os/signal" "path/filepath" "regexp" "runtime" @@ -14,6 +10,7 @@ import ( "strings" humanize "github.com/dustin/go-humanize" + "github.com/zyedidia/micro/cmd/micro/shellwords" ) // A Command contains a action (a function to call) as well as information about how to autocomplete the command @@ -37,6 +34,7 @@ func init() { "Set": Set, "SetLocal": SetLocal, "Show": Show, + "ShowKey": ShowKey, "Run": Run, "Bind": Bind, "Quit": Quit, @@ -55,7 +53,10 @@ func init() { "Pwd": Pwd, "Open": Open, "TabSwitch": TabSwitch, + "Term": Term, "MemUsage": MemUsage, + "Retab": Retab, + "Raw": Raw, } } @@ -90,8 +91,9 @@ func MakeCommand(name, function string, completions ...Completion) { func DefaultCommands() map[string]StrCommand { return map[string]StrCommand{ "set": {"Set", []Completion{OptionCompletion, OptionValueCompletion}}, - "setlocal": {"SetLocal", []Completion{OptionCompletion, NoCompletion}}, + "setlocal": {"SetLocal", []Completion{OptionCompletion, OptionValueCompletion}}, "show": {"Show", []Completion{OptionCompletion, NoCompletion}}, + "showkey": {"ShowKey", []Completion{NoCompletion}}, "bind": {"Bind", []Completion{NoCompletion}}, "run": {"Run", []Completion{NoCompletion}}, "quit": {"Quit", []Completion{NoCompletion}}, @@ -110,7 +112,32 @@ func DefaultCommands() map[string]StrCommand { "pwd": {"Pwd", []Completion{NoCompletion}}, "open": {"Open", []Completion{FileCompletion}}, "tabswitch": {"TabSwitch", []Completion{NoCompletion}}, + "term": {"Term", []Completion{NoCompletion}}, "memusage": {"MemUsage", []Completion{NoCompletion}}, + "retab": {"Retab", []Completion{NoCompletion}}, + "raw": {"Raw", []Completion{NoCompletion}}, + } +} + +// CommandEditAction returns a bindable function that opens a prompt with +// the given string and executes the command when the user presses +// enter +func CommandEditAction(prompt string) func(*View, bool) bool { + return func(v *View, usePlugin bool) bool { + input, canceled := messenger.Prompt("> ", prompt, "Command", CommandCompletion) + if !canceled { + HandleCommand(input) + } + return false + } +} + +// CommandAction returns a bindable function which executes the +// given command +func CommandAction(cmd string) func(*View, bool) bool { + return func(v *View, usePlugin bool) bool { + HandleCommand(cmd) + return false } } @@ -149,7 +176,7 @@ func PluginCmd(args []string) { continue } } - if !IsSpaces(removed) { + if !IsSpaces([]byte(removed)) { messenger.Message("Removed ", removed) } else { messenger.Error("The requested plugins do not exist") @@ -197,6 +224,34 @@ func PluginCmd(args []string) { } } +// Retab changes all spaces to tabs or all tabs to spaces +// depending on the user's settings +func Retab(args []string) { + CurView().Retab(true) +} + +// Raw opens a new raw view which displays the escape sequences micro +// is receiving in real-time +func Raw(args []string) { + buf := NewBufferFromString("", "Raw events") + + view := NewView(buf) + view.Buf.Insert(view.Cursor.Loc, "Warning: Showing raw event escape codes\n") + view.Buf.Insert(view.Cursor.Loc, "Use CtrlQ to exit\n") + view.Type = vtRaw + tab := NewTabFromView(view) + tab.SetNum(len(tabs)) + tabs = append(tabs, tab) + curTab = len(tabs) - 1 + if len(tabs) == 2 { + for _, t := range tabs { + for _, v := range t.views { + v.ToggleTabbar() + } + } + } +} + // TabSwitch switches to a given tab either by name or by number func TabSwitch(args []string) { if len(args) > 0 { @@ -230,10 +285,18 @@ func TabSwitch(args []string) { func Cd(args []string) { if len(args) > 0 { path := ReplaceHome(args[0]) - os.Chdir(path) + err := os.Chdir(path) + if err != nil { + messenger.Error("Error with cd: ", err) + return + } + wd, _ := os.Getwd() for _, tab := range tabs { for _, view := range tab.views { - wd, _ := os.Getwd() + if len(view.Buf.name) == 0 { + continue + } + 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 @@ -272,7 +335,12 @@ 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), " ") + args, err := shellwords.Split(filename) + if err != nil { + messenger.Error("Error parsing args ", err) + return + } + filename = strings.Join(args, " ") CurView().Open(filename) } else { @@ -470,6 +538,20 @@ func Show(args []string) { messenger.Message(option) } +// ShowKey displays the action that a key is bound to +func ShowKey(args []string) { + if len(args) < 1 { + messenger.Error("Please provide a key to show") + return + } + + if action, ok := bindingsStr[args[0]]; ok { + messenger.Message(action) + } else { + messenger.Message(args[0], " has no binding") + } +} + // Bind creates a new keybinding func Bind(args []string) { if len(args) < 2 { @@ -482,7 +564,7 @@ func Bind(args []string) { // Run runs a shell command in the background func Run(args []string) { // Run a shell command in the background (openTerm is false) - HandleShellCommand(JoinCommandArgs(args...), false, true) + HandleShellCommand(shellwords.Join(args...), false, true) } // Quit closes the main view @@ -503,24 +585,35 @@ func Save(args []string) { // Replace runs search and replace func Replace(args []string) { - if len(args) < 2 || len(args) > 3 { + if len(args) < 2 || len(args) > 4 { // We need to find both a search and replace expression messenger.Error("Invalid replace statement: " + strings.Join(args, " ")) return } - allAtOnce := false - if len(args) == 3 { - // user added -a flag - if args[2] == "-a" { - allAtOnce = true - } else { - messenger.Error("Invalid replace flag: " + args[2]) - return + all := false + noRegex := false + + if len(args) > 2 { + for _, arg := range args[2:] { + switch arg { + case "-a": + all = true + case "-l": + noRegex = true + default: + messenger.Error("Invalid flag: " + arg) + return + } } } search := string(args[0]) + + if noRegex { + search = regexp.QuoteMeta(search) + } + replace := string(args[1]) regex, err := regexp.Compile("(?m)" + search) @@ -556,7 +649,7 @@ func Replace(args []string) { view.Buf.MultipleReplace(deltas) } - if allAtOnce { + if all { replaceAll() } else { for { @@ -614,93 +707,27 @@ func ReplaceAll(args []string) { Replace(append(args, "-a")) } -// RunShellCommand executes a shell command and returns the output/error -func RunShellCommand(input string) (string, error) { - inputCmd := SplitCommandArgs(input)[0] - args := SplitCommandArgs(input)[1:] - - cmd := exec.Command(inputCmd, args...) - outputBytes := &bytes.Buffer{} - cmd.Stdout = outputBytes - cmd.Stderr = outputBytes - cmd.Start() - err := cmd.Wait() // wait for command to finish - outstring := outputBytes.String() - return outstring, err -} - -// 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, 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...") - go func() { - output, err := RunShellCommand(input) - totalLines := strings.Split(output, "\n") - - if len(totalLines) < 3 { - if err == nil { - messenger.Message(inputCmd, " exited without error") - } else { - messenger.Message(inputCmd, " exited with error: ", err, ": ", output) - } - } else { - messenger.Message(output) - } - // We have to make sure to redraw - RedrawAll() - }() +// Term opens a terminal in the current view +func Term(args []string) { + var err error + if len(args) == 0 { + err = CurView().StartTerminal([]string{os.Getenv("SHELL"), "-i"}, true, false, "") } else { - // Shut down the screen because we're going to interact directly with the shell - screen.Fini() - screen = nil - - 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 = io.MultiWriter(os.Stdout, &outputBuf) - cmd.Stderr = os.Stderr - - // This is a trap for Ctrl-C so that it doesn't kill micro - // Instead we trap Ctrl-C to kill the program we're running - c := make(chan os.Signal, 1) - signal.Notify(c, os.Interrupt) - go func() { - for range c { - cmd.Process.Kill() - } - }() - - cmd.Start() - err := cmd.Wait() - - output := outputBuf.String() - if err != nil { - output = err.Error() - } - - 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 + err = CurView().StartTerminal(args, true, false, "") + } + if err != nil { + messenger.Error(err) } - return "" } // HandleCommand handles input from the user func HandleCommand(input string) { - args := SplitCommandArgs(input) + args, err := shellwords.Split(input) + if err != nil { + messenger.Error("Error parsing args ", err) + return + } + inputCmd := args[0] if _, ok := commands[inputCmd]; !ok {