import time

from selenium import webdriver
from selenium.webdriver.firefox.service import Service
from selenium.webdriver.firefox.options import Options
from webdriver_manager.firefox import GeckoDriverManager


BASE_URL = "https://soyjak.st/poll.php?id={}"
START_ID = 23586
END_ID = 23864

KEYWORDS = ["Iron", "Ruby", "Coal"]
CHALLENGE_TITLE = "McChallenge"

def create_driver():
    options = Options()
    options.set_preference("dom.webdriver.enabled", False)
    options.set_preference("useAutomationExtension", False)

    driver = webdriver.Firefox(
        service=Service(GeckoDriverManager().install()),
        options=options,
    )

    driver.set_page_load_timeout(30)
    return driver


def is_challenge_page(driver):
    return CHALLENGE_TITLE.lower() in driver.title.lower()

def contains_all_keywords(page_source):
    html_lower = page_source.lower()
    return all(k.lower() in html_lower for k in KEYWORDS)

def wait_for_manual_solve(driver):
    print("\nCaptcha detected o algo.")
    input()

def main():
    driver = create_driver()

    try:
        for poll_id in range(START_ID, END_ID + 1):
            url = BASE_URL.format(poll_id)
            driver.get(url)

            if is_challenge_page(driver):
                print(f"Poll ID {poll_id}: challenge detected")
                wait_for_manual_solve(driver)

                driver.get(url)

            page_source = driver.page_source

            if contains_all_keywords(page_source):
                print(f"Poll ID {poll_id}: keyword match found")

            time.sleep(0.1)

    except Exception as e:
        print("Error:", e)

    finally:
        print("Finished scanning.")
        driver.quit()


if __name__ == "__main__":
    main()