import json
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager
import time
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.common.exceptions import NoSuchElementException
def blur_elements_by_id(browser, element_ids):
        for element in element_ids:
            try:
                element = browser.find_element(By.ID, element)
                browser.execute_script("arguments[0].style.filter = 'blur(7px)';", element)
            except NoSuchElementException:
                print("Issue finding:", element, "on page")
def get_adsbx_screenshot(file_path, url_params, enable_labels=False, enable_track_labels=False, overrides={}, conceal_ac_id=False, conceal_pia=False):
    import os
    import platform
    chrome_options = webdriver.ChromeOptions()
    chrome_options.headless = True
    chrome_options.add_argument('window-size=800,800')
    chrome_options.add_argument('ignore-certificate-errors')
    chrome_options.add_experimental_option('excludeSwitches', ['enable-logging'])
    if platform.system() == "Linux":
        chrome_options.add_argument('crash-dumps-dir=/tmp/plane-notify/chrome')

    #Plane images issue loading when in headless setting agent fixes.
    chrome_options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.54 Safari/537.36")
    if platform.system() == "Linux" and os.geteuid()==0:
        chrome_options.add_argument('--no-sandbox') # required when running as root user. otherwise you would get no sandbox errors.
    browser = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
    url = f"https://globe.theairtraffic.com/?{url_params}"
    print(f"Getting Screenshot of {url}")
    browser.set_page_load_timeout(80)
    browser.get(url)
    WebDriverWait(browser, 40).until(lambda d: d.execute_script("return jQuery.active == 0"))
    remove_id_elements = ["show_trace", "credits", 'infoblock_close', 'selected_photo_link', "history_collapse"]
    for element in remove_id_elements:
        try:
            element = browser.find_element(By.ID, element)
            browser.execute_script("""var element = arguments[0];    element.parentNode.removeChild(element); """, element)
        except:
            print("Issue finding:", element, "on page")
    #Remove watermark on data
    try:
        browser.execute_script("document.getElementById('selected_infoblock').className = 'none';")
    except:
        print("Couldn't remove watermark from map")
    #Disable slidebar
    try:
        browser.execute_script("$('#infoblock-container').css('overflow', 'hidden');")
    except:
        print("Couldn't disable sidebar on map")
    #Remove Google Ads
    try:
        element = browser.find_element(By.XPATH, "//*[contains(@id, 'FIOnDemandWrapper_')]")
        browser.execute_script("""var element = arguments[0];    element.parentNode.removeChild(element); """, element)
    except:
        print("Couldn't remove Google Ads")
    #Remove Copy Link
    try:
        element = browser.find_element(By.XPATH, "//*[@id='selected_icao']/span[2]/a")
        browser.execute_script("""var element = arguments[0];    element.parentNode.removeChild(element); """, element)
    except Exception as e:
        print("Couldn't remove copy link button from map", e)
    #browser.execute_script("toggleFollow()")
    if conceal_pia or conceal_ac_id:
        blur_elements_by_id(browser, ["selected_callsign", "selected_icao", "selected_squawk1"])
    if conceal_ac_id:
        blur_elements_by_id(browser, ["selected_registration", "selected_country", "selected_dbFlags", "selected_ownop", "selected_typelong", "selected_icaotype", "airplanePhoto", "silhouette", "copyrightInfo"])
    if enable_labels:
        browser.find_element(By.TAG_NAME, 'body').send_keys('l')
    if enable_track_labels:
        browser.find_element(By.TAG_NAME, 'body').send_keys('k')
    from selenium.webdriver.support import expected_conditions as EC
    time.sleep(15)

    if 'reg' in overrides.keys():
        element = browser.find_element(By.ID, "selected_registration")
        browser.execute_script(f"arguments[0].innerText = '* {overrides['reg']}'", element)
        reg = overrides['reg']
    else:
        try:
            reg = browser.find_element(By.ID, "selected_registration").get_attribute("innerHTML")
            print("Reg from tar1090 is", reg)
        except Exception as e:
            print("Couldn't find reg in tar1090", e)
            reg = None
    if reg is not None:
        try:
            try:
                photo_box = browser.find_element(By.ID, "silhouette")
            except NoSuchElementException:
                photo_box = browser.find_element(By.ID, "airplanePhoto")
            finally:
                import requests, json
                photo_list = json.loads(requests.get("https://raw.githubusercontent.com/Jxck-S/aircraft-photos/main/photo-list.json", timeout=20).text)
                if reg in photo_list.keys():
                    browser.execute_script("arguments[0].id = 'airplanePhoto';", photo_box)
                    browser.execute_script("arguments[0].removeAttribute('width')", photo_box)
                    browser.execute_script("arguments[0].style.width = '200px';", photo_box)
                    browser.execute_script("arguments[0].style.float = 'left';", photo_box)
                    browser.execute_script(f"arguments[0].src = 'https://raw.githubusercontent.com/Jxck-S/aircraft-photos/main/images/{reg}.jpg';", photo_box)
                    image_copy_right = browser.find_element(By.ID, "copyrightInfo")
                    browser.execute_cdp_cmd('Emulation.setScriptExecutionDisabled', {'value': True})
                    copy_right_children = image_copy_right.find_elements(By.XPATH, "*")
                    if len(copy_right_children) > 0:
                        browser.execute_script(f"arguments[0].innerText = 'Image © {photo_list[reg]['photographer']}'", copy_right_children[0])
                    else:
                        browser.execute_script(f"arguments[0].appendChild(document.createTextNode('Image © {photo_list[reg]['photographer']}'))", image_copy_right)
        except Exception as e:
            print("Error on changing photo", e)
    if 'type' in overrides.keys():
        element = browser.find_element(By.ID, "selected_icaotype")
        browser.execute_script(f"arguments[0].innerText = '* {overrides['type']}'", element)
    if 'typelong' in overrides.keys():
        element = browser.find_element(By.ID, "selected_typelong")
        browser.execute_script(f"arguments[0].innerText = '* {overrides['typelong']}'", element)
    if 'ownop' in overrides.keys():
        element = browser.find_element(By.ID, "selected_ownop")
        browser.execute_script(f"arguments[0].innerText = '* {overrides['ownop']}'", element)
    time.sleep(5)
    browser.save_screenshot(file_path)
    browser.quit()
def generate_adsbx_screenshot_time_params(timestamp):
    from datetime import datetime
    from datetime import timedelta
    timestamp_dt = datetime.utcfromtimestamp(timestamp)
    print(timestamp_dt)
    start_time = timestamp_dt - timedelta(minutes=1)
    time_params = "&showTrace=" + timestamp_dt.strftime("%Y-%m-%d")  + "&startTime=" + start_time.strftime("%H:%M:%S") + "&endTime=" + timestamp_dt.strftime("%H:%M:%S")
    return time_params