import re
from vulkanbot.music.Types import Provider
from vulkanbot.music.Spotify import SpotifySearch


class Searcher():
    """Turn the user input into list of musics names, support youtube and spotify"""

    def __init__(self) -> None:
        self.__Spotify = SpotifySearch()
        print(f'Spotify Connected: {self.__Spotify.connect()}')

    def search(self, music: str) -> list:
        """Return a list with the track name of a music or playlist

        Return -> A list of musics names
        """
        url_type = self.__identify_source(music)

        if url_type == Provider.YouTube:
            return [music], Provider.YouTube

        elif url_type == Provider.Spotify:
            musics = self.__Spotify.search(music)
            return musics, Provider.Name

        elif url_type == Provider.Name:
            return [music], Provider.Name

    def __identify_source(self, music) -> Provider:
        """Identify the provider of a music"""
        if not self.__is_url(music):
            return Provider.Name

        if "https://www.youtu" in music or "https://youtu.be" in music:
            return Provider.YouTube

        if "https://open.spotify.com" in music:
            return Provider.Spotify

        # If no match
        return Provider.Unknown

    def __is_url(self, string) -> bool:
        """Verify if a string is a url"""
        regex = re.compile(
            "http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+")

        if re.search(regex, string):
            return True
        else:
            return False