From fdf2d28f5165be29bc39d6ef8d6d8978b49e8cf9 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Mon, 9 Dec 2024 23:24:12 +0800
Subject: [PATCH] :zap: A separate API to only get notification count

---
 .idea/workspace.xml                        |  7 ++++---
 pkg/internal/http/api/index.go             |  1 +
 pkg/internal/http/api/notifications_api.go | 19 +++++++++++++++++++
 3 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 6b83b28..ef95159 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -4,9 +4,10 @@
     <option name="autoReloadType" value="ALL" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":zap: Add cache into querying user">
+    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":bug: Fix missing api endpoint">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/pkg/internal/http/api/index.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/http/api/index.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/http/api/notifications_api.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/http/api/notifications_api.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -159,7 +160,6 @@
   </component>
   <component name="VcsManagerConfiguration">
     <option name="CHECK_CODE_SMELLS_BEFORE_PROJECT_COMMIT" value="false" />
-    <MESSAGE value=":bug: Fix notifiable is empty when user do not set" />
     <MESSAGE value=":bug: Fix notification push batch emitted twice" />
     <MESSAGE value=":sparkles: Support jwks.json according OIDC stranded" />
     <MESSAGE value=":bug: Fix get user grpc method didn't extend group's permission" />
@@ -184,7 +184,8 @@
     <MESSAGE value=":bug: Bug fix directory service wasn't registered" />
     <MESSAGE value=":bug: Fix random panic" />
     <MESSAGE value=":zap: Add cache into querying user" />
-    <option name="LAST_COMMIT_MESSAGE" value=":zap: Add cache into querying user" />
+    <MESSAGE value=":bug: Fix missing api endpoint" />
+    <option name="LAST_COMMIT_MESSAGE" value=":bug: Fix missing api endpoint" />
     <option name="GROUP_MULTIFILE_MERGE_BY_DIRECTORY" value="true" />
   </component>
   <component name="VgoProject">
diff --git a/pkg/internal/http/api/index.go b/pkg/internal/http/api/index.go
index 6374142..55fdbeb 100644
--- a/pkg/internal/http/api/index.go
+++ b/pkg/internal/http/api/index.go
@@ -20,6 +20,7 @@ func MapAPIs(app *fiber.App, baseURL string) {
 		notify := api.Group("/notifications").Name("Notifications API")
 		{
 			notify.Get("/", getNotifications)
+			notify.Get("/count", getNotificationCount)
 			notify.Get("/subscription", getNotifySubscriber)
 			notify.Post("/subscription", addNotifySubscriber)
 			notify.Delete("/subscription/:deviceId", removeNotifySubscriber)
diff --git a/pkg/internal/http/api/notifications_api.go b/pkg/internal/http/api/notifications_api.go
index 907de5d..204bada 100644
--- a/pkg/internal/http/api/notifications_api.go
+++ b/pkg/internal/http/api/notifications_api.go
@@ -43,6 +43,25 @@ func getNotifications(c *fiber.Ctx) error {
 	})
 }
 
+func getNotificationCount(c *fiber.Ctx) error {
+	if err := exts.EnsureAuthenticated(c); err != nil {
+		return err
+	}
+	user := c.Locals("user").(models.Account)
+
+	tx := database.C.Where(&models.Notification{AccountID: user.ID}).Model(&models.Notification{})
+
+	var count int64
+	if err := tx.
+		Count(&count).Error; err != nil {
+		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
+	}
+
+	return c.JSON(fiber.Map{
+		"count": count,
+	})
+}
+
 func markNotificationRead(c *fiber.Ctx) error {
 	if err := exts.EnsureAuthenticated(c); err != nil {
 		return err