# coding: utf-8
from __future__ import unicode_literals

import string
import random
import time

from .common import InfoExtractor
from ..utils import (
    js_to_json,
    urljoin,
)


def doodExe(crp, crs):
    if crp == 'N_crp':
        return crs
    sorted_crp = ''.join(sorted(crp))
    result = ''
    for c in crs:
        i = crp.find(c)
        if i >= 0:
            result += sorted_crp[i]
    result = result.replace('+.+', '(')
    result = result.replace('+..+', ')')
    result = result.replace('+-+', '[')
    result = result.replace('+--+', ']')
    result = result.replace('+', ' ')
    return result


class DoodStreamIE(InfoExtractor):
    _VALID_URL = r'https?://(?:www\.)?(?:doodstream\.com|dood\.(?:so|to|watch))/[de]/(?P<id>[^/?#]+)'
    _TESTS = [{
        'url': 'https://dood.to/d/wpyp2mgwi2kb',
        'md5': '2aaf633bcd5fefb64b27344f55022bf9',
        'info_dict': {
            'id': 'wpyp2mgwi2kb',
            'ext': 'mp4',
            'title': 'Big Buck Bunny Trailer',
            'thumbnail': r're:^https?://.*\.jpg$',
            'filesize': 4447915,
            'duration': 33,
        },
    }]

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

        url = urljoin(url, '/e/' + video_id)
        referer = {'Referer': url}
        webpage = self._download_webpage(url, video_id)

        metadata_url = self._html_search_regex(r"('/cptr/[^']*')", webpage,
                                               'video metadata')
        metadata_url = self._parse_json(metadata_url, video_id,
                                        transform_source=js_to_json)
        metadata_url = urljoin(url, metadata_url)
        metadata = self._download_json(metadata_url, video_id, headers=referer)

        thumb = self._og_search_thumbnail(webpage)
        try:
            filesize = int(doodExe(**metadata['siz']), 10)
        except (KeyError, ValueError):
            filesize = None
        try:
            duration = int(doodExe(**metadata['len']), 10)
        except (KeyError, ValueError):
            duration = None
        try:
            title = doodExe(**metadata['ttl'])
        except KeyError:
            title = video_id

        token = self._html_search_regex(r'[?&]token=([a-z0-9]+)[&\']', webpage, 'token')
        auth_url = self._html_search_regex(r"('/pass_md5.*?')", webpage,
                                           'pass_md5')
        auth_url = self._parse_json(auth_url, video_id,
                                    transform_source=js_to_json)
        auth_url = urljoin(url, auth_url)

        webpage = self._download_webpage(auth_url, video_id, headers=referer)
        final_url = webpage + ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(10)) + "?token=" + token + "&expiry=" + str(int(time.time() * 1000))

        return {
            'id': video_id,
            'title': title,
            'url': final_url,
            'http_headers': referer,
            'ext': 'mp4',
            'thumbnail': thumb,
            'filesize': filesize,
            'duration': duration,
        }