# coding: utf-8
from __future__ import unicode_literals

import json
import re

from .common import InfoExtractor
from ..utils import urljoin


class TelevizeSeznamIE(InfoExtractor):
    _VALID_URL = r'https?://(?:www\.)?televizeseznam\.cz/.+/(?P<display_id>.+)-(?P<id>[0-9]+)'

    _GRAPHQL_URL = 'https://www.televizeseznam.cz/api/graphql'
    _GRAPHQL_QUERY = '''query LoadEpisode($urlName : String){ episode(urlName: $urlName){ ...VideoDetailFragmentOnEpisode } }
        fragment VideoDetailFragmentOnEpisode on Episode {
            id
            spl
            urlName
            name
            perex
        }
'''

    _TEST = {
        'url': 'https://www.televizeseznam.cz/video/lajna/buh-57953890',
        'md5': '40c41ade1464a390a0b447e333df4239',
        'info_dict': {
            'id': '57953890',
            'display_id': 'buh',
            'title': 'Bůh',
            'description': 'Trenér Hrouzek je plný rozporů. Na pomoc si povolá i toho nejvyššího. Kdo to ale je? Pomůže mu vyřešit několik dilemat, která se mu v poslední době v životě nahromadila?',
            'ext': 'mp4',
        }
    }

    def extract_subtitles(self, spl_url, play_list):

        if not play_list:
            return None

        subtitles = {}
        for k, v in play_list.items():
            subtitles.update({
                v['language']: {
                    'ext': 'srt',
                    'url': urljoin(spl_url, v['urls']['srt'])
                }
            })
        return subtitles

    def extract_formats(self, spl_url, play_list, subtitles):
        formats = []
        for r, v in play_list.items():
            format = {
                'format_id': r,
                'url': urljoin(spl_url, v.get('url')),
                'protocol': 'https',
                'ext': 'mp4',
                'subtitles': subtitles,
            }
            if v.get('resolution'):
                format.update({ 'width': v['resolution'][0], 'height': v['resolution'][1] })

            formats.append(format)

        return formats

    def _real_extract(self, url):
        display_id, video_id = re.match(self._VALID_URL, url).groups()

        data = self._download_json(
            self._GRAPHQL_URL, video_id, 'Downloading GraphQL result',
            data=json.dumps({
                'query': self._GRAPHQL_QUERY,
                'variables': {'urlName': video_id}
            }).encode('utf-8'),
            headers={'Content-Type': 'application/json;charset=UTF-8'}
        )['data']

        spl_url = data['episode']['spl'] + 'spl2,3'
        metadata = self._download_json(spl_url, video_id, 'Downloading playlist')
        if 'Location' in metadata and 'data' not in metadata:
            # they sometimes wants to redirect
            spl_url = metadata['Location']
            metadata = self._download_json(spl_url, video_id, 'Redirected -> Downloading playlist')
        play_list = metadata['data']
        subtitles = self.extract_subtitles(spl_url, play_list.get('subtitles'))
        formats = self.extract_formats(spl_url, play_list['mp4'], subtitles)

        return {
            'id': video_id,
            'display_id': display_id,
            'title': data['episode'].get('name'),
            'description': data['episode'].get('perex'),
            'formats': formats
        }