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

import re
import time

from pyload.plugin.Account import Account


class Keep2ShareCc(Account):
    __name    = "Keep2ShareCc"
    __type    = "account"
    __version = "0.05"

    __description = """Keep2Share.cc account plugin"""
    __license     = "GPLv3"
    __authors     = [("aeronaut", "aeronaut@pianoguy.de"),
                       ("Walter Purcaro", "vuolter@gmail.com")]


    VALID_UNTIL_PATTERN  = r'Premium expires:\s*<b>(.+?)<'
    TRAFFIC_LEFT_PATTERN = r'Available traffic \(today\):\s*<b><a href="/user/statistic.html">(.+?)<'

    LOGIN_FAIL_PATTERN = r'Please fix the following input errors'


    def loadAccountInfo(self, user, req):
        validuntil  = None
        trafficleft = -1
        premium     = False

        html = req.load("http://keep2share.cc/site/profile.html", decode=True)

        m = re.search(self.VALID_UNTIL_PATTERN, html)
        if m:
            expiredate = m.group(1).strip()
            self.logDebug("Expire date: " + expiredate)

            if expiredate == "LifeTime":
                premium    = True
                validuntil = -1
            else:
                try:
                    validuntil = time.mktime(time.strptime(expiredate, "%Y.%m.%d"))

                except Exception, e:
                    self.logError(e)

                else:
                    premium = validuntil > time.mktime(time.gmtime())

            m = re.search(self.TRAFFIC_LEFT_PATTERN, html)
            if m:
                try:
                    trafficleft = self.parseTraffic(m.group(1))

                except Exception, e:
                    self.logError(e)

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


    def login(self, user, data, req):
        req.cj.setCookie("keep2share.cc", "lang", "en")

        html = req.load("http://keep2share.cc/login.html",
                        post={'LoginForm[username]'  : user,
                              'LoginForm[password]'  : data['password'],
                              'LoginForm[rememberMe]': 1,
                              'yt0'                  : ""},
                        decode=True)

        if re.search(self.LOGIN_FAIL_PATTERN, html):
            self.wrongPassword()