# import json
# import re
import requests
from engine.plugins import YDownload, BatchCheckBase
from common import logger

headers = {
    'client-id': 'jzkbprff40iqj646a697cyrvl0zt2m6'
}
VALID_URL_BASE = r'(?:https?://)?(?:(?:www|go|m)\.)?twitch\.tv/(?P<id>[0-9_a-zA-Z]+)'
API_ROOMS = 'https://api.twitch.tv/helix/streams'
_API_USER = 'https://api.twitch.tv/helix/users'


class Twitch(YDownload):
    def __init__(self, fname, url, suffix='mp4'):
        YDownload.__init__(self, fname, url, suffix=suffix)

    def dl(self):
        info_list = self.get_sinfo()

        if self.fname not in ['星际2PartinG跳跳胖丁神族天梯第一视角']:
            pass
        elif '720p' in info_list:
            self.ydl_opts['format'] = '720p'
        elif '720p60' in info_list:
            self.ydl_opts['format'] = '720p60'

        super().dl()


# def check_stream(self):
#
#     check_url = re.sub(r'.*twitch.tv', 'https://api.twitch.tv/kraken/streams', self.url)
#     try:
#         res = requests.get(check_url, headers=headers)
#         res.close()
#     except requests.exceptions.SSLError:
#         logger.error('获取流信息发生错误')
#         logger.error(requests.exceptions.SSLError, exc_info=True)
#         return None
#     except requests.exceptions.ConnectionError:
#         logger.exception('During handling of the above exception, another exception occurred:')
#         return None
#
#     try:
#         s = json.loads(res.text)
#         # s = res.json()  https://api.twitch.tv/kraken/streams/
#     except json.decoder.JSONDecodeError:
#         logger.exception('Expecting value')
#         return None
#     print(self.fname)
#     try:
#         stream = s['stream']
#     except KeyError:
#         logger.error(KeyError, exc_info=True)
#         return None
#     return stream

class BatchCheck(BatchCheckBase):
    def __init__(self, urls):
        BatchCheckBase.__init__(self, pattern_id=VALID_URL_BASE, urls=urls)
        self.use_id = {}
        if self.usr_list:
            login = requests.get(_API_USER, headers=headers, params={'login': self.usr_list}, timeout=5)
            login.close()
        else:
            logger.debug('无twitch主播')
            return
        try:
            for pair in login.json()['data']:
                self.use_id[pair['id']] = pair['login']
        except KeyError:
            logger.info(login.json())
            return

    def check(self):

        live = []
        usr_list = self.usr_list
        if not usr_list:
            logger.debug('无用户列表')
            return
        # url = 'https://api.twitch.tv/kraken/streams/sc2_ragnarok'

        stream = requests.get(API_ROOMS, headers=headers, params={'user_login': usr_list}, timeout=5)
        stream.close()

        data = stream.json()['data']
        if data:
            for i in data:
                live.append(self.use_id[i['user_id']])
        else:
            logger.debug('twitch无开播')

        return map(lambda x: self.usr_dict.get(x.lower()), live)


__plugin__ = Twitch