82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""消息路由"""
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.dependencies import get_db, get_current_user
|
|
from app.models.user import User
|
|
from app.schemas.message import MessageSend, MessagePage, MarkReadRequest
|
|
from app.services.message_service import MessageService
|
|
from app.services.conversation_service import ConversationService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/{conversation_id}/messages", response_model=dict)
|
|
async def get_messages(
|
|
conversation_id: str,
|
|
before: str | None = Query(None),
|
|
limit: int = Query(50, ge=1, le=100),
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""获取消息列表（游标分页）"""
|
|
service = MessageService(db)
|
|
try:
|
|
return await service.get_messages(conversation_id, user.id, before, limit)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=403, detail=str(e))
|
|
|
|
|
|
@router.get("/{conversation_id}/messages/search")
|
|
async def search_messages(
|
|
conversation_id: str,
|
|
q: str = Query(..., min_length=1, max_length=200),
|
|
limit: int = Query(20, ge=1, le=50),
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""在会话内搜索消息（用户级）"""
|
|
# 验证成员身份
|
|
conv_service = ConversationService(db)
|
|
detail = await conv_service.get_conversation_detail(conversation_id, user.id)
|
|
if not detail:
|
|
raise HTTPException(status_code=403, detail="无权访问该会话")
|
|
|
|
service = MessageService(db)
|
|
results = await service.search_messages(
|
|
conversation_id=conversation_id,
|
|
keyword=q,
|
|
limit=limit,
|
|
)
|
|
return {"results": results}
|
|
|
|
|
|
@router.put("/{conversation_id}/messages/{message_id}/read")
|
|
async def mark_as_read(
|
|
conversation_id: str,
|
|
message_id: str,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""标记消息已读"""
|
|
service = MessageService(db)
|
|
await service.mark_as_read(conversation_id, user.id, message_id)
|
|
return {"success": True}
|
|
|
|
|
|
@router.delete("/{conversation_id}/messages/{message_id}")
|
|
async def delete_message(
|
|
conversation_id: str,
|
|
message_id: str,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""删除消息"""
|
|
service = MessageService(db)
|
|
try:
|
|
await service.soft_delete(message_id, user.id)
|
|
return {"success": True}
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=403, detail=str(e))
|