66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package handler
|
|
|
|
import (
|
|
"math/rand"
|
|
"strings"
|
|
|
|
"git.kill0.net/chill9/beepboop/bot"
|
|
"git.kill0.net/chill9/beepboop/lib"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type (
|
|
ReactionHandler struct {
|
|
Config bot.Config
|
|
}
|
|
)
|
|
|
|
func NewReactionHandler() *ReactionHandler {
|
|
return new(ReactionHandler)
|
|
}
|
|
|
|
func (h *ReactionHandler) SetConfig(config bot.Config) {
|
|
h.Config = config
|
|
}
|
|
|
|
func (h *ReactionHandler) Handle(s *discordgo.Session, m *discordgo.MessageCreate) {
|
|
if m.Author.ID == s.State.User.ID {
|
|
return
|
|
}
|
|
|
|
emojis := h.Config.Handler.Reaction.Emojis
|
|
channels := h.Config.Handler.Reaction.Channels
|
|
|
|
if len(emojis) == 0 {
|
|
log.Warning("emoji list is empty")
|
|
return
|
|
}
|
|
|
|
channel, err := s.Channel(m.ChannelID)
|
|
if err != nil {
|
|
log.Fatalf("unable to get channel name: %v", err)
|
|
}
|
|
|
|
if len(channels) > 0 && !lib.Contains(channels, channel.Name) {
|
|
return
|
|
}
|
|
|
|
for _, a := range m.Attachments {
|
|
if strings.HasPrefix(a.ContentType, "image/") {
|
|
for i := 1; i <= lib.RandInt(1, len(emojis)); i++ {
|
|
r := emojis[rand.Intn(len(emojis))]
|
|
s.MessageReactionAdd(m.ChannelID, m.ID, r)
|
|
}
|
|
}
|
|
}
|
|
|
|
for range m.Embeds {
|
|
for i := 1; i <= lib.RandInt(1, len(emojis)); i++ {
|
|
r := emojis[rand.Intn(len(emojis))]
|
|
s.MessageReactionAdd(m.ChannelID, m.ID, r)
|
|
}
|
|
}
|
|
}
|