40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
from fastapi import APIRouter, Body, Depends, Response, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.auth.service import get_current_user
|
|
from app.database.session import get_db
|
|
from app.notifications.schemas import NotificationRead, PushTokenDeleteRequest, PushTokenUpsertRequest
|
|
from app.notifications.service import get_notifications_for_user, register_push_token, unregister_push_token
|
|
from app.users.models import User
|
|
|
|
router = APIRouter(prefix="/notifications", tags=["notifications"])
|
|
|
|
|
|
@router.get("", response_model=list[NotificationRead])
|
|
async def list_my_notifications(
|
|
limit: int = 50,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
) -> list[NotificationRead]:
|
|
return await get_notifications_for_user(db, user_id=current_user.id, limit=limit)
|
|
|
|
|
|
@router.post("/push-token", status_code=status.HTTP_204_NO_CONTENT)
|
|
async def upsert_my_push_token(
|
|
payload: PushTokenUpsertRequest,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
) -> Response:
|
|
await register_push_token(db, user_id=current_user.id, payload=payload)
|
|
return Response(status_code=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
@router.api_route("/push-token", methods=["DELETE"], status_code=status.HTTP_204_NO_CONTENT)
|
|
async def delete_my_push_token(
|
|
payload: PushTokenDeleteRequest = Body(...),
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
) -> Response:
|
|
await unregister_push_token(db, user_id=current_user.id, payload=payload)
|
|
return Response(status_code=status.HTTP_204_NO_CONTENT)
|