From 134b11e7f0bce1f6a4102709d24e2f03fea4ecdf Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Thu, 14 Aug 2025 17:39:20 +0800
Subject: [PATCH] :bug: Fix notification missing websocket

---
 .../Notification/PushService.cs                | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/DysonNetwork.Pusher/Notification/PushService.cs b/DysonNetwork.Pusher/Notification/PushService.cs
index c56e4d1..5f39e2e 100644
--- a/DysonNetwork.Pusher/Notification/PushService.cs
+++ b/DysonNetwork.Pusher/Notification/PushService.cs
@@ -111,7 +111,7 @@ public class PushService
         return subscription;
     }
 
-    public async Task SendNotification(Account account,
+    public void SendNotification(Account account,
         string topic,
         string? title = null,
         string? subtitle = null,
@@ -152,6 +152,12 @@ public class PushService
             notification.Id,
             notification.Meta
         );
+        
+        _ws.SendPacketToAccount(notification.AccountId.ToString(), new Connection.WebSocketPacket
+        {
+            Type = "notifications.new",
+            Data = notification
+        });
 
         // Pushing the notification
         var subscribers = await _db.PushSubscriptions
@@ -205,6 +211,16 @@ public class PushService
             notification.Id,
             notification.Meta
         );
+        
+        foreach (var account in accounts)
+        {
+            notification.AccountId = account;
+            _ws.SendPacketToAccount(account.ToString(), new Connection.WebSocketPacket
+            {
+                Type = "notifications.new",
+                Data = notification
+            });
+        }
 
         var subscribers = await _db.PushSubscriptions
             .Where(s => accounts.Contains(s.AccountId))