# coding: utf-8
from __future__ import unicode_literals
from collections import OrderedDict
from datetime import datetime, timedelta
import re
import subprocess as sp
import time

from .common import InfoExtractor
from ..utils import compat_str, int_or_none


class PewTubeIE(InfoExtractor):
    _VALID_URL = r'https?://(?:www\.)?pewtube\.com/user/[^/]+/(?P<id>[^/]+)'
    _TEST = {
        'url': 'https://pew.tube/user/MrBond/4jLJf06',
        'md5': 'TODO: md5 sum of the first 10241 bytes of the video file (use --test)',
        'info_dict': {
            'id': '4jLJf06',
            'ext': 'mp4',
            'title': 'Mr. Bond - Good Old Nationalist',
            'thumbnail': r're:^https?://.*\.jpg$',
            # TODO more properties, either as:
            # * A value
            # * MD5 checksum; start the string with md5:
            # * A regular expression; start the string with re:
            # * Any Python type (for example int or float)
        }
    }

    def _real_extract(self, url):
        video_id = self._match_id(url)
        webpage = self._download_webpage(url, video_id, tries=5, timeout=15)

        title = self._html_search_regex(r'<h2 .*upload\-title\-value[^>]+>([^<]+)</h2>', webpage, 'title')
        thumbnail = self._html_search_regex(r'<video .*poster="(https[^"]+)"', webpage, 'thumbnail', fatal=False)
        if not thumbnail:
            thumbnail = self._og_search_thumbnail(webpage)
        video_url = self._html_search_regex(r'<(?:audio|video)[^>]+>.*<source.*src="([^"]+)".*</(?:audio|video)>', webpage, 'video URL')
        if not video_url:
            video_url = self._og_search_video_url(webpage)
        uploader = self._html_search_regex(r'<h3.*class="uploader-name"[^>]+>.*<a href="/user[^>]+>([^<]+)', webpage, 'uploader')
        description = self._html_search_meta('description', webpage, 'description')
        view_count = self._html_search_regex(r'<h3(?:[^>]+)?>(\d+)(?:\s+)[vV]iews</h3>', webpage, 'view count')

        like_count = 0
        for like_class in ('like', 'laugh', 'love',):
            like_count += int(self._html_search_regex(r'<p class="{}">(\d+)</p>'.format(like_class), webpage, '{} count'))

        dislike_count = 0
        for dislike_class in ('dislike', 'sad', 'discount',):
            dislike_count += int(self._html_search_regex(r'<p class="{}">(\d+)</p>'.format(like_class), webpage, '{} count'))

        like_count = None if not like_count else like_count
        dislike_count = None if not dislike_count else dislike_count

        if not thumbnail:
            thumbnail = self._og_search_thumbnail(webpage)
        if not description:
            description = self._og_search_description(webpage)

        upload_date_s = ou = self._html_search_regex(r'<h4>Uploaded ([^<]+)</h4>', webpage, 'upload date')
        today = datetime.today()
        upload_date_s = re.sub(r'(?:\s+)ago\s+\&nbsp;?$', '', upload_date_s)
        n, unit = re.search('^(?P<n>(?:\d+|a))\s(?:\s+)?(?P<unit>.*)', upload_date_s).groups()
        if n == 'a':
            n = 1
        else:
            n = int(n)
        unit = unit.lower().encode('utf-8')
        total_seconds = 0
        if not unit.endswith('s'):
            unit += 's'
        if unit == 'months':
            unit = 'weeks'
            n *= 4
        elif unit == 'years':
            unit = 'weeks'
            n *= 52
        kwargs = dict()
        kwargs[unit] = n
        td = timedelta(**kwargs)
        upload_date = (today - td).strftime('%Y%m%d')

        return {
            'id': video_id,
            'title': title,
            'description': description,
            'uploader': uploader,
            'creator': uploader,
            'uploader_url': '/'.join(url.split('/')[:-1]),
            'thumbnail': thumbnail,
            'view_count': int_or_none(view_count),
            'like_count': like_count,
            'dislike_count': dislike_count,
            'upload_date': upload_date,
            'url': video_url,
        }