#!/usr/bin/env python
# -*- coding: utf-8 -*-

import re
from time import time
from module.Plugin import Plugin

class HotfileCom(Plugin):

    def __init__(self, parent):
        Plugin.__init__(self, parent)
        props = {}
        props['name'] = "HotfileCom"
        props['type'] = "hoster"
        props['pattern'] = r"http://hotfile.com/dl/"
        props['version'] = "0.1"
        props['description'] = """Hotfile.com Download Plugin"""
        props['author_name'] = ("sitacuisses","spoob","mkaay")
        props['author_mail'] = ("sitacuisses@yhoo.de","spoob@pyload.org","mkaay@mkaay.de")
        self.props = props
        self.parent = parent
        self.html = [None, None]
        self.want_reconnect = False
        self.multi_dl = False
        self.htmlwithlink = None
        self.url = None
        self.read_config()
        if self.config['premium']:
            self.multi_dl = True
            self.req.canContinue = True
        else:
            self.multi_dl = False
        
    def prepare(self, thread):
        pyfile = self.parent
        self.want_reconnect = False
        
        self.download_html()
        pyfile.status.exists = self.file_exists()

        if not pyfile.status.exists:
            return False

        pyfile.status.filename = self.get_file_name()

        if self.config['premium']:
            pyfile.status.url = self.get_file_url()
            return True
            
        self.get_wait_time()
        pyfile.status.waituntil = self.time_plus_wait
        pyfile.status.want_reconnect = self.want_reconnect

        thread.wait(self.parent)
        
        pyfile.status.url = self.get_file_url()
        return True

    def download_html(self):
        if self.config['premium']:
            self.req.add_auth(self.config['username'], self.config['password'])
        self.url = self.parent.url + "?lang=en"
        print self.url
        self.html[0] = self.req.load(self.url, cookies=True)

    def get_file_url(self):
        if self.config['premium']:
            file_url_pattern = r'<td><a href="(http://hotfile.com/get/.+?)" class="click_download">'
            file_url = re.search(file_url_pattern, self.html[0]).group(1)
        else:
            form_content = re.search(r"<form style=.*(\n<.*>\s*)*?\n<tr>", self.html[0]).group(0)
            form_posts = re.findall(r"<input\stype=hidden\sname=(\S*)\svalue=(\S*)>", form_content)
            self.html[1] = self.req.load(self.url, post=form_posts, cookies=True)
            file_url = re.search("a href=\"(http://hotfile\.com/get/\S*?)\"", self.html[1]).group(1)
        return file_url

    def get_file_name(self):
        file_name = re.search(r':</strong> (.+?) <span>\|</span>', self.html[0]).group(1)
        return file_name

    def file_exists(self):
        if re.search(r"404 - Not Found", self.html[0]) != None or self.html[0] == "":
            return False
        return True
    
    def get_wait_time(self):
        free_limit_pattern = re.compile(r"timerend=d\.getTime\(\)\+(\d+);")
        matches = free_limit_pattern.findall(self.html[0])
        if matches:
            for match in matches:
                if int(match) == 60000:
                    continue
                if int(match) == 0:
                    continue
                else:
                    self.time_plus_wait = time() + int(match)/1000 + 65
                    self.want_reconnect = True
                    return True
            self.time_plus_wait = time() + 65