# -*- coding: utf-8 -*-

import re

from module.common.json_layer import json_loads
from module.plugins.internal.MultiHoster import MultiHoster, create_getInfo
from module.plugins.internal.SimpleHoster import secondsToMidnight


class UnrestrictLi(MultiHoster):
    __name__    = "UnrestrictLi"
    __type__    = "hoster"
    __version__ = "0.22"

    __pattern__ = r'https?://(?:www\.)?(unrestrict|unr)\.li/dl/[\w^_]+'
    __config__  = [("use_premium", "bool", "Use premium account if available", True)]

    __description__ = """Unrestrict.li multi-hoster plugin"""
    __license__     = "GPLv3"
    __authors__     = [("stickell", "l.stickell@yahoo.it")]


    LOGIN_ACCOUNT = False


    def setup(self):
        self.chunkLimit     = 16
        self.resumeDownload = True


    def handleFree(self, pyfile):
        for _i in xrange(5):
            self.html = self.load('https://unrestrict.li/unrestrict.php',
                             post={'link': pyfile.url, 'domain': 'long'})

            self.logDebug("JSON data: " + self.html)

            if self.html:
                break
        else:
            self.logInfo(_("Unable to get API data, waiting 1 minute and retry"))
            self.retry(5, 60, "Unable to get API data")

        if 'Expired session' in self.html \
           or ("You are not allowed to download from this host" in self.html and self.premium):
            self.account.relogin(self.user)
            self.retry()

        elif "File offline" in self.html:
            self.offline()

        elif "You are not allowed to download from this host" in self.html:
            self.fail(_("You are not allowed to download from this host"))

        elif "You have reached your daily limit for this host" in self.html:
            self.logWarning(_("Reached daily limit for this host"))
            self.retry(5, secondsToMidnight(gmt=2), "Daily limit for this host reached")

        elif "ERROR_HOSTER_TEMPORARILY_UNAVAILABLE" in self.html:
            self.logInfo(_("Hoster temporarily unavailable, waiting 1 minute and retry"))
            self.retry(5, 60, "Hoster is temporarily unavailable")

        self.html     = json_loads(self.html)
        self.link     = self.html.keys()[0]
        self.api_data = self.html[self.link]

        if hasattr(self, 'api_data'):
            self.setNameSize()


    def checkFile(self, rules={}):
        super(UnrestrictLi, self).checkFile(rules)

        if self.getConfig('history'):
            self.load("https://unrestrict.li/history/", get={'delete': "all"})
            self.logInfo(_("Download history deleted"))


    def setNameSize(self):
        if 'name' in self.api_data:
            self.pyfile.name = self.api_data['name']
        if 'size' in self.api_data:
            self.pyfile.size = self.api_data['size']


getInfo = create_getInfo(UnrestrictLi)