diff options
Diffstat (limited to 'commands/stats.go')
| -rw-r--r-- | commands/stats.go | 224 |
1 files changed, 224 insertions, 0 deletions
diff --git a/commands/stats.go b/commands/stats.go new file mode 100644 index 0000000..f93f6a0 --- /dev/null +++ b/commands/stats.go @@ -0,0 +1,224 @@ +package commands + +import ( + "bytes" + "errors" + "gorm.io/gorm" + "image/color" + "io" + "math" + "math/rand/v2" + "slices" + "time" + + "git.anhgelus.world/anhgelus/les-copaings-bot/config" + "git.anhgelus.world/anhgelus/les-copaings-bot/exp" + "git.anhgelus.world/anhgelus/les-copaings-bot/user" + "github.com/anhgelus/gokord" + "github.com/anhgelus/gokord/cmd" + "github.com/anhgelus/gokord/logger" + "github.com/jackc/pgx/v5/pgtype" + discordgo "github.com/nyttikord/gokord" + "gonum.org/v1/plot" + "gonum.org/v1/plot/plotter" + "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 +} + +func Stats(s *discordgo.Session, i *discordgo.InteractionCreate, opt cmd.OptionMap, resp *cmd.ResponseBuilder) { + cfg := config.GetGuildConfig(i.GuildID) + days := cfg.DaysXPRemains + if v, ok := opt["days"]; ok { + in := v.IntValue() + if in < 0 || uint(in) > days { + if err := resp.SetMessage("Nombre de jours invalide").IsEphemeral().Send(); err != nil { + logger.Alert("commands/stats.go - Sending invalid days", err.Error()) + } + return + } + days = uint(in) + } + err := resp.IsDeferred().Send() + if err != nil { + logger.Alert("commands/stats.go - Sending deferred", err.Error()) + return + } + go func() { + var w io.WriterTo + if v, ok := opt["user"]; ok { + w, err = statsMember(s, i, days, v.UserValue(s).ID) + } else { + w, err = statsAll(s, i, days) + } + if err != nil { + if err = resp.IsEphemeral().SetMessage("Il y a eu une erreur...").Send(); err != nil { + logger.Alert("commands/stats.go - Sending error occurred", err.Error()) + } + return + } + b := new(bytes.Buffer) + _, err = w.WriteTo(b) + if err != nil { + logger.Alert("commands/stats.go - Writing png", err.Error()) + } + err = resp.AddFile(&discordgo.File{ + Name: "plot.png", + ContentType: "image/png", + Reader: b, + }).Send() + if err != nil { + logger.Alert("commands/stats.go - Sending response", err.Error()) + } + }() +} + +func statsAll(s *discordgo.Session, i *discordgo.InteractionCreate, days uint) (io.WriterTo, error) { + return stats(s, i, days, func(before, after string) *gorm.DB { + return gokord.DB.Raw(before+"WHERE guild_id = ? and created_at > ?"+after, i.GuildID, exp.TimeStampNDaysBefore(days)) + }) +} + +func statsMember(s *discordgo.Session, i *discordgo.InteractionCreate, days uint, discordID string) (io.WriterTo, error) { + _, err := s.GuildMember(i.GuildID, discordID) + if err != nil { + return nil, err + } + return stats(s, i, days, func(before, after string) *gorm.DB { + return gokord.DB.Raw( + before+"WHERE guild_id = ? and created_at > ? and copaing_id = ?"+after, + i.GuildID, exp.TimeStampNDaysBefore(days), user.GetCopaing(discordID, i.GuildID).ID, + ) + }) +} + +func stats(s *discordgo.Session, i *discordgo.InteractionCreate, days uint, execSql func(before, after string) *gorm.DB) (io.WriterTo, error) { + var rawData []*data + if gokord.Debug { + var rawCopaingData []*user.CopaingXP + if err := execSql("SELECT * FROM copaing_xps ", "").Scan(&rawCopaingData).Error; err != nil { + logger.Alert("commands/stats.go - Fetching result", err.Error()) + return nil, err + } + 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 { + var rawDbData []dbData + if err := execSql( + `SELECT "created_at"::date::text, sum("xp") as xp, "copaing_id" FROM copaing_xps `, ` GROUP BY "created_at"::date, "copaing_id"`, + ).Scan(&rawDbData).Error; err != nil { + logger.Alert("commands/stats.go - Fetching result", err.Error()) + return nil, err + } + 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{} + stats := map[int][]plotter.XY{} + + for _, raw := range rawData { + _, ok := copaings[raw.CopaingID] + if !ok { + var cp user.Copaing + if err := gokord.DB.First(&cp, raw.CopaingID).Error; err != nil { + if !errors.Is(err, gorm.ErrRecordNotFound) { + logger.Alert("commands/stats.go - Finding copaing", err.Error(), "id", raw.CopaingID) + return nil, err + } + logger.Warn("Copaing not found, skipping entry", "old_id", raw.CopaingID) + continue + } + copaings[raw.CopaingID] = &cp + } + pts, ok := stats[raw.CopaingID] + now := time.Now().Unix() + if !ok { + pts = make([]plotter.XY, days+1) + for i := 0; i < len(pts); i++ { + pts[i] = plotter.XY{ + X: float64(i - int(days)), + Y: 0, + } + } + stats[raw.CopaingID] = pts + } + t := raw.CreatedAt.Unix() - now + if !gokord.Debug { + t = int64(math.Ceil(float64(t) / (24 * 60 * 60))) + } + pts[int64(days)+t] = plotter.XY{ // because t <= 0 + X: float64(t), + Y: float64(raw.XP), + } + } + return generatePlot(s, i, copaings, stats) +} + +func generatePlot(s *discordgo.Session, i *discordgo.InteractionCreate, copaings map[int]*user.Copaing, stats map[int][]plotter.XY) (io.WriterTo, error) { + p := plot.New() + p.Title.Text = "Évolution de l'XP" + p.X.Label.Text = "Jours" + if gokord.Debug { + p.X.Label.Text = "Secondes" + } + 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 nil, err + } + 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 + }) + l, err := plotter.NewLine(plotter.XYs(stats[in])) + if err != nil { + logger.Alert("commands/stats.go - Adding line points", err.Error()) + return nil, err + } + 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(8*vg.Inch, 6*vg.Inch, "png") + if err != nil { + logger.Alert("commands/stats.go - Generating png", err.Error()) + return nil, err + } + return w, nil +} |