bb/bot/reaction.go

package bot

import (
	"math/rand"
	"strings"

	"git.kill0.net/chill9/beepboop/lib"

	"github.com/bwmarrin/discordgo"
	log "github.com/sirupsen/logrus"
)

func (b *Bot) ReactionHandler() func(*discordgo.Session, *discordgo.MessageCreate) {
	return func(s *discordgo.Session, m *discordgo.MessageCreate) {
		if m.Author.ID == s.State.User.ID {
			return
		}

		emojis := b.Config.Handler.Reaction.Emojis
		channels := b.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)
			}
		}
	}
}