brian/helperbot
1
0
Fork 0
Code Issues 1 Pull Requests Releases Wiki Activity

Update to use nlopes-slack

Browse Source
This commit is contained in:
Brian Buller 2019-11-21 17:45:04 -06:00
parent 8909975cce
commit 52e0fca780
9 changed files with 77 additions and 113 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
cmd/app.go
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)
}

11
cmd/helper_plugin.go
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
}

2
cmd/helpers.go
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 {

13
cmd/main.go
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)
}

2
cmd/message.go
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 {

10
cmd/model.go
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
}

24
cmd/model_slack.go
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
}

2
interfaces.go
View File

@ -1,6 +1,6 @@
package helperbot
import slack "git.bullercodeworks.com/brian/go-slack"
import "github.com/nlopes/slack"
type Model interface {
SendMessage(src, dest string, message slack.Message)

111
plugins_src/plugin_aoc.go
View File

@ -14,8 +14,8 @@ import (
"time"
aoc "git.bullercodeworks.com/brian/go-adventofcode"
slack "git.bullercodeworks.com/brian/go-slack"
"git.bullercodeworks.com/brian/helperbot"
"github.com/nlopes/slack"
)
/* Plugin State */
@ -76,11 +76,7 @@ func (s *AoCState) ProcessMessage(m helperbot.Message) {
case 'D':
admin, err := s.model.GetSlackAdminDMId()
if err != nil {
s.model.SendMessage(s.Name(), "error", slack.Message{
Type: "error",
Text: "Error getting Admin DM Id",
Time: time.Now(),
})
s.SendAdminIdError()
return
}
if slackMsg.Channel == admin {
@ -129,11 +125,8 @@ func (s *AoCState) runLoop() {
}
time.Sleep(time.Minute)
}
s.model.SendMessage(s.Name(), "main", slack.Message{
Type: "status",
Text: "done",
Time: time.Now(),
})
s.model.SendMessage(s.Name(), "main", s.BuildMessage("status", "done", ""))
}
func (s *AoCState) ProcessDirectMessage(slackMsg slack.Message) {
@ -183,30 +176,18 @@ func (s *AoCState) DoHelpCmd(slackMsg slack.Message) {
txt := fmt.Sprint(":christmas_tree: AoC Help :christmas_tree:\n",
"-- WiP --",
)
s.model.SendMessage(s.Name(), "slack", slack.Message{
Type: "message",
Channel: slackMsg.Channel,
Text: txt,
})
s.SendSlackMessage(txt, slackMsg.Channel)
}
func (s *AoCState) DoHelpAdminCmd(slackMsg slack.Message) {
txt := fmt.Sprint(":christmas_tree: AoC Help :christmas_tree:\n",
"-- WiP --",
)
s.model.SendMessage(s.Name(), "slack", slack.Message{
Type: "message",
Channel: slackMsg.Channel,
Text: txt,
})
s.SendSlackMessage(txt, slackMsg.Channel)
}
func (s *AoCState) DoPingCmd(slackMsg slack.Message) {
s.model.SendMessage(s.Name(), "slack", slack.Message{
Type: "message",
Channel: slackMsg.Channel,
Text: ":christmas_tree: PONG :christmas_tree:",
})
s.SendSlackMessage(":christmas_tree: PONG :christmas_tree:", slackMsg.Channel)
}
func (s *AoCState) DoSessionCmd(slackMsg slack.Message) {
@ -215,39 +196,23 @@ func (s *AoCState) DoSessionCmd(slackMsg slack.Message) {
// Set the session cookie
admin, err := s.model.GetSlackAdminDMId()
if err != nil {
s.model.SendMessage(s.Name(), "error", slack.Message{
Type: "error",
Text: "Error getting Admin DM Id",
Time: time.Now(),
})
s.SendAdminIdError()
return
}
aocSession := msgPts[2]
s.setAoCSessionCookie(strings.TrimSpace(aocSession))
s.sessionCookie = aocSession
s.sessionNeedsUpdate = false
s.model.SendMessage(s.Name(), "slack", slack.Message{
Type: "message",
Channel: admin,
Text: ":christmas_tree: New Session: " + s.sessionCookie,
})
s.SendSlackMessage(":christmas_tree: New Session: "+s.sessionCookie, admin)
} else if len(msgPts) == 2 && msgPts[1] == "session" {
// Print the session cookie
admin, err := s.model.GetSlackAdminDMId()
if err != nil {
s.model.SendMessage(s.Name(), "error", slack.Message{
Type: "error",
Text: "Error getting Admin DM Id",
Time: time.Now(),
})
s.SendAdminIdError()
return
}
// We only send the session cookie to the admin
s.model.SendMessage(s.Name(), "slack", slack.Message{
Type: "message",
Channel: admin,
Text: ":christmas_tree: session: " + s.sessionCookie,
})
s.SendSlackMessage(":christmas_tree: session: "+s.sessionCookie, admin)
}
}
@ -275,11 +240,7 @@ func (s *AoCState) DoTopCmd(slackMsg slack.Message) {
} else {
txt = s.DoTopForYear(yr)
}
s.model.SendMessage(s.Name(), "slack", slack.Message{
Type: "message",
Channel: slackMsg.Channel,
Text: txt,
})
s.SendSlackMessage(txt, slackMsg.Channel)
}
func (s *AoCState) DoTopForAll() (string, error) {
@ -379,30 +340,18 @@ func (s *AoCState) AoCBoardCheckAndUpdate(yr int) {
if err != nil {
admin, adminErr := s.model.GetSlackAdminDMId()
if adminErr != nil {
s.model.SendMessage(s.Name(), "error", slack.Message{
Type: "error",
Text: "Error getting Admin DM Id",
Time: time.Now(),
})
s.SendAdminIdError()
return
}
if err.Error() == "Invalid Session Cookie" {
s.sessionNeedsUpdate = true
}
s.model.SendMessage(s.Name(), "slack", slack.Message{
Type: "message",
Channel: admin,
Text: fmt.Sprintf(":warning: AoC Error processing leaderboard (%d) - %s", yr, err.Error()),
})
s.SendSlackMessage(fmt.Sprintf(":warning: AoC Error processing leaderboard (%d) - %s", yr, err.Error()), admin)
return
}
msg := slack.Message{
Type: "success",
Text: fmt.Sprintf("Received leaderboard (%d)", yr),
Time: time.Now(),
}
s.model.SendMessage(s.Name(), "main", msg)
s.model.SendMessage(s.Name(), "main",
s.BuildMessage("success", fmt.Sprintf("Received leaderboard (%d)", yr), ""))
// Compare the new leaderboard to the saved one
for _, v := range l.Members {
mbr, err := s.getMember(l.Event, v.ID)
@ -410,11 +359,7 @@ func (s *AoCState) AoCBoardCheckAndUpdate(yr int) {
continue
}
if mbr.Stars != v.Stars {
s.model.SendMessage(s.Name(), "slack", slack.Message{
Type: "message",
Channel: channelId,
Text: ":christmas_tree: " + v.Name + " now has " + strconv.Itoa(v.Stars) + " stars! :christmas_tree:",
})
s.SendSlackMessage(":christmas_tree: "+v.Name+" now has "+strconv.Itoa(v.Stars)+" stars! :christmas_tree:", channelId)
}
}
// Save the leaderboard to the db
@ -499,11 +444,7 @@ func (s *AoCState) saveLeaderboard(l *aoc.Leaderboard) error {
}
for _, v := range l.Members {
if err = s.saveMember(l.Event, &v); err != nil {
s.model.SendMessage(s.Name(), "error", slack.Message{
Type: "error",
Text: fmt.Sprintf("Error Saving Member (%s)", v.Name),
Time: time.Now(),
})
s.model.SendMessage(s.Name(), "error", s.BuildMessage("error", fmt.Sprintf("Error Saving Member (%s)", v.Name), ""))
}
}
return nil
@ -579,3 +520,19 @@ func (s *AoCState) getMember(event string, memberId string) (*aoc.Member, error)
mbr.GlobalScore, err = strconv.Atoi(wrk)
return mbr, nil
}
func (s *AoCState) SendAdminIdError() {
s.model.SendMessage(s.Name(), "error", s.BuildMessage("error", "Error getting Admin DM Id", ""))
}
func (s *AoCState) SendSlackMessage(text, dest string) {
s.model.SendMessage(s.Name(), "slack", s.BuildMessage("message", text, dest))
}
func (s *AoCState) BuildMessage(tp, text, ch string) slack.Message {
ret := slack.Message{}
ret.Type = tp
ret.Text = text
ret.Channel = ch
return ret
}