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

import re
from pyload.plugin.Crypter import Crypter


class UlozTo(Crypter):
    __name    = "UlozTo"
    __type    = "crypter"
    __version = "0.20"

    __pattern = r'http://(?:www\.)?(uloz\.to|ulozto\.(cz|sk|net)|bagruj\.cz|zachowajto\.pl)/(m|soubory)/.+'
    __config  = [("use_subfolder"     , "bool", "Save package to subfolder"          , True),
                   ("subfolder_per_pack", "bool", "Create a subfolder for each package", True)]

    __description = """Uloz.to folder decrypter plugin"""
    __license     = "GPLv3"
    __authors     = [("zoidberg", "zoidberg@mujmail.cz")]


    FOLDER_PATTERN = r'<ul class="profile_files">(.*?)</ul>'
    LINK_PATTERN = r'<br /><a href="/(.+?)">.+?</a>'
    NEXT_PAGE_PATTERN = r'<a class="next " href="/(.+?)">&nbsp;</a>'


    def decrypt(self, pyfile):
        html = self.load(pyfile.url)

        new_links = []
        for i in xrange(1, 100):
            self.logInfo(_("Fetching links from page %i") % i)
            m = re.search(self.FOLDER_PATTERN, html, re.S)
            if m is None:
                self.error(_("FOLDER_PATTERN not found"))

            new_links.extend(re.findall(self.LINK_PATTERN, m.group(1)))
            m = re.search(self.NEXT_PAGE_PATTERN, html)
            if m:
                html = self.load("http://ulozto.net/" + m.group(1))
            else:
                break
        else:
            self.logInfo(_("Limit of 99 pages reached, aborting"))

        if new_links:
            self.urls = [map(lambda s: "http://ulozto.net/%s" % s, new_links)]