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

import re

from pycurl import HTTPHEADER

from module.network.RequestFactory import getRequest
from module.plugins.internal.SimpleHoster import SimpleHoster, parseFileInfo


def getInfo(urls):
    h = getRequest()
    h.c.setopt(HTTPHEADER,
               ["Accept: text/html",
                "User-Agent: Mozilla/5.0 (Windows NT 6.1; Win64; x64; rv:25.0) Gecko/20100101 Firefox/25.0"])

    for url in urls:
        html = h.load(url, decode=True)
        yield parseFileInfo(MegaRapidCz, url, html)


class MegaRapidCz(SimpleHoster):
    __name__    = "MegaRapidCz"
    __type__    = "hoster"
    __version__ = "0.54"

    __pattern__ = r'http://(?:www\.)?(share|mega)rapid\.cz/soubor/\d+/.+'

    __description__ = """MegaRapid.cz hoster plugin"""
    __license__     = "GPLv3"
    __authors__     = [("MikyWoW", "mikywow@seznam.cz"),
                       ("zoidberg", "zoidberg@mujmail.cz"),
                       ("stickell", "l.stickell@yahoo.it"),
                       ("Walter Purcaro", "vuolter@gmail.com")]


    NAME_PATTERN = r'<h1[^>]*><span[^>]*>(?:<a[^>]*>)?(?P<N>[^<]+)'
    SIZE_PATTERN = r'<td class="i">Velikost:</td>\s*<td class="h"><strong>\s*(?P<S>[\d.,]+) (?P<U>[\w^_]+)</strong></td>'
    OFFLINE_PATTERN = ur'Nastala chyba 404|Soubor byl smazán'

    FORCE_CHECK_TRAFFIC = True

    LINK_PATTERN = r'<a href="([^"]+)" title="Stahnout">([^<]+)</a>'
    ERR_LOGIN_PATTERN = ur'<div class="error_div"><strong>Stahování je přístupné pouze přihlášeným uživatelům'
    ERR_CREDIT_PATTERN = ur'<div class="error_div"><strong>Stahování zdarma je možné jen přes náš'


    def setup(self):
        self.chunkLimit = 1


    def handlePremium(self):
        try:
            self.html = self.load(self.pyfile.url, decode=True)
        except BadHeader, e:
            self.account.relogin(self.user)
            self.retry(wait_time=60, reason=str(e))

        m = re.search(self.LINK_PATTERN, self.html)
        if m:
            link = m.group(1)
            self.logDebug("Premium link: %s" % link)
            self.download(link, disposition=True)
        else:
            if re.search(self.ERR_LOGIN_PATTERN, self.html):
                self.relogin(self.user)
                self.retry(wait_time=60, reason=_("User login failed"))
            elif re.search(self.ERR_CREDIT_PATTERN, self.html):
                self.fail(_("Not enough credit left"))
            else:
                self.fail(_("Download link not found"))
