from sites.default import DefaultSite
from sites.helper.query import Query
import requests

# TODO - Make compatible, add artist path
class MusicId(DefaultSite):
    def __init__(self):
        super().__init__()
        self.url = 'https://musicid.ru/'
        self.query = Query(self, 'GET')\
            .add_parameter('song', '{0}') \
            .add_parameter('s', 'f')
        self.structure\
            .set_container_path('div.results')\
            .set_item_path('div.chkd')\
            .set_artist_path('b.artist')\
            .set_title_path('a.trackLink')\
            .set_url_path('a.link')
        self.request\
            .add_header('Referer', self.url + '/')\
            .add_header('Origin', self.url)

    def format_url(self, field):
        link = super().format_url(field)

        request = requests.get(link, cookies=self.request.cookie, allow_redirects=True)

        return request.url