import random
import threading
import time
from http.cookies import SimpleCookie

import pika

from queue_message.CookiesPublisher import CookiesPublisher, QUEUE_HOST, credentials, REQUEST_DATA_QUEUE, \
    REQUEST_DATA_QUEUE_TEST
from workers.captcha_result_getter import CaptchaResultGetter


class CookiesGenerator(threading.Thread):
    def __init__(self, cookiesPublisher: CookiesPublisher):
        super().__init__()
        self.connection = None
        self.cookiesPublisher = cookiesPublisher
        self.channel = None
        self.valid_csrf = None

    def set_up_connection(self):
        self.connection = pika.BlockingConnection(
            pika.ConnectionParameters(host=QUEUE_HOST, port=5672, credentials=credentials))
        self.channel = self.connection.channel()

    def listen_to_queue(self, callback):
        self.channel.basic_qos(prefetch_count=1)
        self.channel.basic_consume(queue=REQUEST_DATA_QUEUE, auto_ack=False, on_message_callback=callback)
        self.channel.start_consuming()

    def message_count(self):
        return self.channel.queue_declare(queue=REQUEST_DATA_QUEUE, durable=True).method.message_count

    def on_message(self, ch, method, properties, body):
        print(f" [x] Received {body}")
        _message_count = self.cookiesPublisher.message_count()
        print("message count in queue is {}".format(_message_count))

        #     generate 10 cookies to new_queue
        captchaResultGetter = CaptchaResultGetter()
        _cookies_count = random.randint(2, 5)
        if _message_count < 10:
            for i in range(1, _cookies_count):
                _received_cookies = body.decode("UTF-8")
                new_cookie = captchaResultGetter.get_valid_ch_cookie()
                _app_sig = get_app_and_app_sig(_received_cookies)
                if new_cookie is not None and _app_sig is not None:
                    new_cookie = _app_sig + "policy=accepted;lang=fr;" + new_cookie
                    new_cookie = new_cookie.replace("Domain=.hermes.com;", "").replace("Path=/;", "").replace(
                        "Secure; SameSite=None", "").replace("Max-Age=31536000;", "").replace(" ", "")
                    print("new_cookie is " + new_cookie)
                    self.cookiesPublisher.publish_body(new_cookie)
                new_cookie = captchaResultGetter.get_valid_cookie(new_cookie)
                if new_cookie is not None and _app_sig is not None:
                    new_cookie = _app_sig + "policy=accepted;lang=fr;" + new_cookie
                    new_cookie = new_cookie.replace("Domain=.hermes.com;", "").replace("Path=/;", "").replace(
                        "Secure; SameSite=None", "").replace("Max-Age=31536000;", "").replace(" ", "")
                    self.cookiesPublisher.publish_body(new_cookie)
            ch.basic_ack(delivery_tag=method.delivery_tag)
        else:
            time.sleep(30)
            ch.basic_reject(delivery_tag=method.delivery_tag, requeue=True)

    def run(self):
        print(threading.currentThread().name + " starts")
        self.set_up_connection()
        self.listen_to_queue(self.on_message)
        self.channel.start_consuming()


def get_app_and_app_sig(cookies: str):
    _local_cookies = SimpleCookie()
    _local_cookies.load(cookies)
    new_coolies_str = ""
    new_cookies = {k: v.value for k, v in _local_cookies.items()}
    for key in _local_cookies:
        if key == "app" or key == "app.sig":
            new_coolies_str = new_coolies_str + key + "=" + new_cookies[key] + ";"
    return new_coolies_str


if __name__ == '__main__':
    # _list = get_app_and_app_sig(
    #     "app.sig=axk5aJ2c83dgV56DswNhw98y4SY;datadome=2dTtQDIHCadAbzh8JWv7MFKGH8~fkOoGlA1O3J7llv_sdYOSZYbaKruHPBbARchV1nnrstsaMX5E_XwbJmGiXzt5qZXOL0UCAV9TTHqBRh10JvO~GCGJv2JVO~6RnrcU;app=eyJmbGFzaCI6e30sImNhY2hlZmxhc2giOltdLCJjc3JmU2VjcmV0IjoiZEE2bEdnN04yd2t0eWVmZmVjVkxMY1dTIiwiYXBwb2ludG1lbnRfY29kZSI6Ik5DTlVaQiIsImJsb2NrX3JlZ2lzdHJhdGlvbiI6ZmFsc2V9;policy=accepted;lang=fr;")
    # print(_list)
    cookiesPublisher = CookiesPublisher(queue_name=REQUEST_DATA_QUEUE_TEST)
    cookiesPublisher.set_up_connection()
    cookieGenerator = CookiesGenerator(cookiesPublisher)
    cookieGenerator.run()