# -*- 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
from module.common.json_layer import json_loads
from time import time

class FilepostCom(SimpleHoster):
    __name__ = "FilepostCom"
    __type__ = "hoster"
    __pattern__ = r"https?://(?:www\.)?(?:filepost\.com/files|fp.io)/([^/]+).*"
    __version__ = "0.26"
    __description__ = """Filepost.com plugin - free only"""
    __author_name__ = ("zoidberg")
    __author_mail__ = ("zoidberg@mujmail.cz")

    FILE_INFO_PATTERN = r'<input type="text" id="url" value=\'<a href[^>]*>(?P<N>[^>]+?) - (?P<S>[0-9\.]+ [kKMG]i?B)</a>\' class="inp_text"/>'
    #FILE_INFO_PATTERN = r'<h1>(?P<N>[^<]+)</h1>\s*<div class="ul">\s*<ul>\s*<li><span>Size:</span> (?P<S>[0-9.]+) (?P<U>[kKMG])i?B</li>'
    FILE_OFFLINE_PATTERN = r'class="error_msg_title"> Invalid or Deleted File. </div>|<div class="file_info file_info_deleted">'
    RECAPTCHA_KEY_PATTERN = r"Captcha.init\({\s*key:\s*'([^']+)'"
    FLP_TOKEN_PATTERN = r"set_store_options\({token: '([^']+)'"

    def handleFree(self):
        # Find token and captcha key
        file_id = re.search(self.__pattern__, self.pyfile.url).group(1)

        found = re.search(self.FLP_TOKEN_PATTERN, self.html)
        if not found: self.parseError("Token")
        flp_token = found.group(1)

        found = re.search(self.RECAPTCHA_KEY_PATTERN, self.html)
        if not found: self.parseError("Captcha key")
        captcha_key = found.group(1)

        # Get wait time
        get_dict = {'SID' : self.req.cj.getCookie('SID'), 'JsHttpRequest' : str(int(time()*10000)) + '-xml'}
        post_dict = {'action' : 'set_download', 'token' : flp_token, 'code' : file_id}                    
        wait_time = int(self.getJsonResponse(get_dict, post_dict, 'wait_time'))     

        if wait_time > 0:
            self.setWait(wait_time)
            self.wait()                               
        
        post_dict = {"token" : flp_token, "code" : file_id, "file_pass" : ''}
        
        if 'var is_pass_exists = true;' in self.html:
            # Solve password            
            for file_pass in self.getPassword().splitlines():
                get_dict['JsHttpRequest'] = str(int(time()*10000)) + '-xml'           
                post_dict['file_pass'] = file_pass
                self.logInfo("Password protected link, trying " + file_pass)                
                                    
                download_url = self.getJsonResponse(get_dict, post_dict, 'link')               
                if download_url: 
                    break
                      
            else: self.fail("No or incorrect password")         
        
        else:
            # Solve recaptcha
            recaptcha = ReCaptcha(self)
            
            for pokus in range(5):
                get_dict['JsHttpRequest'] = str(int(time()*10000)) + '-xml'
                if pokus:
                    post_dict["recaptcha_challenge_field"], post_dict["recaptcha_response_field"] = recaptcha.challenge(captcha_key)
                    self.logDebug(u"RECAPTCHA: %s : %s : %s" % (captcha_key, post_dict["recaptcha_challenge_field"], post_dict["recaptcha_response_field"]))                
                 
                download_url = self.getJsonResponse(get_dict, post_dict, 'link')
                if download_url:
                    if pokus: self.correctCaptcha()
                    break
                elif pokus:
                    self.invalidCaptcha()
                                
            else: self.fail("Invalid captcha")
        
        # Download
        self.download(download_url)       
    
    def getJsonResponse(self, get_dict, post_dict, field):     
        json_response = json_loads(self.load('https://filepost.com/files/get/', get = get_dict, post = post_dict))
        self.logDebug(json_response)
        
        if not 'js' in json_response: self.parseError('JSON %s 1' % field)       
        
        if 'error' in json_response['js']:
            if json_response['js']['error'] == 'download_delay':
                self.retry(js_answer['params']['next_download'])
            elif 'Wrong file password' in json_response['js']['error']:
                return None 
            elif 'You entered a wrong CAPTCHA code' in json_response['js']['error']:
                return None  
            else:
                self.fail(js_answer['error'])
        
        if not 'answer' in json_response['js'] or not field in json_response['js']['answer']: 
            self.parseError('JSON %s 2' % field)
            
        return json_response['js']['answer'][field]
        
getInfo = create_getInfo(FilepostCom)