import zipfile
import os
import json
from selenium.webdriver.common.by import By


def replace_variables_in_js(js_content: str, variables_dict: dict):
    for variable, value in variables_dict.items():
        js_content = js_content.replace('{{ ' + variable + ' }}', value)
    return js_content


def generate_proxy_auth_extension(
        proxy_host: str, proxy_port: str, proxy_user: str, proxy_pass: str,
        extension_file: str):
    """Generate a Chrome extension that modify proxy settings based on desired host, port, username and password.

    If you are using --headless in chromedriver, you must use --headless=new to support extensions in headless mode.
    """
    current_dir = os.path.dirname(os.path.abspath(__file__))
    manifest_json_path = os.path.join(current_dir, 'manifest.json')
    background_js_path = os.path.join(current_dir, 'background.js')
    with open(manifest_json_path, 'r', encoding='utf-8') as f:
        manifest_json = f.read()
    with open(background_js_path, 'r', encoding='utf-8') as f:
        background_js = f.read()

    variables_dict = {
        'proxy_host': proxy_host,
        'proxy_port': proxy_port,
        'proxy_user': proxy_user,
        'proxy_pass': proxy_pass
    }
    background_js = replace_variables_in_js(background_js, variables_dict)

    with zipfile.ZipFile(extension_file, 'w') as zp:
        zp.writestr('manifest.json', manifest_json)
        zp.writestr('background.js', background_js)


def get_my_ip(driver):
    origin_tab = driver.current_window_handle
    driver.execute_script("window.open('', '_blank');")
    driver.switch_to.window(driver.window_handles[-1])

    driver.get('https://api.ipify.org/?format=json')

    ip_row = driver.find_element(By.XPATH, '//body').text
    ip = json.loads(ip_row)['ip']

    driver.close()
    driver.switch_to.window(origin_tab)

    return ip


def proxy_is_working(driver, host: str):
    ip = get_my_ip(driver)

    if ip == host:
        return True
    else:
        return False