首个可运行的版本

2026-06-12 23:14:12 +08:00
commit b3d90c65f8
86 changed files with 4808 additions and 0 deletions
@@ -0,0 +1,179 @@
+"""消息服务"""
+
+import uuid
+from datetime import datetime, timezone
+
+from sqlalchemy import select, func, and_
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.message import Message
+from app.models.conversation import Conversation
+from app.models.conversation_member import ConversationMember
+
+
+class MessageService:
+    def __init__(self, db: AsyncSession):
+        self.db = db
+
+    async def send_message(self, conversation_id: str, sender_id: str,
+                           content: str, msg_type: str = "text",
+                           reply_to_id: str | None = None) -> Message:
+        """发送消息"""
+        message = Message(
+            id=str(uuid.uuid4()),
+            conversation_id=conversation_id,
+            sender_id=sender_id,
+            type=msg_type,
+            content=content,
+            reply_to_id=reply_to_id,
+        )
+        self.db.add(message)
+
+        # 更新会话的最后消息
+        conv_result = await self.db.execute(
+            select(Conversation).where(Conversation.id == conversation_id)
+        )
+        conv = conv_result.scalars().first()
+        if conv:
+            conv.last_message_at = datetime.utcnow()
+            preview = content[:200] if len(content) > 200 else content
+            conv.last_message_preview = preview
+            conv.updated_at = datetime.utcnow()
+
+        return message
+
+    async def get_messages(self, conversation_id: str, user_id: str,
+                           before: str | None = None, limit: int = 50) -> dict:
+        """获取消息列表（游标分页）"""
+        # 验证成员身份
+        member_result = await self.db.execute(
+            select(ConversationMember).where(
+                ConversationMember.conversation_id == conversation_id,
+                ConversationMember.user_id == user_id,
+            )
+        )
+        if not member_result.scalars().first():
+            raise ValueError("无权访问该会话")
+
+        query = (
+            select(Message)
+            .where(Message.conversation_id == conversation_id, Message.is_deleted == False)
+            .order_by(Message.created_at.desc())
+        )
+
+        # 游标分页
+        if before:
+            before_msg = await self.db.execute(
+                select(Message).where(Message.id == before)
+            )
+            before_msg_obj = before_msg.scalars().first()
+            if before_msg_obj:
+                query = query.where(Message.created_at < before_msg_obj.created_at)
+
+        query = query.limit(limit + 1)
+        result = await self.db.execute(query)
+        messages = list(result.scalars().all())
+
+        has_more = len(messages) > limit
+        messages = messages[:limit]
+
+        # 获取发送者信息
+        from app.models.user import User
+        message_list = []
+        for msg in reversed(messages):
+            sender_result = await self.db.execute(
+                select(User).where(User.id == msg.sender_id)
+            )
+            sender = sender_result.scalars().first()
+            message_list.append({
+                "id": msg.id,
+                "conversation_id": msg.conversation_id,
+                "sender_id": msg.sender_id,
+                "sender_name": sender.username if sender else "未知",
+                "sender_avatar": sender.avatar_url if sender else None,
+                "type": msg.type,
+                "content": msg.content,
+                "reply_to_id": msg.reply_to_id,
+                "is_deleted": msg.is_deleted,
+                "created_at": msg.created_at,
+            })
+
+        return {
+            "messages": message_list,
+            "has_more": has_more,
+            "next_cursor": messages[-1].id if has_more and messages else None,
+        }
+
+    async def mark_as_read(self, conversation_id: str, user_id: str, message_id: str):
+        """标记消息已读"""
+        result = await self.db.execute(
+            select(ConversationMember).where(
+                ConversationMember.conversation_id == conversation_id,
+                ConversationMember.user_id == user_id,
+            )
+        )
+        member = result.scalars().first()
+        if member:
+            member.last_read_message_id = message_id
+
+    async def soft_delete(self, message_id: str, user_id: str):
+        """软删除消息（仅能删除自己的）"""
+        result = await self.db.execute(
+            select(Message).where(Message.id == message_id, Message.sender_id == user_id)
+        )
+        message = result.scalars().first()
+        if not message:
+            raise ValueError("消息不存在或无权删除")
+        message.is_deleted = True
+
+    async def get_total_count(self) -> int:
+        """获取消息总数"""
+        result = await self.db.execute(select(func.count(Message.id)))
+        return result.scalar() or 0
+
+    async def get_today_count(self) -> int:
+        """获取今日消息数"""
+        today = datetime.utcnow().replace(hour=0, minute=0, second=0, microsecond=0)
+        result = await self.db.execute(
+            select(func.count(Message.id)).where(Message.created_at >= today)
+        )
+        return result.scalar() or 0
+
+    async def search_messages(self, user_id: str | None = None,
+                              conversation_id: str | None = None,
+                              keyword: str | None = None,
+                              date_from: str | None = None,
+                              date_to: str | None = None,
+                              limit: int = 50) -> list[dict]:
+        """管理后台搜索消息"""
+        query = select(Message).where(Message.is_deleted == False)
+
+        if user_id:
+            query = query.where(Message.sender_id == user_id)
+        if conversation_id:
+            query = query.where(Message.conversation_id == conversation_id)
+        if keyword:
+            query = query.where(Message.content.ilike(f"%{keyword}%"))
+        if date_from:
+            query = query.where(Message.created_at >= date_from)
+        if date_to:
+            query = query.where(Message.created_at <= date_to)
+
+        query = query.order_by(Message.created_at.desc()).limit(limit)
+        result = await self.db.execute(query)
+
+        from app.models.user import User
+        messages = []
+        for msg in result.scalars().all():
+            sender = await self.db.execute(select(User).where(User.id == msg.sender_id))
+            s = sender.scalars().first()
+            messages.append({
+                "id": msg.id,
+                "conversation_id": msg.conversation_id,
+                "sender_id": msg.sender_id,
+                "sender_name": s.username if s else "未知",
+                "type": msg.type,
+                "content": msg.content[:200],
+                "created_at": msg.created_at,
+            })
+        return messages