first commit

app/messages/router.py

@@ -0,0 +1,49 @@
+from fastapi import APIRouter, Depends, status
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.auth.service import get_current_user
+from app.database.session import get_db
+from app.messages.schemas import MessageCreateRequest, MessageRead, MessageUpdateRequest
+from app.messages.service import create_chat_message, delete_message, get_messages, update_message
+from app.users.models import User
+
+router = APIRouter(prefix="/messages", tags=["messages"])
+
+
+@router.post("", response_model=MessageRead, status_code=status.HTTP_201_CREATED)
+async def create_message(
+    payload: MessageCreateRequest,
+    db: AsyncSession = Depends(get_db),
+    current_user: User = Depends(get_current_user),
+) -> MessageRead:
+    return await create_chat_message(db, sender_id=current_user.id, payload=payload)
+
+
+@router.get("/{chat_id}", response_model=list[MessageRead])
+async def list_messages(
+    chat_id: int,
+    limit: int = 50,
+    before_id: int | None = None,
+    db: AsyncSession = Depends(get_db),
+    current_user: User = Depends(get_current_user),
+) -> list[MessageRead]:
+    return await get_messages(db, chat_id=chat_id, user_id=current_user.id, limit=limit, before_id=before_id)
+
+
+@router.put("/{message_id}", response_model=MessageRead)
+async def edit_message(
+    message_id: int,
+    payload: MessageUpdateRequest,
+    db: AsyncSession = Depends(get_db),
+    current_user: User = Depends(get_current_user),
+) -> MessageRead:
+    return await update_message(db, message_id=message_id, user_id=current_user.id, payload=payload)
+
+
+@router.delete("/{message_id}", status_code=status.HTTP_204_NO_CONTENT)
+async def remove_message(
+    message_id: int,
+    db: AsyncSession = Depends(get_db),
+    current_user: User = Depends(get_current_user),
+) -> None:
+    await delete_message(db, message_id=message_id, user_id=current_user.id)