@learnpack/learnpack 5.0.178 → 5.0.180

package/src/utils/creatorSocket.ts

@@ -1,6 +1,8 @@
 import { Server as SocketIOServer, Socket } from "socket.io"
 
 const courseSocketMap = new Map<string, Set<string>>() // slug -> Set<socket.id>
+
+const notificationSocketMap = new Map<string, Set<string>>() // notificationId -> Set<socket.id>
 const socketStore = new Map<string, Socket>() // socket.id -> socket
 
 let io: SocketIOServer | null = null
@@ -31,6 +33,21 @@ return
       console.log(`📦 Socket ${socket.id} registered to course: ${courseSlug}`)
     })
 
+    socket.on("registerNotification", (data: { notificationId: string }) => {
+      const { notificationId } = data
+      if (!notificationId) 
+return
+
+      if (!notificationSocketMap.has(notificationId)) {
+        notificationSocketMap.set(notificationId, new Set())
+      }
+
+      notificationSocketMap.get(notificationId)?.add(socket.id)
+      console.log(
+        `📧 Socket ${socket.id} registered to notification: ${notificationId}`
+      )
+    })
+
     socket.on("disconnect", () => {
       console.log("🔥 Socket disconnected:", socket.id)
       socketStore.delete(socket.id)
@@ -56,6 +73,19 @@ socket.emit(event, payload)
   }
 }
 
+export function emitToNotification(notificationId: string, payload: any) {
+  console.log("Emitting to notification", notificationId, payload)
+  const socketIds = notificationSocketMap.get(notificationId)
+  if (!socketIds || socketIds.size === 0) 
+return
+
+  for (const id of socketIds) {
+    const socket = socketStore.get(id)
+    if (socket) 
+socket.emit(notificationId, payload)
+  }
+}
+
 export function getSocketIO() {
   if (!io) 
 throw new Error("Socket.IO not initialized")