app/routers/notifications.py

from fastapi import APIRouter, Depends, BackgroundTasks, HTTPException
from sqlalchemy.ext.asyncio import AsyncSession
from pydantic import BaseModel
from typing import List, Optional
from datetime import datetime

from app.database import get_db
from app.dependencies import require_user
from app.db_models import User
from app.db_storage import db_storage

router = APIRouter(prefix="/api/v1", tags=["notifications"])


class NotificationResponse(BaseModel):
    id: int
    type: str
    title: str
    message: str
    severity: str
    created_at: datetime
    read: bool

    class Config:
        from_attributes = True


async def create_notification(
    db: AsyncSession,
    user_id: int,
    notification_type: str,
    title: str,
    message: str,
    severity: str = "info",
):
    await db_storage.create_notification(
        db,
        user_id=user_id,
        notification_type=notification_type,
        title=title,
        message=message,
        severity=severity,
    )


@router.get("/notifications", response_model=List[NotificationResponse])
async def get_notifications(
    current_user: User = Depends(require_user),
    unread_only: bool = False,
    db: AsyncSession = Depends(get_db),
):
    return await db_storage.get_notifications(
        db, user_id=current_user.id, unread_only=unread_only
    )


@router.post("/notifications/{notification_id}/read")
async def mark_notification_read(
    notification_id: int,
    current_user: User = Depends(require_user),
    db: AsyncSession = Depends(get_db),
):
    success = await db_storage.mark_notification_read(
        db, user_id=current_user.id, notification_id=notification_id
    )
    if not success:
        raise HTTPException(status_code=404, detail="Notification not found")

    return {"message": "Notification marked as read"}


@router.post("/notifications/read-all")
async def mark_all_read(
    current_user: User = Depends(require_user), db: AsyncSession = Depends(get_db)
):
    count = await db_storage.mark_all_notifications_read(db, user_id=current_user.id)
    return {"message": f"Marked {count} notifications as read"}