121 lines
2.6 KiB
Go
121 lines
2.6 KiB
Go
package xp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/anhgelus/gokord"
|
|
"github.com/anhgelus/gokord/utils"
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/redis/go-redis/v9"
|
|
"gorm.io/gorm"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type Copaing struct {
|
|
gorm.Model
|
|
DiscordID string `gorm:"not null"`
|
|
XP uint `gorm:"default:0"`
|
|
GuildID string `gorm:"not null"`
|
|
}
|
|
|
|
var redisClient *redis.Client
|
|
|
|
const (
|
|
LastEvent = "last_event"
|
|
)
|
|
|
|
func GetCopaing(discordID string, guildID string) *Copaing {
|
|
c := Copaing{DiscordID: discordID, GuildID: guildID}
|
|
return c.Load()
|
|
}
|
|
|
|
func (c *Copaing) Load() *Copaing {
|
|
gokord.DB.Where("discord_id = ? and guild_id = ?", c.DiscordID, c.GuildID).FirstOrCreate(c)
|
|
return c
|
|
}
|
|
|
|
func (c *Copaing) Save() {
|
|
gokord.DB.Save(c)
|
|
}
|
|
|
|
func (c *Copaing) AddXP(s *discordgo.Session, m *discordgo.Member, xp uint, fn func(uint, uint)) {
|
|
pastLevel := Level(c.XP)
|
|
c.XP += xp
|
|
c.Save()
|
|
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
|
|
}
|
|
u := c.GetUserBase()
|
|
t := time.Now().Unix()
|
|
err = client.Set(context.Background(), fmt.Sprintf(
|
|
"%s:%s",
|
|
u.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", u.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
|
|
}
|
|
u := c.GetUserBase()
|
|
res := client.Get(context.Background(), fmt.Sprintf("%s:%s", u.GenKey(), LastEvent))
|
|
if res.Err() != nil {
|
|
utils.SendAlert("xp/member.go - Getting last event", res.Err().Error(), "base_key", u.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",
|
|
res.Err().Error(),
|
|
"base_key",
|
|
u.GenKey(),
|
|
"val",
|
|
res.Val(),
|
|
)
|
|
return 0
|
|
}
|
|
return utils.HoursOfUnix(t - int64(last))
|
|
}
|
|
|
|
func (c *Copaing) GetUserBase() *gokord.UserBase {
|
|
return &gokord.UserBase{DiscordID: c.DiscordID, GuildID: c.GuildID}
|
|
}
|
|
|
|
func getRedisClient() (*redis.Client, error) {
|
|
if redisClient == nil {
|
|
var err error
|
|
redisClient, err = gokord.BaseCfg.Redis.Get()
|
|
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())
|
|
}
|
|
}
|