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

import re
import time

from pyload.plugin.Account import Account


class HellshareCz(Account):
    __name    = "HellshareCz"
    __type    = "account"
    __version = "0.16"

    __description = """Hellshare.cz account plugin"""
    __license     = "GPLv3"
    __authors     = [("zoidberg", "zoidberg@mujmail.cz")]


    CREDIT_LEFT_PATTERN = r'<div class="credit-link">\s*<table>\s*<tr>\s*<th>(\d+|\d\d\.\d\d\.)</th>'


    def loadAccountInfo(self, user, req):
        self.relogin(user)
        html = req.load("http://www.hellshare.com/")

        m = re.search(self.CREDIT_LEFT_PATTERN, html)
        if m is None:
            trafficleft = None
            validuntil = None
            premium = False
        else:
            credit = m.group(1)
            premium = True
            try:
                if "." in credit:
                    #Time-based account
                    vt = [int(x) for x in credit.split('.')[:2]]
                    lt = time.localtime()
                    year = lt.tm_year + int(vt[1] < lt.tm_mon or (vt[1] == lt.tm_mon and vt[0] < lt.tm_mday))
                    validuntil = time.mktime(time.strptime("%s%d 23:59:59" % (credit, year), "%d.%m.%Y %H:%M:%S"))
                    trafficleft = -1
                else:
                    #Traffic-based account
                    trafficleft = self.parseTraffic(credit + "MB")
                    validuntil = -1
            except Exception, e:
                self.logError(_("Unable to parse credit info"), e)
                validuntil = -1
                trafficleft = -1

        return {"validuntil": validuntil, "trafficleft": trafficleft, "premium": premium}


    def login(self, user, data, req):
        html = req.load('http://www.hellshare.com/', decode=True)
        if req.lastEffectiveURL != 'http://www.hellshare.com/':
            #Switch to English
            self.logDebug("Switch lang - URL: %s" % req.lastEffectiveURL)

            json = req.load("%s?do=locRouter-show" % req.lastEffectiveURL)
            hash = re.search(r"(\-\-[0-9a-f]+\-)", json).group(1)

            self.logDebug("Switch lang - HASH: %s" % hash)

            html = req.load('http://www.hellshare.com/%s/' % hash, decode=True)

        if re.search(self.CREDIT_LEFT_PATTERN, html):
            self.logDebug("Already logged in")
            return

        html = req.load('http://www.hellshare.com/login?do=loginForm-submit',
                        post={"login": "Log in",
                              "password": data['password'],
                              "username": user,
                              "perm_login": "on"},
                        decode=True)

        if "<p>You input a wrong user name or wrong password</p>" in html:
            self.wrongPassword()