aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAnhgelus Morhtuuzh <anhgelus@anhgelus.world>2025-05-13 16:15:47 +0200
committerAnhgelus Morhtuuzh <anhgelus@anhgelus.world>2025-05-13 16:15:47 +0200
commit01bafe9bf1de5be4e770b9500480807d4973d8d6 (patch)
treec6eafab11fef21c221136019ea490176185cec81
parent799df74fcda5266fd295b49fc759c605c815cad9 (diff)
feat(top): implements new kind of tops
-rw-r--r--Dockerfile2
-rw-r--r--commands/rank.go32
-rw-r--r--commands/reset.go11
-rw-r--r--commands/top.go57
-rw-r--r--main.go6
-rw-r--r--user/member.go30
-rw-r--r--user/xp.go99
7 files changed, 179 insertions, 58 deletions
diff --git a/Dockerfile b/Dockerfile
index ca8d3f8..0d8aee4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM docker.io/golang:1.23-alpine
+FROM docker.io/golang:1.24-alpine
WORKDIR /app
diff --git a/commands/rank.go b/commands/rank.go
index c659f2b..785b8c8 100644
--- a/commands/rank.go
+++ b/commands/rank.go
@@ -11,7 +11,6 @@ import (
func Rank(s *discordgo.Session, i *discordgo.InteractionCreate) {
optMap := utils.GenerateOptionMap(i)
c := user.GetCopaing(i.Member.User.ID, i.GuildID) // current user = member who used /rank
- user.LastEventUpdate(s, c) // update exp and reset last event
msg := "Votre niveau"
m := i.Member
var err error
@@ -21,13 +20,13 @@ func Rank(s *discordgo.Session, i *discordgo.InteractionCreate) {
if u.Bot {
err = resp.Message("Imagine si les bots avaient un niveau :rolling_eyes:").IsEphemeral().Send()
if err != nil {
- utils.SendAlert("rank.go - Reply error user is a bot", err.Error())
+ utils.SendAlert("commands/rank.go - Reply error user is a bot", err.Error())
}
}
m, err = s.GuildMember(i.GuildID, u.ID)
if err != nil {
utils.SendAlert(
- "rank.go - Fetching guild member",
+ "commands/rank.go - Fetching guild member",
err.Error(),
"discord_id",
u.ID,
@@ -36,24 +35,39 @@ func Rank(s *discordgo.Session, i *discordgo.InteractionCreate) {
)
err = resp.Message("Erreur : impossible de récupérer le membre").IsEphemeral().Send()
if err != nil {
- utils.SendAlert("rank.go - Reply error fetching guild member", err.Error())
+ utils.SendAlert("commands/rank.go - Reply error fetching guild member", err.Error())
}
return
}
c = user.GetCopaing(u.ID, i.GuildID) // current user = member targeted by member who wrote /rank
- user.UpdateXP(s, c) // update exp without resetting event
msg = fmt.Sprintf("Le niveau de %s", m.DisplayName())
}
- lvl := exp.Level(c.XP)
+ xp, err := c.GetXP()
+ if err != nil {
+ utils.SendAlert(
+ "commands/rank.go - Fetching XP",
+ err.Error(),
+ "discord_id",
+ c.ID,
+ "guild_id",
+ i.GuildID,
+ )
+ err = resp.Message("Erreur : impossible de récupérer l'XP").IsEphemeral().Send()
+ if err != nil {
+ utils.SendAlert("commands/rank.go - Reply error fetching xp", err.Error())
+ }
+ return
+ }
+ lvl := exp.Level(xp)
nxtLvlXP := exp.LevelXP(lvl + 1)
err = resp.Message(fmt.Sprintf(
"%s : **%d**\n> XP : %d\n> Prochain niveau dans %d XP",
msg,
lvl,
- c.XP,
- nxtLvlXP-c.XP,
+ xp,
+ nxtLvlXP-xp,
)).Send()
if err != nil {
- utils.SendAlert("rank.go - Sending rank", err.Error())
+ utils.SendAlert("commands/rank.go - Sending rank", err.Error())
}
}
diff --git a/commands/reset.go b/commands/reset.go
index 1e8df6a..3f8af03 100644
--- a/commands/reset.go
+++ b/commands/reset.go
@@ -34,8 +34,15 @@ func ResetUser(s *discordgo.Session, i *discordgo.InteractionCreate) {
}
return
}
- user.GetCopaing(m.ID, i.GuildID).Reset()
- if err := resp.Message("Le user bien été reset.").Send(); err != nil {
+ err := user.GetCopaing(m.ID, i.GuildID).Delete()
+ if err != nil {
+ utils.SendAlert("commands/reset.go - Copaing not deleted", err.Error(), "discord_id", m.ID, "guild_id", i.GuildID)
+ err = resp.Message("Erreur : impossible de reset l'utilisateur").Send()
+ if err != nil {
+ utils.SendAlert("commands/reset.go - Error deleting", err.Error())
+ }
+ }
+ if err = resp.Message("Le user bien été reset.").Send(); err != nil {
utils.SendAlert("commands/reset.go - Sending success (user)", err.Error())
}
}
diff --git a/commands/top.go b/commands/top.go
index 22574ce..a43e055 100644
--- a/commands/top.go
+++ b/commands/top.go
@@ -2,15 +2,14 @@ package commands
import (
"fmt"
- "github.com/anhgelus/gokord"
"github.com/anhgelus/gokord/utils"
"github.com/anhgelus/les-copaings-bot/exp"
"github.com/anhgelus/les-copaings-bot/user"
"github.com/bwmarrin/discordgo"
+ "sync"
)
func Top(s *discordgo.Session, i *discordgo.InteractionCreate) {
- user.LastEventUpdate(s, user.GetCopaing(i.Member.User.ID, i.GuildID))
resp := utils.ResponseBuilder{C: s, I: i}
err := resp.IsDeferred().Send()
if err != nil {
@@ -18,26 +17,48 @@ func Top(s *discordgo.Session, i *discordgo.InteractionCreate) {
return
}
resp.NotDeferred().IsEdit()
- go func() {
- var tops []user.Copaing
- gokord.DB.Where("guild_id = ?", i.GuildID).Limit(10).Order("exp desc").Find(&tops)
- msg := ""
- for i, c := range tops {
- if i == 9 {
- msg += fmt.Sprintf("%d. **<@%s>** - niveau %d", i+1, c.DiscordID, exp.Level(c.XP))
- } else {
- msg += fmt.Sprintf("%d. **<@%s>** - niveau %d\n", i+1, c.DiscordID, exp.Level(c.XP))
+ embeds := make([]*discordgo.MessageEmbed, 3)
+ wg := sync.WaitGroup{}
+
+ fn := func(s string, n uint, d int, id int) {
+ defer wg.Done()
+ tops, err := user.GetBestXP(i.GuildID, n, d)
+ if err != nil {
+ utils.SendAlert("commands/top.go - Fetching best xp", err.Error(), "n", n, "d", d, "id", id, "guild_id", i.GuildID)
+ embeds[id] = &discordgo.MessageEmbed{
+ Title: s,
+ Description: "Erreur : impossible de récupérer la liste",
+ Color: utils.Error,
}
+ return
}
- err = resp.Embeds([]*discordgo.MessageEmbed{
- {
- Title: "Top",
- Description: msg,
- Color: utils.Success,
- },
- }).Send()
+ embeds[id] = &discordgo.MessageEmbed{
+ Title: s,
+ Description: genTopsMessage(tops),
+ Color: utils.Success,
+ }
+ }
+
+ wg.Add(3)
+ go fn("Top full time", 10, -1, 0)
+ go fn("Top 30 jours", 5, 30, 1)
+ go fn("Top 7 jours", 5, 7, 2)
+ go func() {
+ wg.Wait()
+ err = resp.Embeds(embeds).Send()
if err != nil {
utils.SendAlert("commands/top.go - Sending response top", err.Error())
}
}()
}
+
+func genTopsMessage(tops []user.CopaingAccess) string {
+ msg := ""
+ for i, c := range tops {
+ msg += fmt.Sprintf("%d. **<@%s>** - niveau %d", i+1, c.ToCopaing().DiscordID, exp.Level(c.GetXP()))
+ if i != len(tops)-1 {
+ msg += "\n"
+ }
+ }
+ return msg
+}
diff --git a/main.go b/main.go
index 7b8e75f..b1f137d 100644
--- a/main.go
+++ b/main.go
@@ -17,10 +17,10 @@ var (
//go:embed updates.json
updatesData []byte
Version = gokord.Version{
- Major: 2,
- Minor: 4,
+ Major: 3,
+ Minor: 0,
Patch: 0,
- } // git version: 0.4.0 (it's the v2 of the bot)
+ }
stopPeriodicReducer chan<- interface{}
)
diff --git a/user/member.go b/user/member.go
index b61aa76..cf7c4a4 100644
--- a/user/member.go
+++ b/user/member.go
@@ -4,7 +4,6 @@ import (
"fmt"
"github.com/anhgelus/gokord"
"github.com/anhgelus/gokord/utils"
- "github.com/anhgelus/les-copaings-bot/config"
"time"
)
@@ -23,6 +22,11 @@ type CopaingXP struct {
CreatedAt time.Time
}
+type CopaingAccess interface {
+ ToCopaing() *Copaing
+ GetXP() uint
+}
+
const (
LastEvent = "last_event"
AlreadyRemoved = "already_removed"
@@ -52,30 +56,6 @@ func (c *Copaing) Load() error {
Error
}
-func (c *Copaing) GetXP() (uint, error) {
- cfg := config.GetGuildConfig(c.GuildID)
- xp := uint(0)
- y, m, d := time.Unix(time.Now().Unix()-int64(cfg.DaysXPRemains*24*60*60), 0).Date()
- rows, err := gokord.DB.
- Model(&CopaingXP{}).
- Where(fmt.Sprintf("created_at >= '%d-%d-%d' and guild_id = ? and discord_id = ?", y, m, d), c.GuildID, c.DiscordID).
- Rows()
- defer rows.Close()
- if err != nil {
- return 0, err
- }
- for rows.Next() {
- var cXP CopaingXP
- err = gokord.DB.ScanRows(rows, &cXP)
- if err != nil {
- utils.SendAlert("user/member.go - Scaning rows", err.Error(), "discord_id", c.DiscordID, "guild_id", c.GuildID)
- continue
- }
- xp += cXP.XP
- }
- return xp, nil
-}
-
func (c *Copaing) Save() error {
return gokord.DB.Save(c).Error
}
diff --git a/user/xp.go b/user/xp.go
index f8bcd06..c2e4d08 100644
--- a/user/xp.go
+++ b/user/xp.go
@@ -1,11 +1,30 @@
package user
import (
+ "fmt"
+ "github.com/anhgelus/gokord"
"github.com/anhgelus/gokord/utils"
+ "github.com/anhgelus/les-copaings-bot/config"
"github.com/anhgelus/les-copaings-bot/exp"
"github.com/bwmarrin/discordgo"
+ "slices"
+ "sync"
+ "time"
)
+type cXP struct {
+ Cxp uint
+ *Copaing
+}
+
+func (c *cXP) ToCopaing() *Copaing {
+ return c.Copaing
+}
+
+func (c *cXP) GetXP() uint {
+ return c.Cxp
+}
+
func (c *Copaing) AddXP(s *discordgo.Session, m *discordgo.Member, xp uint, fn func(uint, uint)) {
old, err := c.GetXP()
pastLevel := exp.Level(old)
@@ -29,3 +48,83 @@ func (c *Copaing) AddXP(s *discordgo.Session, m *discordgo.Member, xp uint, fn f
onNewLevel(s, m, newLevel)
}
}
+
+func (c *Copaing) GetXP() (uint, error) {
+ cfg := config.GetGuildConfig(c.GuildID)
+ return c.GetXPForDays(cfg.DaysXPRemains)
+}
+
+func (c *Copaing) GetXPForDays(n uint) (uint, error) {
+ xp := uint(0)
+ y, m, d := time.Unix(time.Now().Unix()-int64(n*24*60*60), 0).Date()
+ rows, err := gokord.DB.
+ Model(&CopaingXP{}).
+ Where(fmt.Sprintf("created_at >= '%d-%d-%d' and guild_id = ? and discord_id = ?", y, m, d), c.GuildID, c.DiscordID).
+ Rows()
+ defer rows.Close()
+ if err != nil {
+ return 0, err
+ }
+ for rows.Next() {
+ var cXP CopaingXP
+ err = gokord.DB.ScanRows(rows, &cXP)
+ if err != nil {
+ utils.SendAlert("user/xp.go - Scanning rows", err.Error(), "discord_id", c.DiscordID, "guild_id", c.GuildID)
+ continue
+ }
+ xp += cXP.XP
+ }
+ return xp, nil
+}
+
+// GetBestXP returns n Copaing with the best XP within d days (d <= cfg.DaysXPRemain; d < 0 <=> d = cfg.DaysXPRemain)
+//
+// This function is slow
+func GetBestXP(guildId string, n uint, d int) ([]CopaingAccess, error) {
+ if d < 0 {
+ cfg := config.GetGuildConfig(guildId)
+ d = int(cfg.DaysXPRemains)
+ }
+ rows, err := gokord.DB.Model(&Copaing{}).Where("guild_id = ?", guildId).Rows()
+ defer rows.Close()
+ if err != nil {
+ return nil, err
+ }
+ var l []*cXP
+ wg := sync.WaitGroup{}
+ for rows.Next() {
+ var c Copaing
+ err = gokord.DB.ScanRows(rows, &c)
+ if err != nil {
+ utils.SendAlert("user/xp.go - Scanning rows", err.Error(), "discord_id", c.DiscordID, "guild_id", guildId)
+ continue
+ }
+ wg.Add(1)
+ go func() {
+ defer wg.Done()
+ xp, err := c.GetXPForDays(uint(d))
+ if err != nil {
+ utils.SendAlert("user/xp.go - Fetching xp", err.Error(), "discord_id", c.DiscordID, "guild_id", guildId)
+ return
+ }
+ l = append(l, &cXP{Cxp: xp, Copaing: &c})
+ }()
+ }
+ wg.Wait()
+ slices.SortFunc(l, func(a, b *cXP) int {
+ // desc order
+ if a.Cxp < b.Cxp {
+ return 1
+ }
+ if a.Cxp > b.Cxp {
+ return -1
+ }
+ return 0
+ })
+ m := min(len(l), int(n))
+ cs := make([]CopaingAccess, m)
+ for i, c := range l[:m] {
+ cs[i] = c
+ }
+ return cs, nil
+}