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

import re

from urlparse import urljoin
from time import mktime, strptime

from module.plugins.Account import Account
from module.plugins.internal.SimpleHoster import parseHtmlForm, set_cookies
from module.utils import parseFileSize


class XFSPAccount(Account):
    __name__ = "XFSPAccount"
    __type__ = "account"
    __version__ = "0.12"

    __description__ = """XFileSharingPro account plugin"""
    __license__ = "GPLv3"
    __authors__ = [("zoidberg", "zoidberg@mujmail.cz"),
                   ("Walter Purcaro", "vuolter@gmail.com")]


    HOSTER_URL = None

    COOKIES = None  #: or list of tuples [(domain, name, value)]

    VALID_UNTIL_PATTERN = r'>Premium.[Aa]ccount expire:.*?<b>(.+?)</b>'

    TRAFFIC_LEFT_PATTERN = r'>Traffic available today:.*?<b>\s*(?P<S>[\d.,]+)\s*(?:(?P<U>[\w^_]+)\s*)?</b>'
    TRAFFIC_LEFT_UNIT = "MB"  #: used only if no group <U> was found

    LOGIN_FAIL_PATTERN = r'>(Incorrect Login or Password|Error<)'

    # PREMIUM_PATTERN = r'>Renew premium<'


    def loadAccountInfo(self, user, req):
        html = req.load(self.HOSTER_URL, get={'op': "my_account"}, decode=True)

        validuntil = None
        trafficleft = None
        premium = None

        if hasattr(self, "PREMIUM_PATTERN"):
            premium = True if re.search(self.PREMIUM_PATTERN, html) else False

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

            try:
                validuntil = mktime(strptime(expiredate, "%d %B %Y"))
            except Exception, e:
                self.logError(e)
            else:
                if validuntil > mktime(gmtime()):
                    premium = True
                    trafficleft = -1
                else:
                    if premium is False:  #: registered account type (not premium)
                        validuntil = -1
                    premium = False

        try:
            traffic = re.search(self.TRAFFIC_LEFT_PATTERN, html).groupdict()
            if "Unlimited" in traffic['S']:
                trafficleft = -1
                if premium is None:
                    premium = True
            else:
                if 'U' in traffic:
                    unit = traffic['U']
                elif isinstance(self.TRAFFIC_LEFT_UNIT, basestring):
                    unit = self.TRAFFIC_LEFT_UNIT
                else:
                    unit = None
                trafficleft = parseFileSize(traffic['S'], unit)
        except:
            pass

        if premium is None:
            premium = False

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


    def login(self, user, data, req):
        if isinstance(self.COOKIES, list):
            set_cookies(req.cj, self.COOKIES)

        url = urljoin(self.HOSTER_URL, "login.html")
        html = req.load(url, decode=True)

        action, inputs = parseHtmlForm('name="FL"', html)
        if not inputs:
            inputs = {'op': "login",
                      'redirect': self.HOSTER_URL}

        inputs.update({'login': user,
                       'password': data['password']})

        html = req.load(self.HOSTER_URL, post=inputs, decode=True)

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

@@ -361,9 +361,9 @@
 %220.1
-2
+3
 %22%0A%0A 
@@ -572,45 +572,351 @@
     
-HOSTER_URL = None%0A%0A    COOKIES = None
+%22%22%22%0A    Following patterns should be defined by each hoster:%0A%0A      HOSTER_URL: (optional)%0A        example: HOSTER_URL = r'linestorage.com'%0A%0A      PREMIUM_PATTERN: (optional) Checks if the account is premium%0A        example: PREMIUM_PATTERN = r'%3ERenew premium%3C'%0A    %22%22%22%0A%0A    HOSTER_NAME = None%0A%0A    COOKIES = %5B(HOSTER_NAME, %22lang%22, %22english%22)%5D
   #:
@@ -1281,50 +1281,134 @@
 )'%0A%0A
+%0A
     
-# PREMIUM_PATTERN = r'%3ERenew premium%3C'
+def init(self):%0A        if not hasattr(self, %22HOSTER_URL%22):%0A            self.HOSTER_URL = %22http://%25s/%22 %25 self.HOSTER_NAME
 %0A%0A%0A 

Set HOSTER_NAME (instead HOSTER_URL) as the mandatory attribute