# coding: utf-8
from __future__ import unicode_literals
import re

from .common import InfoExtractor
from ..compat import compat_str
from ..utils import (
    determine_ext,
    int_or_none,
    parse_iso8601,
    try_get,
    ExtractorError,
)


class RumbleIE(InfoExtractor):

    RE_DICT = {
        'iframe_url': {
            're': r'https?://(?:www\.)?rumble\.com/embed/(?:[0-9a-z]+\.)?(?P<id>[0-9a-z]+)',
            'compiled': None},
        'jscript_url': {
            're': r'https?://rumble\.com/[a-zA-Z0-9-_.]*\.html',
            'compiled': None},
        'list_url': {
            're': r'https?://rumble.com/(?:c|user)/(?P<id>[^/]+)',
            'compiled': None},
        'jscript_id': {
            're': r'Rumble *\( *["\']play["\'], *\{[^}]*["\']video["\'] *: *["\'](?P<id>[^"\']+)',
            'compiled': None}
    }

    _TESTS = [
        {
            'url': 'https://rumble.com/embed/v5pv5f',
            'md5': '36a18a049856720189f30977ccbb2c34',
            'info_dict': {
                'id': 'v5pv5f',
                'ext': 'mp4',
                'title': 'WMAR 2 News Latest Headlines | October 20, 6pm',
                'timestamp': 1571611968,
                'upload_date': '20191020',
            }
        },
        {
            'url': 'https://rumble.com/v8c1bt-wmar-2-news-latest-headlines-october-20-6pm.html',
            'md5': '36a18a049856720189f30977ccbb2c34',
            'info_dict': {
                'id': 'v5pv5f',
                'ext': 'mp4',
                'title': 'WMAR 2 News Latest Headlines | October 20, 6pm',
                'timestamp': 1571611968,
                'upload_date': '20191020',
            }
        },
        {
            'url': 'https://rumble.com/c/PeakProsperity',
            'playlist_mincount': 25,
            'info_dict': {
                'id': 'PeakProsperity',
            }
        },
        {
            'url': 'https://rumble.com/embed/ufe9n.v5pv5f',
            'only_matching': True,
        }
    ]

    @classmethod
    def get_re(cls, tag):
        if cls.RE_DICT[tag]['compiled'] is None:
            cls.RE_DICT[tag]['compiled'] = re.compile(cls.RE_DICT[tag]['re'])
        return cls.RE_DICT[tag]['compiled']

    @classmethod
    def suitable(cls, url):
        return (cls.get_re('jscript_url').match(url) is not None or
                cls.get_re('list_url').match(url) is not None or
                cls.get_re('iframe_url').match(url) is not None)

    @staticmethod
    def rumble_embedded_id(page_data):
        '''For use by extractors of sites which use emedded Rumble videos. Given
        a webpage as a string returns a list of url result dicts for each embedded
        rumble video found. None is returned if no embeds were found. Duplicates
        are not removed'''

        embeds = []
        # The JS embeds
        for mobj in RumbleIE.get_re('jscript_id').finditer(page_data):
            embeds.append(InfoExtractor.url_result('https://rumble.com/embed/' + mobj.group('id'), 'Rumble', mobj.group('id')))

        # The iframes embeds
        for mobj in RumbleIE.get_re('iframe_url').finditer(page_data):
            embeds.append(InfoExtractor.url_result('https://rumble.com/embed/' + mobj.group('id'), 'Rumble', mobj.group('id')))

        return embeds if embeds else None

    def rumble_video_info(self, video_id):
        video = self._download_json(
            'https://rumble.com/embedJS/', video_id,
            query={'request': 'video', 'v': video_id})
        if not video:
            raise ExtractorError('Unable to locate video information.', expected=True)

        title = video['title']

        formats = []
        for height, ua in (video.get('ua') or {}).items():
            for i in range(2):
                f_url = try_get(ua, lambda x: x[i], compat_str)
                if f_url:
                    ext = determine_ext(f_url)
                    f = {
                        'ext': ext,
                        'format_id': '%s-%sp' % (ext, height),
                        'height': int_or_none(height),
                        'url': f_url,
                    }
                    bitrate = try_get(ua, lambda x: x[i + 2]['bitrate'])
                    if bitrate:
                        f['tbr'] = int_or_none(bitrate)
                    formats.append(f)
        self._sort_formats(formats)

        author = video.get('author') or {}

        return {
            'id': video_id,
            'title': title,
            'formats': formats,
            'thumbnail': video.get('i'),
            'timestamp': parse_iso8601(video.get('pubDate')),
            'channel': author.get('name'),
            'channel_url': author.get('url'),
            'duration': int_or_none(video.get('duration')),
        }

    def _real_extract(self, url):
        if self.get_re('jscript_url').match(url) is not None:
            page = self._download_webpage(url, 'Rumble Page')
            video_id = self._search_regex(self.get_re('jscript_id'), page, "id")
            return self.rumble_video_info(video_id)

        mobj = self.get_re('list_url').match(url)
        if mobj is not None:
            urls = []
            id = mobj.group('id')
            page = self._download_webpage(url, id)
            for mobj in re.finditer(r'<a class=video-item--a href=\/(?P<href>[a-zA-Z0-9\-.]+)>', page):
                urls.append('https://rumble.com/' + mobj.group('href'))

            return self.playlist_from_matches(urls, id)

        mobj = self.get_re('iframe_url').match(url)
        if mobj is not None:
            return self.rumble_video_info(mobj.group('id'))