wexa-chat 0.1.21 → 0.1.23

package/dist/index.mjs

@@ -498,7 +498,7 @@ function createMessagesService(models, hooks = {}) {
         }
       );
       if (onMessageCreated) {
-        onMessageCreated(message);
+        await onMessageCreated(message);
       }
       return message;
     },
@@ -657,6 +657,8 @@ function createMessagesService(models, hooks = {}) {
 // src/transport/localSubs.ts
 var convSubs = /* @__PURE__ */ new Map();
 var socketSubs = /* @__PURE__ */ new Map();
+var userSubs = /* @__PURE__ */ new Map();
+var socketUserSubs = /* @__PURE__ */ new Map();
 function subscribeLocal(conversationId, socketId) {
   var _a, _b;
   if (!convSubs.has(conversationId)) {
@@ -698,6 +700,19 @@ function cleanupLocal(socketId) {
     }
     socketSubs.delete(socketId);
   }
+  const users = socketUserSubs.get(socketId);
+  if (users) {
+    for (const userId of users) {
+      const sockets = userSubs.get(userId);
+      if (sockets) {
+        sockets.delete(socketId);
+        if (sockets.size === 0) {
+          userSubs.delete(userId);
+        }
+      }
+    }
+    socketUserSubs.delete(socketId);
+  }
 }
 function fanoutLocal(conversationId, payload, emitter) {
   const sockets = convSubs.get(conversationId);
@@ -711,6 +726,52 @@ function fanoutLocal(conversationId, payload, emitter) {
     }
   }
 }
+function subscribeToUser(userId, socketId) {
+  var _a, _b;
+  if (!userSubs.has(userId)) {
+    userSubs.set(userId, /* @__PURE__ */ new Set());
+  }
+  (_a = userSubs.get(userId)) == null ? void 0 : _a.add(socketId);
+  if (!socketUserSubs.has(socketId)) {
+    socketUserSubs.set(socketId, /* @__PURE__ */ new Set());
+  }
+  (_b = socketUserSubs.get(socketId)) == null ? void 0 : _b.add(userId);
+}
+function unsubscribeFromUser(userId, socketId) {
+  const sockets = userSubs.get(userId);
+  if (sockets) {
+    sockets.delete(socketId);
+    if (sockets.size === 0) {
+      userSubs.delete(userId);
+    }
+  }
+  const users = socketUserSubs.get(socketId);
+  if (users) {
+    users.delete(userId);
+    if (users.size === 0) {
+      socketUserSubs.delete(socketId);
+    }
+  }
+}
+function isSubscribedToUser(userId, socketId) {
+  var _a;
+  return ((_a = socketUserSubs.get(socketId)) == null ? void 0 : _a.has(userId)) || false;
+}
+function getSocketsForUser(userId) {
+  return userSubs.get(userId) || /* @__PURE__ */ new Set();
+}
+function fanoutToUser(userId, payload, emitter) {
+  const sockets = userSubs.get(userId);
+  if (sockets) {
+    for (const socketId of sockets) {
+      try {
+        emitter(socketId, payload);
+      } catch (error) {
+        console.error(`Error emitting to socket ${socketId}:`, error);
+      }
+    }
+  }
+}
 var SHARED_CLIENT = null;
 var SHARED_CFG_FINGERPRINT = null;
 function createRedisClient(config) {
@@ -807,6 +868,8 @@ var SocketTransport = class {
     this.sockets = /* @__PURE__ */ new Map();
     this.users = /* @__PURE__ */ new Map();
     this.joinedConversations = /* @__PURE__ */ new Map();
+    // socketId -> convIds
+    this.subscribedToUsers = /* @__PURE__ */ new Map();
     // Socket.IO-like shim for getIO().sockets.sockets.get(id).emit('chat-event', payload)
     this.ioShim = {
       sockets: {
@@ -916,6 +979,7 @@ var SocketTransport = class {
         this.joinedConversations.set(socketId, /* @__PURE__ */ new Set());
         syncShim();
         ws.on("message", (raw) => {
+          var _a;
           try {
             const msg = JSON.parse(raw.toString());
             const type = msg == null ? void 0 : msg.type;
@@ -986,6 +1050,17 @@ var SocketTransport = class {
                 }
               });
               this.transport.publishToConversation(convId, payload);
+            } else if (type === "subscribe-to-user" && typeof msg.userId === "string") {
+              const userId2 = msg.userId;
+              this.transport.subscribeToUser(userId2, socketId);
+              if (!this.subscribedToUsers.has(socketId)) {
+                this.subscribedToUsers.set(socketId, /* @__PURE__ */ new Set());
+              }
+              this.subscribedToUsers.get(socketId).add(userId2);
+            } else if (type === "unsubscribe-from-user" && typeof msg.userId === "string") {
+              const userId2 = msg.userId;
+              this.transport.unsubscribeFromUser(userId2, socketId);
+              (_a = this.subscribedToUsers.get(socketId)) == null ? void 0 : _a.delete(userId2);
             }
           } catch {
           }
@@ -1014,6 +1089,13 @@ var SocketTransport = class {
               });
             }
           }
+          const subscribedUsers = this.subscribedToUsers.get(socketId);
+          if (subscribedUsers) {
+            for (const userId2 of subscribedUsers) {
+              this.transport.unsubscribeFromUser(userId2, socketId);
+            }
+            this.subscribedToUsers.delete(socketId);
+          }
           this.transport.cleanupLocal(socketId);
           this.sockets.delete(socketId);
           this.users.delete(socketId);
@@ -1078,6 +1160,12 @@ function createTransport(config, server) {
     unsubscribeLocal,
     cleanupLocal,
     fanoutLocal,
+    // User subscription methods
+    subscribeToUser,
+    unsubscribeFromUser,
+    isSubscribedToUser,
+    getSocketsForUser,
+    fanoutToUser,
     // Redis methods
     publishToConversation: (conversationId, payload) => publishToConversation(redisClient, conversationId, payload),
     startRedisListener: (onEvent) => startRedisListener(redisClient, onEvent)
@@ -1246,7 +1334,7 @@ async function initChat(mongooseInstance, options, server) {
   const services = {
     conversations: createConversationsService(models),
     messages: createMessagesService(models, {
-      onMessageCreated: (message) => {
+      onMessageCreated: async (message) => {
         const payload = { type: "message:created", message };
         transport.fanoutLocal(message.conversationId.toString(), payload, (socketId, event) => {
           if (transport.socket) {
@@ -1256,6 +1344,26 @@ async function initChat(mongooseInstance, options, server) {
             }
           }
         });
+        try {
+          const ConversationModel = models.Conversation;
+          const conversation = await ConversationModel.findById(message.conversationId);
+          if (conversation && conversation.participants) {
+            for (const participant of conversation.participants) {
+              if (participant.entityModel === "User") {
+                transport.fanoutToUser(participant.entityId, payload, (socketId, event) => {
+                  if (transport.socket) {
+                    const socket = transport.socket.getIO().sockets.sockets.get(socketId);
+                    if (socket) {
+                      socket.emit("chat-event", event);
+                    }
+                  }
+                });
+              }
+            }
+          }
+        } catch (error) {
+          console.error("Error fanning out to user subscribers:", error);
+        }
         transport.publishToConversation(message.conversationId.toString(), payload);
       },
       onMessageRead: (args) => {