Update to use nlopes-slack

This commit is contained in:
2019-11-21 17:45:04 -06:00
parent 8909975cce
commit 52e0fca780
9 changed files with 77 additions and 113 deletions

View File

@@ -23,12 +23,11 @@ func NewApp() (*App, error) {
}
a.DebugMode = DebugMode
a.running = true
err := a.initialize()
if err != nil {
a.running = false
return nil, err
}
a.running = true
go a.MonitorSlackMessages()
return a, nil
@@ -39,6 +38,10 @@ func (a *App) initialize() error {
if a.m, err = NewBotModel(); err != nil {
return err
}
bt, bterr := a.m.GetString([]string{"config", "plugin_dir"})
if bterr == nil {
fmt.Println(bt)
}
reader := bufio.NewReader(os.Stdin)
// Load up the plugins
@@ -65,12 +68,10 @@ func (a *App) initialize() error {
a.m.setSlackAdminDMId(strings.TrimSpace(slackDMid))
}
fmt.Println("Creating Slack Model")
if err = a.m.NewSlack(); err != nil {
return err
}
fmt.Println("Starting gofunc watchMessageChannel")
go a.watchMessageChannel()
return nil
}
@@ -79,10 +80,7 @@ func (a *App) watchMessageChannel() {
for a.running {
msg := <-a.m.messages
slackMsg := msg.GetMessage()
if slackMsg.Type != "user_typing" {
fmt.Println(slackMsg)
}
if slackMsg.Type == "control" && slackMsg.Name == "quit" {
if slackMsg.Type == "control" && slackMsg.Text == "quit" {
a.running = false
break
} else if msg.GetDestination() == "error" {
@@ -95,4 +93,5 @@ func (a *App) watchMessageChannel() {
v.State.ProcessMessage(msg)
}
}
close(a.m.messages)
}

View File

@@ -5,6 +5,7 @@ import (
"io/ioutil"
"os"
"plugin"
"strings"
"git.bullercodeworks.com/brian/helperbot"
)
@@ -24,22 +25,26 @@ func (a *App) LoadPluginsFromDirectory(dir string) 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.Printf("Error loading plugin (%s)\n", f.Name())
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.Printf("Error loading plugin (%s)\n", f.Name())
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)
fmt.Printf("Plugin loaded: %s\n", f.Name())
}
return nil
}

View File

@@ -3,7 +3,7 @@ package main
import (
"encoding/json"
slack "git.bullercodeworks.com/brian/go-slack"
"github.com/nlopes/slack"
)
func GetMessageJson(msg *slack.Message) string {

View File

@@ -7,7 +7,7 @@ import (
"syscall"
"time"
slack "git.bullercodeworks.com/brian/go-slack"
"github.com/nlopes/slack"
)
var DebugMode = false
@@ -26,6 +26,11 @@ func main() {
os.Exit(1)
}
// Monitor the Advent of Code Boards
//go m.MonitorAoCBoards()
// Monitor incoming Slack messages
//go m.MonitorSlackMessages()
// Set up a channel to intercept Ctrl+C for graceful shutdowns
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
@@ -33,11 +38,15 @@ func main() {
<-c
// Save the changes when the app quits
fmt.Println("\nFinishing up...")
a.m.messages <- NewBotMessage("main", "main", slack.Message{Type: "control", Name: "quit"})
msg := slack.Message{}
msg.Type = "control"
msg.Text = "quit"
a.m.messages <- NewBotMessage("main", "main", msg)
}()
for a.running {
time.Sleep(time.Second * 2)
}
fmt.Println("Model has stopped running")
fmt.Println("Done")
os.Exit(0)
}

View File

@@ -1,6 +1,6 @@
package main
import slack "git.bullercodeworks.com/brian/go-slack"
import "github.com/nlopes/slack"
// This message type is for communications over the messages channel
type BotMessage struct {

View File

@@ -6,9 +6,10 @@ import (
"os"
"strings"
slack "git.bullercodeworks.com/brian/go-slack"
goslack "git.bullercodeworks.com/brian/go-slack"
"git.bullercodeworks.com/brian/helperbot"
"github.com/br0xen/boltease"
"github.com/nlopes/slack"
)
type BotModel struct {
@@ -16,7 +17,7 @@ type BotModel struct {
messages chan helperbot.Message
slack *slack.Slack
slack *goslack.Slack
cache map[string][]byte
}
@@ -30,8 +31,6 @@ func NewBotModel() (*BotModel, error) {
if err != nil {
return nil, err
}
m.db.MkBucketPath([]string{"slack", "users"})
m.db.MkBucketPath([]string{"slack", "channels"})
return m, nil
}
@@ -93,6 +92,9 @@ func (m *BotModel) GetString(path []string) (string, error) {
if err != nil {
return "", err
}
if len(bts) == 0 {
return "", nil
}
return string(bts), nil
}

View File

@@ -1,11 +1,10 @@
package main
import (
slack "git.bullercodeworks.com/brian/go-slack"
goslack "git.bullercodeworks.com/brian/go-slack"
"github.com/nlopes/slack"
)
// TODO: Look into switching to: https://github.com/nlopes/slack
/* DB Functions */
func (m *BotModel) setSlackToken(token string) error {
return m.SetBytes([]string{"slack", "config", "token"}, []byte(token))
@@ -30,18 +29,16 @@ func (m *BotModel) NewSlack() error {
if err != nil {
return err
}
if m.slack, err = slack.CreateSlack(token); err != nil {
if m.slack, err = goslack.CreateSlack(token); err != nil {
return err
}
m.slack.StartRTM()
return nil
}
func (a *App) MonitorSlackMessages() {
for a.running {
msg, err := a.m.slack.GetMessage()
if err == nil {
a.m.SendMessage("slack", "main", msg)
}
for msg := range a.m.slack.IncomingMessages {
a.m.SendMessage("slack", "main", slack.Message(*msg))
}
}
@@ -50,10 +47,7 @@ func (m *BotModel) SendSlackChannelMessage(msg *slack.Message) error {
return m.SendSlackAdminMessage(msg)
}
// Send message to slack channel
var err error
if err = m.slack.PostMessage(*msg); err != nil {
return err
}
m.slack.PostMessage(msg)
return nil
}
@@ -64,8 +58,6 @@ func (m *BotModel) SendSlackAdminMessage(msg *slack.Message) error {
if err != nil {
return err
}
if err = m.slack.PostMessage(*msg); err != nil {
return err
}
m.slack.PostMessage(msg)
return nil
}