151 lines
4.9 KiB
Python
151 lines
4.9 KiB
Python
"""会话路由"""
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.dependencies import get_db, get_current_user
|
|
from app.models.user import User
|
|
from app.schemas.conversation import (
|
|
ConversationCreate, ConversationRead, ConversationDetail,
|
|
GroupCreate, GroupUpdate, MemberAdd, RoleUpdate,
|
|
)
|
|
from app.services.conversation_service import ConversationService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/", response_model=list[dict])
|
|
async def list_conversations(
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""获取会话列表"""
|
|
service = ConversationService(db)
|
|
return await service.get_user_conversations(user.id)
|
|
|
|
|
|
@router.post("/", response_model=dict)
|
|
async def create_conversation(
|
|
req: ConversationCreate,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""创建会话（私聊或群聊）"""
|
|
service = ConversationService(db)
|
|
if req.type == "private":
|
|
if len(req.member_ids) != 1:
|
|
raise HTTPException(status_code=400, detail="私聊只能选择一个用户")
|
|
conv = await service.get_or_create_private(user.id, req.member_ids[0])
|
|
else:
|
|
conv = await service.create_group(user.id, req.name or "群聊", req.member_ids)
|
|
|
|
detail = await service.get_conversation_detail(conv.id, user.id)
|
|
return detail
|
|
|
|
|
|
@router.post("/group", response_model=dict)
|
|
async def create_group(
|
|
req: GroupCreate,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""创建群聊"""
|
|
service = ConversationService(db)
|
|
conv = await service.create_group(user.id, req.name, req.member_ids, req.description)
|
|
detail = await service.get_conversation_detail(conv.id, user.id)
|
|
return detail
|
|
|
|
|
|
@router.get("/{conversation_id}", response_model=dict)
|
|
async def get_conversation(
|
|
conversation_id: str,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""获取会话详情"""
|
|
service = ConversationService(db)
|
|
detail = await service.get_conversation_detail(conversation_id, user.id)
|
|
if not detail:
|
|
raise HTTPException(status_code=404, detail="会话不存在或无权访问")
|
|
return detail
|
|
|
|
|
|
@router.put("/{conversation_id}")
|
|
async def update_group(
|
|
conversation_id: str,
|
|
req: GroupUpdate,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""更新群聊信息（群主/管理员）"""
|
|
service = ConversationService(db)
|
|
try:
|
|
await service.update_group(conversation_id, user.id, **req.model_dump(exclude_none=True))
|
|
return {"success": True, "message": "群信息已更新"}
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.post("/{conversation_id}/members")
|
|
async def add_members(
|
|
conversation_id: str,
|
|
req: MemberAdd,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""添加群成员（群主/管理员）"""
|
|
service = ConversationService(db)
|
|
try:
|
|
await service.add_members(conversation_id, user.id, req.user_ids)
|
|
return {"success": True, "message": "成员已添加"}
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.delete("/{conversation_id}/members/{target_user_id}")
|
|
async def remove_member(
|
|
conversation_id: str,
|
|
target_user_id: str,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""移除群成员（群主/管理员）"""
|
|
service = ConversationService(db)
|
|
try:
|
|
await service.remove_member(conversation_id, user.id, target_user_id)
|
|
return {"success": True, "message": "成员已移除"}
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.post("/{conversation_id}/leave")
|
|
async def leave_group(
|
|
conversation_id: str,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""退出群聊"""
|
|
service = ConversationService(db)
|
|
try:
|
|
await service.leave_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,
|
|
target_user_id: str,
|
|
req: RoleUpdate,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""修改群成员角色（群主）"""
|
|
service = ConversationService(db)
|
|
try:
|
|
await service.update_member_role(conversation_id, user.id, target_user_id, req.role)
|
|
return {"success": True, "message": "角色已更新"}
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|