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)