From 2e1489c6d460007bbc04d3b2298b3329cccab702 Mon Sep 17 00:00:00 2001
From: MingxuanGame <MingxuanGame@outlook.com>
Date: Fri, 25 Jul 2025 17:21:16 +0800
Subject: [PATCH] feat(beatmap): implement GET `/beatmaps`

---
 app/router/beatmap.py | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/app/router/beatmap.py b/app/router/beatmap.py
index f6aef2c..d232d4d 100644
--- a/app/router/beatmap.py
+++ b/app/router/beatmap.py
@@ -10,8 +10,9 @@ from app.dependencies.user import get_current_user
 
 from .api_router import router
 
-from fastapi import Depends, HTTPException
-from sqlmodel import Session, select
+from fastapi import Depends, HTTPException, Query
+from pydantic import BaseModel
+from sqlmodel import Session, col, select
 
 
 @router.get("/beatmaps/{bid}", tags=["beatmap"], response_model=BeatmapResp)
@@ -24,3 +25,27 @@ async def get_beatmap(
     if not beatmap:
         raise HTTPException(status_code=404, detail="Beatmap not found")
     return BeatmapResp.from_db(beatmap)
+
+
+class BatchGetResp(BaseModel):
+    beatmaps: list[BeatmapResp]
+
+
+@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),
+    current_user: DBUser = Depends(get_current_user),
+    db: Session = Depends(get_db),
+):
+    if not b_ids:
+        # select 50 beatmaps by last_updated
+        beatmaps = db.exec(
+            select(Beatmap).order_by(col(Beatmap.last_updated).desc()).limit(50)
+        ).all()
+    else:
+        beatmaps = db.exec(
+            select(Beatmap).where(col(Beatmap.id).in_(b_ids)).limit(50)
+        ).all()
+
+    return BatchGetResp(beatmaps=[BeatmapResp.from_db(bm) for bm in beatmaps])