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

from time import strptime, mktime
import hashlib

from module.plugins.Account import Account


class HotfileCom(Account):
    __name__ = "HotfileCom"
    __type__ = "account"
    __version__ = "0.2"

    __description__ = """Hotfile.com account plugin"""
    __author_name__ = ("mkaay", "JoKoT3")
    __author_mail__ = ("mkaay@mkaay.de", "jokot3@gmail.com")


    def loadAccountInfo(self, user, req):
        resp = self.apiCall("getuserinfo", user=user)
        if resp.startswith("."):
            self.core.debug("HotfileCom API Error: %s" % resp)
            raise Exception
        info = {}
        for p in resp.split("&"):
            key, value = p.split("=")
            info[key] = value

        if info['is_premium'] == '1':
            info['premium_until'] = info['premium_until'].replace("T", " ")
            zone = info['premium_until'][19:]
            info['premium_until'] = info['premium_until'][:19]
            zone = int(zone[:3])

            validuntil = int(mktime(strptime(info['premium_until'], "%Y-%m-%d %H:%M:%S"))) + (zone * 60 * 60)
            tmp = {"validuntil": validuntil, "trafficleft": -1, "premium": True}

        elif info['is_premium'] == '0':
            tmp = {"premium": False}

        return tmp

    def apiCall(self, method, post={}, user=None):
        if user:
            data = self.getAccountData(user)
        else:
            user, data = self.selectAccount()

        req = self.getAccountRequest(user)

        digest = req.load("http://api.hotfile.com/", post={"action": "getdigest"})
        h = hashlib.md5()
        h.update(data['password'])
        hp = h.hexdigest()
        h = hashlib.md5()
        h.update(hp)
        h.update(digest)
        pwhash = h.hexdigest()

        post.update({"action": method})
        post.update({"username": user, "passwordmd5dig": pwhash, "digest": digest})
        resp = req.load("http://api.hotfile.com/", post=post)
        req.close()
        return resp

    def login(self, user, data, req):
        cj = self.getAccountCookies(user)
        cj.setCookie("hotfile.com", "lang", "en")
        req.load("http://hotfile.com/", cookies=True)
        page = req.load("http://hotfile.com/login.php", post={"returnto": "/", "user": user, "pass": data['password']},
                        cookies=True)

        if "Bad username/password" in page:
            self.wrongPassword()