from attrs import define

from .engine import SoundCloudEngine


@define
class SoundCloudDriver:
    engine: SoundCloudEngine

    async def get_track(self, track_id: int | str):
        return await self.engine.call(
            f'tracks/{track_id}'
        )

    async def search(self, query: str, limit: int = 30):
        return (await self.engine.call(
            'search/tracks',
            params={
                'q': query,
                'limit': limit
            }
        ))['collection']

    async def resolve_url(self, url: str):
        return await self.engine.call(
            'resolve',
            params={
                'url': url
            }
        )