plugin manager qui fonctionne

plugin_manager/manager.go

@@ -7,6 +7,7 @@ package plugin_manager
 import (
 	tb "gopkg.in/tucnak/telebot.v2"
 	"log"
+	"strings"
 	"sync"
 )
 
@@ -38,6 +39,42 @@ func Init(_pluginDir string, bot *tb.Bot) {
 	}
 }
 
+func GetPluginList() []string {
+	var lst []string
+	for name, _ := range plugins {
+		lst = append(lst, name)
+	}
+	return lst
+}
+
+func IsPluginRunning(name string) bool {
+	if p, ok := plugins[name]; ok {
+		return p.running
+	}
+	return false
+}
+
+func ExistPlugin(name string) bool {
+	if _, ok := plugins[name]; ok {
+		return true
+	}
+	return false
+}
+
+func GetPlugin(name string) Plugin {
+	if p, ok := plugins[name]; ok {
+		return p
+	}
+	return nil
+}
+
+func IsPluginEnable(name string) bool {
+	if p, ok := plugins[name]; ok {
+		return p.enable
+	}
+	return false
+}
+
 func EnablePlugin(name string, enable bool) {
 	if p, ok := plugins[name]; ok {
 		if enable != p.enable {
@@ -69,8 +106,16 @@ func StopPlugins() {
 
 func HandleMessage(msg *tb.Message) {
 	for _, val := range plugins {
-		if val.enable {
-			val.plugin.HandleMessage(context.bot, msg.Text)
+		if val.enable && val.running {
+			if strings.HasPrefix(msg.Text, "/") {
+				split := strings.Split(msg.Text, " ")
+				split[0] = split[0][1:]
+				if Contains(split[0], ExecGetCommands(val.plugin)) {
+					ExecHandleCommand(val.plugin, context.bot, msg, split[0], split[1:])
+				}
+			} else {
+				ExecHandleMessage(val.plugin, context.bot, msg)
+			}
 		}
 	}
 }
@@ -88,9 +133,8 @@ func startPlugin(name string) {
 	if p, ok := plugins[name]; ok {
 		//p.mux.Lock()
 		if !p.running && p.enable {
-			p.running = true
+			p.running = ExecLoad(p.plugin)
 			plugins[name] = p
-			p.plugin.Load()
 		}
 		//p.mux.Unlock()
 	}
@@ -102,7 +146,7 @@ func stopPlugin(name string) {
 		if p.running {
 			p.running = false
 			plugins[name] = p
-			p.plugin.Unload()
+			ExecUnload(p.plugin)
 		}
 		//p.mux.Unlock()
 	}
@@ -111,7 +155,7 @@ func stopPlugin(name string) {
 func Exit() {
 	for _, p := range plugins {
 		if p.running {
-			p.plugin.Unload()
+			ExecUnload(p.plugin)
 		}
 	}
 }

plugin_manager/tools.go

@@ -12,10 +12,32 @@ import (
 	"strings"
 )
 
-type Plugin interface {
-	Load()
-	HandleMessage(bot *tb.Bot, msg string)
-	Unload()
+type TestGetCommands interface {
+	GetCommands() []string
+}
+
+type TestLoad interface {
+	Load() bool
+}
+
+type TestHandleMessage interface {
+	HandleMessage(bot *tb.Bot, msg *tb.Message)
+}
+
+type TestHandleCommand interface {
+	HandleCommand(bot *tb.Bot, msg *tb.Message, cmd string, args []string)
+}
+
+type TestUnload interface {
+	Unload() bool
+}
+
+type Plugin interface { /*
+		GetCommandes() []string
+		Load() bool
+		HandleMessage(bot *tb.Bot, msg *tb.Message)
+		HandleCommand(bot *tb.Bot, msg *tb.Message, cmd string, args []string)
+		Unload() bool*/
 }
 
 func GetSoFiles(dir string) []string {
@@ -48,6 +70,7 @@ func LoadSoFile(file string) Plugin {
 	}
 
 	var plugin Plugin
+	//plugin, _ = symPlugin.(Plugin)
 	plugin, ok := symPlugin.(Plugin)
 	if !ok {
 		log.Fatal(file + ": unexpected type from module symbol")
@@ -55,3 +78,50 @@ func LoadSoFile(file string) Plugin {
 	}
 	return plugin
 }
+
+func ExecGetCommands(plugin Plugin) []string {
+	p, ok := plugin.(TestGetCommands)
+	if ok {
+		return p.GetCommands()
+	}
+	return []string{}
+}
+
+func ExecLoad(plugin Plugin) bool {
+	p, ok := plugin.(TestLoad)
+	if ok {
+		return p.Load()
+	}
+	return true
+}
+
+func ExecHandleMessage(plugin Plugin, bot *tb.Bot, msg *tb.Message) {
+	p, ok := plugin.(TestHandleMessage)
+	if ok {
+		p.HandleMessage(bot, msg)
+	}
+}
+
+func ExecHandleCommand(plugin Plugin, bot *tb.Bot, msg *tb.Message, cmd string, args []string) {
+	p, ok := plugin.(TestHandleCommand)
+	if ok {
+		p.HandleCommand(bot, msg, cmd, args)
+	}
+}
+
+func ExecUnload(plugin Plugin) bool {
+	p, ok := plugin.(TestUnload)
+	if ok {
+		return p.Unload()
+	}
+	return true
+}
+
+func Contains(e string, s []string) bool {
+	for _, a := range s {
+		if a == e {
+			return true
+		}
+	}
+	return false
+}