X-Git-Url: https://git.lizzy.rs/?a=blobdiff_plain;f=cmd%2Fmicro%2Fplugin.go;h=04e7b78b27b448221b6c382551d3934753491a3a;hb=64d574c35cfdbd61c5abb456adf58da3ed328b1d;hp=97e9c6b43c34d5b80c10e2dc188b8d5c06dff758;hpb=0711c29c0a1a68604ea3384fc59c202dedbe8d76;p=micro.git diff --git a/cmd/micro/plugin.go b/cmd/micro/plugin.go index 97e9c6b4..04e7b78b 100644 --- a/cmd/micro/plugin.go +++ b/cmd/micro/plugin.go @@ -6,17 +6,12 @@ import ( "os" "strings" - "github.com/layeh/gopher-luar" "github.com/yuin/gopher-lua" + "github.com/zyedidia/tcell" + "layeh.com/gopher-luar" ) -var loadedPlugins []string - -var preInstalledPlugins = []string{ - "go", - "linter", - "autoclose", -} +var loadedPlugins map[string]string // Call calls the lua function 'function' // If it does not exist nothing happens, if there is an error, @@ -66,76 +61,105 @@ func LuaFunctionBinding(function string) func(*View, bool) bool { } } +func LuaFunctionMouseBinding(function string) func(*View, bool, *tcell.EventMouse) bool { + return func(v *View, _ bool, e *tcell.EventMouse) bool { + _, err := Call(function, e) + if err != nil { + TermMessage(err) + } + return false + } +} + +func unpack(old []string) []interface{} { + new := make([]interface{}, len(old)) + for i, v := range old { + new[i] = v + } + return new +} + // LuaFunctionCommand is the same as LuaFunctionBinding except it returns a normal function // so that a command can be bound to a lua function func LuaFunctionCommand(function string) func([]string) { return func(args []string) { - _, err := Call(function, args) + _, err := Call(function, unpack(args)...) + if err != nil { + TermMessage(err) + } + } +} + +// LuaFunctionComplete returns a function which can be used for autocomplete in plugins +func LuaFunctionComplete(function string) func(string) []string { + return func(input string) (result []string) { + + res, err := Call(function, input) if err != nil { TermMessage(err) } + if tbl, ok := res.(*lua.LTable); !ok { + TermMessage(function, "should return a table of strings") + } else { + for i := 1; i <= tbl.Len(); i++ { + val := tbl.RawGetInt(i) + if v, ok := val.(lua.LString); !ok { + TermMessage(function, "should return a table of strings") + } else { + result = append(result, string(v)) + } + } + } + return result } } func LuaFunctionJob(function string) func(string, ...string) { return func(output string, args ...string) { - _, err := Call(function, append([]string{output}, args...)) + _, err := Call(function, unpack(append([]string{output}, args...))...) if err != nil { TermMessage(err) } } } +// luaPluginName convert a human-friendly plugin name into a valid lua variable name. +func luaPluginName(name string) string { + return strings.Replace(name, "-", "_", -1) +} + // LoadPlugins loads the pre-installed plugins and the plugins located in ~/.config/micro/plugins func LoadPlugins() { - files, _ := ioutil.ReadDir(configDir + "/plugins") - for _, plugin := range files { - if plugin.IsDir() { - pluginName := plugin.Name() - files, _ := ioutil.ReadDir(configDir + "/plugins/" + pluginName) - for _, f := range files { - if f.Name() == pluginName+".lua" { - data, _ := ioutil.ReadFile(configDir + "/plugins/" + pluginName + "/" + f.Name()) - pluginDef := "\nlocal P = {}\n" + pluginName + " = P\nsetmetatable(" + pluginName + ", {__index = _G})\nsetfenv(1, P)\n" - - if err := L.DoString(pluginDef + string(data)); err != nil { - TermMessage(err) - continue - } - loadedPlugins = append(loadedPlugins, pluginName) - } - } + loadedPlugins = make(map[string]string) + + for _, plugin := range ListRuntimeFiles(RTPlugin) { + pluginName := plugin.Name() + if _, ok := loadedPlugins[pluginName]; ok { + continue } - } - for _, pluginName := range preInstalledPlugins { - alreadyExists := false - for _, pl := range loadedPlugins { - if pl == pluginName { - alreadyExists = true - break - } + data, err := plugin.Data() + if err != nil { + TermMessage("Error loading plugin: " + pluginName) + continue } - if !alreadyExists { - plugin := "runtime/plugins/" + pluginName + "/" + pluginName + ".lua" - data, err := Asset(plugin) - if err != nil { - TermMessage("Error loading pre-installed plugin: " + pluginName) - continue - } - pluginDef := "\nlocal P = {}\n" + pluginName + " = P\nsetmetatable(" + pluginName + ", {__index = _G})\nsetfenv(1, P)\n" - if err := L.DoString(pluginDef + string(data)); err != nil { - TermMessage(err) - continue - } - loadedPlugins = append(loadedPlugins, pluginName) + pluginLuaName := luaPluginName(pluginName) + + if err := LoadFile(pluginName, pluginName, string(data)); err != nil { + TermMessage(err) + continue } + + loadedPlugins[pluginName] = pluginLuaName + } if _, err := os.Stat(configDir + "/init.lua"); err == nil { - if err := L.DoFile(configDir + "/init.lua"); err != nil { + data, _ := ioutil.ReadFile(configDir + "/init.lua") + if err := LoadFile("init", configDir+"init.lua", string(data)); err != nil { TermMessage(err) } + loadedPlugins["init"] = "init" } }