collabdocchat 1.2.13 → 2.0.0

package/server/websocket/index.js

@@ -33,6 +33,30 @@ export const setupWebSocket = (wss) => {
           case 'typing':
             handleTyping(ws, data, wss);
             break;
+          case 'recall_message':
+            await handleRecallMessage(ws, data, wss);
+            break;
+          case 'message_read':
+            await handleMessageRead(ws, data, wss);
+            break;
+          case 'user_online':
+            handleUserOnline(ws, data, wss);
+            break;
+          case 'whiteboard_draw':
+            handleWhiteboardDraw(ws, data, wss);
+            break;
+          case 'whiteboard_clear':
+            handleWhiteboardClear(ws, data, wss);
+            break;
+          case 'whiteboard_collaboration_toggle':
+            handleWhiteboardCollaborationToggle(ws, data, wss);
+            break;
+          case 'whiteboard_user_join':
+            handleWhiteboardUserJoin(ws, data, wss);
+            break;
+          case 'whiteboard_user_leave':
+            handleWhiteboardUserLeave(ws, data, wss);
+            break;
           default:
             console.log('未知消息类型:', data.type);
         }
@@ -204,6 +228,109 @@ const handleTyping = (ws, data, wss) => {
   broadcastToGroup(wss, ws.groupId, message, ws);
 };
 
+const handleRecallMessage = async (ws, data, wss) => {
+  if (!ws.userId) {
+    ws.send(JSON.stringify({ type: 'error', message: '未认证' }));
+    return;
+  }
+
+  try {
+    const message = await Message.findById(data.messageId);
+    
+    if (!message) {
+      ws.send(JSON.stringify({ type: 'error', message: '消息不存在' }));
+      return;
+    }
+
+    // 检查是否是消息发送者
+    if (message.sender.toString() !== ws.userId) {
+      ws.send(JSON.stringify({ type: 'error', message: '只能撤回自己的消息' }));
+      return;
+    }
+
+    // 检查是否在2分钟内
+    const timeDiff = Date.now() - new Date(message.timestamp).getTime();
+    const twoMinutes = 2 * 60 * 1000;
+    
+    if (timeDiff > twoMinutes) {
+      ws.send(JSON.stringify({ type: 'error', message: '只能撤回2分钟内的消息' }));
+      return;
+    }
+
+    message.isRecalled = true;
+    message.recalledAt = new Date();
+    message.content = '[消息已撤回]';
+    await message.save();
+
+    // 广播撤回消息
+    broadcastToGroup(wss, message.group.toString(), {
+      type: 'message_recalled',
+      messageId: data.messageId,
+      groupId: message.group.toString(),
+      userId: ws.userId
+    });
+  } catch (error) {
+    console.error('撤回消息失败:', error);
+    ws.send(JSON.stringify({ type: 'error', message: '撤回失败' }));
+  }
+};
+
+const handleMessageRead = async (ws, data, wss) => {
+  if (!ws.userId) {
+    return;
+  }
+
+  try {
+    const message = await Message.findById(data.messageId);
+    
+    if (!message) {
+      return;
+    }
+
+    // 检查是否已标记为已读
+    const alreadyRead = message.readBy.some(
+      r => r.user.toString() === ws.userId
+    );
+
+    if (!alreadyRead) {
+      message.readBy.push({
+        user: ws.userId,
+        readAt: new Date()
+      });
+      await message.save();
+
+      // 通知消息发送者
+      const senderWs = clients.get(message.sender.toString());
+      if (senderWs && senderWs.readyState === 1) {
+        senderWs.send(JSON.stringify({
+          type: 'message_read_receipt',
+          messageId: data.messageId,
+          readBy: ws.userId,
+          readAt: new Date().toISOString()
+        }));
+      }
+    }
+  } catch (error) {
+    console.error('标记已读失败:', error);
+  }
+};
+
+const handleUserOnline = (ws, data, wss) => {
+  if (!ws.userId) {
+    return;
+  }
+
+  // 广播用户在线状态
+  if (ws.groupId) {
+    broadcastToGroup(wss, ws.groupId, {
+      type: 'user_online',
+      userId: ws.userId,
+      username: data.username,
+      isOnline: true
+    });
+  }
+};
+
 const handleDisconnect = (ws, wss) => {
   if (ws.userId) {
     clients.delete(ws.userId);
@@ -211,7 +338,8 @@ const handleDisconnect = (ws, wss) => {
     if (ws.groupId) {
       broadcastToGroup(wss, ws.groupId, {
         type: 'user_offline',
-        userId: ws.userId
+        userId: ws.userId,
+        isOnline: false
       });
     }
   }
@@ -226,3 +354,107 @@ const broadcastToGroup = (wss, groupId, message, excludeWs = null) => {
   });
 };
 
+// 白板协作功能处理
+const handleWhiteboardDraw = (ws, data, wss) => {
+  if (!ws.userId || !ws.groupId) {
+    return;
+  }
+
+  // 广播绘画数据到同一群组的其他用户
+  const message = {
+    type: 'whiteboard_draw',
+    groupId: data.groupId,
+    username: data.username,
+    data: data.data,
+    timestamp: new Date().toISOString()
+  };
+
+  broadcastToGroup(wss, data.groupId, message, ws);
+};
+
+const handleWhiteboardClear = (ws, data, wss) => {
+  if (!ws.userId || !ws.groupId) {
+    return;
+  }
+
+  // 广播清空画布事件
+  const message = {
+    type: 'whiteboard_clear',
+    groupId: data.groupId,
+    username: data.username,
+    timestamp: new Date().toISOString()
+  };
+
+  broadcastToGroup(wss, data.groupId, message, ws);
+};
+
+const handleWhiteboardCollaborationToggle = async (ws, data, wss) => {
+  if (!ws.userId || !ws.groupId) {
+    return;
+  }
+
+  // 验证是否是管理员
+  try {
+    const group = await Group.findById(data.groupId).select('admin');
+    if (!group) {
+      return;
+    }
+
+    const isAdmin = group.admin.toString() === ws.userId || ws.role === 'admin';
+    if (!isAdmin) {
+      ws.send(JSON.stringify({ 
+        type: 'error', 
+        message: '只有管理员可以切换协作模式' 
+      }));
+      return;
+    }
+
+    // 广播协作模式变更
+    const message = {
+      type: 'whiteboard_collaboration_toggle',
+      groupId: data.groupId,
+      enabled: data.enabled,
+      username: data.username,
+      timestamp: new Date().toISOString()
+    };
+
+    broadcastToGroup(wss, data.groupId, message);
+  } catch (error) {
+    console.error('切换协作模式失败:', error);
+  }
+};
+
+const handleWhiteboardUserJoin = (ws, data, wss) => {
+  if (!ws.userId) {
+    return;
+  }
+
+  // 广播用户加入白板
+  const message = {
+    type: 'whiteboard_user_join',
+    groupId: data.groupId,
+    username: data.username,
+    userId: ws.userId,
+    timestamp: new Date().toISOString()
+  };
+
+  broadcastToGroup(wss, data.groupId, message, ws);
+};
+
+const handleWhiteboardUserLeave = (ws, data, wss) => {
+  if (!ws.userId) {
+    return;
+  }
+
+  // 广播用户离开白板
+  const message = {
+    type: 'whiteboard_user_leave',
+    groupId: data.groupId,
+    username: data.username,
+    userId: ws.userId,
+    timestamp: new Date().toISOString()
+  };
+
+  broadcastToGroup(wss, data.groupId, message, ws);
+};
+