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

import re

from ..internal.misc import parse_size
from ..internal.SimpleHoster import SimpleHoster


class CzshareCom(SimpleHoster):
    __name__ = "CzshareCom"
    __type__ = "hoster"
    __version__ = "1.11"
    __status__ = "testing"

    __pattern__ = r'https?://(?:www\.)?(czshare|sdilej)\.(com|cz)/(\d+/|download\.php\?).+'
    __config__ = [("activated", "bool", "Activated", True),
                  ("use_premium", "bool", "Use premium account if available", True),
                  ("fallback", "bool",
                   "Fallback to free download if premium fails", True),
                  ("chk_filesize", "bool", "Check file size", True),
                  ("max_wait", "int", "Reconnect if waiting time is greater than minutes", 10)]

    __description__ = """CZshare.com hoster plugin, now Sdilej.cz"""
    __license__ = "GPLv3"
    __authors__ = [("zoidberg", "zoidberg@mujmail.cz"),
                   ("ondrej", "git@ondrej.it"), ]

    NAME_PATTERN = r'<div class="tab" id="parameters">\s*<p>\s*Cel. n.zev: <a href=.*?>(?P<N>.+?)</a>'
    SIZE_PATTERN = r'<div class="tab" id="category">(?:\s*<p>[^\n]*</p>)*\s*Velikost:\s*(?P<S>[\d .,]+)(?P<U>[\w^_]+)\s*</div>'
    OFFLINE_PATTERN = r'<div class="header clearfix">\s*<h2 class="red">'

    SIZE_REPLACEMENTS = [(' ', '')]
    URL_REPLACEMENTS = [
        (r'http://[^/]*/download.php\?.*?id=(\w+).*',
         r'http://sdilej.cz/\1/x/')]

    CHECK_TRAFFIC = True

    FREE_URL_PATTERN = r'<a href="(.+?)" class="page-download">[^>]*alt="(.+?)" /></a>'
    FREE_FORM_PATTERN = r'<form action="download\.php" method="post">\s*<img src="captcha\.php" id="captcha" />(.*?)</form>'
    PREMIUM_FORM_PATTERN = r'<form action="/profi_down\.php" method="post">(.*?)</form>'
    FORM_INPUT_PATTERN = r'<input[^>]* name="(.+?)" value="(.+?)"[^>]*/>'
    MULTIDL_PATTERN = r'<p><font color=\'red\'>Z.*?PROFI.</font></p>'
    USER_CREDIT_PATTERN = r'<div class="credit">\s*kredit: <strong>([\d .,]+)(\w+)</strong>\s*</div><!-- .credit -->'

    def out_of_traffic(self):
        #: Check if user logged in
        m = re.search(self.USER_CREDIT_PATTERN, self.data)
        if m is None:
            self.account.relogin()
            self.data = self.load(self.pyfile.url)
            m = re.search(self.USER_CREDIT_PATTERN, self.data)
            if m is None:
                return True

        #: Check user credit
        try:
            credit = parse_size(m.group(1).replace(' ', ''), m.group(2))
            self.log_info(
                _("Premium download for %i KiB of Credit") %
                (self.pyfile.size / 1024))
            self.log_info(
                _("User %s has %i KiB left") %
                (self.account.user, credit / 1024))
            if credit < self.pyfile.size:
                self.log_info(
                    _("Not enough credit to download file: %s") %
                    self.pyfile.name)
                return True

        except Exception, e:
            #: let's continue and see what happens...
            self.log_error(e, trace=True)

        return False

    def handle_premium(self, pyfile):
        try:
            form = re.search(
                self.PREMIUM_FORM_PATTERN,
                self.data,
                re.S).group(1)
            inputs = dict(re.findall(self.FORM_INPUT_PATTERN, form))

        except Exception, e:
            self.log_error(e, trace=True)
            self.restart(premium=False)

        #: Download the file, destination is determined by pyLoad
        self.download(
            "http://sdilej.cz/profi_down.php",
            post=inputs,
            disposition=True)

    def handle_free(self, pyfile):
        #: Get free url
        m = re.search(self.FREE_URL_PATTERN, self.data)
        if m is None:
            self.error(_("FREE_URL_PATTERN not found"))

        parsed_url = "http://sdilej.cz" + m.group(1)

        self.log_debug("PARSED_URL:" + parsed_url)

        #: Get download ticket and parse html
        self.data = self.load(parsed_url)
        if re.search(self.MULTIDL_PATTERN, self.data):
            self.retry(5 * 60, 12, _("Download limit reached"))

        try:
            form = re.search(self.FREE_FORM_PATTERN, self.data, re.S).group(1)
            inputs = dict(re.findall(self.FORM_INPUT_PATTERN, form))
            pyfile.size = int(inputs['size'])

        except Exception, e:
            self.log_error(e, trace=True)
            self.error(_("Form"))

        #: Get and decrypt captcha
        captcha_url = 'http://sdilej.cz/captcha.php'
        inputs['captchastring2'] = self.captcha.decrypt(captcha_url)
        self.data = self.load(parsed_url, post=inputs)

        if u"<li>Zadaný ověřovací kód nesouhlasí!</li>" in self.data:
            self.retry_captcha()

        elif re.search(self.MULTIDL_PATTERN, self.data):
            self.retry(5 * 60, 12, _("Download limit reached"))

        else:
            self.captcha.correct()

        m = re.search("countdown_number = (\d+);", self.data)
        self.set_wait(int(m.group(1)) if m else 50)

        #: Download the file, destination is determined by pyLoad
        self.log_debug("WAIT URL", self.req.lastEffectiveURL)

        m = re.search(
            "free_wait.php\?server=(.*?)&(.*)",
            self.req.lastEffectiveURL)
        if m is None:
            self.error(_("Download URL not found"))

        self.link = "http://%s/download.php?%s" % (m.group(1), m.group(2))

        self.wait()

    def check_download(self):
        #: Check download
        check = self.scan_download({
            "temp offline": re.compile(r'^Soubor je do.*asn.* nedostupn.*$'),
            'credit': re.compile(r'^Nem.*te dostate.*n.* kredit.$'),
            "multi-dl": re.compile(self.MULTIDL_PATTERN),
            'captcha': "<li>Zadaný ověřovací kód nesouhlasí!</li>"
        })

        if check == "temp offline":
            self.fail(_("File not available - try later"))

        elif check == "credit":
            self.restart(premium=False)

        elif check == "multi-dl":
            self.retry(5 * 60, 12, _("Download limit reached"))

        elif check == "captcha":
            self.retry_captcha()

        return SimpleHoster.check_download(self)
