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

import re
from module.plugins.internal.SimpleHoster import SimpleHoster, create_getInfo
from module.plugins.ReCaptcha import ReCaptcha

def replace_eval(js_expr):
    return js_expr.replace(r'eval("', '').replace(r"\'", r"'").replace(r'\"', r'"')

class MediafireCom(SimpleHoster):
    __name__ = "MediafireCom"
    __type__ = "hoster"
    __pattern__ = r"http://(?:\w*\.)*mediafire\.com/download.php\?.*"
    __version__ = "0.66"
    __description__ = """Mediafire.com plugin - free only"""
    __author_name__ = ("zoidberg")
    __author_mail__ = ("zoidberg@mujmail.cz")

    PAGE1_FUNCTION_PATTERN = r"function %s\(qk,pk1\)\{if[^']*'loadingicon'\);[^;]*; (.*?)eval"
    PAGE1_KEY_PATTERN = ";break;}\s*(\w+='';\w+=unescape.*?)eval\("
    PAGE1_RESULT_PATTERN = r"(\w+)\('(?P<qk>[^']+)','(?P<pk1>[^']+)'\)"
    PAGE1_DIV_PATTERN = r'getElementById\("(\w{32})"\)'
    PAGE1_PKR_PATTERN = r"pKr='([^']+)';"
    RECAPTCHA_PATTERN = r'src="http://(?:api.recaptcha.net|www.google.com/recaptcha/api)/challenge\?k=([^"]+)">'
    PAGE1_ACTION_PATTERN = r'<link rel="canonical" href="([^"]+)"/>'

    PAGE2_VARS_PATTERN = r'<script language="Javascript"><!--\s*(var.*?unescape.*?)eval\('
    PAGE2_DZ_PATTERN = r'break;case 15:(.*)</script>'
    PAGE2_LINK_PATTERN = r"(\w+='';\w+=unescape.*?)eval\(\w+\);(.{0,10}href=[^>]*>)?"
    FINAL_LINK_PATTERN = r'parent.document.getElementById\(\'(\w{32})\'\).*(http://[^"]+)" \+(\w+)\+ "([^"]+)">'

    FILE_NAME_PATTERN = r'<META NAME="description" CONTENT="(?P<N>[^"]+)"/>'
    FILE_SIZE_PATTERN = r'<input type="hidden" id="sharedtabsfileinfo1-fs" value="(?P<S>[0-9.]+) (?P<U>[kKMG])i?B">'
    FILE_OFFLINE_PATTERN = r'class="error_msg_title"> Invalid or Deleted File. </div>'

    def process(self, pyfile):
        self.html = self.load(pyfile.url, decode = True)
        self.checkCaptcha()
        self.getFileInfo()    
        if self.account:
            self.handlePremium()
        else:
            self.handleFree()
    
    def handleFree(self):
        found = re.search(self.PAGE1_KEY_PATTERN, self.html)
        if found:
            result = self.js.eval(found.group(1))
            found = re.search(self.PAGE1_RESULT_PATTERN, result)
        else:
            self.retry(3, 0, "Parse error (KEY)")

        try:
            param_dict = found.groupdict()
            param_dict['r'] = re.search(self.PAGE1_PKR_PATTERN, self.html).group(1)
            self.logDebug(param_dict)
            key_func = found.group(1)
            self.logDebug("KEY_FUNC: %s" % key_func)

            found = re.search(self.PAGE1_FUNCTION_PATTERN % key_func, self.html)
            result = self.js.eval(found.group(1))
            key_div = re.search(self.PAGE1_DIV_PATTERN, result).group(1)
            self.logDebug("KEY_DIV: %s" % key_div)
        except Exception, e:
            self.logError(e)
            self.retry(3, 0, "Parse error (KEY DIV)")

        self.html = self.load("http://www.mediafire.com/dynamic/download.php", get=param_dict)
        js_expr = replace_eval(re.search(self.PAGE2_VARS_PATTERN, self.html).group(1))
        result = self.js.eval(js_expr)
        var_list = dict(re.findall("([^=]+)='([^']+)';", result))

        page2_dz = replace_eval(re.search(self.PAGE2_DZ_PATTERN, self.html, re.DOTALL).group(1))

        final_link = None
        for link_enc in re.finditer(self.PAGE2_LINK_PATTERN, page2_dz):
            #self.logDebug("LINK_ENC: %s..." % link_enc.group(1)[:20])
            try:
                link_dec = self.js.eval(link_enc.group(1))
            except Exception, e:
                self.logError("Unable to decrypt link %s" % link_enc.group(1)[:20])
                self.logError(e)
                self.logDebug(link_enc.group(1))
                continue

            #self.logDebug("LINK_DEC: %s" % link_dec)
            if link_enc.group(2): link_dec = link_dec + replace_eval(link_enc.group(2))

            found = re.search(self.FINAL_LINK_PATTERN, link_dec)
            if found:
                if found.group(1) == key_div:
                    final_link = found.group(2) + var_list[found.group(3)] + found.group(4)
                    break
            else:
                self.logDebug("Link not found in %s..." % link_dec)
        else:
            self.fail("Final link not found")

        self.logDebug("FINAL LINK: %s" % final_link)
        self.download(final_link)
        
    def checkCaptcha(self):
        for i in range(5):
            found = re.search(self.RECAPTCHA_PATTERN, self.html)
            if found:
                captcha_action = re.search(self.PAGE1_ACTION_PATTERN, self.html).group(1)
                captcha_key = found.group(1)
                recaptcha = ReCaptcha(self)
                captcha_challenge, captcha_response = recaptcha.challenge(captcha_key)
                self.html = self.load(captcha_action, post = {
                    "recaptcha_challenge_field": captcha_challenge,
                    "recaptcha_response_field": captcha_response
                    })
            else:
                break
        else:
            self.fail("No valid recaptcha solution received")

getInfo = create_getInfo(MediafireCom)