refactor(router): restructure

2025-08-11 14:38:45 +00:00
parent 867b99cca5
commit d8c607137a
14 changed files with 36 additions and 33 deletions
--- a/app/router/v2/beatmapset.py
+++ b/app/router/v2/beatmapset.py
@@ -0,0 +1,95 @@
+from __future__ import annotations
+
+from typing import Literal
+
+from app.database import Beatmap, Beatmapset, BeatmapsetResp, FavouriteBeatmapset, User
+from app.dependencies.database import get_db
+from app.dependencies.fetcher import get_fetcher
+from app.dependencies.user import get_current_user
+from app.fetcher import Fetcher
+
+from .router import router
+
+from fastapi import Depends, Form, HTTPException, Query, Security
+from fastapi.responses import RedirectResponse
+from httpx import HTTPError
+from sqlmodel import select
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+
+@router.get("/beatmapsets/lookup", tags=["beatmapset"], response_model=BeatmapsetResp)
+async def lookup_beatmapset(
+    beatmap_id: int = Query(),
+    current_user: User = Security(get_current_user, scopes=["public"]),
+    db: AsyncSession = Depends(get_db),
+    fetcher: Fetcher = Depends(get_fetcher),
+):
+    beatmap = await Beatmap.get_or_fetch(db, fetcher, bid=beatmap_id)
+    resp = await BeatmapsetResp.from_db(
+        beatmap.beatmapset, session=db, user=current_user
+    )
+    return resp
+
+
+@router.get("/beatmapsets/{sid}", tags=["beatmapset"], response_model=BeatmapsetResp)
+async def get_beatmapset(
+    sid: int,
+    current_user: User = Security(get_current_user, scopes=["public"]),
+    db: AsyncSession = Depends(get_db),
+    fetcher: Fetcher = Depends(get_fetcher),
+):
+    try:
+        beatmapset = await Beatmapset.get_or_fetch(db, fetcher, sid)
+        return await BeatmapsetResp.from_db(
+            beatmapset, session=db, include=["recent_favourites"], user=current_user
+        )
+    except HTTPError:
+        raise HTTPException(status_code=404, detail="Beatmapset not found")
+
+
+@router.get("/beatmapsets/{beatmapset}/download", tags=["beatmapset"])
+async def download_beatmapset(
+    beatmapset: int,
+    no_video: bool = Query(True, alias="noVideo"),
+    current_user: User = Security(get_current_user, scopes=["*"]),
+):
+    if current_user.country_code == "CN":
+        return RedirectResponse(
+            f"https://txy1.sayobot.cn/beatmaps/download/"
+            f"{'novideo' if no_video else 'full'}/{beatmapset}?server=auto"
+        )
+    else:
+        return RedirectResponse(
+            f"https://api.nerinyan.moe/d/{beatmapset}?noVideo={no_video}"
+        )
+
+
+@router.post("/beatmapsets/{beatmapset}/favourites", tags=["beatmapset"])
+async def favourite_beatmapset(
+    beatmapset: int,
+    action: Literal["favourite", "unfavourite"] = Form(),
+    current_user: User = Security(get_current_user, scopes=["*"]),
+    db: AsyncSession = Depends(get_db),
+):
+    existing_favourite = (
+        await db.exec(
+            select(FavouriteBeatmapset).where(
+                FavouriteBeatmapset.user_id == current_user.id,
+                FavouriteBeatmapset.beatmapset_id == beatmapset,
+            )
+        )
+    ).first()
+
+    if (action == "favourite" and existing_favourite) or (
+        action == "unfavourite" and not existing_favourite
+    ):
+        return
+
+    if action == "favourite":
+        favourite = FavouriteBeatmapset(
+            user_id=current_user.id, beatmapset_id=beatmapset
+        )
+        db.add(favourite)
+    else:
+        await db.delete(existing_favourite)
+    await db.commit()