From bfddfa220198276d86096012b9563088c50615d5 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Mon, 2 Sep 2024 20:07:19 +0800
Subject: [PATCH] :sparkles: Can get today's daily sign record

---
 .idea/workspace.xml                 |  8 +++++---
 pkg/internal/server/api/index.go    |  1 +
 pkg/internal/server/api/sign_api.go | 13 +++++++++++++
 pkg/internal/services/sign.go       | 18 ++++++++++++++++++
 4 files changed, 37 insertions(+), 3 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index aac6aa9..81822e6 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -4,8 +4,10 @@
     <option name="autoReloadType" value="ALL" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":sparkles: Daily signs">
+    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":bug: Fix daily check issue">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/server/api/index.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/server/api/index.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/server/api/sign_api.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/server/api/sign_api.go" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/pkg/internal/services/sign.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/sign.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
@@ -153,7 +155,6 @@
     </option>
   </component>
   <component name="VcsManagerConfiguration">
-    <MESSAGE value=":sparkles: Default user group" />
     <MESSAGE value=":bug: Fix permissions in groups" />
     <MESSAGE value=":wastebasket: Clean up code" />
     <MESSAGE value=":sparkles: Use capital to deal with links" />
@@ -178,7 +179,8 @@
     <MESSAGE value=":bug: Fix bot related key api issue" />
     <MESSAGE value=":bug: Fix query statement column issue" />
     <MESSAGE value=":sparkles: Daily signs" />
-    <option name="LAST_COMMIT_MESSAGE" value=":sparkles: Daily signs" />
+    <MESSAGE value=":bug: Fix daily check issue" />
+    <option name="LAST_COMMIT_MESSAGE" value=":bug: Fix daily check issue" />
   </component>
   <component name="VgoProject">
     <settings-migrated>true</settings-migrated>
diff --git a/pkg/internal/server/api/index.go b/pkg/internal/server/api/index.go
index a6c9831..3fa6beb 100644
--- a/pkg/internal/server/api/index.go
+++ b/pkg/internal/server/api/index.go
@@ -12,6 +12,7 @@ func MapAPIs(app *fiber.App, baseURL string) {
 		daily := api.Group("/daily").Name("Daily Sign API")
 		{
 			daily.Get("/", listDailySignRecord)
+			daily.Get("/today", getTodayDailySign)
 			daily.Post("/", doDailySign)
 		}
 
diff --git a/pkg/internal/server/api/sign_api.go b/pkg/internal/server/api/sign_api.go
index c8765be..dcb9d03 100644
--- a/pkg/internal/server/api/sign_api.go
+++ b/pkg/internal/server/api/sign_api.go
@@ -40,6 +40,19 @@ func listDailySignRecord(c *fiber.Ctx) error {
 	})
 }
 
+func getTodayDailySign(c *fiber.Ctx) error {
+	if err := exts.EnsureAuthenticated(c); err != nil {
+		return err
+	}
+	user := c.Locals("user").(models.Account)
+
+	if record, err := services.GetTodayDailySign(user); err != nil {
+		return fiber.NewError(fiber.StatusNotFound, err.Error())
+	} else {
+		return c.JSON(record)
+	}
+}
+
 func doDailySign(c *fiber.Ctx) error {
 	if err := exts.EnsureAuthenticated(c); err != nil {
 		return err
diff --git a/pkg/internal/services/sign.go b/pkg/internal/services/sign.go
index 8db8ac9..aeb9069 100644
--- a/pkg/internal/services/sign.go
+++ b/pkg/internal/services/sign.go
@@ -23,6 +23,16 @@ func CheckDailyCanSign(user models.Account) error {
 	return fmt.Errorf("daliy sign record exists")
 }
 
+func GetTodayDailySign(user models.Account) (models.SignRecord, error) {
+	probe := time.Now().Format("YYYY-MM-DD")
+
+	var record models.SignRecord
+	if err := database.C.Where("account_id = ? AND created_at::date = ?", user.ID, probe).First(&record).Error; err != nil {
+		return record, fmt.Errorf("unable get daliy sign record: %v", err)
+	}
+	return record, nil
+}
+
 func DailySign(user models.Account) (models.SignRecord, error) {
 	tier := rand.Intn(5)
 	record := models.SignRecord{
@@ -35,6 +45,14 @@ func DailySign(user models.Account) (models.SignRecord, error) {
 		return record, fmt.Errorf("today already signed")
 	}
 
+	var profile models.AccountProfile
+	if err := database.C.Where("account_id = ?", user.ID).First(&profile).Error; err != nil {
+		return record, fmt.Errorf("unable get account profile: %v", err)
+	} else {
+		profile.Experience += uint64(record.ResultExperience)
+		database.C.Save(&profile)
+	}
+
 	if err := database.C.Save(&record).Error; err != nil {
 		return record, fmt.Errorf("unable do daliy sign: %v", err)
 	}