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

import re

from urlparse import urlparse, urljoin

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


class Keep2shareCC(SimpleHoster):
    __name__ = "Keep2shareCC"
    __type__ = "hoster"
    __version__ = "0.12"

    __pattern__ = r'https?://(?:www\.)?(keep2share|k2s|keep2s)\.cc/file/(?P<ID>\w+)'

    __description__ = """Keep2share.cc hoster plugin"""
    __license__ = "GPLv3"
    __authors__ = [("stickell", "l.stickell@yahoo.it"),
                   ("Walter Purcaro", "vuolter@gmail.com")]


    FILE_NAME_PATTERN = r'File: <span>(?P<N>.+)</span>'
    FILE_SIZE_PATTERN = r'Size: (?P<S>[^<]+)</div>'
    OFFLINE_PATTERN = r'File not found or deleted|Sorry, this file is blocked or deleted|Error 404'

    LINK_PATTERN = r'To download this file with slow speed, use <a href="([^"]+)">this link</a>'
    CAPTCHA_PATTERN = r'src="(/file/captcha\.html.+?)"'
    WAIT_PATTERN = r'Please wait ([\d:]+) to download this file'
    MULTIDL_ERROR = r'Free account does not allow to download more than one file at the same time'


    def handleFree(self):
        self.sanitize_url()
        self.html = self.load(self.pyfile.url)

        self.fid = re.search(r'<input type="hidden" name="slow_id" value="([^"]+)">', self.html).group(1)
        self.html = self.load(self.pyfile.url, post={'yt0': '', 'slow_id': self.fid})

        m = re.search(r"function download\(\){.*window\.location\.href = '([^']+)';", self.html, re.DOTALL)
        if m:  # Direct mode
            self.startDownload(m.group(1))
        else:
            self.handleCaptcha()

            self.wait(30)

            self.html = self.load(self.pyfile.url, post={'uniqueId': self.fid, 'free': 1})

            m = re.search(self.WAIT_PATTERN, self.html)
            if m:
                self.logDebug("Hoster told us to wait for %s" % m.group(1))
                # string to time convert courtesy of https://stackoverflow.com/questions/10663720
                ftr = [3600, 60, 1]
                wait_time = sum([a * b for a, b in zip(ftr, map(int, m.group(1).split(':')))])
                self.wait(wait_time, reconnect=True)
                self.retry()

            m = re.search(self.MULTIDL_ERROR, self.html)
            if m:
                # if someone is already downloading on our line, wait 30min and retry
                self.logDebug("Already downloading, waiting for 30 minutes")
                self.wait(30 * 60, reconnect=True)
                self.retry()

            m = re.search(self.LINK_PATTERN, self.html)
            if m is None:
                self.error("Unable to detect direct link")
            self.startDownload(m.group(1))


    def handleCaptcha(self):
        m = re.search(self.CAPTCHA_PATTERN, self.html)
        if m is None:
            self.error("Captcha key not found")

        captcha_url = urljoin(self.base_url, m.group(1))

        for _ in xrange(5):
            post_data = {'CaptchaForm[code]': self.decryptCaptcha(captcha_url),
                         'free': 1,
                         'freeDownloadRequest': 1,
                         'uniqueId': self.fid,
                         'yt0': ''}

            self.html = self.load(self.pyfile.url, post=post_data)

            if 'recaptcha' not in self.html:
                self.correctCaptcha()
                break
            else:
                self.logInfo("Wrong captcha")
                self.invalidCaptcha()
        else:
            self.fail("All captcha attempts failed")


    def startDownload(self, url):
        d = urljoin(self.base_url, url)
        self.logDebug("Direct Link: " + d)
        self.download(d, disposition=True)


    def sanitize_url(self):
        header = self.load(self.pyfile.url, just_header=True)
        if 'location' in header:
            self.pyfile.url = header['location']
        p = urlparse(self.pyfile.url)
        self.base_url = "%s://%s" % (p.scheme, p.hostname)


getInfo = create_getInfo(Keep2shareCC)

@@ -320,17 +320,17 @@
 _ = %220.1
-2
+3
 %22%0A%0A    _
@@ -2768,32 +2768,272 @@
 eCaptcha(self):%0A
+        recaptcha = ReCaptcha(self)%0A%0A        for _ in xrange(5):%0A            post_data = %7B'free': 1,%0A                         'freeDownloadRequest': 1,%0A                         'uniqueId': self.fid,%0A                         'yt0': ''%7D%0A%0A    
         m = re.s
@@ -3075,36 +3075,32 @@
 ml)%0A        
+    
 if m
- is None
 :%0A          
@@ -3101,49 +3101,8 @@
     
-    self.error(%22Captcha key not found%22)%0A%0A
     
@@ -3158,37 +3158,12 @@
 1))%0A
-%0A
     
-    for _ in xrange(5):%0A
     
@@ -3179,20 +3179,17 @@
 ost_data
- = %7B
+%5B
 'Captcha
@@ -3199,17 +3199,19 @@
 m%5Bcode%5D'
-:
+%5D =
  self.de
@@ -3235,17 +3235,16 @@
 cha_url)
-,
 %0A       
@@ -3237,32 +3237,41 @@
 a_url)%0A         
+   else:%0A
                 
@@ -3274,152 +3274,198 @@
     
-'free': 1,%0A                         'freeDownloadRequest': 1,%0A                         'uniqueId': self.fid,%0A                         'yt0': ''%7D
+challenge, response = recaptcha.challenge()%0A                post_data.update(%7B'recaptcha_challenge_field': challenge,%0A                                  'recaptcha_response_field': response%7D)
 %0A%0A  

Fix handleCaptcha