From ef39c156f94b854796bb1c2adddfd40a5acaeca8 Mon Sep 17 00:00:00 2001
From: Anhgelus Morhtuuzh <anhgelus.morhtuuzh@proton.me>
Date: Tue, 16 Apr 2024 15:23:15 +0200
Subject: [PATCH] feat(reducer): role update on new level

---
 commands/rank.go |  4 ++--
 commands/top.go  |  2 +-
 xp/events.go     |  4 ++--
 xp/level.go      | 30 +++++++++++++++++++++++++++---
 4 files changed, 32 insertions(+), 8 deletions(-)

diff --git a/commands/rank.go b/commands/rank.go
index d056de1..219977d 100644
--- a/commands/rank.go
+++ b/commands/rank.go
@@ -10,7 +10,7 @@ import (
 func Rank(s *discordgo.Session, i *discordgo.InteractionCreate) {
 	optMap := utils.GenerateOptionMap(i)
 	c := xp.GetCopaing(i.Member.User.ID, i.GuildID) // current copaing = member who used /rank
-	xp.LastEventUpdate(c)                           // update xp and reset last event
+	xp.LastEventUpdate(s, c)                        // update xp and reset last event
 	msg := "Votre niveau"
 	m := i.Member
 	var err error
@@ -41,7 +41,7 @@ func Rank(s *discordgo.Session, i *discordgo.InteractionCreate) {
 		}
 		c.DiscordID = u.ID // current copaing = member targeted by member who wrote /rank
 		c.Load()           // reload copaing (change line before)
-		xp.XPUpdate(c)     // update xp without resetting event
+		xp.XPUpdate(s, c)  // update xp without resetting event
 		msg = fmt.Sprintf("Le niveau de %s", m.DisplayName())
 	}
 	lvl := xp.Level(c.XP)
diff --git a/commands/top.go b/commands/top.go
index 6df259f..c0f4dae 100644
--- a/commands/top.go
+++ b/commands/top.go
@@ -9,7 +9,7 @@ import (
 )
 
 func Top(s *discordgo.Session, i *discordgo.InteractionCreate) {
-	xp.LastEventUpdate(xp.GetCopaing(i.User.ID, i.GuildID))
+	xp.LastEventUpdate(s, xp.GetCopaing(i.User.ID, i.GuildID))
 	resp := utils.ResponseBuilder{C: s, I: i}
 	err := resp.IsDeferred().Send()
 	if err != nil {
diff --git a/xp/events.go b/xp/events.go
index 655e9fd..ebb7f40 100644
--- a/xp/events.go
+++ b/xp/events.go
@@ -30,7 +30,7 @@ func OnMessage(s *discordgo.Session, m *discordgo.MessageCreate) {
 		return
 	}
 	c := GetCopaing(m.Author.ID, m.GuildID)
-	LastEventUpdate(c)
+	LastEventUpdate(s, c)
 	// add xp
 	trimmed := utils.TrimMessage(strings.ToLower(m.Content))
 	m.Member.User = m.Author
@@ -63,7 +63,7 @@ func OnVoiceUpdate(s *discordgo.Session, e *discordgo.VoiceStateUpdate) {
 	if e.Member.User.Bot {
 		return
 	}
-	LastEventUpdate(GetCopaing(e.UserID, e.GuildID))
+	LastEventUpdate(s, GetCopaing(e.UserID, e.GuildID))
 	cfg := config.GetGuildConfig(e.GuildID)
 	if cfg.IsDisabled(e.ChannelID) {
 		return
diff --git a/xp/level.go b/xp/level.go
index bbcc71f..12e5a28 100644
--- a/xp/level.go
+++ b/xp/level.go
@@ -43,7 +43,23 @@ func onNewLevel(s *discordgo.Session, m *discordgo.Member, level uint) {
 	}
 }
 
-func LastEventUpdate(c *Copaing) {
+func (c *Copaing) OnNewLevel(s *discordgo.Session, level uint) {
+	m, err := s.GuildMember(c.GuildID, c.DiscordID)
+	if err != nil {
+		utils.SendAlert(
+			"xp/level.go - Getting member for new level",
+			err.Error(),
+			"discord_id",
+			c.DiscordID,
+			"guild_id",
+			c.GuildID,
+		)
+		return
+	}
+	onNewLevel(s, m, level)
+}
+
+func LastEventUpdate(s *discordgo.Session, c *Copaing) {
 	h := c.HourSinceLastEvent()
 	l := Lose(h, c.XP)
 	xp := c.XPAlreadyRemoved()
@@ -61,11 +77,15 @@ func LastEventUpdate(c *Copaing) {
 	}
 	if oldXP != c.XP {
 		c.Save()
+		lvl := Level(c.XP)
+		if Level(oldXP) != Level(c.XP) {
+			c.OnNewLevel(s, lvl)
+		}
 	}
 	c.SetLastEvent()
 }
 
-func XPUpdate(c *Copaing) {
+func XPUpdate(s *discordgo.Session, c *Copaing) {
 	oldXP := c.XP
 	if oldXP == 0 {
 		return
@@ -89,6 +109,10 @@ func XPUpdate(c *Copaing) {
 	if oldXP != c.XP {
 		utils.SendDebug("Save XP", "old", oldXP, "new", c.XP, "user", c.DiscordID)
 		c.Save()
+		lvl := Level(c.XP)
+		if Level(oldXP) != Level(c.XP) {
+			c.OnNewLevel(s, lvl)
+		}
 	}
 }
 
@@ -103,7 +127,7 @@ func XPUpdate(c *Copaing) {
 //			go func() {
 //				utils.SendDebug("Async reducer", "user", m.DisplayName(), "guild", g.Name)
 //				c := GetCopaing(m.User.ID, g.ID)
-//				XPUpdate(c)
+//				XPUpdate(s, c)
 //				wg.Done()
 //			}()
 //		}