mirror of https://github.com/velour/catbase.git
118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package tell
|
|
|
|
import (
|
|
"fmt"
|
|
bh "github.com/timshannon/bolthold"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"github.com/velour/catbase/bot"
|
|
"github.com/velour/catbase/bot/msg"
|
|
)
|
|
|
|
type delayedMsg string
|
|
|
|
type TellPlugin struct {
|
|
b bot.Bot
|
|
store *bh.Store
|
|
}
|
|
|
|
func New(b bot.Bot) *TellPlugin {
|
|
tp := &TellPlugin{b, b.Store()}
|
|
b.Register(tp, bot.Message, tp.message)
|
|
return tp
|
|
}
|
|
|
|
type tell struct {
|
|
ID int `boltholdIndex:"ID"`
|
|
Who string
|
|
What string
|
|
}
|
|
|
|
func (t *TellPlugin) getTells() []tell {
|
|
result := []tell{}
|
|
t.store.Find(&result, &bh.Query{})
|
|
return result
|
|
}
|
|
|
|
func (t *TellPlugin) rmTell(entry tell) {
|
|
if err := t.store.Delete(entry.ID, tell{}); err != nil {
|
|
log.Error().Err(err).Msg("could not remove tell")
|
|
}
|
|
}
|
|
|
|
func (t *TellPlugin) addTell(who, what string) error {
|
|
tell := tell{Who: who, What: what}
|
|
err := t.store.Insert(bh.NextSequence(), tell)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("could not add tell")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (t *TellPlugin) check(who string) []tell {
|
|
result := []tell{}
|
|
tells := t.getTells()
|
|
for _, e := range tells {
|
|
if e.Who == who {
|
|
result = append(result, e)
|
|
t.rmTell(e)
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
func (t *TellPlugin) checkValidTarget(ch, target string) bool {
|
|
users := t.b.Who(ch)
|
|
log.Debug().
|
|
Str("ch", ch).
|
|
Str("target", target).
|
|
Interface("users", users).
|
|
Msg("checking valid target")
|
|
for _, u := range users {
|
|
if u.Name == target {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (t *TellPlugin) troll(who string) bool {
|
|
targets := t.b.Config().GetArray("tell.troll", []string{})
|
|
for _, target := range targets {
|
|
if who == target {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (t *TellPlugin) message(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
|
|
if strings.HasPrefix(strings.ToLower(message.Body), "tell ") ||
|
|
strings.HasPrefix(strings.ToLower(message.Body), "tellah ") {
|
|
parts := strings.Split(message.Body, " ")
|
|
target := strings.ToLower(parts[1])
|
|
if !t.checkValidTarget(message.Channel, target) {
|
|
if t.troll(message.User.Name) {
|
|
t.b.Send(c, bot.Message, message.Channel, fmt.Sprintf("Okay. I'll tell %s.", target))
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
newMessage := strings.Join(parts[2:], " ")
|
|
newMessage = fmt.Sprintf("Hey, %s. %s said: %s", target, message.User.Name, newMessage)
|
|
t.addTell(target, newMessage)
|
|
t.b.Send(c, bot.Message, message.Channel, fmt.Sprintf("Okay. I'll tell %s.", target))
|
|
return true
|
|
}
|
|
uname := strings.ToLower(message.User.Name)
|
|
if tells := t.check(uname); len(tells) > 0 {
|
|
for _, m := range tells {
|
|
t.b.Send(c, bot.Message, message.Channel, m.What)
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|