#!/usr/bin/python
# -*- coding: utf-8 -*-

# for localized messages
from . import _
from . import xstreamity_globals as glob

from .plugin import skin_path, hdr, cfg, common_path, json_file
from .xStaticText import StaticText

from Components.ActionMap import ActionMap
from Components.Pixmap import Pixmap
from Components.Sources.List import List
from datetime import datetime
from enigma import eTimer
from multiprocessing.pool import ThreadPool
from requests.adapters import HTTPAdapter
from Screens.MessageBox import MessageBox
from Screens.Screen import Screen
from Tools.LoadPixmap import LoadPixmap

try:
    from requests.packages.urllib3.util.retry import Retry
except:
    from urllib3.util import Retry

import json
import requests


class XStreamity_Menu(Screen):

    def __init__(self, session):
        Screen.__init__(self, session)
        self.session = session

        skin = skin_path + 'menu.xml'
        with open(skin, 'r') as f:
            self.skin = f.read()

        self.list = []
        self.drawList = []
        self["list"] = List(self.drawList, enableWrapAround=True)

        self.setup_title = (_('Stream Selection'))

        self['key_red'] = StaticText(_('Back'))
        self['key_yellow'] = StaticText(_('Refresh Lists'))

        self['lastchecked'] = StaticText(_("Lists updated: ") + str(glob.current_playlist['data']['last_check']))

        self["splash"] = Pixmap()
        self["splash"].show()

        self['actions'] = ActionMap(['XStreamityActions'], {
            'red': self.quit,
            'cancel': self.quit,
            'yellow': self.updateCategories,
            'ok': self.__next__,
        }, -2)

        self.protocol = glob.current_playlist['playlist_info']['protocol']
        self.domain = glob.current_playlist['playlist_info']['domain']
        self.host = glob.current_playlist['playlist_info']['host']
        self.username = glob.current_playlist['playlist_info']['username']
        self.password = glob.current_playlist['playlist_info']['password']

        self.live_streams = "%s/player_api.php?username=%s&password=%s&action=get_live_streams" % (self.host, self.username, self.password)
        self.p_live_categories_url = "%s/player_api.php?username=%s&password=%s&action=get_live_categories" % (self.host, self.username, self.password)
        self.p_vod_categories_url = "%s/player_api.php?username=%s&password=%s&action=get_vod_categories" % (self.host, self.username, self.password)
        self.p_series_categories_url = "%s/player_api.php?username=%s&password=%s&action=get_series_categories" % (self.host, self.username, self.password)
        self.p_live_streams_url = "%s/player_api.php?username=%s&password=%s&action=get_live_streams" % (self.host, self.username, self.password)

        glob.current_playlist['data']['live_streams'] = []

        self.onFirstExecBegin.append(self.delayedDownload)
        self.onLayoutFinish.append(self.__layoutFinished)

    def __layoutFinished(self):
        self.setTitle(self.setup_title)

    # delay to allow splash screen to show
    def delayedDownload(self):
        if glob.current_playlist['data']['live_categories'] == [] and glob.current_playlist['data']['vod_categories'] == [] and glob.current_playlist['data']['series_categories'] == []:

            glob.current_playlist['data']['last_check'] = datetime.now().strftime("%d/%m/%Y %H:%M")
            self['lastchecked'].setText(_("Lists updated: ") + str(glob.current_playlist['data']['last_check']))

            self.timer = eTimer()
            try:
                self.timer_conn = self.timer.timeout.connect(self.makeUrlList)
            except:
                try:
                    self.timer.callback.append(self.makeUrlList)
                except:
                    self.makeUrlList()
            self.timer.start(5, True)
        else:
            self["splash"].hide()
            self.createSetup()

    def makeUrlList(self):
        self.url_list = []

        if glob.current_playlist['player_info']['showlive'] is True:
            self.url_list.append([self.p_live_categories_url, 0])

        if glob.current_playlist['player_info']['showvod'] is True:
            self.url_list.append([self.p_vod_categories_url, 1])

        if glob.current_playlist['player_info']['showseries'] is True:
            self.url_list.append([self.p_series_categories_url, 2])

        if glob.current_playlist['player_info']['showcatchup'] is True:
            if glob.current_playlist['data']['catchup_checked'] is False:
                self.url_list.append([self.p_live_streams_url, 3])
                glob.current_playlist['data']['catchup_checked'] = True

        self.process_downloads()

    def download_url(self, url):
        timeout = cfg.timeout.getValue()
        category = url[1]
        r = ''

        # retries = Retry(total=3, status_forcelist=[408, 429, 500, 503, 504], backoff_factor = 1)

        adapter = HTTPAdapter(max_retries=0)
        http = requests.Session()
        http.mount("http://", adapter)

        try:
            r = http.get(url[0], headers=hdr, stream=False, timeout=timeout, verify=False)
            r.raise_for_status()
            if r.status_code == requests.codes.ok:
                response = r.json()
                return category, r.json()

        except requests.exceptions.ConnectionError as e:
            print(("Error Connecting: %s" % e))
            return category, ''

        except requests.exceptions.RequestException as e:
            print(e)
            return category, ''

    def process_downloads(self):
        threads = 10
        pool = ThreadPool(threads)
        results = pool.imap_unordered(self.download_url, self.url_list)

        for category, response in results:
            if response != '':
                # add categories to main json file
                if category == 0:
                    glob.current_playlist['data']['live_categories'] = response
                elif category == 1:
                    glob.current_playlist['data']['vod_categories'] = response
                elif category == 2:
                    glob.current_playlist['data']['series_categories'] = response
                elif category == 3:
                    glob.current_playlist['data']['live_streams'] = response

        pool.terminate()
        self["splash"].hide()
        self.createSetup()

    def writeJsonFile(self):
        with open(json_file, "r+") as f:
            self.playlists_all = json.load(f)
            self.playlists_all[glob.current_selection] = glob.current_playlist
            f.seek(0)
            json.dump(self.playlists_all, f)

    def createSetup(self):
        self.list = []
        self.index = 0

        if glob.current_playlist['player_info']['showlive'] is True:
            if glob.current_playlist['data']['live_categories'] != []:
                self.index += 1
                self.list.append([self.index, "Live Streams", 0, ""])

        if glob.current_playlist['player_info']['showvod'] is True:
            if glob.current_playlist['data']['vod_categories'] != []:
                self.index += 1
                self.list.append([self.index, "Vod", 1, ""])

        if glob.current_playlist['player_info']['showseries'] is True:
            if glob.current_playlist['data']['series_categories'] != []:
                self.index += 1
                self.list.append([self.index, "TV Series", 2, ""])

        content = glob.current_playlist['data']['live_streams']
        hascatchup = any(int(item["tv_archive"]) == 1 for item in content if "tv_archive" in item)
        glob.current_playlist['data']['live_streams'] = []

        if hascatchup:
            glob.current_playlist['data']['catchup'] = True

        if glob.current_playlist['player_info']['showcatchup'] is True:
            if glob.current_playlist['data']['catchup'] is True:
                self.index += 1
                self.list.append([self.index, "Catch Up TV", 3, ""])

        self.drawList = []
        self.drawList = [buildListEntry(x[0], x[1], x[2], x[3]) for x in self.list]
        self["list"].setList(self.drawList)

        self.writeJsonFile()

        if len(self.list) == 0:
            self.session.openWithCallback(self.close, MessageBox, (_('No data, blocked or playlist not compatible with XStreamity plugin.')), MessageBox.TYPE_WARNING, timeout=5)
        elif len(self.list) == 1:
            self.__next__()
            self.close()

    def quit(self):
        self.writeJsonFile()
        self.close()

    def __next__(self):
        from . import categories
        from . import catchup

        category = self["list"].getCurrent()[2]

        if self["list"].getCurrent():
            if self["list"].getCurrent()[2] == 3:
                self.session.open(catchup.XStreamity_Catchup)
            else:
                self.session.open(categories.XStreamity_Categories, category)

    def updateCategories(self):
        self["splash"].show()
        glob.current_playlist['data']['live_categories'] = []
        glob.current_playlist['data']['vod_categories'] = []
        glob.current_playlist['data']['series_categories'] = []
        glob.current_playlist['data']['catchup'] = False
        glob.current_playlist['data']['catchup_checked'] = False
        glob.current_playlist['data']['last_check'] = datetime.now().strftime("%d/%m/%Y %H:%M")
        self['lastchecked'].setText(_("Lists updated: ") + str(glob.current_playlist['data']['last_check']))
        self.timer = eTimer()

        try:
            self.timer_conn = self.timer.timeout.connect(self.makeUrlList)
        except:
            try:
                self.timer.callback.append(self.makeUrlList)
            except:
                self.makeUrlList()
        self.timer.start(5, True)


def buildListEntry(index, title, category_id, playlisturl):
    png = None

    if category_id == 0:
        png = LoadPixmap(common_path + "live.png")
    if category_id == 1:
        png = LoadPixmap(common_path + "vod.png")
    if category_id == 2:
        png = LoadPixmap(common_path + "series.png")
    if category_id == 3:
        png = LoadPixmap(common_path + "catchup.png")
    return (index, str(title), category_id, str(playlisturl), png)