# coding: utf-8
from __future__ import unicode_literals

import re
import hashlib
from ..aes import aes_cbc_decrypt
from ..compat import (
    compat_b64decode,
)

from .common import InfoExtractor
from ..utils import (
    bytes_to_intlist,
    intlist_to_bytes,
    int_or_none,
)


class GaanaBaseIE(InfoExtractor):
    _BASE_URL = 'https://gaana.com'
    _API_URL = 'https://apiv2.gaana.com/track/stream'
    _COOKIE = ''

    def _Decrypt(self, data):

        key = b'g@1n!(f1#r.0$)&%'
        iv = b'asd!@#!@#@!12312'
        stream_url = intlist_to_bytes(aes_cbc_decrypt(
            bytes_to_intlist(compat_b64decode(data)),
            bytes_to_intlist(key),
            bytes_to_intlist(iv))).decode()
        # unpad
        s = stream_url[:-ord(stream_url[len(stream_url) - 1:])]
        return s

    def _Create_ht(self, track_id):
        if not self._COOKIE:
            self._COOKIE = self._get_cookies(self._BASE_URL)['PHPSESSID'].value

        mess = track_id + '|' + self._COOKIE + '|03:40:31 sec'
        ht = hashlib.md5(mess.encode()).hexdigest() + self._COOKIE[3:9] + '='
        return ht

    def _create_entry(self, raw_data, video_id):
        video_data = raw_data.get('path')
        title = raw_data.get('title')
        thumbnail = raw_data.get('albumartwork').replace("_175x175_", "_480x480_")
        duration = raw_data.get('duration')
        artist = raw_data.get('artist')

        def _format_artist(art):
            r_sample = r'#..(\d+)#..(\w+)[^|,]*'
            res = re.sub(r_sample, '', art)
            return re.sub(r',', ', ', res)

        if artist:
            artist = _format_artist(artist)

        formats = []
        if isinstance(video_data, dict):
            for value in video_data.keys():
                content = video_data.get(value)
                for k in content:
                    format_url = self._Decrypt(k.get('message'))

                    if value == 'auto':
                        format_id = 'normal'
                    else:
                        format_id = value

                    info = {
                        'url': format_url,
                        'format_id': format_id,
                        'ext': 'mp4',
                        'abr': int_or_none(k.get('bitRate')),
                        'format_note': 'mp4-aac'
                    }

                    if format_id == 'normal':
                        formats.insert(0, info)
                    else:
                        formats.append(info)

        else:
            track_id = raw_data.get('track_ids')
            ht = self._Create_ht(track_id)

            for g in ('normal', 'medium', 'high'):
                js = self._download_json(self._API_URL, title, headers={
                    'async': '1',
                    'method': 'POST'}, query={
                    'ht': ht,
                    'request_type': 'web',
                    'track_id': track_id,
                    'quality': g
                })
                format_url = js.get('stream_path')

                formats.append({
                    'url': format_url,
                    'format_id': g,
                    'ext': 'mp4',
                    'abr': int_or_none(js.get('bit_rate')),
                    'format_note': 'mp4-aac'
                })

        return {
            'id': video_id,
            'title': title,
            'duration': int_or_none(duration),
            'formats': formats,
            'album': raw_data.get('albumtitle'),
            'thumbnail': thumbnail,
            'artist': artist,
            'release_date': raw_data.get('release_date'),
            'language': raw_data.get('language')
        }


class GaanaIE(GaanaBaseIE):
    IE_NAME = 'gaana'
    _VALID_URL = r'https?://(?:www\.)?gaana\.com/(?P<idtype>(song|album|artist|playlist))/(?P<id>[^/#?]+)'
    _TESTS = [{
        'url': 'https://gaana.com/song/chamma-chamma-5',
        'md5': 'c94cc6896b24bc2b5bb3e5f496b0c809',
        'info_dict': {
            'id': '24725286',
            'ext': 'mp4',
            'title': 'Chamma Chamma',
            'thumbnail': r'https://a10.gaanacdn.com/images/song/86/24725286/crop_480x480_1544754220.jpg',
            'duration': 196,
            'album': 'Fraud Saiyaan',
            'artist': 'Neha Kakkar, Romi, Arun, Ikka',
            'language': 'Hindi',
            'release_date': 'Dec 14, 2018'
        }
    }, {
        'url': 'https://gaana.com/song/saaun-di-jhadi-1',
        'md5': 'f484453440d9c4edd2d320bd2ce7a069',
        'info_dict': {
            'id': '1742426',
            'ext': 'mp4',
            'title': 'Saaun Di Jhadi',
            'thumbnail': r'https://a10.gaanacdn.com/images/albums/25/151425/crop_480x480_151425.jpg',
            'duration': 301,
            'album': 'Virsa - Romance',
            'artist': None,
            'language': 'Punjabi',
            'release_date': 'Jan 20, 2014'
        }
    }, {
        'url': 'https://gaana.com/album/simmba',
        'playlist_count': 7,
        'info_dict': {
            'id': '2291554',
            'title': 'Simmba'
        }
    }, {
        'url': 'https://gaana.com/artist/dhvani-bhanushali',
        'playlist_min_count': 26,
        'info_dict': {
            'id': '1380321',
            'title': 'Dhvani Bhanushali'
        }
    }, {
        'url': 'https://gaana.com/playlist/gaana-dj-jatt-di-playlist',
        'playlist_min_count': 19,
        'info_dict': {
            'id': '8700656',
            'title': 'Jatt Di Playlist'
        }
    }]

    def _real_extract(self, url):
        r_match = re.match(self._VALID_URL, url)

        video_id = r_match.group('id')
        type_id = r_match.group('idtype')
        self.IE_NAME = 'gaana:' + type_id

        self._set_cookie(self._BASE_URL, 'PHPSESSID', 'val')
        webpage = self._download_webpage(url, video_id)

        matchobj = re.findall(r'class="parentnode sourcelist_\d+">(.*?)</span>', webpage)
        matchobj = [self._parse_json(g, video_id) for g in matchobj]
        entries = []
        if len(matchobj) > 1:
            for g in matchobj:
                entries.append(self._create_entry(g, g['id']))
            data = re.findall(r'id="parent-row-(?:album|artist|playlist)\d+">(.*?)</span>', webpage)
            data = self._parse_json(data[0], video_id)
            return self.playlist_result(entries, data['id'], data['title'])
        else:
            return self._create_entry(matchobj[0], matchobj[0]['id'])