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

from __future__ import with_statement

import base64
import re
import time

import pycurl
from module.network.HTTPRequest import BadHeader
from module.network.RequestFactory import getRequest as get_request

from ..internal.Addon import Addon
from ..internal.misc import fs_encode, json, threaded


class DeathByCaptchaException(Exception):
    DBC_ERRORS = {'not-logged-in': 'Access denied, check your credentials',
                  'invalid-credentials': 'Access denied, check your credentials',
                  'banned': 'Access denied, account is suspended',
                  'insufficient-funds': 'Insufficient account balance to decrypt CAPTCHA',
                  'invalid-captcha': 'CAPTCHA is not a valid image',
                  'service-overload': 'CAPTCHA was rejected due to service overload, try again later',
                  'invalid-request': 'Invalid request',
                  'timed-out': 'No CAPTCHA solution received in time'}

    def __init__(self, err):
        self.err = err

    def get_code(self):
        return self.err

    def get_desc(self):
        if self.err in self.DBC_ERRORS.keys():
            return self.DBC_ERRORS[self.err]
        else:
            return self.err

    def __str__(self):
        return "<DeathByCaptchaException %s>" % self.err

    def __repr__(self):
        return "<DeathByCaptchaException %s>" % self.err


class DeathByCaptcha(Addon):
    __name__ = "DeathByCaptcha"
    __type__ = "hook"
    __version__ = "0.17"
    __status__ = "testing"

    __config__ = [("activated", "bool", "Activated", False),
                  ("username", "str", "Username", ""),
                  ("password", "password", "Password", ""),
                  ("check_client", "bool", "Don't use if client is connected", True)]

    __description__ = """Send captchas to DeathByCaptcha.com"""
    __license__ = "GPLv3"
    __authors__ = [("RaNaN", "RaNaN@pyload.org"),
                   ("zoidberg", "zoidberg@mujmail.cz")]

    API_URL = "http://api.dbcapi.me/api/"

    def api_response(self, api="captcha", post=False, multipart=False):
        req = get_request()
        req.c.setopt(
            pycurl.HTTPHEADER, [
                "Accept: application/json", "User-Agent: pyLoad %s" %
                self.pyload.version])

        if post:
            if not isinstance(post, dict):
                post = {}
            post.update({'username': self.config.get('username'),
                         'password': self.config.get('password')})

        res = None
        try:
            html = self.load("%s%s" % (self.API_URL, api),
                             post=post,
                             multipart=multipart,
                             req=req)

            self.log_debug(html)
            res = json.loads(html)

            if "error" in res:
                raise DeathByCaptchaException(res['error'])
            elif "status" not in res:
                raise DeathByCaptchaException(str(res))

        except BadHeader, e:
            if e.code == 403:
                raise DeathByCaptchaException('not-logged-in')

            elif e.code == 413:
                raise DeathByCaptchaException('invalid-captcha')

            elif e.code == 503:
                raise DeathByCaptchaException('service-overload')

            elif e.code in (400, 405):
                raise DeathByCaptchaException('invalid-request')

            else:
                raise

        finally:
            req.close()

        return res

    def get_credits(self):
        res = self.api_response("user", True)

        if 'is_banned' in res and res['is_banned']:
            raise DeathByCaptchaException('banned')
        elif 'balance' in res and 'rate' in res:
            self.info.update(res)
        else:
            raise DeathByCaptchaException(res)

    def get_status(self):
        res = self.api_response("status", False)

        if 'is_service_overloaded' in res and res['is_service_overloaded']:
            raise DeathByCaptchaException('service-overload')

    def submit(self, captcha, captchaType="file", match=None):
        #@NOTE: Workaround multipart-post bug in HTTPRequest.py
        if re.match("^\w*$", self.config.get('password')):
            multipart = True
            data = (pycurl.FORM_FILE, captcha)
        else:
            multipart = False
            with open(fs_encode(captcha), 'rb') as f:
                data = f.read()
            data = "base64:" + base64.b64encode(data)

        res = self.api_response("captcha", {'captchafile': data}, multipart)

        if "captcha" not in res:
            raise DeathByCaptchaException(res)
        ticket = res['captcha']

        for _i in range(24):
            time.sleep(5)
            res = self.api_response("captcha/%d" % ticket, False)
            if res['text'] and res['is_correct']:
                break
        else:
            raise DeathByCaptchaException('timed-out')

        result = res['text']
        self.log_debug("Result %s : %s" % (ticket, result))

        return ticket, result

    def captcha_task(self, task):
        if "service" in task.data:
            return False

        if not task.isTextual():
            return False

        if not self.config.get('username') or not self.config.get('password'):
            return False

        if self.pyload.isClientConnected() and self.config.get('check_client'):
            return False

        try:
            self.get_status()
            self.get_credits()
        except DeathByCaptchaException, e:
            self.log_error(e.message)
            return False

        balance, rate = self.info['balance'], self.info['rate']
        self.log_info(_("Account balance"),
                      _("US$%.3f (%d captchas left at %.2f cents each)") % (balance / 100,
                                                                            balance // rate, rate))

        if balance > rate:
            task.handler.append(self)
            task.data['service'] = self.classname
            task.setWaiting(180)
            self._process_captcha(task)

    def captcha_invalid(self, task):
        if task.data['service'] == self.classname and "ticket" in task.data:
            try:
                res = self.api_response(
                    "captcha/%d/report" %
                    task.data['ticket'], True)

            except DeathByCaptchaException, e:
                self.log_error(e.message)

            except Exception, e:
                self.log_error(e, trace=True)

    @threaded
    def _process_captcha(self, task):
        c = task.captchaParams['file']
        try:
            ticket, result = self.submit(c)
        except DeathByCaptchaException, e:
            task.error = e.get_code()
            self.log_error(e.message)
            return

        task.data['ticket'] = ticket
        task.setResult(result)
