g0v0-server/app/router/private/beatmapset.py

from typing import Annotated

from app.database.beatmap import Beatmap
from app.database.beatmapset import Beatmapset
from app.database.beatmapset_ratings import BeatmapRating
from app.database.score import Score
from app.dependencies.database import Database
from app.dependencies.user import ClientUser
from app.service.beatmapset_update_service import get_beatmapset_update_service

from .router import router

from fastapi import Body, Depends, HTTPException, Path, Query
from fastapi_limiter.depends import RateLimiter
from sqlmodel import col, exists, select


@router.get(
    "/beatmapsets/{beatmapset_id}/can_rate",
    name="判断用户能否为谱面集打分",
    response_model=bool,
    tags=["谱面集", "g0v0 API"],
)
async def can_rate_beatmapset(
    beatmapset_id: int,
    session: Database,
    current_user: ClientUser,
):
    """检查用户是否可以评价谱面集

    检查当前用户是否可以对指定的谱面集进行评价
    参数:
    - beatmapset_id: 谱面集ID

    错误情况:
    - 404: 找不到指定谱面集

    返回:
    - bool: 用户是否可以评价谱面集
    """
    if await current_user.is_restricted(session):
        return False

    user_id = current_user.id
    prev_ratings = (await session.exec(select(BeatmapRating).where(BeatmapRating.user_id == user_id))).first()
    if prev_ratings is not None:
        return False
    query = select(exists()).where(
        Score.user_id == user_id,
        col(Score.beatmap).has(col(Beatmap.beatmapset_id) == beatmapset_id),
        col(Score.passed).is_(True),
    )
    return (await session.exec(query)).first() or False


@router.post(
    "/beatmapsets/{beatmapset_id}/ratings", name="上传对谱面集的打分", status_code=201, tags=["谱面集", "g0v0 API"]
)
async def rate_beatmaps(
    beatmapset_id: int,
    session: Database,
    rating: Annotated[int, Body(..., ge=0, le=10)],
    current_user: ClientUser,
):
    """为谱面集评分

    为指定的谱面集添加用户评分，并更新谱面集的评分统计信息

    参数:
    - beatmapset_id: 谱面集ID
    - rating: 评分

    错误情况:
    - 404: 找不到指定谱面集

    返回:
    - 成功: None
    """
    if await current_user.is_restricted(session):
        raise HTTPException(status_code=403, detail="Your account is restricted and cannot perform this action.")

    user_id = current_user.id
    current_beatmapset = (await session.exec(select(exists()).where(Beatmapset.id == beatmapset_id))).first()
    if not current_beatmapset:
        raise HTTPException(404, "Beatmapset Not Found")
    can_rating = await can_rate_beatmapset(beatmapset_id, session, current_user)
    if not can_rating:
        raise HTTPException(403, "User Cannot Rate This Beatmapset")
    new_rating: BeatmapRating = BeatmapRating(beatmapset_id=beatmapset_id, user_id=user_id, rating=rating)
    session.add(new_rating)
    await session.commit()


@router.post(
    "/beatmapsets/{beatmapset_id}/sync",
    name="请求同步谱面集",
    status_code=202,
    tags=["谱面集", "g0v0 API"],
    dependencies=[Depends(RateLimiter(times=50, hours=1))],
)
async def sync_beatmapset(
    beatmapset_id: Annotated[int, Path(..., description="谱面集ID")],
    session: Database,
    current_user: ClientUser,
    immediate: Annotated[bool, Query(description="是否立即同步")] = False,
):
    """请求同步谱面集

    请求将指定的谱面集从 Bancho 同步到服务器

    默认情况下请求会加入同步队列，等待自动同步。
    若设置 `immediate=true`，会尝试立刻同步该谱面集。

    速率限制:
    - 每个用户每小时最多50次请求

    参数:
    - beatmapset_id: 谱面集ID
    - immediate: 是否立即同步（默认 false）

    错误情况:
    - 404: 找不到指定谱面集
    """
    current_beatmapset = (await session.exec(select(exists()).where(Beatmapset.id == beatmapset_id))).first()
    if not current_beatmapset:
        raise HTTPException(404, "Beatmapset Not Found")
    await get_beatmapset_update_service().add_missing_beatmapset(beatmapset_id, immediate)