from threading import Thread
from time import sleep

import pyautogui as p
from requests import get
from vk_api import VkApi
from vk_api.longpoll import VkEventType, VkLongPoll
from clipboard import copy

from solver import Solver


p.FAILSAFE = True
amount = 40
s = Solver()
gid = -144979945
mode = int(input("1. Одноразовый\n2. Пока не оффнешь.\n"))
p.alert("НАЖМИТЕ ПРИ ГОТОВНОСТИ")
threads = []
done = False
p.MINIMUM_DURATION = 0
p.MINIMUM_SLEEP = 0
p.PAUSE = 0


def open_accs():
    p.click(1723, 50)  # Клик на расширение

    sleep(1)

    p.click(1693, 445)  # Клик на шестерёнку
    sleep(1)
    p.click(1588, 502)  # Клик на открытие вкладок

    sleep(10)
    prepare()


def prepare():
    p.click(15, 15)

    for x in range(amount):
        sleep(0.1)
        while not p.locateOnScreen("find.png"): continue
        sleep(0.8)
        p.click(111, 305)
        p.moveRel(0, 100)
        p.hotkey("ctrl", "tab")


def confirm_code():
    global done
    for x in range(amount):
        p.click(930, 520)
        p.hotkey("ctrl", "v")

        p.click(930, 580)

        p.hotkey("ctrl", "tab")

    for x in range(amount):
        p.hotkey("ctrl", "tab")
        sleep(0.05)

    done = False


def download(url, name):
    with open(name + ".png", "wb") as file:
        file.write(get(url, stream=True).content)


def clear_browser():
    p.click(15, 15)
    p.click(137, 82)
    sleep(8)
    prepare()


def start_vk():
    for u in list(open("vk.txt", "r", encoding="utf-8").read().splitlines()):
        def run():
            global done
            session = VkApi(token=u)
            vk = session.get_api()
            lp = VkLongPoll(session)

            while True:
                try:
                    for event in lp.listen():
                        if not done:
                            if event.type == VkEventType.MESSAGE_NEW and event.peer_id == gid and event.attachments and \
                                    event.attachments['attach1_type'] == "photo":
                                download(
                                    vk.messages.getById(message_ids=event.message_id)["items"][0]["attachments"][0][
                                        "photo"][
                                        "sizes"][3]["url"], "img")
                                if not done:
                                    done = True
                                    copy(s.solve("img.png"))
                                    confirm_code()
                                if mode == 1:
                                    exit(0)
                                else:
                                    sleep(3300)
                                    clear_browser()
                        else:
                            if mode == 1:
                                exit(0)

                except Exception as e:
                    print("ОШЫПКА: ", e)

        t = Thread(target=run)
        t.start()
        threads.append(t)


open_accs()
start_vk()

print("\nВсе аккаунты загружены. Ждём промокод...\n")

for x in threads:
    while x.isAlive():
        sleep(1)