xiangqinxiaochengxu/backend/app/routers/activities.py

from fastapi import APIRouter, Depends, HTTPException, Query, status
from sqlalchemy.ext.asyncio import AsyncSession

from app.database import get_db
from app.dependencies import get_current_user_id
from app.schemas.activity import ActivityDetailResponse, ActivityListItem, ActivityRegisterResponse
from app.services.activity_service import (
    cancel_registration,
    get_activity_by_id,
    get_activity_detail,
    list_my_activities,
    list_activities,
    register_activity,
)
from app.services.user_service import get_user_by_id

router = APIRouter()


@router.get("")
async def get_activities(
    status_filter: int | None = Query(default=None, alias="status"),
    page: int = Query(default=1, ge=1),
    page_size: int = Query(default=10, ge=1, le=50),
    current_user_id: int = Depends(get_current_user_id),
    session: AsyncSession = Depends(get_db),
) -> dict:
    current_user = await get_user_by_id(session, current_user_id)
    items = await list_activities(
        session,
        current_user,
        status=status_filter,
        page=page,
        page_size=page_size,
    )
    data = [ActivityListItem(**item).model_dump() for item in items]
    return {"code": 0, "message": "ok", "data": data}


@router.get("/mine")
async def get_my_activities(
    current_user_id: int = Depends(get_current_user_id),
    session: AsyncSession = Depends(get_db),
) -> dict:
    current_user = await get_user_by_id(session, current_user_id)
    if current_user is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="User not found")

    items = await list_my_activities(session, current_user)
    data = [ActivityListItem(**item).model_dump() for item in items]
    return {"code": 0, "message": "ok", "data": data}


@router.get("/{activity_id}")
async def get_activity(
    activity_id: int,
    current_user_id: int = Depends(get_current_user_id),
    session: AsyncSession = Depends(get_db),
) -> dict:
    activity = await get_activity_by_id(session, activity_id)
    if activity is None or activity.is_published != 1:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Activity not found")

    current_user = await get_user_by_id(session, current_user_id)
    detail = await get_activity_detail(session, activity, current_user)
    return {"code": 0, "message": "ok", "data": ActivityDetailResponse(**detail).model_dump()}


@router.post("/{activity_id}/register")
async def register(
    activity_id: int,
    current_user_id: int = Depends(get_current_user_id),
    session: AsyncSession = Depends(get_db),
) -> dict:
    activity = await get_activity_by_id(session, activity_id)
    if activity is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Activity not found")

    current_user = await get_user_by_id(session, current_user_id)
    if current_user is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="User not found")

    try:
        registration = await register_activity(session, activity, current_user)
    except ValueError as exc:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(exc)) from exc

    data = ActivityRegisterResponse(
        registration_id=registration.id,
        status=registration.status,
        message="报名成功，等待确认",
    ).model_dump()
    return {"code": 0, "message": "ok", "data": data}


@router.delete("/{activity_id}/register")
async def cancel_register(
    activity_id: int,
    current_user_id: int = Depends(get_current_user_id),
    session: AsyncSession = Depends(get_db),
) -> dict:
    activity = await get_activity_by_id(session, activity_id)
    if activity is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Activity not found")

    current_user = await get_user_by_id(session, current_user_id)
    if current_user is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="User not found")

    try:
        registration = await cancel_registration(session, activity, current_user)
    except ValueError as exc:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(exc)) from exc

    data = ActivityRegisterResponse(
        registration_id=registration.id,
        status=registration.status,
        message="报名已取消",
    ).model_dump()
    return {"code": 0, "message": "ok", "data": data}