import hashlib
import logging
import re
import time

from streamlink.plugin import Plugin, PluginError, pluginmatcher
from streamlink.plugin.api import validate
from streamlink.stream import HLSStream
from streamlink.utils import parse_json

log = logging.getLogger(__name__)


@pluginmatcher(re.compile(
    r'https?://(?:www\.)?huomao\.(?:tv|com)(?P<path>/|/video/v/)(?P<room_id>\d+)'
))
class Huomao(Plugin):
    magic_val = '6FE26D855E1AEAE090E243EB1AF73685'
    mobile_url = 'https://m.huomao.com/mobile/mob_live/{0}'
    live_data_url = 'https://m.huomao.com/swf/live_data'
    vod_url = 'https://www.huomao.com/video/vreplay/{0}'

    author = None
    category = None
    title = None

    author_re = re.compile(
        r'<p class="nickname_live">\s*<span>\s*(.*?)\s*</span>',
        re.DOTALL,
    )

    title_re = re.compile(
        r'<p class="title-name">\s*(.*?)\s*</p>',
        re.DOTALL,
    )

    video_id_re = re.compile(r'var stream = "([^"]+)"')
    video_res_re = re.compile(r'_([\d]+p?)\.m3u8')
    vod_data_re = re.compile(r'var video = ({.*});')

    _live_data_schema = validate.Schema({
        'roomStatus': validate.transform(lambda x: int(x)),
        'streamList': [{'list_hls': [{
            'url': validate.url(),
        }]}],
    })

    _vod_data_schema = validate.Schema({
        'title': validate.text,
        'username': validate.text,
        'vaddress': validate.all(
            validate.text,
            validate.transform(parse_json),
            [{
                'url': validate.url(),
                'vheight': int,
            }],
        ),
    })

    def _get_live_streams_data(self, video_id):
        client_type = 'huomaomobileh5'
        time_now = str(int(time.time()))

        token_data = "{0}{1}{2}{3}".format(
            video_id,
            client_type,
            time_now,
            self.magic_val,
        )

        token = hashlib.md5(bytes(token_data, 'utf-8')).hexdigest()
        log.debug("Token={0}".format(token))

        post_data = {
            'cdns': 1,
            'streamtype': 'live',
            'VideoIDS': video_id,
            'from': client_type,
            'time': time_now,
            'token': token,
        }
        video_data = self.session.http.post(self.live_data_url, data=post_data)

        return self.session.http.json(
            video_data,
            schema=self._live_data_schema,
        )

    def _get_vod_streams(self, vod_id):
        res = self.session.http.get(self.vod_url.format(vod_id))
        m = self.vod_data_re.search(res.text)
        vod_json = m and m.group(1)

        if vod_json is None:
            raise PluginError("Failed to get VOD data")

        vod_data = parse_json(vod_json, schema=self._vod_data_schema)

        self.author = vod_data['username']
        self.category = 'VOD'
        self.title = vod_data['title']

        vod_data = vod_data['vaddress']

        streams = {}
        for stream in vod_data:
            video_res = stream['vheight']

            if 'p' not in str(video_res):
                video_res = "{0}p".format(video_res)

            if video_res in streams:
                video_res = "{0}_alt".format(video_res)

            streams[video_res] = HLSStream(self.session, stream['url'])

        return streams

    def _get_live_streams(self, room_id):
        res = self.session.http.get(self.mobile_url.format(room_id))

        m = self.author_re.search(res.text)
        if m:
            self.author = m.group(1)

        self.category = 'Live'

        m = self.title_re.search(res.text)
        if m:
            self.title = m.group(1)

        m = self.video_id_re.search(res.text)
        video_id = m and m.group(1)

        if video_id is None:
            raise PluginError("Failed to get video ID")
        else:
            log.debug("Video ID={0}".format(video_id))

        streams_data = self._get_live_streams_data(video_id)

        if streams_data['roomStatus'] == 0:
            log.info("This room is currently inactive: {0}".format(room_id))
            return

        streams_data = streams_data['streamList'][0]['list_hls']

        streams = {}
        for stream in streams_data:
            m = self.video_res_re.search(stream['url'])
            video_res = m and m.group(1)
            if video_res is None:
                continue

            if 'p' not in video_res:
                video_res = "{0}p".format(video_res)

            if video_res in streams:
                video_res = "{0}_alt".format(video_res)

            streams[video_res] = HLSStream(self.session, stream['url'])

        return streams

    def get_author(self):
        if self.author is not None:
            return self.author

    def get_category(self):
        if self.category is not None:
            return self.category

    def get_title(self):
        if self.title is not None:
            return self.title

    def _get_streams(self):
        path, url_id = self.match.groups()
        log.debug("Path={0}".format(path))
        log.debug("URL ID={0}".format(url_id))

        if path != '/':
            return self._get_vod_streams(url_id)
        else:
            return self._get_live_streams(url_id)


__plugin__ = Huomao
