Cheng Zhou
3e77127687
增加通用提醒状态、数据库迁移和定时同步，覆盖风险时效、文件回执、项目里程碑、访视窗口与协作申请。 新增网页端和桌面端提醒中心、真实投递诊断与固定隐私通知正文，并补齐登录来源聚合、测试和说明文档。
87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
import uuid
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.deps import get_current_user, get_db_session
|
|
from app.schemas.notification import (
|
|
DesktopNotificationAckRequest,
|
|
DesktopNotificationClaimRequest,
|
|
DesktopNotificationClaimResponse,
|
|
DesktopNotificationSubscriptionRead,
|
|
DesktopNotificationSubscriptionUpdate,
|
|
)
|
|
from app.services import desktop_notification_service
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/subscription", response_model=DesktopNotificationSubscriptionRead)
|
|
async def read_subscription(
|
|
db: AsyncSession = Depends(get_db_session),
|
|
current_user=Depends(get_current_user),
|
|
) -> DesktopNotificationSubscriptionRead:
|
|
subscription = await desktop_notification_service.get_subscription(db, current_user.id)
|
|
return DesktopNotificationSubscriptionRead(
|
|
enabled=bool(subscription and subscription.enabled),
|
|
enabled_at=subscription.enabled_at if subscription else None,
|
|
)
|
|
|
|
|
|
@router.put("/subscription", response_model=DesktopNotificationSubscriptionRead)
|
|
async def update_subscription(
|
|
payload: DesktopNotificationSubscriptionUpdate,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
current_user=Depends(get_current_user),
|
|
) -> DesktopNotificationSubscriptionRead:
|
|
subscription = await desktop_notification_service.set_subscription(
|
|
db, current_user.id, payload.enabled
|
|
)
|
|
return DesktopNotificationSubscriptionRead(
|
|
enabled=subscription.enabled,
|
|
enabled_at=subscription.enabled_at,
|
|
)
|
|
|
|
|
|
@router.post("/claim", response_model=DesktopNotificationClaimResponse)
|
|
async def claim_notifications(
|
|
payload: DesktopNotificationClaimRequest,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
current_user=Depends(get_current_user),
|
|
) -> DesktopNotificationClaimResponse:
|
|
if payload.limit < 1 or payload.limit > 50:
|
|
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail="limit 必须在 1 到 50 之间")
|
|
claim_token, lease_expires_at, items = await desktop_notification_service.claim_notifications(
|
|
db, current_user.id, payload.limit
|
|
)
|
|
return DesktopNotificationClaimResponse(
|
|
claim_token=claim_token,
|
|
lease_expires_at=lease_expires_at,
|
|
items=items,
|
|
)
|
|
|
|
|
|
@router.post("/ack", status_code=status.HTTP_204_NO_CONTENT)
|
|
async def acknowledge_notifications(
|
|
payload: DesktopNotificationAckRequest,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
current_user=Depends(get_current_user),
|
|
) -> None:
|
|
await desktop_notification_service.acknowledge_notifications(
|
|
db,
|
|
current_user.id,
|
|
payload.claim_token,
|
|
payload.delivered_ids,
|
|
)
|
|
|
|
|
|
@router.post("/{notification_id}/read", status_code=status.HTTP_204_NO_CONTENT)
|
|
async def mark_notification_read(
|
|
notification_id: uuid.UUID,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
current_user=Depends(get_current_user),
|
|
) -> None:
|
|
await desktop_notification_service.mark_notification_read(
|
|
db, current_user.id, notification_id
|
|
)
|