X-Git-Url: https://git.lizzy.rs/?a=blobdiff_plain;f=cmd%2Fmicro%2Fplugin.go;h=04e7b78b27b448221b6c382551d3934753491a3a;hb=64d574c35cfdbd61c5abb456adf58da3ed328b1d;hp=c68faa0e463139d6e2a34925cfc0f3b509f19e2a;hpb=d2277a376a616b98f30c46f4cc641eaebd81b53b;p=micro.git

diff --git a/cmd/micro/plugin.go b/cmd/micro/plugin.go
index c68faa0e..04e7b78b 100644
--- a/cmd/micro/plugin.go
+++ b/cmd/micro/plugin.go
@@ -3,25 +3,33 @@ package main
 import (
 	"errors"
 	"io/ioutil"
+	"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",
-}
+var loadedPlugins map[string]string
 
 // Call calls the lua function 'function'
 // If it does not exist nothing happens, if there is an error,
 // the error is returned
-func Call(function string, args []string) error {
-	luaFunc := L.GetGlobal(function)
+func Call(function string, args ...interface{}) (lua.LValue, error) {
+	var luaFunc lua.LValue
+	if strings.Contains(function, ".") {
+		plugin := L.GetGlobal(strings.Split(function, ".")[0])
+		if plugin.String() == "nil" {
+			return nil, errors.New("function does not exist: " + function)
+		}
+		luaFunc = L.GetField(plugin, strings.Split(function, ".")[1])
+	} else {
+		luaFunc = L.GetGlobal(function)
+	}
+
 	if luaFunc.String() == "nil" {
-		return errors.New("function does not exist: " + function)
+		return nil, errors.New("function does not exist: " + function)
 	}
 	var luaArgs []lua.LValue
 	for _, v := range args {
@@ -29,19 +37,23 @@ func Call(function string, args []string) error {
 	}
 	err := L.CallByParam(lua.P{
 		Fn:      luaFunc,
-		NRet:    0,
+		NRet:    1,
 		Protect: true,
 	}, luaArgs...)
-	return err
+	ret := L.Get(-1) // returned value
+	if ret.String() != "nil" {
+		L.Pop(1) // remove received value
+	}
+	return ret, err
 }
 
 // LuaFunctionBinding is a function generator which takes the name of a lua function
 // and creates a function that will call that lua function
 // Specifically it creates a function that can be called as a binding because this is used
 // to bind keys to lua functions
-func LuaFunctionBinding(function string) func(*View) bool {
-	return func(v *View) bool {
-		err := Call(function, nil)
+func LuaFunctionBinding(function string) func(*View, bool) bool {
+	return func(v *View, _ bool) bool {
+		_, err := Call(function, nil)
 		if err != nil {
 			TermMessage(err)
 		}
@@ -49,56 +61,105 @@ func LuaFunctionBinding(function string) func(*View) 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" {
-					if err := L.DoFile(configDir + "/plugins/" + pluginName + "/" + f.Name()); 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 {
-		plugin := "runtime/plugins/" + pluginName + "/" + pluginName + ".lua"
-		data, err := Asset(plugin)
+		data, err := plugin.Data()
 		if err != nil {
-			TermMessage("Error loading pre-installed plugin: " + pluginName)
+			TermMessage("Error loading plugin: " + pluginName)
 			continue
 		}
-		if err := L.DoString(string(data)); err != nil {
+
+		pluginLuaName := luaPluginName(pluginName)
+
+		if err := LoadFile(pluginName, pluginName, string(data)); err != nil {
 			TermMessage(err)
 			continue
 		}
-		loadedPlugins = append(loadedPlugins, pluginName)
+
+		loadedPlugins[pluginName] = pluginLuaName
+
+	}
+
+	if _, err := os.Stat(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"
 	}
 }