feat(command): better plot for stats

1 files changed, 79 insertions, 11 deletions

diff --git a/commands/stats.go b/commands/stats.go
index 116e89f..e464890 100644
--- a/commands/stats.go
+++ b/commands/stats.go
@@ -2,6 +2,10 @@ package commands
 
 import (
 	"bytes"
+	"image/color"
+	"math"
+	"math/rand/v2"
+	"slices"
 	"time"
 
 	"git.anhgelus.world/anhgelus/les-copaings-bot/config"
@@ -14,11 +18,16 @@ import (
 	"github.com/jackc/pgx/v5/pgtype"
 	"gonum.org/v1/plot"
 	"gonum.org/v1/plot/plotter"
-	"gonum.org/v1/plot/plotutil"
 	"gonum.org/v1/plot/vg"
 )
 
 type data struct {
+	CreatedAt time.Time
+	XP        int
+	CopaingID int
+}
+
+type dbData struct {
 	CreatedAt *pgtype.Date
 	XP        int
 	CopaingID int
@@ -38,13 +47,40 @@ func Stats(s *discordgo.Session, i *discordgo.InteractionCreate, opt cmd.OptionM
 		days = uint(in)
 	}
 	var rawData []*data
-	res := gokord.DB.Raw(
-		`SELECT "created_at"::date::text, sum("xp") as xp, "copaing_id" FROM copaing_xps WHERE "guild_id" = ? and "created_at" > ? GROUP BY "created_at"::date, "copaing_id"`,
-		i.GuildID, exp.TimeStampNDaysBefore(days),
-	)
-	if err := res.Scan(&rawData).Error; err != nil {
-		logger.Alert("commands/stats.go - Fetching result", err.Error())
-		return
+	if gokord.Debug {
+		var rawCopaingData []*user.CopaingXP
+		err := gokord.DB.
+			Where("guild_id = ? and created_at > ?", i.GuildID, exp.TimeStampNDaysBefore(days)).
+			Find(&rawCopaingData).
+			Error
+		if err != nil {
+			logger.Alert("commands/stats.go - Fetching result", err.Error())
+			return
+		}
+		rawData = make([]*data, len(rawCopaingData))
+		for in, d := range rawCopaingData {
+			rawData[in] = &data{
+				CreatedAt: d.CreatedAt,
+				XP:        int(d.XP),
+				CopaingID: int(d.CopaingID),
+			}
+		}
+	} else {
+		sql := `SELECT "created_at"::date::text, sum("xp") as xp, "copaing_id" FROM copaing_xps WHERE "guild_id" = ? and "created_at" > ? GROUP BY "created_at"::date, "copaing_id"`
+		var rawDbData []dbData
+		res := gokord.DB.Raw(sql, i.GuildID, exp.TimeStampNDaysBefore(days))
+		if err := res.Scan(&rawDbData).Error; err != nil {
+			logger.Alert("commands/stats.go - Fetching result", err.Error())
+			return
+		}
+		rawData = make([]*data, len(rawDbData))
+		for in, d := range rawDbData {
+			rawData[in] = &data{
+				CreatedAt: d.CreatedAt.Time,
+				XP:        d.XP,
+				CopaingID: d.CopaingID,
+			}
+		}
 	}
 
 	copaings := map[int]*user.Copaing{}
@@ -65,7 +101,10 @@ func Stats(s *discordgo.Session, i *discordgo.InteractionCreate, opt cmd.OptionM
 			pts = &[]plotter.XY{}
 			stats[raw.CopaingID] = pts
 		}
-		t := float64(raw.CreatedAt.Time.Unix()-time.Now().Unix()) / (24 * 60 * 60)
+		t := float64(raw.CreatedAt.Unix() - time.Now().Unix())
+		if !gokord.Debug {
+			t = math.Ceil(t / (24 * 60 * 60))
+		}
 		*pts = append(*pts, plotter.XY{
 			X: t,
 			Y: float64(raw.XP),
@@ -77,19 +116,48 @@ func Stats(s *discordgo.Session, i *discordgo.InteractionCreate, opt cmd.OptionM
 	p.X.Label.Text = "Jours"
 	p.Y.Label.Text = "XP"
 
+	p.Add(plotter.NewGrid())
+
+	r := rand.New(rand.NewPCG(uint64(time.Now().Unix()), uint64(time.Now().Unix())))
 	for in, c := range copaings {
 		m, err := s.GuildMember(i.GuildID, c.DiscordID)
 		if err != nil {
 			logger.Alert("commands/stats.go - Fetching guild member", err.Error())
 			return
 		}
-		err = plotutil.AddLinePoints(p, m.DisplayName(), plotter.XYs(*stats[in]))
+		slices.SortFunc(*stats[in], func(a, b plotter.XY) int {
+			if a.X < b.X {
+				return -1
+			}
+			if a.X > b.X {
+				return 1
+			}
+			return 0
+		})
+		first := (*stats[in])[0]
+		if first.X > float64(-days) {
+			*stats[in] = append([]plotter.XY{{
+				X: first.X - 1, Y: 0,
+			}}, *stats[in]...)
+		}
+		last := (*stats[in])[len(*stats[in])-1]
+		if last.X <= -1 {
+			*stats[in] = append(*stats[in], plotter.XY{
+				X: last.X + 1, Y: 0,
+			})
+		}
+		l, err := plotter.NewLine(plotter.XYs(*stats[in]))
 		if err != nil {
 			logger.Alert("commands/stats.go - Adding line points", err.Error())
 			return
 		}
+		l.LineStyle.Width = vg.Points(1)
+		l.LineStyle.Dashes = []vg.Length{vg.Points(5), vg.Points(5)}
+		l.LineStyle.Color = color.RGBA{R: uint8(r.UintN(255)), G: uint8(r.UintN(255)), B: uint8(r.UintN(255)), A: 255}
+		p.Add(l)
+		p.Legend.Add(m.DisplayName(), l)
 	}
-	w, err := p.WriterTo(4*vg.Inch, 4*vg.Inch, "png")
+	w, err := p.WriterTo(8*vg.Inch, 6*vg.Inch, "png")
 	if err != nil {
 		logger.Alert("commands/stats.go - Generating png", err.Error())
 		return