from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import TimeoutException as STimeout
from http.cookies import SimpleCookie
import json
import time

def get_bing_api_token(
    username,
    password,
    headless=True,
    remote_selenium="http://localhost:4444",
    wait=10,
    json_outfile="bing_cookie.json",
):
    options = webdriver.ChromeOptions()
    options.set_capability("goog:loggingPrefs", {"performance": "ALL"})

    if headless:
        options.add_argument("headless")  # Run browser in headless mode

    # Create a new WebDriver instance with Docker
    driver = webdriver.Remote(remote_selenium, options=options)

    # Open Bing
    driver.get("https://www.bing.com")

    # <button id="bnp_btn_accept" class="bnp_btn_accept">
    # respond to cookies
    try:
        cookie = WebDriverWait(driver, wait).until(
            EC.element_to_be_clickable((By.ID, "bnp_btn_accept"))
        )
        cookie.click()
    except STimeout:
        pass

    try:
        login_button = WebDriverWait(driver, wait).until(
            EC.element_to_be_clickable((By.ID, "id_l"))
        )
        login_button.click()

        username_field = WebDriverWait(driver, wait).until(
            EC.element_to_be_clickable((By.ID, "i0116"))
        )
        username_field.send_keys(username)
        username_field.send_keys(Keys.ENTER)

        password_field = WebDriverWait(driver, wait).until(
            EC.element_to_be_clickable((By.ID, "i0118"))
        )
        password_field.send_keys(password)
        password_field.send_keys(Keys.ENTER)

        signed_in_no = WebDriverWait(driver, wait).until(
            EC.element_to_be_clickable((By.ID, "idBtn_Back"))
        )
        signed_in_no.click()

        cookie = SimpleCookie()
        interesting = []
        # loop until some appropriate cookies are found
        while len(interesting) == 0:
            time.sleep(1)

            logs_raw = driver.get_log("performance")
            for r in logs_raw:
                try:
                    hdr = json.loads(r["message"])["message"]["params"][
                        "headers"
                    ]
                    if "cookie" in hdr.keys():
                        cookie.load(hdr["cookie"])
                        hdr["cookie"] = {
                            k: v.value for k, v in cookie.items()
                        }
                        if "_U" in hdr["cookie"].keys():
                            interesting.append(hdr)
                except KeyError:
                    pass

        if json_outfile is not None:
            with open("bing_cookie.json", "w") as f:
                json.dump(interesting, f, indent=2)

        return interesting[0]["cookie"]["_U"]
    except STimeout:
        return "selenium timeout"
    finally:
        # Close the WebDriver session
        driver.quit()