twitch-bot/actions.go

80 lines
1.9 KiB
Go
Raw Normal View History

2020-12-21 00:32:39 +00:00
package main
import (
"sync"
"github.com/Luzifer/twitch-bot/plugins"
2020-12-21 00:32:39 +00:00
"github.com/go-irc/irc"
"github.com/pkg/errors"
log "github.com/sirupsen/logrus"
)
var (
availableActions []plugins.ActorCreationFunc
2020-12-21 00:32:39 +00:00
availableActionsLock = new(sync.RWMutex)
)
// Compile-time assertion
var _ plugins.ActorRegistrationFunc = registerAction
func registerAction(af plugins.ActorCreationFunc) {
2020-12-21 00:32:39 +00:00
availableActionsLock.Lock()
defer availableActionsLock.Unlock()
availableActions = append(availableActions, af)
}
func triggerActions(c *irc.Client, m *irc.Message, rule *plugins.Rule, ra *plugins.RuleAction, eventData plugins.FieldCollection) (preventCooldown bool, err error) {
2020-12-21 00:32:39 +00:00
availableActionsLock.RLock()
defer availableActionsLock.RUnlock()
for _, acf := range availableActions {
var (
a = acf()
logger = log.WithField("actor", a.Name())
)
if err := ra.Unmarshal(a); err != nil {
logger.WithError(err).Trace("Unable to unmarshal config")
continue
}
if a.IsAsync() {
go func() {
if _, err := a.Execute(c, m, rule, eventData); err != nil {
logger.WithError(err).Error("Error in async actor")
}
}()
continue
}
apc, err := a.Execute(c, m, rule, eventData)
preventCooldown = preventCooldown || apc
if err != nil {
return preventCooldown, errors.Wrap(err, "execute action")
2020-12-21 00:32:39 +00:00
}
}
return preventCooldown, nil
2020-12-21 00:32:39 +00:00
}
func handleMessage(c *irc.Client, m *irc.Message, event *string, eventData plugins.FieldCollection) {
for _, r := range config.GetMatchingRules(m, event, eventData) {
var preventCooldown bool
2020-12-21 00:32:39 +00:00
for _, a := range r.Actions {
apc, err := triggerActions(c, m, r, a, eventData)
if err != nil {
2020-12-21 00:32:39 +00:00
log.WithError(err).Error("Unable to trigger action")
break // Break execution when one action fails
2020-12-21 00:32:39 +00:00
}
preventCooldown = preventCooldown || apc
2020-12-21 00:32:39 +00:00
}
// Lock command
if !preventCooldown {
r.SetCooldown(timerStore, m, eventData)
}
2020-12-21 00:32:39 +00:00
}
}