From 0f9bde25335b2359f898657568c3a1304dae6716 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Sun, 9 Mar 2025 01:30:30 +0800
Subject: [PATCH] :necktie: Update check in result caluclate expr

---
 pkg/authkit/models/check_in.go    |  1 +
 pkg/internal/services/check_in.go | 34 +++++++++++++++++++++++--------
 2 files changed, 26 insertions(+), 9 deletions(-)

diff --git a/pkg/authkit/models/check_in.go b/pkg/authkit/models/check_in.go
index 8e7e325..59ff6a5 100644
--- a/pkg/authkit/models/check_in.go
+++ b/pkg/authkit/models/check_in.go
@@ -8,6 +8,7 @@ type CheckInRecord struct {
 	ResultTier       int     `json:"result_tier"`
 	ResultExperience int     `json:"result_experience"`
 	ResultCoin       float64 `json:"result_coin"`
+	CurrentStreak    int     `json:"current_streak"`
 
 	// The result modifiers are some random tips that will show up in the client;
 	// This field is to use to make sure the tips will be the same when the client is reloaded.
diff --git a/pkg/internal/services/check_in.go b/pkg/internal/services/check_in.go
index 8250019..87134d6 100644
--- a/pkg/internal/services/check_in.go
+++ b/pkg/internal/services/check_in.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"math"
 	"math/rand"
 	"time"
 
@@ -17,10 +18,8 @@ import (
 )
 
 func CheckCanCheckIn(user models.Account) error {
-	probe := time.Now().Format("2006-01-02")
-
 	var record models.CheckInRecord
-	if err := database.C.Where("account_id = ? AND created_at::date = ?", user.ID, probe).First(&record).Error; err != nil {
+	if err := database.C.Where("account_id = ? AND created_at::date = CURRENT_DATE", user.ID).First(&record).Error; err != nil {
 		if errors.Is(err, gorm.ErrRecordNotFound) {
 			return nil
 		}
@@ -29,11 +28,27 @@ func CheckCanCheckIn(user models.Account) error {
 	return fmt.Errorf("today's check in record exists")
 }
 
-func GetTodayCheckIn(user models.Account) (models.CheckInRecord, error) {
-	probe := time.Now().Format("2006-01-02")
+func GetCheckInStreak(user models.Account) (int64, error) {
+	var streaks int64
+	if err := database.C.Raw(`WITH dates AS (
+			SELECT DISTINCT created_at::DATE AS created_date
+			FROM check_in_records
+			WHERE created_at::DATE <= CURRENT_DATE
+		),
+		streak AS (
+			SELECT created_date,
+				   created_date - INTERVAL '1 day' * ROW_NUMBER() OVER (ORDER BY created_date) AS grp
+			FROM dates
+		)
+		SELECT COUNT(*) FROM streak WHERE grp = (SELECT MIN(grp) FROM streak);`).Scan(&streaks).Error; err != nil {
+		return streaks, err
+	}
+	return streaks, nil
+}
 
+func GetTodayCheckIn(user models.Account) (models.CheckInRecord, error) {
 	var record models.CheckInRecord
-	if err := database.C.Where("account_id = ? AND created_at::date = ?", user.ID, probe).First(&record).Error; err != nil {
+	if err := database.C.Where("account_id = ? AND created_at::date = CURRENT_DATE", user.ID).First(&record).Error; err != nil {
 		return record, fmt.Errorf("unable get check in record: %v", err)
 	}
 	return record, nil
@@ -48,19 +63,20 @@ func CheckIn(user models.Account) (models.CheckInRecord, error) {
 	}
 
 	tier := rand.Intn(5)
+	streak, _ := GetCheckInStreak(user)
 
-	expMax := 100 * (tier + 1)
 	expMin := 100
-	exp := rand.Intn(expMax+1-expMin) + expMin
+	exp := expMin + int(math.Max(float64(streak)*50, 10*50))
 
 	coinMax := 10.0 * float64(tier+1)
 	coinMin := 10.0
-	rawCoins := coinMax + rand.Float64()*(coinMax-coinMin)
+	rawCoins := coinMax + rand.Float64()*(coinMax-coinMin) + math.Max(float64(streak)*10, float64(100*10))
 
 	record = models.CheckInRecord{
 		ResultTier:       tier,
 		ResultExperience: exp,
 		ResultCoin:       float64(int(rawCoins*100)) / 100,
+		CurrentStreak:    int(streak),
 		AccountID:        user.ID,
 	}