feat(command): reset and reset-user

3 files changed, 56 insertions, 0 deletions

diff --git a/commands/reset.go b/commands/reset.go
new file mode 100644
index 0000000..f0c6d9a
--- /dev/null
+++ b/commands/reset.go
@@ -0,0 +1,35 @@
+package commands
+
+import (
+	"github.com/anhgelus/gokord"
+	"github.com/anhgelus/gokord/utils"
+	"github.com/anhgelus/les-copaings-bot/xp"
+	"github.com/bwmarrin/discordgo"
+)
+
+func Reset(s *discordgo.Session, i *discordgo.InteractionCreate) {
+	var copaings []*xp.Copaing
+	gokord.DB.Where("guild_id = ?", i.GuildID).Delete(&copaings)
+	resp := utils.ResponseBuilder{C: s, I: i}
+	if err := resp.IsEphemeral().Message("L'XP a été reset.").Send(); err != nil {
+		utils.SendAlert("commands/reset.go - Sending success (all)", err.Error())
+	}
+}
+
+func ResetUser(s *discordgo.Session, i *discordgo.InteractionCreate) {
+	resp := utils.ResponseBuilder{C: s, I: i}
+	resp.IsEphemeral()
+	optMap := utils.GenerateOptionMap(i)
+	v, ok := optMap["copaing"]
+	if !ok {
+		if err := resp.Message("Le copaing n'a pas été renseigné.").Send(); err != nil {
+			utils.SendAlert("commands/reset.go - Copaing not set", err.Error())
+		}
+		return
+	}
+	m := v.UserValue(s)
+	xp.GetCopaing(m.ID, i.GuildID).Reset()
+	if err := resp.Message("Le a bien été reset.").Send(); err != nil {
+		utils.SendAlert("commands/reset.go - Sending success (copaing)", err.Error())
+	}
+}
diff --git a/main.go b/main.go
index d8cdbc2..e23d0ea 100644
--- a/main.go
+++ b/main.go
@@ -100,6 +100,21 @@ func main() {
 		HasOption().
 		SetHandler(commands.Top)
 
+	resetCmd := gokord.NewCommand("reset", "Reset l'xp").
+		HasOption().
+		SetHandler(commands.Reset).
+		SetPermission(gokord.AdminPermission)
+
+	resetUserCmd := gokord.NewCommand("reset-user", "Reset l'xp d'un utilisation").
+		HasOption().
+		AddOption(gokord.NewOption(
+			discordgo.ApplicationCommandOptionUser,
+			"copaing",
+			"Copaing a reset",
+		).IsRequired()).
+		SetHandler(commands.ResetUser).
+		SetPermission(gokord.AdminPermission)
+
 	bot := gokord.Bot{
 		Token: token,
 		Status: []*gokord.Status{
@@ -113,6 +128,8 @@ func main() {
 			rankCmd,
 			configCmd,
 			topCmd,
+			resetCmd,
+			resetUserCmd,
 		},
 		AfterInit: afterInit,
 	}
diff --git a/xp/member.go b/xp/member.go
index bbf013c..9859ca2 100644
--- a/xp/member.go
+++ b/xp/member.go
@@ -185,6 +185,10 @@ func (c *Copaing) GetUserBase() *gokord.UserBase {
 	return &gokord.UserBase{DiscordID: c.DiscordID, GuildID: c.GuildID}
 }
 
+func (c *Copaing) Reset() {
+	gokord.DB.Where("guild_id = ? AND discord_id", c.GuildID, c.DiscordID).Delete(c)
+}
+
 func getRedisClient() (*redis.Client, error) {
 	if redisClient == nil {
 		var err error