1 files changed, 143 insertions, 0 deletions
diff --git a/xp/member.go b/xp/member.go
index 4f1e6ee..bbf013c 100644
--- a/xp/member.go
+++ b/xp/member.go
@@ -1,11 +1,17 @@
 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 {
@@ -17,6 +23,11 @@ type Copaing struct {
 
 var redisClient *redis.Client
 
+const (
+	LastEvent      = "last_event"
+	AlreadyRemoved = "already_removed"
+)
+
 func GetCopaing(discordID string, guildID string) *Copaing {
 	c := Copaing{DiscordID: discordID, GuildID: guildID}
 	return c.Load()
@@ -42,6 +53,138 @@ func (c *Copaing) AddXP(s *discordgo.Session, m *discordgo.Member, xp uint, fn f
 	}
 }
 
+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
+	}
+	err = client.Set(context.Background(), fmt.Sprintf(
+		"%s:%s",
+		u.GenKey(),
+		AlreadyRemoved,
+	), "0", 0).Err()
+	if err != nil {
+		utils.SendAlert("xp/member.go - Setting already removed to 0", 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 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", 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 (last event)",
+			err.Error(),
+			"base_key",
+			u.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
+	}
+	u := c.GetUserBase()
+	exp := xp + c.XPAlreadyRemoved()
+	err = client.Set(context.Background(), fmt.Sprintf(
+		"%s:%s",
+		u.GenKey(),
+		AlreadyRemoved,
+	), exp, 0).Err()
+	if err != nil {
+		utils.SendAlert(
+			"xp/member.go - Setting already removed",
+			err.Error(),
+			"xp already removed",
+			exp,
+			"base_key",
+			u.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
+	}
+	u := c.GetUserBase()
+	res := client.Get(context.Background(), fmt.Sprintf("%s:%s", u.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", u.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",
+			u.GenKey(),
+			"val",
+			res.Val(),
+		)
+		return 0
+	}
+	if xp < 0 {
+		utils.SendAlert(
+			"xp/member.go - Assertion xp >= 0",
+			"xp is negative",
+			"base_key",
+			u.GenKey(),
+			"xp",
+			xp,
+		)
+		return 0
+	}
+	return uint(xp)
+}
+
+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