from attrs import define

from .engine import DeezerEngine
from .util import clean_query


@define
class DeezerDriver:
    engine: DeezerEngine

    async def get_track(self, track_id: int | str):
        data = await self.engine.call_legacy_api(f"track/{track_id}")

        return data

    async def reverse_get_track(self, track_id: str):
        return await self.engine.call_api("song.getData", params={"SNG_ID": track_id})

    async def search(self, query: str, limit: int = 30):
        data = await self.engine.call_legacy_api(
            "search/track", params={"q": clean_query(query), "limit": limit}
        )

        return data["data"]

    async def renew_engine(self):
        self.engine = await self.engine.from_arl(self.engine.arl)