Helperbot V2

app/app.go

@@ -0,0 +1,99 @@
+package app
+
+import (
+	"fmt"
+	"os"
+
+	"git.bullercodeworks.com/brian/helperbot/models"
+	"github.com/spf13/viper"
+)
+
+type App struct {
+	debug   bool
+	Running bool
+
+	m *models.BotModel
+
+	plugins []HelperPlugin
+}
+
+func NewApp(debugMode bool) (*App, error) {
+	a := new(App)
+	if debugMode {
+		fmt.Println("Running in Debug Mode. All messages will be sent to Admin DM")
+	}
+	a.debug = debugMode
+
+	err := a.initialize()
+	if err != nil {
+		return nil, err
+	}
+	a.Running = true
+	return a, err
+}
+
+func (a *App) initialize() error {
+	var err error
+	viper.SetConfigName("config")
+	viper.SetConfigType("yaml")
+	viper.AddConfigPath(".")
+	if err = viper.ReadInConfig(); err != nil {
+		fmt.Println("Config Read Error...")
+		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
+			fmt.Println("Config not found, creating")
+			// Config file not found; create it
+			viper.WriteConfigAs("config.yaml")
+		} else {
+			// Config file was found but another error was produced
+			fmt.Println("Other config err:", err)
+			return err
+		}
+	}
+
+	if a.m, err = models.NewBotModel(a.debug); err != nil {
+		return err
+	}
+
+	// Load up the plugins
+	pluginDir := a.m.GetPluginDir()
+	a.loadPluginsFromDirectory(pluginDir)
+	if err != nil {
+		fmt.Println("Error loading plugins")
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	a.setupMessageWatchers()
+
+	a.m.SendSlackAdminMessage(":robot_face: Helperbot Initialized :robot_face:")
+	return err
+}
+
+func (a *App) SendQuitMessage() { a.m.SendMessage(models.QuitMessage) }
+
+func (a *App) setupMessageWatchers() {
+	a.m.AddMessageWatcher(func(msg models.BotMessage) bool {
+		if msg.Is(models.QuitMessage) {
+			// App is shutting down
+			a.Running = false
+		} else if msg.IsError() {
+			// Received an Error Message
+			fmt.Printf("ERROR: %s\n", msg)
+		} else if msg.Dest == models.MsgSrcSlack {
+			// Sending a message to a slack channel
+			a.m.SendSlackChannelMessage(msg.Text, msg.Target)
+			return true
+		}
+		return false
+	})
+
+	for _, v := range a.plugins {
+		a.m.AddMessageWatcher(v.State.ProcessMessage)
+		a.m.AddRTMWatcher(v.State.ProcessRTMEvent)
+	}
+}
+
+func (a *App) watchMessageChannel() {
+	for a.Running {
+		a.m.ProcessMessageChannel()
+	}
+}

app/plugins.go

@@ -0,0 +1,61 @@
+package app
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"plugin"
+	"strings"
+
+	"git.bullercodeworks.com/brian/helperbot/models"
+)
+
+type HelperPlugin struct {
+	p     *plugin.Plugin
+	State models.PluginState
+}
+
+func (a *App) loadPluginsFromDirectory(dir string) error {
+	fmt.Println("Loading Plugins (", dir, ")")
+	files, err := ioutil.ReadDir(dir)
+	if err != nil {
+		fmt.Println("Error loading plugins")
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	for _, f := range files {
+		if !strings.HasSuffix(f.Name(), ".so") {
+			fmt.Printf("Skipping file (%s)\n", f.Name())
+			continue
+		}
+		p, err := plugin.Open(dir + f.Name())
+		if err != nil {
+			fmt.Println(fmt.Sprintf("Error loading plugin (%s)\n", f.Name()))
+			fmt.Println(err.Error())
+			os.Exit(1)
+		}
+		hp, err := NewHelperPlugin(p)
+		if err != nil {
+			fmt.Println(fmt.Sprintf("Error loading plugin (%s)\n", f.Name()))
+			fmt.Println(err.Error())
+			os.Exit(1)
+		}
+		hp.State.Initialize(a.m)
+		hp.State.Run()
+		fmt.Printf("Plugin Loaded (%s)\n", f.Name())
+		a.plugins = append(a.plugins, *hp)
+	}
+	return nil
+}
+
+func NewHelperPlugin(p *plugin.Plugin) (*HelperPlugin, error) {
+	h := &HelperPlugin{p: p}
+
+	// Parse the plugin's state
+	pluginStateSymbol, err := p.Lookup("State")
+	if err != nil {
+		return nil, err
+	}
+	h.State = pluginStateSymbol.(models.PluginState)
+	return h, nil
+}