les-copaings-bot/commands/config.go

455 lines
14 KiB
Go

package commands
import (
"fmt"
"github.com/anhgelus/gokord"
"github.com/anhgelus/gokord/cmd"
"github.com/anhgelus/gokord/component"
"github.com/anhgelus/gokord/logger"
"github.com/anhgelus/les-copaings-bot/config"
"github.com/anhgelus/les-copaings-bot/exp"
"github.com/bwmarrin/discordgo"
"strconv"
"strings"
"time"
)
const (
ConfigModify = "config_modify"
ConfigModifyXpRole = "xp_role"
ConfigModifyDisChannel = "disabled_channel"
ConfigModifyFallbackChannel = "fallback_channel"
ConfigModifyTimeReduce = "time_reduce"
// xp role related
XpRoleAdd = "xp_role_add"
XpRoleAddLevel = "xp_role_add_level"
XpRoleAddRole = "xp_role_add_role"
XpRoleDel = "xp_role_del"
XpRoleDelRole = "xp_role_del_role"
XpRoleEdit = "xp_role_edit"
XpRoleEditLevel = "xp_role_edit_level"
XpRoleEditRole = "xp_role_edit_role"
)
var (
configModifyMap = map[string]uint{}
)
func Config(_ *discordgo.Session, i *discordgo.InteractionCreate, _ cmd.OptionMap, resp *cmd.ResponseBuilder) {
cfg := config.GetGuildConfig(i.GuildID)
roles := ""
l := len(cfg.XpRoles) - 1
for i, r := range cfg.XpRoles {
if i == l {
roles += fmt.Sprintf("> Niveau %d - <@&%s>", exp.Level(r.XP), r.RoleID)
} else {
roles += fmt.Sprintf("> Niveau %d - <@&%s>\n", exp.Level(r.XP), r.RoleID)
}
}
if len(roles) == 0 {
roles = "Aucun rôle configuré :("
}
disChans := strings.Split(cfg.DisabledChannels, ";")
l = len(disChans) - 1
chans := ""
for i, c := range disChans {
if i == l-1 {
chans += fmt.Sprintf("> <#%s>", c)
} else if i != l {
chans += fmt.Sprintf("> <#%s>\n", c)
}
}
if len(chans) == 0 {
chans = "Aucun salon désactivé :)"
}
var defaultChan string
if len(cfg.FallbackChannel) == 0 {
defaultChan = "Pas de valeur"
} else {
defaultChan = fmt.Sprintf("<#%s>", cfg.FallbackChannel)
}
err := resp.AddEmbed(&discordgo.MessageEmbed{
Type: discordgo.EmbedTypeRich,
Title: "Config",
Color: 0x10E6AD,
Fields: []*discordgo.MessageEmbedField{
{
Name: "Salon par défaut",
Value: defaultChan,
Inline: false,
},
{
Name: "Rôles liés aux niveaux",
Value: roles,
Inline: false,
},
{
Name: "Salons désactivés",
Value: chans,
Inline: false,
},
{
Name: "Jours avant la réduction",
Value: fmt.Sprintf("%d", cfg.DaysXPRemains),
Inline: false,
},
},
}).SetComponents(component.New().Add(component.NewActionRow().Add(
component.NewStringSelect(ConfigModify).SetPlaceholder("Modifier...").
AddOption(
component.NewSelectOption("Rôles liés à l'XP", ConfigModifyXpRole).
SetDescription("Gère les rôles liés à l'XP").
SetEmoji(&discordgo.ComponentEmoji{Name: "🏅"}),
).
AddOption(
component.NewSelectOption("Salons désactivés", ConfigModifyDisChannel).
SetDescription("Gère les salons désactivés").
SetEmoji(&discordgo.ComponentEmoji{Name: "❌"}),
).
AddOption(
// I don't have a better idea for this...
component.NewSelectOption("Salons de repli", ConfigModifyFallbackChannel).
SetDescription("Spécifie le salon de repli").
SetEmoji(&discordgo.ComponentEmoji{Name: "💾"}),
).
AddOption(
component.NewSelectOption("Temps avec la réduction", ConfigModifyTimeReduce).
SetDescription("Gère le temps avant la réduction d'XP").
SetEmoji(&discordgo.ComponentEmoji{Name: "⌛"}),
),
))).IsEphemeral().Send()
if err != nil {
logger.Alert("config/guild.go - Sending config", err.Error())
}
}
func ConfigModifyXPRole(_ *discordgo.Session, _ *discordgo.InteractionCreate, _ discordgo.MessageComponentInteractionData, resp *cmd.ResponseBuilder) {
err := resp.IsEphemeral().
SetMessage("Action à réaliser").
SetComponents(component.New().Add(component.NewActionRow().Add(
component.NewStringSelect(ConfigModify).SetPlaceholder("Action").
AddOption(
component.NewSelectOption("Ajouter", XpRoleAdd).
SetDescription("Ajouter un rôle à XP").
SetEmoji(&discordgo.ComponentEmoji{Name: "⬆️"}),
).
AddOption(
component.NewSelectOption("Modifier", XpRoleEdit).
SetDescription("Modifier un rôle à XP").
SetEmoji(&discordgo.ComponentEmoji{Name: "📝"}),
).
AddOption(
component.NewSelectOption("Supprimer", XpRoleDel).
SetDescription("Supprimer un rôle à XP").
SetEmoji(&discordgo.ComponentEmoji{Name: "❌"}),
),
))).Send()
if err != nil {
logger.Alert("config/guild.go - Sending config", err.Error())
}
}
func ConfigXPRoleAddEdit(_ *discordgo.Session, _ *discordgo.InteractionCreate, data discordgo.MessageComponentInteractionData, resp *cmd.ResponseBuilder) {
cID := XpRoleAddLevel
if data.CustomID == XpRoleEdit {
cID = XpRoleEditLevel
}
err := resp.IsModal().
SetTitle("Role").
SetComponents(component.New().ForModal().Add(component.NewActionRow().Add(
component.NewTextInput(cID, "Niveau", discordgo.TextInputShort).
SetPlaceholder("5").
IsRequired().
SetMinLength(0).
SetMaxLength(5),
))).
Send()
if err != nil {
logger.Alert("config/guild.go - Sending modal to add/edit", err.Error())
}
}
func ConfigXPRoleAddRole(_ *discordgo.Session, i *discordgo.InteractionCreate, data discordgo.MessageComponentInteractionData, resp *cmd.ResponseBuilder) {
cfg := config.GetGuildConfig(i.GuildID)
roleId := data.Values[0]
for _, r := range cfg.XpRoles {
if r.RoleID == roleId {
err := resp.SetMessage("Le rôle est déjà présent dans la config").Send()
if err != nil {
logger.Alert("commands/config.go - Role already in config", err.Error())
}
return
}
}
cfg.XpRoles = append(cfg.XpRoles, config.XpRole{
XP: configModifyMap[getKeyConfigRole(i)],
RoleID: roleId,
})
err := cfg.Save()
if err != nil {
logger.Alert(
"commands/config.go - Saving config",
err.Error(),
"guild_id", i.GuildID,
"role_id", roleId,
"type", "add",
)
}
if err = resp.IsEphemeral().SetMessage("Rôle ajouté.").Send(); err != nil {
logger.Alert("commands/config.go - Sending success", err.Error())
}
}
func ConfigXPRoleEditRole(_ *discordgo.Session, i *discordgo.InteractionCreate, data discordgo.MessageComponentInteractionData, resp *cmd.ResponseBuilder) {
cfg := config.GetGuildConfig(i.GuildID)
roleId := data.Values[0]
_, r := cfg.FindXpRole(roleId)
if r == nil {
err := resp.SetMessage("Le rôle n'a pas été trouvé dans la config.").Send()
if err != nil {
logger.Alert("commands/config.go - Role not found (edit)", err.Error())
}
return
}
r.XP = configModifyMap[getKeyConfigRole(i)]
err := gokord.DB.Save(r).Error
if err != nil {
logger.Alert(
"commands/config.go - Saving config",
err.Error(),
"guild_id", i.GuildID,
"role_id", roleId,
"type", "edit",
)
}
if err = resp.IsEphemeral().SetMessage("Rôle modifié.").Send(); err != nil {
logger.Alert("commands/config.go - Sending success", err.Error())
}
}
func ConfigXPRoleDel(_ *discordgo.Session, _ *discordgo.InteractionCreate, _ discordgo.MessageComponentInteractionData, resp *cmd.ResponseBuilder) {
err := resp.IsEphemeral().
SetMessage("Rôle à supprimer").
SetComponents(component.New().Add(component.NewActionRow().Add(component.NewRoleSelect(XpRoleDelRole)))).
Send()
if err != nil {
logger.Alert("config/guild.go - Sending response to del", err.Error())
}
}
func ConfigXPRoleDelRole(_ *discordgo.Session, i *discordgo.InteractionCreate, data discordgo.MessageComponentInteractionData, resp *cmd.ResponseBuilder) {
cfg := config.GetGuildConfig(i.GuildID)
roleId := data.Values[0]
_, r := cfg.FindXpRole(roleId)
if r == nil {
err := resp.SetMessage("Le rôle n'a pas été trouvé dans la config.").Send()
if err != nil {
logger.Alert("commands/config.go - Sending role not found (del)", err.Error())
}
return
}
err := gokord.DB.Delete(r).Error
if err != nil {
logger.Alert(
"commands/config.go - Deleting entry",
err.Error(),
"guild_id", i.GuildID,
"role_id", roleId,
"type", "del",
)
}
if err = resp.IsEphemeral().SetMessage("Rôle supprimé.").Send(); err != nil {
logger.Alert("commands/config.go - Sending success", err.Error())
}
}
func ConfigXPRoleLevel(_ *discordgo.Session, i *discordgo.InteractionCreate, data discordgo.ModalSubmitInteractionData, resp *cmd.ResponseBuilder) {
input := data.Components[0].(*discordgo.ActionsRow).Components[0].(*discordgo.TextInput)
k := getKeyConfigRole(i)
in, err := strconv.Atoi(input.Value)
if err != nil || in < 0 {
if err = resp.IsEphemeral().
SetMessage("Impossible de lire le nombre. Il doit s'agit d'un nombre entier positif.").
Send(); err != nil {
logger.Alert("command/config.go - Sending bad number", err.Error())
}
return
}
configModifyMap[k] = uint(in)
go func(i *discordgo.InteractionCreate, k string) {
time.Sleep(5 * time.Minute)
delete(configModifyMap, k)
}(i, k)
cID := XpRoleAddRole
resp.SetMessage("Rôle à ajouter")
if data.CustomID == XpRoleEditLevel {
cID = XpRoleEditLevel
resp.SetMessage("Rôle à modifier")
}
err = resp.IsEphemeral().
SetMessage("Rôle à supprimer").
SetComponents(component.New().Add(component.NewActionRow().Add(component.NewRoleSelect(cID)))).
Send()
if err != nil {
logger.Alert("config/guild.go - Sending response to add/edit", err.Error())
}
}
func getKeyConfigRole(i *discordgo.InteractionCreate) string {
return fmt.Sprintf("r:%s:%s", i.GuildID, i.User.ID)
}
func ConfigChannel(s *discordgo.Session, i *discordgo.InteractionCreate, optMap cmd.OptionMap, resp *cmd.ResponseBuilder) {
resp.IsEphemeral()
// verify every args
t, ok := optMap["type"]
if !ok {
err := resp.SetMessage("Le type d'action n'a pas été renseigné.").Send()
if err != nil {
logger.Alert("commands/config.go - Action type not set", err.Error())
}
return
}
ts := t.StringValue()
salon, ok := optMap["channel"]
if !ok {
err := resp.SetMessage("Le salon n'a pas été renseigné.").Send()
if err != nil {
logger.Alert("commands/config.go - Channel not set (disabled)", err.Error())
}
return
}
channel := salon.ChannelValue(s)
cfg := config.GetGuildConfig(i.GuildID)
switch ts {
case "add":
if strings.Contains(cfg.DisabledChannels, channel.ID) {
err := resp.SetMessage("Le salon est déjà dans la liste des salons désactivés").Send()
if err != nil {
logger.Alert("commands/config.go - Channel already disabled", err.Error())
}
return
}
cfg.DisabledChannels += channel.ID + ";"
case "del":
if !strings.Contains(cfg.DisabledChannels, channel.ID) {
err := resp.SetMessage("Le salon n'est pas désactivé").Send()
if err != nil {
logger.Alert("commands/config.go - Channel not disabled", err.Error())
}
return
}
cfg.DisabledChannels = strings.ReplaceAll(cfg.DisabledChannels, channel.ID+";", "")
default:
err := resp.SetMessage("Le type d'action n'est pas valide.").Send()
if err != nil {
logger.Alert("commands/config.go - Invalid action type", err.Error())
}
return
}
// save
err := cfg.Save()
if err != nil {
logger.Alert(
"commands/config.go - Saving config",
err.Error(),
"guild_id",
i.GuildID,
"type",
ts,
"channel_id",
channel.ID,
)
err = resp.SetMessage("Il y a eu une erreur lors de la modification de de la base de données.").Send()
} else {
err = resp.SetMessage("Modification sauvegardé.").Send()
}
if err != nil {
logger.Alert("commands/config.go - Modification saved message", err.Error())
}
}
func ConfigFallbackChannel(s *discordgo.Session, i *discordgo.InteractionCreate, optMap cmd.OptionMap, resp *cmd.ResponseBuilder) {
resp.IsEphemeral()
// verify every args
salon, ok := optMap["channel"]
if !ok {
err := resp.SetMessage("Le salon n'a pas été renseigné.").Send()
if err != nil {
logger.Alert("commands/config.go - Channel not set (fallback)", err.Error())
}
return
}
channel := salon.ChannelValue(s)
if channel.Type != discordgo.ChannelTypeGuildText {
err := resp.SetMessage("Le salon n'est pas un salon textuel.").Send()
if err != nil {
logger.Alert("commands/config.go - Invalid channel type", err.Error())
}
return
}
cfg := config.GetGuildConfig(i.GuildID)
cfg.FallbackChannel = channel.ID
// save
err := cfg.Save()
if err != nil {
logger.Alert(
"commands/config.go - Saving config",
err.Error(),
"guild_id",
i.GuildID,
"channel_id",
channel.ID,
)
err = resp.SetMessage("Il y a eu une erreur lors de la modification de de la base de données.").Send()
} else {
err = resp.SetMessage("Salon enregistré.").Send()
}
if err != nil {
logger.Alert("commands/config.go - Channel saved message", err.Error())
}
}
func ConfigPeriodBeforeReduce(s *discordgo.Session, i *discordgo.InteractionCreate, optMap cmd.OptionMap, resp *cmd.ResponseBuilder) {
resp.IsEphemeral()
// verify every args
days, ok := optMap["days"]
if !ok {
err := resp.SetMessage("Le nombre de jours n'a pas été renseigné.").Send()
if err != nil {
logger.Alert("commands/config.go - Days not set (fallback)", err.Error())
}
return
}
d := days.IntValue()
if d < 30 {
err := resp.SetMessage("Le nombre de jours est inférieur à 30.").Send()
if err != nil {
logger.Alert("commands/config.go - Days < 30 (fallback)", err.Error())
}
return
}
// save
cfg := config.GetGuildConfig(i.GuildID)
cfg.DaysXPRemains = uint(d)
err := cfg.Save()
if err != nil {
logger.Alert(
"commands/config.go - Saving config",
err.Error(),
"guild_id",
i.GuildID,
"days",
d,
)
err = resp.SetMessage("Il y a eu une erreur lors de la modification de de la base de données.").Send()
} else {
err = resp.SetMessage("Nombre de jours enregistré.").Send()
}
if err != nil {
logger.Alert("commands/config.go - Days saved message", err.Error())
}
}