# -*- 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: mkaay
"""

from module.plugins.Account import Account
import re
from time import strptime, mktime

class UploadedTo(Account):
    __name__ = "UploadedTo"
    __version__ = "0.1"
    __type__ = "account"
    __description__ = """ul.to account plugin"""
    __author_name__ = ("mkaay")
    __author_mail__ = ("mkaay@mkaay.de")
    
    def getAccountInfo(self, user):
        try:
            data = None
            for account in self.accounts.items():
                if account[0] == user:
                    data = account[1]
            if not data:
                raise Exception
            req = self.core.requestFactory.getRequest(self.__name__, user)
            html = req.load("http://ul.to/", cookies=True)
            raw_traffic = re.search(r"Traffic left: </span><span class=.*?>(.*?)</span>", html).group(1)
            raw_valid = re.search(r"Valid until: </span> <span class=.*?>(.*?)</span>", html).group(1)
            traffic = int(self.parseTraffic(raw_traffic))
            validuntil = int(mktime(strptime(raw_valid.strip(), "%d-%m-%Y %H:%M")))
            out = Account.getAccountInfo(self, user)
            tmp =  {"validuntil":validuntil, "trafficleft":traffic, "maxtraffic":100*1024*1024}
            out.update(tmp)
            return out
        except:
            return Account.getAccountInfo(self, user)
        
    def login(self, user, data):
        req = self.core.requestFactory.getRequest(self.__name__, user)
        req.load("http://ul.to/login", post={ "email" : user, "password" : data["password"]}, cookies=True)