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

"""
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License,
    or (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
    See the GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, see <http://www.gnu.org/licenses/>.

    @author: RaNaN
"""

import re
from time import mktime, strptime

from module.plugins.Account import Account

try:
    from json import loads as json_loads
except ImportError: # pragma: no cover
    from module.lib.simplejson import loads as json_loads


class FilesonicCom(Account):
    __name__ = "FilesonicCom"
    __version__ = "0.31"
    __type__ = "account"
    __description__ = """filesonic.com account plugin"""
    __author_name__ = ("RaNaN","Paul King")
    __author_mail__ = ("RaNaN@pyload.org","")
	
    def getDomain(self, req):
        xml = req.load("http://api.filesonic.com/utility?method=getFilesonicDomainForCurrentIp&format=json").decode("utf8")
        return json_loads(xml)["FSApi_Utility"]["getFilesonicDomainForCurrentIp"]["response"]
        
    def loadAccountInfo(self, user, req):
        xml = req.load("http://api.filesonic.com/user?method=getInfo&format=json", 
                       post = {"u": user, 
                               "p" : self.accounts[user]["password"]}
                      ).decode("utf8")
        self.core.log.debug(_("%s: account status retrieved from api %s" % (self.__name__,xml)))
        json = json_loads(xml)
        if json["FSApi_User"]["getInfo"]["status"] != "success":
           self.core.log.error(_("%s: Invalid login retrieving user details" % self.__name__))
           return {"validuntil": -1, "trafficleft": -1, "premium" : False}
        premium = json["FSApi_User"]["getInfo"]["response"]["users"]["user"]["is_premium"]
        if premium:
            validuntil = json["FSApi_User"]["getInfo"]["response"]["users"]["user"]["premium_expiration"]
            validuntil = int(mktime(strptime(validuntil, "%Y-%m-%d %H:%M:%S")))
        else:
            validuntil = -1
        return {"validuntil": validuntil, "trafficleft": -1, "premium" : premium}

    def login(self, user, data, req):
        domain = self.getDomain(req)

        post_vars = {
            "email": user,
            "password": data["password"],
            "rememberMe" : 1
        }
        page = req.load("http://www%s/user/login" % domain, cookies=True, post=post_vars).decode("utf8")

        if "Provided password does not match." in page or "You must be logged in to view this page." in page:
            self.wrongPassword()