From 6ec48aaa8a8ea290b53da413f15214c56e76abd1 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Sun, 28 Apr 2024 19:59:27 +0800
Subject: [PATCH] :sparkles: Batch mark notify as read API

---
 .idea/workspace.xml             |  2 ++
 pkg/server/notifications_api.go | 28 ++++++++++++++++++++++++----
 pkg/server/startup.go           |  3 ++-
 3 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 08ecef9..4ab9e55 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -6,6 +6,8 @@
   <component name="ChangeListManager">
     <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":sparkles: Bug fixes">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/server/notifications_api.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/server/notifications_api.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/server/startup.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/server/startup.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
diff --git a/pkg/server/notifications_api.go b/pkg/server/notifications_api.go
index e2c76d6..4ad4f5c 100644
--- a/pkg/server/notifications_api.go
+++ b/pkg/server/notifications_api.go
@@ -48,17 +48,37 @@ func markNotificationRead(c *fiber.Ctx) error {
 	user := c.Locals("principal").(models.Account)
 	id, _ := c.ParamsInt("notificationId", 0)
 
-	var data models.Notification
+	var notify models.Notification
 	if err := database.C.Where(&models.Notification{
 		BaseModel:   models.BaseModel{ID: uint(id)},
 		RecipientID: user.ID,
-	}).First(&data).Error; err != nil {
+	}).First(&notify).Error; err != nil {
 		return fiber.NewError(fiber.StatusNotFound, err.Error())
 	}
 
-	data.ReadAt = lo.ToPtr(time.Now())
+	notify.ReadAt = lo.ToPtr(time.Now())
 
-	if err := database.C.Save(&data).Error; err != nil {
+	if err := database.C.Save(&notify).Error; err != nil {
+		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
+	} else {
+		return c.SendStatus(fiber.StatusOK)
+	}
+}
+
+func markNotificationReadBatch(c *fiber.Ctx) error {
+	user := c.Locals("principal").(models.Account)
+
+	var data struct {
+		MessageIDs []uint `json:"messages"`
+	}
+
+	if err := utils.BindAndValidate(c, &data); err != nil {
+		return fiber.NewError(fiber.StatusBadRequest, err.Error())
+	}
+
+	if err := database.C.Model(&models.Notification{}).
+		Where("recipient_id = ? AND id IN ?", user.ID, data.MessageIDs).
+		Updates(&models.Notification{ReadAt: lo.ToPtr(time.Now())}).Error; err != nil {
 		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
 	} else {
 		return c.SendStatus(fiber.StatusOK)
diff --git a/pkg/server/startup.go b/pkg/server/startup.go
index 3b0452e..9775b20 100644
--- a/pkg/server/startup.go
+++ b/pkg/server/startup.go
@@ -71,8 +71,9 @@ func NewServer() {
 		notify := api.Group("/notifications").Name("Notifications API")
 		{
 			notify.Get("/", authMiddleware, getNotifications)
-			notify.Put("/:notificationId/read", authMiddleware, markNotificationRead)
 			notify.Post("/subscribe", authMiddleware, addNotifySubscriber)
+			notify.Put("/batch/read", authMiddleware, markNotificationReadBatch)
+			notify.Put("/:notificationId/read", authMiddleware, markNotificationRead)
 
 			notify.Get("/listen", authMiddleware, websocket.New(listenNotifications))
 		}