From 78604db54ec58fe0287bd26161493cb9e3529e5d Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Sun, 30 Jun 2024 11:32:01 +0800
Subject: [PATCH] :sparkles: Admin notify all API

---
 .idea/workspace.xml                     |  9 ++--
 pkg/internal/server/admin/index.go      |  2 +
 pkg/internal/server/admin/notify_api.go | 64 +++++++++++++++++++++++++
 3 files changed, 71 insertions(+), 4 deletions(-)
 create mode 100644 pkg/internal/server/admin/notify_api.go

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 6d5a957..5c6e168 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=":bug: Fix status query condition">
+    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":bug: Fix disturbable condition">
+      <change afterPath="$PROJECT_DIR$/pkg/internal/server/admin/notify_api.go" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/internal/services/statuses.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/statuses.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/server/admin/index.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/server/admin/index.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -152,7 +153,6 @@
     </option>
   </component>
   <component name="VcsManagerConfiguration">
-    <MESSAGE value=":arrow_up: Fix notification listen" />
     <MESSAGE value=":bug: Fix magic token's foreign key" />
     <MESSAGE value=":sparkles: Better avatar and banner APIs" />
     <MESSAGE value=":bug: Fix avatar and banner APIs" />
@@ -177,7 +177,8 @@
     <MESSAGE value=":bug: Fix status validation issue" />
     <MESSAGE value=":bug: Fix bugs in status" />
     <MESSAGE value=":bug: Fix status query condition" />
-    <option name="LAST_COMMIT_MESSAGE" value=":bug: Fix status query condition" />
+    <MESSAGE value=":bug: Fix disturbable condition" />
+    <option name="LAST_COMMIT_MESSAGE" value=":bug: Fix disturbable condition" />
   </component>
   <component name="VgoProject">
     <settings-migrated>true</settings-migrated>
diff --git a/pkg/internal/server/admin/index.go b/pkg/internal/server/admin/index.go
index 2445d42..33ffbe8 100644
--- a/pkg/internal/server/admin/index.go
+++ b/pkg/internal/server/admin/index.go
@@ -9,5 +9,7 @@ func MapAdminEndpoints(app *fiber.App) {
 	{
 		admin.Post("/badges", grantBadge)
 		admin.Delete("/badges/:badgeId", revokeBadge)
+
+		admin.Post("/notify/all", notifyAllUser)
 	}
 }
diff --git a/pkg/internal/server/admin/notify_api.go b/pkg/internal/server/admin/notify_api.go
new file mode 100644
index 0000000..d2adc34
--- /dev/null
+++ b/pkg/internal/server/admin/notify_api.go
@@ -0,0 +1,64 @@
+package admin
+
+import (
+	"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
+	"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
+	"git.solsynth.dev/hydrogen/passport/pkg/internal/server/exts"
+	"git.solsynth.dev/hydrogen/passport/pkg/internal/services"
+	"github.com/gofiber/fiber/v2"
+	"github.com/rs/zerolog/log"
+)
+
+func notifyAllUser(c *fiber.Ctx) error {
+	var data struct {
+		ClientID     string                    `json:"client_id" validate:"required"`
+		ClientSecret string                    `json:"client_secret" validate:"required"`
+		Type         string                    `json:"type" validate:"required"`
+		Subject      string                    `json:"subject" validate:"required,max=1024"`
+		Content      string                    `json:"content" validate:"required,max=4096"`
+		Metadata     map[string]any            `json:"metadata"`
+		Links        []models.NotificationLink `json:"links"`
+		IsForcePush  bool                      `json:"is_force_push"`
+		IsRealtime   bool                      `json:"is_realtime"`
+		UserID       uint                      `json:"user_id" validate:"required"`
+	}
+
+	if err := exts.BindAndValidate(c, &data); err != nil {
+		return err
+	}
+
+	if err := exts.EnsureGrantedPerm(c, "AdminNotifyAll", true); err != nil {
+		return err
+	}
+
+	var users []models.Account
+	if err := database.C.Find(&users).Error; err != nil {
+		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
+	}
+
+	go func() {
+		for _, user := range users {
+			notification := models.Notification{
+				Type:        data.Type,
+				Subject:     data.Subject,
+				Content:     data.Content,
+				Links:       data.Links,
+				IsRealtime:  data.IsRealtime,
+				IsForcePush: data.IsForcePush,
+				RecipientID: user.ID,
+			}
+
+			if data.IsRealtime {
+				if err := services.PushNotification(notification); err != nil {
+					log.Error().Err(err).Uint("user", user.ID).Msg("Failed to push notification...")
+				}
+			} else {
+				if err := services.NewNotification(notification); err != nil {
+					log.Error().Err(err).Uint("user", user.ID).Msg("Failed to create notification...")
+				}
+			}
+		}
+	}()
+
+	return c.SendStatus(fiber.StatusOK)
+}