From bee5b676fa3b9096eecec2ee837482003209f470 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Sun, 1 Sep 2024 16:51:13 +0800
Subject: [PATCH] :bug: Fix daily check issue

---
 .idea/workspace.xml           | 12 ++++--------
 pkg/internal/services/sign.go |  5 ++++-
 2 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 490c578..aac6aa9 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -4,13 +4,9 @@
     <option name="autoReloadType" value="ALL" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":bug: Fix query statement column issue">
-      <change afterPath="$PROJECT_DIR$/pkg/internal/models/sign_record.go" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/pkg/internal/server/api/sign_api.go" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/pkg/internal/services/sign.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.idea/dataSources.local.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/dataSources.local.xml" afterDir="false" />
+    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":sparkles: Daily signs">
       <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/services/sign.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/sign.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -157,7 +153,6 @@
     </option>
   </component>
   <component name="VcsManagerConfiguration">
-    <MESSAGE value=":sparkles: Account groups" />
     <MESSAGE value=":sparkles: Default user group" />
     <MESSAGE value=":bug: Fix permissions in groups" />
     <MESSAGE value=":wastebasket: Clean up code" />
@@ -182,7 +177,8 @@
     <MESSAGE value=":bug: Fix preloading issue" />
     <MESSAGE value=":bug: Fix bot related key api issue" />
     <MESSAGE value=":bug: Fix query statement column issue" />
-    <option name="LAST_COMMIT_MESSAGE" value=":bug: Fix query statement column issue" />
+    <MESSAGE value=":sparkles: Daily signs" />
+    <option name="LAST_COMMIT_MESSAGE" value=":sparkles: Daily signs" />
   </component>
   <component name="VgoProject">
     <settings-migrated>true</settings-migrated>
diff --git a/pkg/internal/services/sign.go b/pkg/internal/services/sign.go
index 9b3d824..8db8ac9 100644
--- a/pkg/internal/services/sign.go
+++ b/pkg/internal/services/sign.go
@@ -7,11 +7,14 @@ import (
 	"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
 	"gorm.io/gorm"
 	"math/rand"
+	"time"
 )
 
 func CheckDailyCanSign(user models.Account) 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).First(&record).Error; err != nil {
+	if err := database.C.Where("account_id = ? AND created_at::date = ?", user.ID, probe).First(&record).Error; err != nil {
 		if errors.Is(err, gorm.ErrRecordNotFound) {
 			return nil
 		}