from contributor import config
from contributor.connectors.apple_music import (
    AppleArtist,
    AppleMusicClient,
    AppleMusicClientCredentialsManager,
)

_apple_music_client = AppleMusicClient(
    client_credentials_manager=AppleMusicClientCredentialsManager(
        team_id=config.APPLE_MUSIC_TEAM_ID,
        client_id=config.APPLE_MUSIC_CLIENT_ID,
        private_key=config.APPLE_MUSIC_PRIVATE_KEY,
    )
)


def artist_search(
    query: str, limit: int, offset: int, localization: str | None
) -> list[AppleArtist]:
    return _apple_music_client.search(
        query=query,
        limit=limit,
        offset=offset,
        localization=localization,
    )


def get_artist_by_id(artist_id: str) -> AppleArtist | None:
    return _apple_music_client.get_artist_by_id(artist_id=artist_id)