1.4
This commit is contained in:
@@ -10,6 +10,8 @@ from app.schemas.conversation import (
|
||||
GroupCreate, GroupUpdate, MemberAdd, RoleUpdate,
|
||||
)
|
||||
from app.services.conversation_service import ConversationService
|
||||
from app.websocket.events import EventType
|
||||
from app.websocket.manager import manager
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
@@ -97,6 +99,16 @@ async def add_members(
|
||||
service = ConversationService(db)
|
||||
try:
|
||||
await service.add_members(conversation_id, user.id, req.user_ids)
|
||||
# 通知新成员被加入群聊
|
||||
detail = await service.get_conversation_detail(conversation_id, user.id)
|
||||
group_name = detail.get("name", "群聊") if detail else "群聊"
|
||||
for mid in req.user_ids:
|
||||
await manager.send_to_user(mid, EventType.CONVERSATION_MEMBER_ADDED, {
|
||||
"conversation_id": conversation_id,
|
||||
"group_name": group_name,
|
||||
"added_by_user_id": user.id,
|
||||
"added_by_username": user.username,
|
||||
})
|
||||
return {"success": True, "message": "成员已添加"}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
@@ -113,6 +125,12 @@ async def remove_member(
|
||||
service = ConversationService(db)
|
||||
try:
|
||||
await service.remove_member(conversation_id, user.id, target_user_id)
|
||||
# 通知被移除的用户
|
||||
await manager.send_to_user(target_user_id, EventType.CONVERSATION_MEMBER_REMOVED, {
|
||||
"conversation_id": conversation_id,
|
||||
"removed_by_user_id": user.id,
|
||||
"removed_by_username": user.username,
|
||||
})
|
||||
return {"success": True, "message": "成员已移除"}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
@@ -133,6 +151,21 @@ async def leave_group(
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
|
||||
@router.post("/{conversation_id}/dissolve")
|
||||
async def dissolve_group(
|
||||
conversation_id: str,
|
||||
user: User = Depends(get_current_user),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
"""解散群聊(仅群主)"""
|
||||
service = ConversationService(db)
|
||||
try:
|
||||
await service.dissolve_group(conversation_id, user.id)
|
||||
return {"success": True, "message": "群聊已解散"}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
|
||||
@router.put("/{conversation_id}/members/{target_user_id}/role")
|
||||
async def update_member_role(
|
||||
conversation_id: str,
|
||||
|
||||
@@ -7,6 +7,8 @@ from app.dependencies import get_db, get_current_user
|
||||
from app.models.user import User
|
||||
from app.schemas.friend import FriendRequestCreate, FriendRead, FriendRequestRead, RemarkUpdate
|
||||
from app.services.friend_service import FriendService
|
||||
from app.websocket.events import EventType
|
||||
from app.websocket.manager import manager
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
@@ -41,6 +43,14 @@ async def send_friend_request(
|
||||
service = FriendService(db)
|
||||
try:
|
||||
await service.send_request(user.id, req.to_user_id, req.message)
|
||||
# 通知接收者
|
||||
await manager.send_to_user(req.to_user_id, EventType.FRIEND_REQUEST, {
|
||||
"from_user_id": user.id,
|
||||
"from_username": user.username,
|
||||
"from_nickname": user.nickname,
|
||||
"from_avatar": user.avatar_url,
|
||||
"message": req.message,
|
||||
})
|
||||
return {"success": True, "message": "好友请求已发送"}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
@@ -86,7 +96,24 @@ async def accept_friend_request(
|
||||
"""接受好友请求"""
|
||||
service = FriendService(db)
|
||||
try:
|
||||
# Get request details before accepting to know who sent it
|
||||
request = await service.get_pending_requests(user.id)
|
||||
from_user_id = None
|
||||
for r in request:
|
||||
if r["id"] == request_id:
|
||||
from_user_id = r["from_user_id"]
|
||||
break
|
||||
|
||||
await service.accept_request(request_id, user.id)
|
||||
|
||||
# Notify the requester that their request was accepted
|
||||
if from_user_id:
|
||||
await manager.send_to_user(from_user_id, EventType.FRIEND_ACCEPTED, {
|
||||
"accepted_by_user_id": user.id,
|
||||
"accepted_by_username": user.username,
|
||||
"accepted_by_nickname": user.nickname,
|
||||
})
|
||||
|
||||
return {"success": True, "message": "已添加好友"}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
@@ -21,6 +21,8 @@ class MessageRead(BaseModel):
|
||||
type: str
|
||||
content: str
|
||||
reply_to_id: str | None = None
|
||||
reply_to_content: str | None = None
|
||||
reply_to_sender_name: str | None = None
|
||||
is_deleted: bool = False
|
||||
created_at: datetime
|
||||
|
||||
|
||||
@@ -115,6 +115,33 @@ class ConversationService:
|
||||
raise ValueError("群主不能退出,请先转让群主身份")
|
||||
member.left_at = datetime.utcnow()
|
||||
|
||||
async def dissolve_group(self, conv_id: str, user_id: str):
|
||||
"""解散群聊(仅群主)"""
|
||||
member = await self._get_member(conv_id, user_id)
|
||||
if not member:
|
||||
raise ValueError("你不在该群中")
|
||||
if member.role != "owner":
|
||||
raise ValueError("只有群主可以解散群聊")
|
||||
|
||||
# 验证会话存在且为群聊
|
||||
conv_result = await self.db.execute(
|
||||
select(Conversation).where(Conversation.id == conv_id)
|
||||
)
|
||||
conv = conv_result.scalars().first()
|
||||
if not conv or conv.type != "group":
|
||||
raise ValueError("群聊不存在")
|
||||
|
||||
# 软删除所有成员(设置 left_at)
|
||||
members_result = await self.db.execute(
|
||||
select(ConversationMember).where(
|
||||
ConversationMember.conversation_id == conv_id,
|
||||
ConversationMember.left_at.is_(None),
|
||||
)
|
||||
)
|
||||
now = datetime.utcnow()
|
||||
for m in members_result.scalars().all():
|
||||
m.left_at = now
|
||||
|
||||
async def update_member_role(self, conv_id: str, user_id: str, target_user_id: str, role: str):
|
||||
"""修改成员角色(仅群主)"""
|
||||
member = await self._get_member(conv_id, user_id)
|
||||
|
||||
@@ -85,6 +85,23 @@ class MessageService:
|
||||
select(User).where(User.id == msg.sender_id)
|
||||
)
|
||||
sender = sender_result.scalars().first()
|
||||
|
||||
# 获取被引用消息的信息
|
||||
reply_to_content = None
|
||||
reply_to_sender_name = None
|
||||
if msg.reply_to_id:
|
||||
reply_msg_result = await self.db.execute(
|
||||
select(Message).where(Message.id == msg.reply_to_id)
|
||||
)
|
||||
reply_msg = reply_msg_result.scalars().first()
|
||||
if reply_msg:
|
||||
reply_to_content = reply_msg.content[:200] if reply_msg.content else None
|
||||
reply_sender_result = await self.db.execute(
|
||||
select(User).where(User.id == reply_msg.sender_id)
|
||||
)
|
||||
reply_sender = reply_sender_result.scalars().first()
|
||||
reply_to_sender_name = reply_sender.username if reply_sender else None
|
||||
|
||||
message_list.append({
|
||||
"id": msg.id,
|
||||
"conversation_id": msg.conversation_id,
|
||||
@@ -94,6 +111,8 @@ class MessageService:
|
||||
"type": msg.type,
|
||||
"content": msg.content,
|
||||
"reply_to_id": msg.reply_to_id,
|
||||
"reply_to_content": reply_to_content,
|
||||
"reply_to_sender_name": reply_to_sender_name,
|
||||
"is_deleted": msg.is_deleted,
|
||||
"created_at": msg.created_at,
|
||||
})
|
||||
|
||||
@@ -4,6 +4,7 @@ import json
|
||||
from datetime import datetime, timezone
|
||||
|
||||
from fastapi import WebSocket
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.websocket.events import EventType
|
||||
@@ -34,6 +35,20 @@ async def handle_chat_send(ws: WebSocket, user_id: str, data: dict, db: AsyncSes
|
||||
conv_service = ConversationService(db)
|
||||
detail = await conv_service.get_conversation_detail(data["conversation_id"], user_id)
|
||||
|
||||
# 获取被引用消息的信息
|
||||
reply_to_content = None
|
||||
reply_to_sender_name = None
|
||||
if message.reply_to_id:
|
||||
from app.models.message import Message
|
||||
reply_msg_result = await db.execute(
|
||||
select(Message).where(Message.id == message.reply_to_id)
|
||||
)
|
||||
reply_msg = reply_msg_result.scalars().first()
|
||||
if reply_msg:
|
||||
reply_to_content = reply_msg.content[:200] if reply_msg.content else None
|
||||
reply_sender = await user_service.get_by_id(reply_msg.sender_id)
|
||||
reply_to_sender_name = reply_sender.username if reply_sender else None
|
||||
|
||||
msg_data = {
|
||||
"id": message.id,
|
||||
"conversation_id": message.conversation_id,
|
||||
@@ -42,6 +57,9 @@ async def handle_chat_send(ws: WebSocket, user_id: str, data: dict, db: AsyncSes
|
||||
"sender_avatar": sender.avatar_url if sender else None,
|
||||
"type": message.type,
|
||||
"content": message.content,
|
||||
"reply_to_id": message.reply_to_id,
|
||||
"reply_to_content": reply_to_content,
|
||||
"reply_to_sender_name": reply_to_sender_name,
|
||||
"created_at": message.created_at.isoformat(),
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user