# coding: utf-8
from __future__ import unicode_literals

from .common import InfoExtractor

import re

from ..utils import extract_attributes

class VTVIE(InfoExtractor):
    _VALID_URL = r'https?://..\.tvnet\.gov\.vn/[^/]*/(?P<id>[0-9]+)/?.*'
    _TESTS = [{
        'url': 'http://us.tvnet.gov.vn/kenh-truyen-hinh/1011/vtv1',
        'info_dict': {
            'id': '1011',
            'ext': 'mp4',
            'title': r're:VTV1 | LiveTV - TV Net .*',
            'thumbnail': r're:https?://.*\.png$',
        }
    }, {
        'url': 'http://us.tvnet.gov.vn/video/109788/vtv1---bac-tuyet-tai-lao-cai-va-ha-giang/tin-nong-24h',
        'md5': '5263c63d738569ed507980f1e49ebc03',
        'info_dict': {
            'id': '109788',
            'ext': 'mp4',
            'title': 'VTV1 - Bắc tuyết tại Lào Cai và Hà Giang - TV Net',
            'thumbnail': r're:https?://.*\.JPG$',
        }
    }, {
        'url': 'http://us.tvnet.gov.vn/kenh-truyen-hinh/1014',
        'info_dict': {
            'id': '1014',
            'ext': 'm4a',
            'title': r're:VOV1 | LiveTV - TV Net .*',
            'thumbnail': r're:https?://.*\.png$',
        }

    }]

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

        title = self._html_search_regex(r'<title>(.+?)</title>', webpage, 'title', fatal=False)
        if title is None:
            title = self._og_search_title(webpage)
        title.strip()

        mediaplayer_div = self._search_regex(r'(<div[^>]*id="mediaplayer"[^>]*>)', webpage, 'mediaplayer element')
        mediaplayer_div_attributes = extract_attributes(mediaplayer_div)

        thumbnail = mediaplayer_div_attributes.get("data-image")

        json_url = mediaplayer_div_attributes.get("data-file")
        video_streams = self._download_json(json_url, video_id)

        # little hack to better support radio streams
        if title.startswith("VOV"):
            ext = "m4a"
        else:
            ext = "mp4"

        # get any working playlist from streams. Currently there's 2 and the first always works,
        # but you never know in the future
        for stream in video_streams:
            formats = self._extract_m3u8_formats(stream.get("url"), video_id, ext=ext, fatal=False)
            if len(formats) != 0:
                break

        if re.match(r'https?://[^/]*/video/.*', url) is not None:
            is_live = False
        elif re.match(r'https?://[^/]*/kenh-truyen-hinh/.*', url) is not None:
            is_live = True
        else:
            is_live = None

        if is_live:
            title = self._live_title(title)

        return {
            'id': video_id,
            'title': title,
            'thumbnail': thumbnail,
            'formats': formats,
            'is_live': is_live,
        }