# coding: utf-8

from __future__ import unicode_literals

import re

from .common import InfoExtractor


class MyTaratataIE(InfoExtractor):
    _VALID_URL = r'https?://(?:www\.)?mytaratata\.com/taratata/(?P<id>[a-zA-Z0-9_\-/]+)'
    _TEST = {
        'url': 'http://mytaratata.com/taratata/519/shaka-ponk-camille-et-julie-berthollet-smells-like-teen-spirit-nirvana',
        'md5': 'c2876e18716b350c9de69cfda2662919',
        'info_dict': {
            'id': '519/shaka-ponk-camille-et-julie-berthollet-smells-like-teen-spirit-nirvana',
            'ext': 'mp4',
            'title': 'Taratata - Shaka Ponk / Camille et Julie Berthollet "Smells Like Teen Spirit" (Nirvana)',
            'uploader': 'Taratata',
            'description': 'Shaka Ponk / Camille et Julie Berthollet "Smells Like Teen Spirit" (Nirvana)',
            # '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)

        # TODO more code goes here, for example ...
        title = self._html_search_regex(r'<title>(.+?)</title>', webpage, 'title')

        formats = []

        video_source_re = re.compile(
            r'data-source="(?P<url>http://videos.air-productions.cdn.sfr.net'
            r'/mytaratata/Taratata[^"]+\.mp4)"'
        )

        last_vid = None
        for url in video_source_re.findall(webpage):
            info_m = re.match(r'.*(?P<vid>[0-9]+)-[a-f0-9]+-(?P<w>[0-9]+)x(?P<h>[0-9]+)\.mp4', url)
            if info_m is None:
                continue
            vid = info_m.group('vid')
            w = info_m.group('w')
            h = info_m.group('h')
            if last_vid is None:
                last_vid = vid
            if vid != last_vid:
                break

            formats.append({'url': url, 'width': int(w), 'height': int(h)})

        formats = list(sorted(formats, key=lambda f: f['width']))

        return {
            'id': video_id,
            'title': title,
            'description': self._og_search_description(webpage),
            'uploader': "Taratata",
            # TODO more properties (see youtube_dl/extractor/common.py)
            'formats': formats,
        }