spider 2.51.63

use crate::configuration::Configuration;
use crate::features::webdriver_args::get_browser_args;
use crate::features::webdriver_common::{WebDriverBrowser, WebDriverConfig};
use std::sync::Arc;
use std::time::Duration;
use thirtyfour::common::capabilities::desiredcapabilities::Capabilities;
use thirtyfour::prelude::*;
use tokio::task::JoinHandle;

/// Stealth scripts from spider_fingerprint - cleans up automation markers.
#[cfg(feature = "webdriver_stealth")]
pub use spider_fingerprint::spoofs::{
    spoof_device_memory, CLEANUP_CDP_MARKERS, HIDE_SELENIUM_MARKERS, HIDE_WEBDRIVER,
};

/// Legacy stealth script (fallback when spider_fingerprint not available).
#[cfg(all(feature = "webdriver_stealth", not(feature = "serde")))]
pub const STEALTH_SCRIPT: &str = r#"
Object.defineProperty(navigator, 'webdriver', { get: () => undefined });
Object.defineProperty(navigator, 'plugins', { get: () => [1, 2, 3, 4, 5] });
Object.defineProperty(navigator, 'languages', { get: () => ['en-US', 'en'] });
window.chrome = { runtime: {} };
"#;

/// WebDriver control tuple type alias (mirrors BrowserControl).
pub type WebDriverControl = (Arc<WebDriver>, Option<JoinHandle<()>>, Option<String>);

/// WebDriver controller with RAII cleanup (mirrors BrowserController).
pub struct WebDriverController {
    /// The WebDriver instance.
    pub driver: WebDriverControl,
    /// Whether the driver has been closed.
    pub closed: bool,
}

impl WebDriverController {
    /// Create a new WebDriver controller.
    pub fn new(driver: WebDriverControl) -> Self {
        Self {
            driver,
            closed: false,
        }
    }

    /// Get a reference to the WebDriver.
    pub fn driver(&self) -> &Arc<WebDriver> {
        &self.driver.0
    }

    /// Dispose the WebDriver and close the session.
    pub fn dispose(&mut self) {
        if !self.closed {
            self.closed = true;
            if let Some(handler) = self.driver.1.take() {
                handler.abort();
            }
        }
    }
}

impl Drop for WebDriverController {
    fn drop(&mut self) {
        self.dispose();
    }
}

/// Launch a WebDriver session with the provided configuration.
pub async fn launch_driver(config: &Configuration) -> Option<WebDriverController> {
    let webdriver_config = config.webdriver_config.as_ref()?;
    launch_driver_base(webdriver_config, config).await
}

/// Launch a WebDriver session with the base configuration.
pub async fn launch_driver_base(
    webdriver_config: &WebDriverConfig,
    config: &Configuration,
) -> Option<WebDriverController> {
    let server_url = &webdriver_config.server_url;

    let caps = build_capabilities(webdriver_config, config).await?;

    let mut attempts = 0;
    let max_retries = 10;
    let mut driver: Option<WebDriver> = None;

    while attempts <= max_retries {
        match WebDriver::new(server_url, caps.clone()).await {
            Ok(d) => {
                driver = Some(d);
                break;
            }
            Err(err) => {
                log::error!("WebDriver connection error: {:?}", err);
                attempts += 1;
                if attempts > max_retries {
                    log::error!("Exceeded maximum retry attempts for WebDriver connection");
                    break;
                }
                tokio::time::sleep(Duration::from_millis(500)).await;
            }
        }
    }

    let driver = driver?;
    let driver_arc = Arc::new(driver);

    // Set up viewport if configured
    if let (Some(width), Some(height)) = (
        webdriver_config.viewport_width,
        webdriver_config.viewport_height,
    ) {
        if let Err(e) = driver_arc.set_window_rect(0, 0, width, height).await {
            log::warn!("Failed to set viewport: {:?}", e);
        }
    }

    // Set up timeouts
    if let Some(timeout) = webdriver_config.timeout {
        let timeouts = TimeoutConfiguration::new(Some(timeout), Some(timeout), Some(timeout));
        if let Err(e) = driver_arc.update_timeouts(timeouts).await {
            log::warn!("Failed to set timeouts: {:?}", e);
        }
    }

    Some(WebDriverController::new((
        driver_arc,
        None,
        Some(server_url.clone()),
    )))
}

/// Build browser capabilities based on configuration.
async fn build_capabilities(
    webdriver_config: &WebDriverConfig,
    config: &Configuration,
) -> Option<Capabilities> {
    match webdriver_config.browser {
        WebDriverBrowser::Chrome => build_chrome_capabilities(webdriver_config, config).await,
        WebDriverBrowser::Firefox => build_firefox_capabilities(webdriver_config, config).await,
        WebDriverBrowser::Edge => build_edge_capabilities(webdriver_config, config).await,
    }
}

/// Build Chrome capabilities.
async fn build_chrome_capabilities(
    webdriver_config: &WebDriverConfig,
    config: &Configuration,
) -> Option<Capabilities> {
    let mut caps = DesiredCapabilities::chrome();

    // Set accept insecure certs
    if webdriver_config.accept_insecure_certs {
        if let Err(e) = caps.accept_insecure_certs(true) {
            log::warn!("Failed to set accept_insecure_certs: {:?}", e);
        }
    }

    // Set page load strategy
    if let Some(ref strategy) = webdriver_config.page_load_strategy {
        let strategy = match strategy.as_str() {
            "eager" => thirtyfour::PageLoadStrategy::Eager,
            "none" => thirtyfour::PageLoadStrategy::None,
            _ => thirtyfour::PageLoadStrategy::Normal,
        };
        if let Err(e) = caps.set_page_load_strategy(strategy) {
            log::warn!("Failed to set page_load_strategy: {:?}", e);
        }
    }

    // Collect all arguments
    let mut args: Vec<String> = Vec::new();

    // Add default browser args
    let default_args = get_browser_args(&WebDriverBrowser::Chrome);
    for arg in default_args {
        args.push(arg.to_string());
    }

    // Add custom browser args
    if let Some(ref custom_args) = webdriver_config.browser_args {
        args.extend(custom_args.clone());
    }

    // Add headless argument if needed
    if webdriver_config.headless && !args.iter().any(|a| a.contains("headless")) {
        args.push("--headless".to_string());
    }

    // Add user agent
    if let Some(ref ua) = webdriver_config.user_agent {
        args.push(format!("--user-agent={}", ua));
    } else if let Some(ref ua) = config.user_agent {
        args.push(format!("--user-agent={}", ua));
    }

    // Add proxy
    if let Some(ref proxy) = webdriver_config.proxy {
        args.push(format!("--proxy-server={}", proxy));
    }

    // Add viewport
    if let (Some(width), Some(height)) = (
        webdriver_config.viewport_width,
        webdriver_config.viewport_height,
    ) {
        args.push(format!("--window-size={},{}", width, height));
    }

    // Add all args to capabilities
    for arg in args {
        if let Err(e) = caps.add_arg(&arg) {
            log::warn!("Failed to add Chrome arg '{}': {:?}", arg, e);
        }
    }

    Some(caps.into())
}

/// Build Firefox capabilities.
async fn build_firefox_capabilities(
    webdriver_config: &WebDriverConfig,
    _config: &Configuration,
) -> Option<Capabilities> {
    let mut caps = DesiredCapabilities::firefox();

    // Set accept insecure certs
    if webdriver_config.accept_insecure_certs {
        if let Err(e) = caps.accept_insecure_certs(true) {
            log::warn!("Failed to set accept_insecure_certs: {:?}", e);
        }
    }

    // Set page load strategy
    if let Some(ref strategy) = webdriver_config.page_load_strategy {
        let strategy = match strategy.as_str() {
            "eager" => thirtyfour::PageLoadStrategy::Eager,
            "none" => thirtyfour::PageLoadStrategy::None,
            _ => thirtyfour::PageLoadStrategy::Normal,
        };
        if let Err(e) = caps.set_page_load_strategy(strategy) {
            log::warn!("Failed to set page_load_strategy: {:?}", e);
        }
    }

    // Collect all arguments
    let mut args: Vec<String> = Vec::new();

    // Add default browser args
    let default_args = get_browser_args(&WebDriverBrowser::Firefox);
    for arg in default_args {
        args.push(arg.to_string());
    }

    // Add custom browser args
    if let Some(ref custom_args) = webdriver_config.browser_args {
        args.extend(custom_args.clone());
    }

    // Add headless argument if needed
    if webdriver_config.headless && !args.iter().any(|a| a.contains("headless")) {
        args.push("-headless".to_string());
    }

    // Add all args to capabilities
    for arg in args {
        if let Err(e) = caps.add_arg(&arg) {
            log::warn!("Failed to add Firefox arg '{}': {:?}", arg, e);
        }
    }

    Some(caps.into())
}

/// Build Edge capabilities.
async fn build_edge_capabilities(
    webdriver_config: &WebDriverConfig,
    config: &Configuration,
) -> Option<Capabilities> {
    let mut caps = DesiredCapabilities::edge();

    // Set accept insecure certs
    if webdriver_config.accept_insecure_certs {
        if let Err(e) = caps.accept_insecure_certs(true) {
            log::warn!("Failed to set accept_insecure_certs: {:?}", e);
        }
    }

    // Set page load strategy
    if let Some(ref strategy) = webdriver_config.page_load_strategy {
        let strategy = match strategy.as_str() {
            "eager" => thirtyfour::PageLoadStrategy::Eager,
            "none" => thirtyfour::PageLoadStrategy::None,
            _ => thirtyfour::PageLoadStrategy::Normal,
        };
        if let Err(e) = caps.set_page_load_strategy(strategy) {
            log::warn!("Failed to set page_load_strategy: {:?}", e);
        }
    }

    // Collect all arguments
    let mut args: Vec<String> = Vec::new();

    // Add default browser args
    let default_args = get_browser_args(&WebDriverBrowser::Edge);
    for arg in default_args {
        args.push(arg.to_string());
    }

    // Add custom browser args
    if let Some(ref custom_args) = webdriver_config.browser_args {
        args.extend(custom_args.clone());
    }

    // Add headless argument if needed
    if webdriver_config.headless && !args.iter().any(|a| a.contains("headless")) {
        args.push("--headless".to_string());
    }

    // Add user agent
    if let Some(ref ua) = webdriver_config.user_agent {
        args.push(format!("--user-agent={}", ua));
    } else if let Some(ref ua) = config.user_agent {
        args.push(format!("--user-agent={}", ua));
    }

    // Add proxy
    if let Some(ref proxy) = webdriver_config.proxy {
        args.push(format!("--proxy-server={}", proxy));
    }

    // Add viewport
    if let (Some(width), Some(height)) = (
        webdriver_config.viewport_width,
        webdriver_config.viewport_height,
    ) {
        args.push(format!("--window-size={},{}", width, height));
    }

    // Add all args to capabilities
    for arg in args {
        if let Err(e) = caps.add_arg(&arg) {
            log::warn!("Failed to add Edge arg '{}': {:?}", arg, e);
        }
    }

    Some(caps.into())
}

/// Setup WebDriver events and stealth mode.
/// Injects spider_fingerprint stealth scripts to:
/// - Hide navigator.webdriver property
/// - Spoof navigator.deviceMemory (WebDriver doesn't have CDP for this)
/// - Clean up CDP markers (cdc_, $cdc_)
/// - Clean up Selenium markers
#[cfg(feature = "webdriver_stealth")]
pub async fn setup_driver_events(driver: &WebDriver, _config: &Configuration) {
    // Inject webdriver hiding script
    if let Err(e) = driver.execute(HIDE_WEBDRIVER, vec![]).await {
        log::warn!("Failed to inject webdriver hiding script: {:?}", e);
    }

    // Spoof device memory (WebDriver doesn't have CDP emulation for this)
    // Use realistic values: 4 or 8 GB for desktop
    let device_memory_script = spoof_device_memory(8);
    if let Err(e) = driver.execute(&device_memory_script, vec![]).await {
        log::warn!("Failed to inject device memory script: {:?}", e);
    }

    // Clean up CDP markers (cdc_, $cdc_, etc.)
    if let Err(e) = driver.execute(CLEANUP_CDP_MARKERS, vec![]).await {
        log::warn!("Failed to inject CDP marker cleanup script: {:?}", e);
    }

    // Clean up Selenium-specific markers
    if let Err(e) = driver.execute(HIDE_SELENIUM_MARKERS, vec![]).await {
        log::warn!("Failed to inject Selenium marker cleanup script: {:?}", e);
    }
}

/// Setup WebDriver events (no-op without stealth feature).
#[cfg(not(feature = "webdriver_stealth"))]
pub async fn setup_driver_events(_driver: &WebDriver, _config: &Configuration) {
    // No stealth injection without the feature
}

/// Attempt to navigate to a URL.
pub async fn attempt_navigation(
    url: &str,
    driver: &WebDriver,
    timeout: &Option<Duration>,
) -> Result<(), WebDriverError> {
    let nav_future = driver.goto(url);

    match timeout {
        Some(t) => match tokio::time::timeout(*t, nav_future).await {
            Ok(result) => result,
            Err(_) => Err(WebDriverError::Timeout("Navigation timeout".to_string())),
        },
        None => nav_future.await,
    }
}

/// Get the page content (HTML source).
pub async fn get_page_content(driver: &WebDriver) -> Result<String, WebDriverError> {
    driver.source().await
}

/// Get the current URL.
pub async fn get_current_url(driver: &WebDriver) -> Result<String, WebDriverError> {
    driver.current_url().await.map(|u| u.to_string())
}

/// Get the page title.
pub async fn get_page_title(driver: &WebDriver) -> Result<String, WebDriverError> {
    driver.title().await
}

/// Take a screenshot of the page.
#[cfg(feature = "webdriver_screenshot")]
pub async fn take_screenshot(driver: &WebDriver) -> Result<Vec<u8>, WebDriverError> {
    driver.screenshot_as_png().await
}

/// Take a screenshot (stub without feature).
#[cfg(not(feature = "webdriver_screenshot"))]
pub async fn take_screenshot(_driver: &WebDriver) -> Result<Vec<u8>, WebDriverError> {
    Err(WebDriverError::FatalError(
        "Screenshot feature not enabled".to_string(),
    ))
}

/// Execute JavaScript on the page and return the result as a JSON value.
pub async fn execute_script(
    driver: &WebDriver,
    script: &str,
) -> Result<serde_json::Value, WebDriverError> {
    let result = driver.execute(script, vec![]).await?;
    Ok(result.json().clone())
}

/// Wait for an element to be present.
pub async fn wait_for_element(
    driver: &WebDriver,
    selector: &str,
    timeout: Duration,
) -> Result<WebElement, WebDriverError> {
    driver
        .query(By::Css(selector))
        .wait(timeout, Duration::from_millis(100))
        .first()
        .await
}

/// Close the WebDriver session (consumes the driver).
pub async fn close_driver(driver: WebDriver) {
    if let Err(e) = driver.quit().await {
        log::warn!("Failed to close WebDriver session: {:?}", e);
    }
}

/// Get a random viewport for stealth purposes.
#[cfg(feature = "real_browser")]
pub fn get_random_webdriver_viewport() -> (u32, u32) {
    use super::chrome_viewport::get_random_viewport;
    let vp = get_random_viewport();
    (vp.width, vp.height)
}

/// Get a default viewport.
#[cfg(not(feature = "real_browser"))]
pub fn get_random_webdriver_viewport() -> (u32, u32) {
    (1920, 1080)
}

// ============================================================================
// WebDriver Automation Support
// ============================================================================

use crate::features::chrome_common::WebAutomation;

/// Run a single WebAutomation action on the WebDriver.
pub async fn run_automation(driver: &WebDriver, action: &WebAutomation) -> bool {
    let mut valid = false;

    match action {
        WebAutomation::Evaluate(js) => {
            valid = driver.execute(js.as_str(), vec![]).await.is_ok();
        }
        WebAutomation::Click(selector) => {
            if let Ok(ele) = driver.find(By::Css(selector)).await {
                valid = ele.click().await.is_ok();
            }
        }
        WebAutomation::ClickAll(selector) => {
            if let Ok(eles) = driver.find_all(By::Css(selector)).await {
                for ele in eles {
                    valid = ele.click().await.is_ok();
                }
            }
        }
        WebAutomation::ClickPoint { x, y } => {
            // WebDriver doesn't have direct click at coordinates, use JS
            let js = format!("document.elementFromPoint({}, {})?.click()", x, y);
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::ClickHold { selector, hold_ms } => {
            // Simulate with JS since WebDriver doesn't have native click-hold
            let js = format!(
                r#"
                const el = document.querySelector('{}');
                if (el) {{
                    const evt = new MouseEvent('mousedown', {{ bubbles: true }});
                    el.dispatchEvent(evt);
                    await new Promise(r => setTimeout(r, {}));
                    el.dispatchEvent(new MouseEvent('mouseup', {{ bubbles: true }}));
                }}
                "#,
                selector.replace('\'', "\\'"),
                hold_ms
            );
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::ClickHoldPoint { x, y, hold_ms } => {
            let js = format!(
                r#"
                const el = document.elementFromPoint({}, {});
                if (el) {{
                    el.dispatchEvent(new MouseEvent('mousedown', {{ bubbles: true }}));
                    await new Promise(r => setTimeout(r, {}));
                    el.dispatchEvent(new MouseEvent('mouseup', {{ bubbles: true }}));
                }}
                "#,
                x, y, hold_ms
            );
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::ClickDrag {
            from,
            to,
            modifier: _,
        } => {
            // Simulate drag with JS
            let js = format!(
                r#"
                const fromEl = document.querySelector('{}');
                const toEl = document.querySelector('{}');
                if (fromEl && toEl) {{
                    const fromRect = fromEl.getBoundingClientRect();
                    const toRect = toEl.getBoundingClientRect();
                    fromEl.dispatchEvent(new MouseEvent('mousedown', {{ bubbles: true, clientX: fromRect.x, clientY: fromRect.y }}));
                    toEl.dispatchEvent(new MouseEvent('mousemove', {{ bubbles: true, clientX: toRect.x, clientY: toRect.y }}));
                    toEl.dispatchEvent(new MouseEvent('mouseup', {{ bubbles: true, clientX: toRect.x, clientY: toRect.y }}));
                }}
                "#,
                from.replace('\'', "\\'"),
                to.replace('\'', "\\'")
            );
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::ClickDragPoint {
            from_x,
            from_y,
            to_x,
            to_y,
            modifier: _,
        } => {
            let js = format!(
                r#"
                const fromEl = document.elementFromPoint({}, {});
                const toEl = document.elementFromPoint({}, {});
                if (fromEl) {{
                    fromEl.dispatchEvent(new MouseEvent('mousedown', {{ bubbles: true, clientX: {}, clientY: {} }}));
                    (toEl || fromEl).dispatchEvent(new MouseEvent('mousemove', {{ bubbles: true, clientX: {}, clientY: {} }}));
                    (toEl || fromEl).dispatchEvent(new MouseEvent('mouseup', {{ bubbles: true, clientX: {}, clientY: {} }}));
                }}
                "#,
                from_x, from_y, to_x, to_y, from_x, from_y, to_x, to_y, to_x, to_y
            );
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::ClickAllClickable() => {
            let clickable_selector =
                "a, button, input[type='button'], input[type='submit'], [onclick], [role='button']";
            if let Ok(eles) = driver.find_all(By::Css(clickable_selector)).await {
                for ele in eles {
                    let _ = ele.click().await;
                    valid = true;
                }
            }
        }
        WebAutomation::Wait(ms) => {
            tokio::time::sleep(Duration::from_millis(*ms)).await;
            valid = true;
        }
        WebAutomation::WaitForNavigation => {
            // Wait for page to finish loading
            let js = r#"
                return new Promise(resolve => {
                    if (document.readyState === 'complete') {
                        resolve(true);
                    } else {
                        window.addEventListener('load', () => resolve(true));
                    }
                });
            "#;
            valid = driver.execute(js, vec![]).await.is_ok();
        }
        WebAutomation::WaitForLoad { timeout } => {
            let js = r#"
                return new Promise(resolve => {
                    if (document.readyState === 'complete') {
                        resolve(true);
                    } else {
                        window.addEventListener('load', () => resolve(true));
                    }
                });
            "#;
            let dur = Duration::from_millis(*timeout);
            valid = tokio::time::timeout(dur, driver.execute(js, vec![]))
                .await
                .map(|r| r.is_ok())
                .unwrap_or(false);
        }
        WebAutomation::WaitForNetworkIdle { timeout } => {
            // Webdriver doesn't have native network idle — approximate via JS
            let js = format!(
                r#"return new Promise(r => {{
                    let t; const c = () => {{ clearTimeout(t); t = setTimeout(() => r(true), 500); }};
                    const o = new PerformanceObserver(l => {{ l.getEntries().forEach(c); }});
                    o.observe({{ type: 'resource', buffered: false }});
                    c(); setTimeout(() => {{ o.disconnect(); r(true); }}, {});
                }})"#,
                timeout
            );
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::WaitForNetworkAlmostIdle { timeout } => {
            // Same approach as NetworkIdle for webdriver — no CDP lifecycle
            let js = format!(
                r#"return new Promise(r => {{
                    let t; const c = () => {{ clearTimeout(t); t = setTimeout(() => r(true), 500); }};
                    const o = new PerformanceObserver(l => {{ l.getEntries().forEach(c); }});
                    o.observe({{ type: 'resource', buffered: false }});
                    c(); setTimeout(() => {{ o.disconnect(); r(true); }}, {});
                }})"#,
                timeout
            );
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::WaitForDom { selector, timeout } => {
            let timeout_duration = Duration::from_millis(*timeout as u64);
            if let Some(sel) = selector {
                valid = driver
                    .query(By::Css(sel))
                    .wait(timeout_duration, Duration::from_millis(100))
                    .first()
                    .await
                    .is_ok();
            } else {
                // Wait for DOM to be stable
                tokio::time::sleep(timeout_duration).await;
                valid = true;
            }
        }
        WebAutomation::WaitFor(selector) => {
            valid = driver
                .query(By::Css(selector))
                .wait(Duration::from_secs(60), Duration::from_millis(100))
                .first()
                .await
                .is_ok();
        }
        WebAutomation::WaitForWithTimeout { selector, timeout } => {
            valid = driver
                .query(By::Css(selector))
                .wait(Duration::from_millis(*timeout), Duration::from_millis(100))
                .first()
                .await
                .is_ok();
        }
        WebAutomation::WaitForAndClick(selector) => {
            if let Ok(ele) = driver
                .query(By::Css(selector))
                .wait(Duration::from_secs(60), Duration::from_millis(100))
                .first()
                .await
            {
                valid = ele.click().await.is_ok();
            }
        }
        WebAutomation::ScrollX(px) => {
            let js = format!("window.scrollBy({}, 0)", px);
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::ScrollY(px) => {
            let js = format!("window.scrollBy(0, {})", px);
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::InfiniteScroll(duration) => {
            let timeout = (*duration).min(300); // Cap at 5 minutes
            let js = format!(
                r#"
                const endTime = Date.now() + {} * 1000;
                const scroll = () => {{
                    window.scrollBy(0, window.innerHeight);
                    if (Date.now() < endTime) {{
                        setTimeout(scroll, 500);
                    }}
                }};
                scroll();
                "#,
                timeout
            );
            valid = driver.execute(&js, vec![]).await.is_ok();
            // Wait for the scroll to complete
            tokio::time::sleep(Duration::from_secs(timeout as u64)).await;
        }
        WebAutomation::Fill { selector, value } => {
            if let Ok(ele) = driver.find(By::Css(selector)).await {
                // Clear and type
                if ele.clear().await.is_ok() {
                    valid = ele.send_keys(value).await.is_ok();
                }
            }
        }
        WebAutomation::Type { value, modifier: _ } => {
            // Type into the active element
            let js = format!(
                r#"
                const el = document.activeElement;
                if (el && (el.tagName === 'INPUT' || el.tagName === 'TEXTAREA' || el.isContentEditable)) {{
                    el.value = (el.value || '') + '{}';
                    el.dispatchEvent(new Event('input', {{ bubbles: true }}));
                }}
                "#,
                value.replace('\'', "\\'").replace('\n', "\\n")
            );
            valid = driver.execute(&js, vec![]).await.is_ok();
        }
        WebAutomation::Screenshot {
            full_page: _,
            omit_background: _,
            output,
        } => {
            #[cfg(feature = "webdriver_screenshot")]
            {
                if let Ok(png_data) = driver.screenshot_as_png().await {
                    valid = crate::utils::uring_fs::write_file(output.clone(), png_data)
                        .await
                        .is_ok();
                }
            }
            #[cfg(not(feature = "webdriver_screenshot"))]
            {
                let _ = output;
                log::warn!("Screenshot feature not enabled");
            }
        }
        WebAutomation::ValidateChain => {
            // This is a control flow marker, always returns current valid state
            valid = true;
        }
    }

    valid
}

/// Run a list of WebAutomation actions on the WebDriver.
pub async fn run_automation_scripts(driver: &WebDriver, scripts: &[WebAutomation]) -> bool {
    let mut valid = false;

    for script in scripts {
        if script == &WebAutomation::ValidateChain && !valid {
            break;
        }
        match tokio::time::timeout(Duration::from_secs(60), run_automation(driver, script)).await {
            Ok(result) => valid = result,
            Err(_) => {
                log::warn!("Automation script timed out: {:?}", script.name());
                valid = false;
            }
        }
    }

    valid
}

/// Run execution scripts (JavaScript) for a specific URL.
pub async fn run_execution_scripts(
    driver: &WebDriver,
    url: &str,
    execution_scripts: &Option<crate::features::chrome_common::ExecutionScripts>,
) {
    if let Some(scripts) = execution_scripts {
        if let Some(js) = scripts.search(url) {
            if let Err(e) = driver.execute(js.as_str(), vec![]).await {
                log::warn!("Failed to execute script for {}: {:?}", url, e);
            }
        }
    }
}

/// Run automation scripts for a specific URL.
pub async fn run_url_automation_scripts(
    driver: &WebDriver,
    url: &str,
    automation_scripts: &Option<crate::features::chrome_common::AutomationScripts>,
) -> bool {
    if let Some(scripts) = automation_scripts {
        if let Some(actions) = scripts.search(url) {
            return run_automation_scripts(driver, actions).await;
        }
    }
    true
}