from __future__ import annotations

from app.database.beatmap import BeatmapResp

from ._base import BaseFetcher

from httpx import AsyncClient


class BeatmapFetcher(BaseFetcher):
    async def get_beatmap(
        self, beatmap_id: int | None = None, beatmap_checksum: str | None = None
    ) -> BeatmapResp:
        if beatmap_id:
            params = {"id": beatmap_id}
        elif beatmap_checksum:
            params = {"checksum": beatmap_checksum}
        else:
            raise ValueError("Either beatmap_id or beatmap_checksum must be provided.")
        async with AsyncClient() as client:
            response = await client.get(
                "https://osu.ppy.sh/api/v2/beatmaps/lookup",
                headers=self.header,
                params=params,
            )
            response.raise_for_status()
            return BeatmapResp.model_validate(response.json())