首个可运行的版本

backend/app/websocket/handlers.py

@@ -0,0 +1,106 @@
+"""WebSocket 事件处理器"""
+
+import json
+from datetime import datetime, timezone
+
+from fastapi import WebSocket
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.websocket.events import EventType
+from app.websocket.manager import manager
+
+
+async def handle_chat_send(ws: WebSocket, user_id: str, data: dict, db: AsyncSession):
+    """处理发送消息事件"""
+    from app.services.message_service import MessageService
+    service = MessageService(db)
+    try:
+        message = await service.send_message(
+            conversation_id=data["conversation_id"],
+            sender_id=user_id,
+            content=data["content"],
+            msg_type=data.get("type", "text"),
+            reply_to_id=data.get("reply_to_id"),
+        )
+        await db.commit()
+
+        # 获取发送者信息
+        from app.services.user_service import UserService
+        user_service = UserService(db)
+        sender = await user_service.get_by_id(user_id)
+
+        # 获取会话成员列表
+        from app.services.conversation_service import ConversationService
+        conv_service = ConversationService(db)
+        detail = await conv_service.get_conversation_detail(data["conversation_id"], user_id)
+
+        msg_data = {
+            "id": message.id,
+            "conversation_id": message.conversation_id,
+            "sender_id": user_id,
+            "sender_name": sender.username if sender else "未知",
+            "sender_avatar": sender.avatar_url if sender else None,
+            "type": message.type,
+            "content": message.content,
+            "created_at": message.created_at.isoformat(),
+        }
+
+        # 广播给会话中的所有成员
+        if detail and "members" in detail:
+            member_ids = [m["user_id"] for m in detail["members"]]
+            await manager.broadcast_to_conversation(
+                member_ids, EventType.CHAT_MESSAGE, msg_data
+            )
+    except Exception as e:
+        await manager.send_to_user(user_id, EventType.ERROR, {"message": str(e)})
+
+
+async def handle_chat_typing(ws: WebSocket, user_id: str, data: dict, db: AsyncSession):
+    """处理输入中事件"""
+    from app.services.user_service import UserService
+    user_service = UserService(db)
+    user = await user_service.get_by_id(user_id)
+
+    from app.services.conversation_service import ConversationService
+    conv_service = ConversationService(db)
+    detail = await conv_service.get_conversation_detail(data["conversation_id"], user_id)
+
+    if detail and "members" in detail:
+        member_ids = [m["user_id"] for m in detail["members"]]
+        await manager.broadcast_to_conversation(
+            member_ids, EventType.CHAT_TYPING_INDICATOR,
+            {"conversation_id": data["conversation_id"], "user_id": user_id,
+             "username": user.username if user else "未知"},
+            exclude_user=user_id,
+        )
+
+
+async def handle_chat_read(ws: WebSocket, user_id: str, data: dict, db: AsyncSession):
+    """处理已读事件"""
+    from app.services.message_service import MessageService
+    service = MessageService(db)
+    await service.mark_as_read(data["conversation_id"], user_id, data["message_id"])
+    await db.commit()
+
+    from app.services.conversation_service import ConversationService
+    conv_service = ConversationService(db)
+    detail = await conv_service.get_conversation_detail(data["conversation_id"], user_id)
+
+    if detail and "members" in detail:
+        member_ids = [m["user_id"] for m in detail["members"]]
+        await manager.broadcast_to_conversation(
+            member_ids, EventType.CHAT_READ_RECEIPT,
+            {"conversation_id": data["conversation_id"], "user_id": user_id,
+             "read_up_to": data["message_id"]},
+        )
+
+
+async def handle_presence_update(ws: WebSocket, user_id: str, data: dict, db: AsyncSession):
+    """处理在线状态更新"""
+    from app.services.user_service import UserService
+    user_service = UserService(db)
+    await user_service.update_status(user_id, data["status"])
+    await db.commit()
+
+    event = EventType.PRESENCE_ONLINE if data["status"] == "online" else EventType.PRESENCE_OFFLINE
+    await manager.broadcast(event, {"user_id": user_id})