From 0b68bdc0c11fc75a6a76e97a4f874c98a784c639 Mon Sep 17 00:00:00 2001
From: MingxuanGame <MingxuanGame@outlook.com>
Date: Sat, 2 Aug 2025 01:55:30 +0000
Subject: [PATCH] fix(beatmap,beatmapset): fix lookup

---
 app/router/beatmap.py    |  2 +-
 app/router/beatmapset.py | 28 +++++++++++++++++++++++++++-
 2 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/app/router/beatmap.py b/app/router/beatmap.py
index 9574bdb..7dfd0f9 100644
--- a/app/router/beatmap.py
+++ b/app/router/beatmap.py
@@ -74,7 +74,7 @@ class BatchGetResp(BaseModel):
 @router.get("/beatmaps", tags=["beatmap"], response_model=BatchGetResp)
 @router.get("/beatmaps/", tags=["beatmap"], response_model=BatchGetResp)
 async def batch_get_beatmaps(
-    b_ids: list[int] = Query(alias="id", default_factory=list),
+    b_ids: list[int] = Query(alias="ids[]", default_factory=list),
     current_user: User = Depends(get_current_user),
     db: AsyncSession = Depends(get_db),
 ):
diff --git a/app/router/beatmapset.py b/app/router/beatmapset.py
index b4d2e4c..f77c2ed 100644
--- a/app/router/beatmapset.py
+++ b/app/router/beatmapset.py
@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from typing import Literal
 
-from app.database import Beatmapset, BeatmapsetResp, FavouriteBeatmapset, User
+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
@@ -17,6 +17,32 @@ 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 = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+    fetcher: Fetcher = Depends(get_fetcher),
+):
+    beatmapset_id = (
+        await db.exec(select(Beatmap.beatmapset_id).where(Beatmap.id == beatmap_id))
+    ).first()
+    if not beatmapset_id:
+        try:
+            resp = await fetcher.get_beatmap(beatmap_id)
+            await Beatmap.from_resp(db, resp)
+            await db.refresh(current_user)
+        except HTTPStatusError:
+            raise HTTPException(status_code=404, detail="Beatmapset not found")
+    beatmapset = (
+        await db.exec(select(Beatmapset).where(Beatmapset.id == beatmapset_id))
+    ).first()
+    if not beatmapset:
+        raise HTTPException(status_code=404, detail="Beatmapset not found")
+    resp = await BeatmapsetResp.from_db(beatmapset, session=db, user=current_user)
+    return resp
+
+
 @router.get("/beatmapsets/{sid}", tags=["beatmapset"], response_model=BeatmapsetResp)
 async def get_beatmapset(
     sid: int,