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

import re
from time import time

from module.plugins.Hoster import Hoster

class StorageTo(Hoster):
    __name__ = "StorageTo"
    __type__ = "hoster"
    __pattern__ = r"http://(?:www)?\.storage\.to/get/.*"
    __version__ = "0.1"
    __description__ = """Storage.to Download Hoster"""
    __author_name__ = ("mkaay")

    def __init__(self, parent):
        Hoster.__init__(self, parent)
        self.parent = parent
        self.time_plus_wait = None
        self.want_reconnect = False
        self.api_data = None
        self.html = None
        self.read_config()
        self.multi_dl = False

        self.start_dl = False

    def prepare(self, thread):
        pyfile = self.parent
        
        self.req.clear_cookies()

        self.want_reconnect = False

        pyfile.status.exists = self.file_exists()

        if not pyfile.status.exists:
            return False

        pyfile.status.filename = self.get_file_name()
        
        self.get_wait_time()
        pyfile.status.waituntil = self.time_plus_wait
        pyfile.status.want_reconnect = self.want_reconnect

        while self.want_reconnect:
            thread.wait(self.parent)
            self.download_api_data()
            self.get_wait_time()
            pyfile.status.waituntil = self.time_plus_wait
            pyfile.status.want_reconnect = self.want_reconnect
        
        pyfile.status.url = self.get_file_url()

        return True
    
    def download_html(self):
        url = self.parent.url
        self.html = self.load(url, cookies=True)

    def download_api_data(self):
        url = self.parent.url
        info_url = url.replace("/get/", "/getlink/")
        src = self.load(info_url, cookies=True)
        pattern = re.compile(r"'(\w+)' : (.*?)[,|\}]")
        self.api_data = {}
        for pair in pattern.findall(src):
            self.api_data[pair[0]] = pair[1].strip("'")
        print self.api_data

    def get_wait_time(self):
        if not self.api_data:
            self.download_api_data()
        if self.api_data["state"] == "wait":
            self.want_reconnect = True
        else:
            self.want_reconnect = False
        
        self.time_plus_wait = time() + int(self.api_data["countdown"]) + 3
            
            

    def file_exists(self):
        """ returns True or False
        """
        if not self.api_data:
            self.download_api_data()
        if self.api_data["state"] == "failed":
            return False
        else:
            return True

    def get_file_url(self):
        """ returns the absolute downloadable filepath
        """
        if not self.api_data:
            self.download_api_data()
        return self.api_data["link"]

    def get_file_name(self):
        if not self.html:
            self.download_html()
        file_name_pattern = r"<span class=\"orange\">Downloading:</span>(.*?)<span class=\"light\">(.*?)</span>"
        return re.search(file_name_pattern, self.html).group(1).strip()