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

import re

from module.plugins.internal.SimpleHoster import SimpleHoster


class QuickshareCz(SimpleHoster):
    __name__    = "QuickshareCz"
    __type__    = "hoster"
    __version__ = "0.61"
    __status__  = "testing"

    __pattern__ = r'http://(?:[^/]*\.)?quickshare\.cz/stahnout-soubor/.+'
    __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__ = """Quickshare.cz hoster plugin"""
    __license__     = "GPLv3"
    __authors__     = [("zoidberg", "zoidberg@mujmail.cz")]


    NAME_PATTERN = r'<th width="145px">Název:</th>\s*<td style="word-wrap:break-word;">(?P<N>.+?)</td>'
    SIZE_PATTERN = r'<th>Velikost:</th>\s*<td>(?P<S>[\d.,]+) (?P<U>[\w^_]+)</td>'
    OFFLINE_PATTERN = r'<script type="text/javascript">location\.href=\'/chyba\';</script>'


    def process(self, pyfile):
        self.data = self.load(pyfile.url)
        self.get_fileInfo()

        #: Parse js variables
        self.jsvars = dict((x, y.strip("'")) for x, y in re.findall(r"var (\w+) = ([\d.]+|'.+?')", self.data))
        self.log_debug(self.jsvars)
        pyfile.name = self.jsvars['ID3']

        #: Determine download type - free or premium
        if self.premium:
            if 'UU_prihlasen' in self.jsvars:
                if self.jsvars['UU_prihlasen'] == "0":
                    self.log_warning(_("User not logged in"))
                    self.relogin()
                    self.retry()
                elif float(self.jsvars['UU_kredit']) < float(self.jsvars['kredit_odecet']):
                    self.log_warning(_("Not enough credit left"))
                    self.premium = False

        if self.premium:
            self.handle_premium(pyfile)
        else:
            self.handle_free(pyfile)

        if self.scan_download({'error': re.compile(r'\AChyba!')}, read_size=100):
            self.fail(_("File not m or plugin defect"))


    def handle_free(self, pyfile):
        #: Get download url
        download_url = '%s/download.php' % self.jsvars['server']
        data = dict((x, self.jsvars[x]) for x in self.jsvars if x in ("ID1", "ID2", "ID3", "ID4"))
        self.log_debug("FREE URL1:" + download_url, data)

        header = self.load(download_url, post=data, just_header=True)

        self.link = header.get('location')
        if not self.link:
            elf.fail(_("File not found"))

        self.link = m.group(1)
        self.log_debug("FREE URL2:" + self.link)

        #: Check errors
        m = re.search(r'/chyba/(\d+)', self.link)
        if m is not None:
            if m.group(1) == "1":
                self.retry(60, 2 * 60, "This IP is already downloading")
            elif m.group(1) == "2":
                self.retry(60, 60, "No free slots available")
            else:
                self.fail(_("Error %d") % m.group(1))


    def handle_premium(self, pyfile):
        download_url = '%s/download_premium.php' % self.jsvars['server']
        data = dict((x, self.jsvars[x]) for x in self.jsvars if x in ("ID1", "ID2", "ID4", "ID5"))
        self.download(download_url, get=data)