63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
from app.database import (
|
|
Beatmapset,
|
|
BeatmapsetResp,
|
|
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 .api_router import router
|
|
|
|
from fastapi import Depends, HTTPException, Query
|
|
from fastapi.responses import RedirectResponse
|
|
from httpx import HTTPStatusError
|
|
from sqlalchemy.orm import selectinload
|
|
from sqlmodel import select
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
|
|
|
|
|
@router.get("/beatmapsets/{sid}", tags=["beatmapset"], response_model=BeatmapsetResp)
|
|
async def get_beatmapset(
|
|
sid: int,
|
|
current_user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
fetcher: Fetcher = Depends(get_fetcher),
|
|
):
|
|
beatmapset = (
|
|
await db.exec(
|
|
select(Beatmapset)
|
|
.options(selectinload(Beatmapset.beatmaps)) # pyright: ignore[reportArgumentType]
|
|
.where(Beatmapset.id == sid)
|
|
)
|
|
).first()
|
|
if not beatmapset:
|
|
try:
|
|
resp = await fetcher.get_beatmapset(sid)
|
|
await Beatmapset.from_resp(db, resp)
|
|
except HTTPStatusError:
|
|
raise HTTPException(status_code=404, detail="Beatmapset not found")
|
|
else:
|
|
resp = BeatmapsetResp.from_db(beatmapset)
|
|
return resp
|
|
|
|
|
|
@router.get("/beatmapsets/{beatmapset}/download", tags=["beatmapset"])
|
|
async def download_beatmapset(
|
|
beatmapset: int,
|
|
no_video: bool = Query(True, alias="noVideo"),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
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}"
|
|
)
|