import time
import asyncio

import aiohttp
import requests

from python3_anticaptcha import app_key, create_task_url, get_sync_result, get_async_result


class CustomCaptchaTask:
    def __init__(
        self,
        anticaptcha_key: str,
        sleep_time: int = 5,
        assignment: str = None,
        forms: dict = None,
        callbackUrl: str = None,
    ):
        """
        Модуль отвечает за решение CustomCaptchaTask
        :param anticaptcha_key: Ключ от АнтиКапчи
        :param sleep_time: Время ожидания решения
        :param assignment: Опишите что работник должен сделать на английском языке
        :param forms: Произвольная форма в формате JSON
        :param callbackUrl: URL для решения капчи с ответом через callback
        """
        if sleep_time < 5:
            raise ValueError(f"Param `sleep_time` must be greater than 5. U set - {sleep_time}")
        self.sleep_time = sleep_time

        # Пайлоад для создания задачи
        self.task_payload = {
            "clientKey": anticaptcha_key,
            "task": {"type": "CustomCaptchaTask"},
            "softId": app_key,
        }
        # заполняем поле с заданием пользователю
        if assignment:
            self.task_payload["task"].update({"assignment": assignment})
            # заполняем поле с произвольной формой в формате JSON
        if forms:
            self.task_payload["task"].update({"forms": forms})

            # задаём callbackUrl если передан
        if callbackUrl:
            self.task_payload.update({"callbackUrl": callbackUrl})

        # пайлоад для получения ответа сервиса
        self.result_payload = {"clientKey": anticaptcha_key}

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        if exc_type:
            return False
        return True

    # Работа с капчёй
    def captcha_handler(self, imageUrl: str, **kwargs) -> dict:
        """
                Метод получает ссылку изображение для задания
                :param imageUrl: URL картинки
        :param kwargs: Дополнительные параметры для `requests.post(....)`.
                :return: Возвращает ответ сервера в виде JSON(ответ так же можно глянуть в документации антикапчи)
        """
        self.task_payload["task"].update({"imageUrl": imageUrl})
        # Отправляем на антикапча параметры фанкапич,
        # в результате получаем JSON ответ содержащий номер решаемой капчи
        captcha_id = requests.post(create_task_url, json=self.task_payload, verify=False).json()

        # Проверка статуса создания задачи, если создано без ошибок - извлекаем ID задачи, иначе возвращаем ответ сервера
        if captcha_id["errorId"] == 0:
            captcha_id = captcha_id["taskId"]
            # обновляем пайлоад на получение решения капчи
            self.result_payload.update({"taskId": captcha_id})
        else:
            return captcha_id
            # если передан параметр `callbackUrl` - не ждём решения капчи а возвращаем незаполненный ответ
        if self.task_payload.get("callbackUrl"):
            return self.result_payload

        else:
            # Ждем решения капчи
            time.sleep(self.sleep_time)
            return get_sync_result(result_payload=self.result_payload, sleep_time=self.sleep_time)


class aioCustomCaptchaTask:
    def __init__(
        self,
        anticaptcha_key: str,
        sleep_time: int = 5,
        assignment: str = None,
        forms: dict = None,
        callbackUrl: str = None,
    ):
        """
                Модуль отвечает за решение CustomCaptchaTask
                :param anticaptcha_key: Ключ от АнтиКапчи
                :param sleep_time: Время ожидания решения
        :param assignment: Опишите что работник должен сделать на английском языке
        :param forms: Произвольная форма в формате JSON
        :param callbackUrl: URL для решения капчи с ответом через callback
        """
        if sleep_time < 5:
            raise ValueError(f"Param `sleep_time` must be greater than 5. U set - {sleep_time}")
        self.sleep_time = sleep_time

        # Пайлоад для создания задачи
        self.task_payload = {
            "clientKey": anticaptcha_key,
            "task": {"type": "CustomCaptchaTask"},
            "softId": app_key,
        }
        # заполняем поле с заданием пользователю
        if assignment:
            self.task_payload["task"].update({"assignment": assignment})
            # заполняем поле с произвольной формой в формате JSON
        if forms:
            self.task_payload["task"].update({"forms": forms})

            # задаём callbackUrl если передан
        if callbackUrl:
            self.task_payload.update({"callbackUrl": callbackUrl})

        # пайлоад для получения ответа сервиса
        self.result_payload = {"clientKey": anticaptcha_key}

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        if exc_type:
            return False
        return True

    # Работа с капчёй
    async def captcha_handler(self, imageUrl: str) -> dict:
        """
        Метод получает ссылку изображение для задания
        :param imageUrl: URL картинки
        :return: Возвращает ответ сервера в виде JSON(ответ так же можно глянуть в документации антикапчи)
        """
        self.task_payload["task"].update({"imageUrl": imageUrl})
        # Отправляем на антикапча параметры фанкапич,
        # в результате получаем JSON ответ содержащий номер решаемой капчи
        async with aiohttp.ClientSession() as session:
            async with session.post(create_task_url, json=self.task_payload) as resp:
                captcha_id = await resp.json()

        # Проверка статуса создания задачи, если создано без ошибок - извлекаем ID задачи, иначе возвращаем ответ сервера
        if captcha_id["errorId"] == 0:
            captcha_id = captcha_id["taskId"]
            # обновляем пайлоад на получение решения капчи
            self.result_payload.update({"taskId": captcha_id})
        else:
            return captcha_id

            # если передан параметр `callbackUrl` - не ждём решения капчи а возвращаем незаполненный ответ
        if self.task_payload.get("callbackUrl"):
            return self.result_payload

        else:
            # Ждем решения капчи
            await asyncio.sleep(self.sleep_time)
            return await get_async_result(result_payload=self.result_payload, sleep_time=self.sleep_time)