aboutsummaryrefslogtreecommitdiff
path: root/xp/member.go
diff options
context:
space:
mode:
Diffstat (limited to 'xp/member.go')
-rw-r--r--xp/member.go296
1 files changed, 0 insertions, 296 deletions
diff --git a/xp/member.go b/xp/member.go
deleted file mode 100644
index ad9e6c5..0000000
--- a/xp/member.go
+++ /dev/null
@@ -1,296 +0,0 @@
-package xp
-
-import (
- "context"
- "errors"
- "fmt"
- "github.com/anhgelus/gokord"
- "github.com/anhgelus/gokord/utils"
- "github.com/bwmarrin/discordgo"
- "github.com/redis/go-redis/v9"
- "gorm.io/gorm"
- "math"
- "strconv"
- "time"
-)
-
-type Copaing struct {
- gorm.Model
- DiscordID string `gorm:"not null"`
- XP uint `gorm:"default:0"`
- GuildID string `gorm:"not null"`
-}
-
-type leftCopaing struct {
- ID uint
- StopDelete chan<- interface{}
-}
-
-var (
- redisClient *redis.Client
-
- leftCopaingsMap = map[string]*leftCopaing{}
-)
-
-const (
- LastEvent = "last_event"
- AlreadyRemoved = "already_removed"
-)
-
-func GetCopaing(discordID string, guildID string) *Copaing {
- c := Copaing{DiscordID: discordID, GuildID: guildID}
- if err := c.Load(); err != nil {
- utils.SendAlert(
- "xp/member.go - Loading copaing",
- err.Error(),
- "discord_id",
- discordID,
- "guild_id",
- guildID,
- )
- return nil
- }
- return &c
-}
-
-func (c *Copaing) Load() error {
- // check if user left in the past 48 hours
- k := c.GuildID + ":" + c.DiscordID
- l, ok := leftCopaingsMap[k]
- if !ok || l == nil {
- // if not, common first or create
- return gokord.DB.Where("discord_id = ? and guild_id = ?", c.DiscordID, c.GuildID).FirstOrCreate(c).Error
- }
- // else, getting last data
- tmp := Copaing{
- Model: gorm.Model{
- ID: c.ID,
- },
- DiscordID: c.DiscordID,
- GuildID: c.GuildID,
- }
- if err := gokord.DB.Unscoped().Find(&tmp).Error; err != nil {
- // if error, avoid getting old data and use new one
- utils.SendAlert(
- "xp/member.go - Getting copaing in soft delete", err.Error(),
- "discord_id", c.DiscordID,
- "guild_id", c.DiscordID,
- "last_id", l.ID,
- )
- return gokord.DB.Where("discord_id = ? and guild_id = ?", c.DiscordID, c.GuildID).FirstOrCreate(c).Error
- }
- // resetting internal data
- tmp.Model = gorm.Model{}
- l.StopDelete <- true
- leftCopaingsMap[k] = nil
- // creating new data
- err := gokord.DB.Create(&tmp).Error
- if err != nil {
- return err
- }
- // delete old data
- if err = gokord.DB.Unscoped().Delete(&tmp).Error; err != nil {
- utils.SendAlert(
- "xp/member.go - Deleting copaing in soft delete", err.Error(),
- "discord_id", c.DiscordID,
- "guild_id", c.DiscordID,
- "last_id", l.ID,
- )
- }
- return nil
-}
-
-func (c *Copaing) Save() error {
- return gokord.DB.Save(c).Error
-}
-
-func (c *Copaing) GenKey(key string) string {
- return fmt.Sprintf("%s:%s:%s", c.GuildID, c.DiscordID, key)
-}
-
-func (c *Copaing) AddXP(s *discordgo.Session, m *discordgo.Member, xp uint, fn func(uint, uint)) {
- pastLevel := Level(c.XP)
- old := c.XP
- c.XP += xp
- if err := c.Save(); err != nil {
- utils.SendAlert(
- "xp/level.go - Saving copaing",
- err.Error(),
- "xp",
- c.XP,
- "discord_id",
- c.DiscordID,
- "guild_id",
- c.GuildID,
- )
- c.XP = old
- return
- }
- newLevel := Level(c.XP)
- if newLevel > pastLevel {
- fn(c.XP, newLevel)
- onNewLevel(s, m, newLevel)
- }
-}
-
-func (c *Copaing) SetLastEvent() {
- client, err := getRedisClient()
- if err != nil {
- utils.SendAlert("xp/member.go - Getting redis client (set)", err.Error())
- return
- }
- t := time.Now().Unix()
- err = client.Set(context.Background(), c.GenKey(LastEvent), strconv.FormatInt(t, 10), 0).Err()
- if err != nil {
- utils.SendAlert("xp/member.go - Setting last event", err.Error(), "time", t, "base_key", c.GenKey(""))
- return
- }
- err = client.Set(context.Background(), c.GenKey(AlreadyRemoved), "0", 0).Err()
- if err != nil {
- utils.SendAlert(
- "xp/member.go - Setting already removed to 0",
- err.Error(),
- "time",
- t,
- "base_key",
- c.GenKey(""),
- )
- return
- }
-}
-
-func (c *Copaing) HourSinceLastEvent() uint {
- client, err := getRedisClient()
- if err != nil {
- utils.SendAlert("xp/member.go - Getting redis client (get)", err.Error())
- return 0
- }
- res := client.Get(context.Background(), c.GenKey(LastEvent))
- if errors.Is(res.Err(), redis.Nil) {
- return 0
- } else if res.Err() != nil {
- utils.SendAlert("xp/member.go - Getting last event", res.Err().Error(), "base_key", c.GenKey(""))
- return 0
- }
- t := time.Now().Unix()
- last, err := strconv.Atoi(res.Val())
- if err != nil {
- utils.SendAlert(
- "xp/member.go - Converting time fetched into int (last event)",
- err.Error(),
- "base_key",
- c.GenKey(""),
- "val",
- res.Val(),
- )
- return 0
- }
- if gokord.Debug {
- return uint(math.Floor(float64(t-int64(last)) / 60)) // not hours of unix, is minutes of unix
- }
- return utils.HoursOfUnix(t - int64(last))
-}
-
-func (c *Copaing) AddXPAlreadyRemoved(xp uint) uint {
- client, err := getRedisClient()
- if err != nil {
- utils.SendAlert("xp/member.go - Getting redis client (set)", err.Error())
- return 0
- }
- exp := xp + c.XPAlreadyRemoved()
- err = client.Set(context.Background(), c.GenKey(AlreadyRemoved), exp, 0).Err()
- if err != nil {
- utils.SendAlert(
- "xp/member.go - Setting already removed",
- err.Error(),
- "xp already removed",
- exp,
- "base_key",
- c.GenKey(""),
- )
- return 0
- }
- return exp
-}
-
-func (c *Copaing) XPAlreadyRemoved() uint {
- client, err := getRedisClient()
- if err != nil {
- utils.SendAlert("xp/member.go - Getting redis client (xp)", err.Error())
- return 0
- }
- res := client.Get(context.Background(), fmt.Sprintf("%s:%s", c.GenKey(""), AlreadyRemoved))
- if errors.Is(res.Err(), redis.Nil) {
- return 0
- } else if res.Err() != nil {
- utils.SendAlert("xp/member.go - Getting already removed", res.Err().Error(), "base_key", c.GenKey(""))
- return 0
- }
- xp, err := strconv.Atoi(res.Val())
- if err != nil {
- utils.SendAlert(
- "xp/member.go - Converting time fetched into int (already removed)",
- err.Error(),
- "base_key",
- c.GenKey(""),
- "val",
- res.Val(),
- )
- return 0
- }
- if xp < 0 {
- utils.SendAlert(
- "xp/member.go - Assertion xp >= 0",
- "xp is negative",
- "base_key",
- c.GenKey(""),
- "xp",
- xp,
- )
- return 0
- }
- return uint(xp)
-}
-
-func (c *Copaing) Reset() {
- gokord.DB.Where("guild_id = ? AND discord_id = ?", c.GuildID, c.DiscordID).Delete(c)
-}
-
-func (c *Copaing) AfterDelete(db *gorm.DB) error {
- id := c.ID
- dID := c.DiscordID
- gID := c.GuildID
- k := c.GuildID + ":" + c.DiscordID
- ch := utils.NewTimer(48*time.Hour, func(stop chan<- interface{}) {
- if err := db.Unscoped().Where("id = ?", id).Delete(c).Error; err != nil {
- utils.SendAlert(
- "xp/member.go - Removing copaing from database", err.Error(),
- "discord_id", dID,
- "guild_id", gID,
- )
- }
- stop <- true
- leftCopaingsMap[k] = nil
- })
- leftCopaingsMap[k] = &leftCopaing{id, ch}
- return nil
-}
-
-func getRedisClient() (*redis.Client, error) {
- if redisClient == nil {
- var err error
- redisClient, err = gokord.BaseCfg.GetRedisCredentials().Connect()
- return redisClient, err
- }
- return redisClient, nil
-}
-
-func CloseRedisClient() {
- if redisClient == nil {
- return
- }
- err := redisClient.Close()
- if err != nil {
- utils.SendAlert("xp/member.go - Closing redis client", err.Error())
- }
-}
generated by cgit v1.2.3 (git 2.25.1) at 2026-05-03 16:05:46 +0000