# coding: utf-8
from __future__ import unicode_literals

import re

from .common import InfoExtractor

from ..compat import (
    compat_b64decode,
    compat_urllib_parse_urlencode,
)


class SproutVideoIE(InfoExtractor):
    _NOSCHEMA_URL = r'//videos.sproutvideo.com/embed/(?P<id>[a-f0-9]+)/[a-f0-9]+'
    _VALID_URL = r'https?:%s' % _NOSCHEMA_URL
    _TEST = {
        'url': 'https://videos.sproutvideo.com/embed/4c9dddb01910e3c9c4/0fc24387c4f24ee3',
        'md5': 'fbc675bb97437e797d11d14d99563f50',
        'info_dict': {
            'id': '4c9dddb01910e3c9c4',
            'ext': 'mp4',
            'title': 'Adrien Labaeye : Berlin, des communautés aux communs',
        }
    }

    @staticmethod
    def _extract_url(webpage):
        sproutvideo = re.search(
            r'(?:<iframe\s+class=[\'\"]sproutvideo-player.*src|href)=[\'\"](?P<url>(?:https?:|)%s[^\'\"]+)[\'\"]' % SproutVideoIE._NOSCHEMA_URL, webpage)
        if sproutvideo:
            video_url = sproutvideo.group('url')
            # Fix the video URL if the iframe doesn't have a defined schema
            if video_url[:2] == '//':
                video_url = 'https:' + video_url
            return video_url


    def _real_extract(self, url):
        video_id = self._match_id(url)
        webpage = self._download_webpage(url, video_id)

        data = self._search_regex(r'<script[^>]+>var dat = \'([^\']+)\';</script>', webpage, 'data')
        data_decoded = compat_b64decode(data).decode('utf-8')
        parsed_data = self._parse_json(data_decoded, video_id)

        # https://github.com/ytdl-org/youtube-dl/issues/16996#issuecomment-406901324
        # signature->m for manifests
        # signature->k for keys
        # signature->t for segments
        m_sign = self._policy_to_qs(parsed_data, 'm')
        k_sign = self._policy_to_qs(parsed_data, 'k')
        t_sign = self._policy_to_qs(parsed_data, 't')

        resource_url = 'https://{0}.videos.sproutvideo.com/{1}/{2}/video/index.m3u8?{3}'
        resource_url = resource_url.format(parsed_data['base'],
                                           parsed_data['s3_user_hash'],
                                           parsed_data['s3_video_hash'],
                                           m_sign)

        formats = self._extract_m3u8_formats(resource_url, video_id, 'mp4', entry_protocol='m3u8_native',
                                             m3u8_id='hls', fatal=False)
        self._sort_formats(formats)

        for entry in formats:
            entry.update({
                'url': '{0}?{1}'.format(entry['url'], m_sign),
                'extra_param_to_segment_url': t_sign,
                'extra_param_to_key_url': k_sign,
            })

        return {
            'id': video_id,
            'title': parsed_data['title'],
            'formats': formats,
        }

    def _format_qsdata(self, qs_data):
        parsed_dict = dict()
        for key in qs_data:
            parsed_dict[key.replace('CloudFront-', '')] = qs_data[key]
        return parsed_dict

    def _policy_to_qs(self, policy, key):
        sign = self._format_qsdata(policy['signatures'][key])
        sign['sessionID'] = policy['sessionID']
        return compat_urllib_parse_urlencode(sign, doseq=True)