import musicbrainzngs
import string
from modules.console_colors import ULTRASINGER_HEAD, blue_highlighted, red_highlighted


def get_music_infos(search_string: str) -> tuple[str, str, str, str]:
    print(f"{ULTRASINGER_HEAD} Searching song in {blue_highlighted('musicbrainz')}")
    # https://python-musicbrainzngs.readthedocs.io/en/v0.7.1/usage/#searching

    musicbrainzngs.set_useragent("UltraSinger", "0.1", "https://github.com/rakuri255/UltraSinger")

    # search for artist and titel to get release on the first place
    artist = None
    artists = musicbrainzngs.search_artists(search_string)
    if len(artists['artist-list']) != 0:
        artist = artists['artist-list'][0]['name'].strip()
    else:
        print(f"{ULTRASINGER_HEAD} {red_highlighted('No match found')}")
        return None, None, None, None

    release = None
    release_groups = musicbrainzngs.search_release_groups(search_string, artist=artist)
    if len(release_groups['release-group-list']) != 0:
        release = release_groups['release-group-list'][0]

    if release is not None and 'artist-credit-phrase' in release:
        artist = release['artist-credit-phrase'].strip()

    title = None
    if release is not None and 'title' in release:
        clean_search_string = search_string.translate(str.maketrans('', '', string.punctuation)).lower().strip()
        clean_release_title = release['title'].translate(str.maketrans('', '', string.punctuation)).lower().strip()
        clean_artist = artist.translate(str.maketrans('', '', string.punctuation)).lower().strip()

        # prepare search string when title and artist are the same
        if clean_release_title == clean_artist:
            # remove the first appearance of the artist
            clean_search_string = clean_search_string.replace(clean_artist, "", 1)

        if clean_release_title in clean_search_string:
            title = release['title'].strip()
        else:
            print(
                f"{ULTRASINGER_HEAD} cant find title {red_highlighted(clean_release_title)} in {red_highlighted(clean_search_string)}")

    if title is None or artist is None:
        print(f"{ULTRASINGER_HEAD} {red_highlighted('No match found')}")
        return None, None, None, None

    print(f"{ULTRASINGER_HEAD} Found Titel and Artist {blue_highlighted(title)} by {blue_highlighted(artist)}")

    year = None
    if 'first-release-date' in release:
        year = release['first-release-date'].strip()
        print(f"{ULTRASINGER_HEAD} Found release year: {blue_highlighted(year)}")

    genres = None
    if 'tag-list' in release:
        genres = ""
        for tag in release['tag-list']:
            genres += f"{tag['name'].strip()},"
        print(f"{ULTRASINGER_HEAD} Found genres: {blue_highlighted(genres)}")

    return title, artist, year, genres