from typing import Optional
from config import IMAGE_SERVICE_URL


class ImageService:
    ARTIST_IMG_URL_API = IMAGE_SERVICE_URL + "/v2/artists/by_spotify_id/"
    ALBUM_IMG_URL_API = IMAGE_SERVICE_URL + "/v2/albums/by_spotify_id/"
    TRACK_IMG_URL_API = IMAGE_SERVICE_URL + "/v2/albums/by_spotify_track_id/"

    @classmethod
    def get_img_url(cls, spotify_artist_id: str) -> str:
        return f"{cls.ARTIST_IMG_URL_API}{spotify_artist_id}"

    @classmethod
    def get_album_img_url(cls, spotify_album_id: str) -> str:
        return cls.ALBUM_IMG_URL_API + spotify_album_id

    @classmethod
    def get_track_img_url(cls, spotify_track_id: Optional[str]) -> str:
        return f"{cls.TRACK_IMG_URL_API}{spotify_track_id}"