aboutsummaryrefslogtreecommitdiff
path: root/rolereact/manager.go
blob: 3e0101786df433c01ced9726e4b694df8ae00557 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package rolereact

import (
	"context"
	"fmt"
	"slices"
	"strings"
	"time"

	"git.anhgelus.world/anhgelus/les-copaings-bot/common"
	"git.anhgelus.world/anhgelus/les-copaings-bot/config"
	"github.com/nyttikord/gokord/bot"
	"github.com/nyttikord/gokord/channel"
	"github.com/nyttikord/gokord/emoji"
	"github.com/nyttikord/gokord/event"
	"github.com/nyttikord/gokord/interaction"
)

func MessageContent(message *config.RoleReactMessage) string {
	content := "## Réagis pour obtenir un rôle"
	if message.Note != "" {
		content = fmt.Sprintf("%s\n%s", content, message.Note)
	}
	for _, role := range message.Roles {
		if role.Reaction != "" && role.RoleID != 0 {
			content += fmt.Sprintf("\n> -# %s <@&%d>", FormatEmoji(role.Reaction), role.RoleID)
		}
	}
	if len(message.Roles) == 0 {
		content += "\n*Pas de rôles pour le moment*"
	}
	return content
}

func ApplyMessageChange(ctx context.Context, dg bot.Session, i *interaction.Interaction, message *config.RoleReactMessage) string {
	messageContent := MessageContent(message)
	_, err := channel.EditMessageComplex(&channel.MessageEdit{
		Content:         &messageContent,
		AllowedMentions: &channel.MessageAllowedMentions{},
		Channel:         message.ChannelID,
		ID:              message.MessageID,
	}).Do(ctx)
	if err != nil {
		bot.Logger(ctx).Error("unable to update rolereact message", "error", err)
		return "Impossible de mettre à jour le message."
	}
	for _, role := range message.Roles {
		if role.Reaction != "" && role.RoleID != 0 && err == nil {
			err = channel.AddReaction(message.ChannelID, message.MessageID, role.Reaction).Do(ctx)
		}
	}
	if err != nil {
		bot.Logger(ctx).Error("unable to update reactions on rolereact message", "error", err)
		return "Impossible de mettre à jour le message."
	}
	cfg := config.GetGuild(ctx, i.GuildID)
	messageIndex := slices.IndexFunc(cfg.RrMessages, func(m *config.RoleReactMessage) bool {
		return m.MessageID == message.MessageID
	})
	if messageIndex == -1 {
		// should log it?
		return "Impossible de mmettre à jour le message."
	}
	oldMessage := cfg.RrMessages[messageIndex]
	roles := make(map[uint]config.RoleReact, len(message.Roles))
	for _, role := range message.Roles {
		roles[role.ID] = *role
	}
	db := common.GetDB(ctx)
	for _, role := range oldMessage.Roles {
		_, ok := roles[role.ID]
		if !ok {
			err = role.Delete(ctx, db)
			if err != nil {
				bot.Logger(ctx).Error("unable to delete reaction role from database", "error", err)
				return "Impossible de sauvegarder le message de rôle. Merci de contacter l'administrateur du bot."
			}
		}
	}
	for _, role := range message.Roles {
		err = role.Save(ctx, common.GetDB(ctx))
		if err != nil {
			bot.Logger(ctx).Error("unable to save rolereaction role in database", "error", err)
			return "Impossible de sauvegarder le message de rôle. Merci de contacter l'administrateur du bot."
		}
	}
	return "Message de réaction mis à jour avec succès !"
}

func WaitForEmoji(ctx context.Context, dg bot.Session, userID, messageID uint64) (string, bool) {
	ctx, cancel := context.WithTimeout(ctx, 1*time.Minute)
	defer cancel()

	emojiChann := make(chan emoji.Emoji)

	cancelHandler := dg.EventManager().AddHandler(func(_ bot.Session, e *event.MessageReactionAdd) {
		if e.MessageID == messageID && e.UserID == userID {
			emojiChann <- e.Emoji
		}
	})
	defer cancelHandler()

	select {
	case emoji := <-emojiChann:
		emojiName := emoji.APIName()
		return emojiName, true
	case <-ctx.Done():
		return "", false
	}
}

func GetMessageFromEditID(ctx context.Context, i *interaction.Interaction, editID uint) (*config.RoleReactMessage, bool) {
	cfg := config.GetGuild(ctx, i.GuildID)
	m, ok := messageEdits[editID]
	if !ok || m.GuildID != cfg.ID {
		return &config.RoleReactMessage{}, false
	}
	return m, true
}

func FormatEmoji(apiName string) string {
	if strings.Contains(apiName, ":") {
		return fmt.Sprintf("<:%s>", apiName)
	}
	return apiName
}