allwright_surface_web/

lib.rs

mod capture;
pub use capture::capture;
mod dialogs;
mod frames;
pub use frames::resolve_frame;
mod accessibility;
mod accessibility_yaml;
pub use accessibility::{accessibility_snapshot, accessibility_snapshot_with_mode};

use allwright_plugin_sdk::{
    ALLWRIGHT_PLUGIN_API_VERSION, AutomationSessionInfo, BrowserKind, BrowserLaunchInfo,
    BrowserSessionHandle, ChromeLaunchInfo, ChromeTabInfo, ChromiumBidiMapperInfo, ClickInfo,
    DownloadInfo, DownloadSavedInfo, ElementCountInfo, FileChooserFilesSetInfo, FileChooserInfo,
    FillInfo, FocusInfo, HighlightElementsInfo, HookRegistration, HookResult, HookType, HoverInfo,
    PageInfo, PageSessionHandle, PluginCommand, PluginEnvelope, PluginResult, PressKeyInfo,
    ScreenshotInfo, SurfaceFamily, SurfacePlugin, SurfacePluginDescriptor, TabNavigationInfo,
    TextInfo, WaitForSelectorInfo,
};
use base64::Engine as _;
use std::borrow::Cow;
use std::ffi::{CStr, CString, c_char};
use std::fs;
use std::future::Future;
use std::net::TcpListener;
use std::path::PathBuf;
use std::process::{Command, Stdio};
use std::thread;
use std::time::{Duration as StdDuration, SystemTime, UNIX_EPOCH};

use futures_util::{SinkExt, StreamExt};
use serde_json::{Value, json};
use std::collections::{HashMap, VecDeque};
use std::sync::OnceLock;
use std::sync::atomic::{AtomicU64, Ordering};
use tokio::net::TcpStream;
use tokio::sync::{Mutex, MutexGuard};
use tokio::time::{Duration, sleep, timeout};
use tokio_tungstenite::MaybeTlsStream;
use tokio_tungstenite::WebSocketStream;
use tokio_tungstenite::connect_async;
use tokio_tungstenite::tungstenite::Message;

#[derive(Debug, Clone, PartialEq)]
struct DiscoveredElements {
    count: u32,
    first_center: Option<ElementCenter>,
}

#[derive(Debug, Clone, PartialEq)]
struct ElementCenter {
    x: f64,
    y: f64,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum SelectorKind {
    Structured,
    Css,
    XPath,
}

#[derive(Debug, Clone, Copy, Default)]
struct DiscoveryRequirements {
    require_match: bool,
    require_visible: bool,
    require_pointer_interactable: bool,
    require_editable: bool,
    focus_first_match: bool,
    require_focus: bool,
}

#[derive(Debug, Clone)]
struct ParsedSelector<'a> {
    value: Cow<'a, str>,
    kind: SelectorKind,
}

#[derive(Debug, Clone, Copy)]
struct ExplicitSelectorPrefix {
    kind: SelectorKind,
    prefix_len: usize,
}

const CHROMIUM_BIDI_NPM_VERSION: &str = "17.0.2";
const CHROMIUM_BIDI_MAPPER_BUNDLE: &str =
    include_str!("../third_party/chromium-bidi/17.0.2/mapperTab.js");

fn suppressed_file_dialog_behavior() -> Value {
    json!({ "file": "dismiss", "alert": "ignore", "confirm": "ignore", "prompt": "ignore" })
}

struct FirefoxSessionState {
    process_id: u32,
    connection: BidiConnection,
}

static FIREFOX_SESSION_REGISTRY: OnceLock<Mutex<HashMap<String, FirefoxSessionState>>> =
    OnceLock::new();
static FILE_CHOOSER_HOOK_REGISTRY: OnceLock<Mutex<HashMap<String, FileChooserHookState>>> =
    OnceLock::new();
static FILE_CHOOSER_REGISTRY: OnceLock<Mutex<HashMap<String, FileChooserState>>> = OnceLock::new();
static FILE_CHOOSER_COUNTER: AtomicU64 = AtomicU64::new(1);
static DOWNLOAD_HOOK_REGISTRY: OnceLock<Mutex<HashMap<String, DownloadHookState>>> =
    OnceLock::new();
static DOWNLOAD_REGISTRY: OnceLock<Mutex<HashMap<String, DownloadState>>> = OnceLock::new();
static DOWNLOAD_COUNTER: AtomicU64 = AtomicU64::new(1);
static PLUGIN_RUNTIME: OnceLock<tokio::runtime::Runtime> = OnceLock::new();

enum FileChooserHookState {
    Chromium {
        connection: CdpConnection,
        mapper_session_id: String,
        subscription_id: String,
        mapper_target_id: String,
    },
    Firefox {
        connection_id: String,
        subscription_id: String,
    },
}

#[derive(Clone)]
enum FileChooserState {
    Chromium {
        cdp_websocket_url: String,
        mapper_target_id: String,
        browsing_context_id: String,
        shared_id: String,
    },
    Firefox {
        connection_id: String,
        browsing_context_id: String,
        shared_id: String,
    },
}

enum DownloadHookState {
    Chromium {
        connection: CdpConnection,
        mapper_session_id: String,
        subscription_id: String,
        target_id: String,
        browsing_context_id: String,
        download_dir: PathBuf,
    },
    Firefox {
        connection_id: String,
        subscription_id: String,
        browsing_context_id: String,
        download_dir: PathBuf,
    },
}

enum DownloadState {
    Chromium {
        connection: CdpConnection,
        mapper_session_id: String,
        subscription_id: String,
        cdp_websocket_url: String,
        target_id: String,
        protocol_download_id: String,
        download_dir: PathBuf,
        suggested_filename: String,
    },
    Firefox {
        connection_id: String,
        subscription_id: String,
        browsing_context_id: String,
        protocol_download_id: String,
        download_dir: PathBuf,
        suggested_filename: String,
    },
}

#[derive(Debug, Clone, Copy, Default)]
pub struct WebPlugin;

impl SurfacePlugin for WebPlugin {
    fn descriptor(&self) -> SurfacePluginDescriptor {
        SurfacePluginDescriptor {
            id: "web",
            family: SurfaceFamily::Web,
            version: env!("CARGO_PKG_VERSION"),
            description: "Web surface plugin for the allwright engine.",
        }
    }
}

pub fn open_chrome_window(chrome_binary: Option<&str>) -> Result<ChromeLaunchInfo, String> {
    launch_chrome_for_platform(chrome_binary.filter(|value| !value.trim().is_empty()))
}

pub async fn discover_initial_tab(cdp_websocket_url: &str) -> Result<ChromeTabInfo, String> {
    let (pages, _) = chromium_top_level_pages(cdp_websocket_url, None).await?;
    let page = pages.into_iter().next().ok_or_else(|| {
        "WebDriver BiDi returned no initial Chromium browsing context".to_string()
    })?;
    Ok(ChromeTabInfo {
        note: "tracked initial Chromium tab via WebDriver BiDi".to_string(),
        target_id: page.id,
    })
}

pub async fn open_chrome_tab(cdp_websocket_url: &str) -> Result<ChromeTabInfo, String> {
    let mut cdp = CdpConnection::connect(cdp_websocket_url).await?;
    let mapper = cdp.ensure_chromium_bidi_mapper(None).await?;
    let created = cdp
        .send_bidi_command(
            &mapper.mapper_session_id,
            &json!({
                "id": 2,
                "method": "browsingContext.create",
                "params": { "type": "tab", "background": false },
            }),
        )
        .await?;
    let target_id = required_string(&created, "/result/context")?;

    Ok(ChromeTabInfo {
        note: "opened Chromium tab via WebDriver BiDi".to_string(),
        target_id,
    })
}

pub fn close_browser_process(process_id: u32) -> Result<(), String> {
    block_on_plugin_future(async {
        dialogs::close_browser(process_id).await;
        let registry = firefox_session_registry();
        let mut sessions = registry.lock().await;
        sessions.retain(|_, state| state.process_id != process_id);
        Ok::<(), String>(())
    })?;
    close_browser_process_for_platform(process_id)
}

pub fn launch_browser(
    browser_kind: BrowserKind,
    browser_binary: Option<&str>,
) -> Result<BrowserLaunchInfo, String> {
    let result = match browser_kind {
        BrowserKind::Chromium => {
            let launch = open_chrome_window(browser_binary)?;
            let (initial_page, mapper) = block_on_plugin_future(async {
                let (pages, mapper) =
                    chromium_top_level_pages(&launch.cdp_websocket_url, None).await?;
                let page = pages.into_iter().next().ok_or_else(|| {
                    "WebDriver BiDi returned no initial Chromium browsing context".to_string()
                })?;
                Ok::<_, String>((page, mapper))
            })?;
            Ok(BrowserLaunchInfo {
                browser_kind,
                browser: launch.browser,
                note: launch.note,
                user_data_dir: launch.user_data_dir,
                process_id: launch.process_id,
                browser_session: BrowserSessionHandle::Chromium {
                    cdp_websocket_url: launch.cdp_websocket_url,
                },
                initial_page: PageInfo {
                    note: "tracked initial Chromium tab via WebDriver BiDi".to_string(),
                    page_session: PageSessionHandle::Chromium {
                        target_id: initial_page.id.clone(),
                        browsing_context_id: Some(initial_page.id),
                        mapper_target_id: Some(mapper.mapper_target_id),
                    },
                },
            })
        }
        BrowserKind::Firefox => launch_firefox_with_bidi(browser_binary),
    }?;
    block_on_plugin_future(async {
        dialogs::track_browser(&result).await;
        Ok::<(), String>(())
    })?;
    Ok(result)
}

pub async fn open_page(browser_session: &BrowserSessionHandle) -> Result<PageInfo, String> {
    match browser_session {
        BrowserSessionHandle::Chromium { cdp_websocket_url } => {
            let mut cdp = CdpConnection::connect(cdp_websocket_url).await?;
            let mapper = cdp.ensure_chromium_bidi_mapper(None).await?;
            let created = cdp
                .send_bidi_command(
                    &mapper.mapper_session_id,
                    &json!({
                        "id": 2,
                        "method": "browsingContext.create",
                        "params": { "type": "tab", "background": false },
                    }),
                )
                .await?;
            let context = required_string(&created, "/result/context")?;
            Ok(PageInfo {
                note: "opened Chromium tab via WebDriver BiDi".to_string(),
                page_session: PageSessionHandle::Chromium {
                    target_id: context.clone(),
                    browsing_context_id: Some(context),
                    mapper_target_id: Some(mapper.mapper_target_id),
                },
            })
        }
        BrowserSessionHandle::Firefox { connection_id, .. } => {
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let created = bidi
                .send_command(
                    "browsingContext.create",
                    json!({
                        "type": "tab",
                        "background": false,
                    }),
                )
                .await?;
            Ok(PageInfo {
                note: "opened Firefox tab via native WebDriver BiDi".to_string(),
                page_session: PageSessionHandle::Firefox {
                    browsing_context_id: required_string(&created, "/result/context")?,
                },
            })
        }
    }
}

#[derive(Debug)]
struct TopLevelPage {
    id: String,
    opener_id: Option<String>,
}

async fn chromium_top_level_pages(
    cdp_websocket_url: &str,
    mapper_target_id: Option<&str>,
) -> Result<(Vec<TopLevelPage>, MapperConnectionInfo), String> {
    let mut cdp = CdpConnection::connect(cdp_websocket_url).await?;
    let mapper = cdp.ensure_chromium_bidi_mapper(mapper_target_id).await?;
    let tree = cdp
        .send_bidi_command(
            &mapper.mapper_session_id,
            &json!({
                "id": 2,
                "method": "browsingContext.getTree",
                "params": { "maxDepth": 0 },
            }),
        )
        .await?;
    let pages = tree
        .pointer("/result/contexts")
        .and_then(Value::as_array)
        .into_iter()
        .flatten()
        .filter_map(|context| {
            context
                .get("context")
                .and_then(Value::as_str)
                .map(|id| TopLevelPage {
                    id: id.to_string(),
                    opener_id: context
                        .get("originalOpener")
                        .and_then(Value::as_str)
                        .map(str::to_string),
                })
        })
        .collect();
    Ok((pages, mapper))
}

async fn top_level_pages(
    browser_session: &BrowserSessionHandle,
    mapper_target_id: Option<&str>,
) -> Result<Vec<TopLevelPage>, String> {
    match browser_session {
        BrowserSessionHandle::Chromium { cdp_websocket_url } => {
            chromium_top_level_pages(cdp_websocket_url, mapper_target_id)
                .await
                .map(|(pages, _)| pages)
        }
        BrowserSessionHandle::Firefox { connection_id, .. } => {
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let tree = bidi
                .send_command(
                    "browsingContext.getTree",
                    json!({
                        "maxDepth": 0,
                    }),
                )
                .await?;
            Ok(tree
                .pointer("/result/contexts")
                .and_then(Value::as_array)
                .into_iter()
                .flatten()
                .filter_map(|context| {
                    context
                        .get("context")
                        .and_then(Value::as_str)
                        .map(|id| TopLevelPage {
                            id: id.to_string(),
                            opener_id: context
                                .get("originalOpener")
                                .and_then(Value::as_str)
                                .map(str::to_string),
                        })
                })
                .collect())
        }
    }
}

pub async fn register_hook(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    hook_type: HookType,
) -> Result<HookRegistration, String> {
    if hook_type == HookType::Dialog {
        return dialogs::register(browser_session, page_session).await;
    }
    let mapper_target_id = match page_session {
        PageSessionHandle::Chromium {
            mapper_target_id, ..
        } => mapper_target_id.as_deref(),
        PageSessionHandle::Firefox { .. } => None,
    };
    let source_page_id = match (browser_session, page_session) {
        (BrowserSessionHandle::Chromium { .. }, PageSessionHandle::Chromium { target_id, .. }) => {
            target_id
        }
        (
            BrowserSessionHandle::Firefox { .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => browsing_context_id,
        _ => return Err("page session does not belong to the hook browser backend".to_string()),
    };
    let state = match hook_type {
        HookType::Dialog => unreachable!(),
        HookType::NewPage => json!({
            "hook_type": "new_page",
            "source_page_id": source_page_id,
            "mapper_target_id": mapper_target_id,
            "existing_page_ids": top_level_pages(browser_session, mapper_target_id).await?.into_iter().map(|page| page.id).collect::<Vec<_>>(),
        }),
        HookType::FileChooser => {
            let registration_id = next_file_chooser_id("file-chooser-hook");
            let hook_state = match (browser_session, page_session) {
                (
                    BrowserSessionHandle::Chromium { cdp_websocket_url },
                    PageSessionHandle::Chromium {
                        browsing_context_id,
                        mapper_target_id,
                        ..
                    },
                ) => {
                    let (mut connection, mapper, context) = chromium_bidi_session(
                        cdp_websocket_url,
                        mapper_target_id.as_deref(),
                        browsing_context_id.as_deref(),
                    )
                    .await?;
                    let response = connection
                        .send_bidi_command(
                            &mapper.mapper_session_id,
                            &json!({
                                "id": 7001,
                                "method": "session.subscribe",
                                "params": {
                                    "events": ["input.fileDialogOpened"],
                                    "contexts": [context],
                                },
                            }),
                        )
                        .await?;
                    FileChooserHookState::Chromium {
                        connection,
                        mapper_session_id: mapper.mapper_session_id,
                        subscription_id: required_string(&response, "/result/subscription")?,
                        mapper_target_id: mapper.mapper_target_id,
                    }
                }
                (
                    BrowserSessionHandle::Firefox { connection_id, .. },
                    PageSessionHandle::Firefox {
                        browsing_context_id,
                    },
                ) => {
                    let mut sessions = firefox_session_guard(connection_id).await?;
                    let response = sessions
                        .get_mut(connection_id)
                        .expect("Firefox session guard validated connection id")
                        .connection
                        .send_command(
                            "session.subscribe",
                            json!({
                                "events": ["input.fileDialogOpened"],
                                "contexts": [browsing_context_id],
                            }),
                        )
                        .await?;
                    FileChooserHookState::Firefox {
                        connection_id: connection_id.clone(),
                        subscription_id: required_string(&response, "/result/subscription")?,
                    }
                }
                _ => unreachable!("hook backend was validated above"),
            };
            file_chooser_hook_registry()
                .lock()
                .await
                .insert(registration_id.clone(), hook_state);
            json!({
                "hook_type": "file_chooser",
                "registration_id": registration_id,
                "source_page_id": source_page_id,
            })
        }
        HookType::Download => {
            let registration_id = next_download_id("download-hook");
            let download_dir = create_download_dir()?;
            let hook_state = match (browser_session, page_session) {
                (
                    BrowserSessionHandle::Chromium { cdp_websocket_url },
                    PageSessionHandle::Chromium {
                        target_id,
                        browsing_context_id,
                        mapper_target_id,
                    },
                ) => {
                    let (mut connection, mapper, context) = chromium_bidi_session(
                        cdp_websocket_url,
                        mapper_target_id.as_deref(),
                        browsing_context_id.as_deref(),
                    )
                    .await?;
                    let response = connection
                        .send_bidi_command(
                            &mapper.mapper_session_id,
                            &json!({
                                "id": 7201,
                                "method": "session.subscribe",
                                "params": {
                                    "events": [
                                        "browsingContext.downloadWillBegin",
                                        "browsingContext.downloadEnd"
                                    ],
                                    "contexts": [context],
                                },
                            }),
                        )
                        .await?;
                    connection
                        .send_bidi_command(
                            &mapper.mapper_session_id,
                            &json!({
                                "id": 7202,
                                "method": "browser.setDownloadBehavior",
                                "params": {
                                    "downloadBehavior": {
                                        "type": "allowed",
                                        "destinationFolder": download_dir.to_string_lossy(),
                                    }
                                },
                            }),
                        )
                        .await?;
                    DownloadHookState::Chromium {
                        connection,
                        mapper_session_id: mapper.mapper_session_id,
                        subscription_id: required_string(&response, "/result/subscription")?,
                        target_id: target_id.clone(),
                        browsing_context_id: context,
                        download_dir,
                    }
                }
                (
                    BrowserSessionHandle::Firefox { connection_id, .. },
                    PageSessionHandle::Firefox {
                        browsing_context_id,
                    },
                ) => {
                    let mut sessions = firefox_session_guard(connection_id).await?;
                    let bidi = &mut sessions
                        .get_mut(connection_id)
                        .expect("Firefox session guard validated connection id")
                        .connection;
                    let response = bidi
                        .send_command(
                            "session.subscribe",
                            json!({
                                "events": [
                                    "browsingContext.downloadWillBegin",
                                    "browsingContext.downloadEnd"
                                ],
                                "contexts": [browsing_context_id],
                            }),
                        )
                        .await?;
                    bidi.send_command(
                        "browser.setDownloadBehavior",
                        json!({
                            "downloadBehavior": {
                                "type": "allowed",
                                "destinationFolder": download_dir.to_string_lossy(),
                            }
                        }),
                    )
                    .await?;
                    DownloadHookState::Firefox {
                        connection_id: connection_id.clone(),
                        subscription_id: required_string(&response, "/result/subscription")?,
                        browsing_context_id: browsing_context_id.clone(),
                        download_dir,
                    }
                }
                _ => unreachable!("hook backend was validated above"),
            };
            download_hook_registry()
                .lock()
                .await
                .insert(registration_id.clone(), hook_state);
            json!({
                "hook_type": "download",
                "registration_id": registration_id,
                "source_page_id": source_page_id,
            })
        }
    };
    Ok(HookRegistration {
        opaque_state: state.to_string(),
    })
}

pub async fn poll_hook(
    browser_session: &BrowserSessionHandle,
    registration: &HookRegistration,
) -> Result<HookResult, String> {
    let state: Value = serde_json::from_str(&registration.opaque_state)
        .map_err(|error| format!("failed to decode web hook state: {error}"))?;
    match state.get("hook_type").and_then(Value::as_str) {
        Some("dialog") => dialogs::poll(browser_session, &state).await,
        Some("new_page") => poll_new_page_hook(browser_session, &state).await,
        Some("file_chooser") => poll_file_chooser_hook(browser_session, &state).await,
        Some("download") => poll_download_hook(browser_session, &state).await,
        _ => Err("web hook state has an unsupported hook type".to_string()),
    }
}

async fn poll_new_page_hook(
    browser_session: &BrowserSessionHandle,
    state: &Value,
) -> Result<HookResult, String> {
    let existing_page_ids = state
        .get("existing_page_ids")
        .and_then(Value::as_array)
        .ok_or_else(|| "new page hook state is missing existing_page_ids".to_string())?;
    let source_page_id = state
        .get("source_page_id")
        .and_then(Value::as_str)
        .ok_or_else(|| "new page hook state is missing source_page_id".to_string())?;
    let mapper_target_id = state.get("mapper_target_id").and_then(Value::as_str);
    let page_id = top_level_pages(browser_session, mapper_target_id)
        .await?
        .into_iter()
        .find(|page| {
            !existing_page_ids
                .iter()
                .any(|existing| existing.as_str() == Some(&page.id))
                && page.opener_id.as_deref() == Some(source_page_id)
        })
        .ok_or_else(|| "new page hook is still waiting for a page".to_string())?;

    let page_session = match browser_session {
        BrowserSessionHandle::Chromium { .. } => PageSessionHandle::Chromium {
            target_id: page_id.id.clone(),
            browsing_context_id: Some(page_id.id),
            mapper_target_id: mapper_target_id.map(str::to_string),
        },
        BrowserSessionHandle::Firefox { .. } => PageSessionHandle::Firefox {
            browsing_context_id: page_id.id,
        },
    };

    Ok(HookResult::NewPage(PageInfo {
        note: "new page hook completed".to_string(),
        page_session,
    }))
}

async fn poll_file_chooser_hook(
    browser_session: &BrowserSessionHandle,
    state: &Value,
) -> Result<HookResult, String> {
    let registration_id = state
        .get("registration_id")
        .and_then(Value::as_str)
        .ok_or_else(|| "file chooser hook state is missing registration_id".to_string())?;
    let mut hooks = file_chooser_hook_registry().lock().await;
    let hook = hooks
        .get_mut(registration_id)
        .ok_or_else(|| "file chooser hook registration is no longer available".to_string())?;

    let (chooser_state, is_multiple) = match hook {
        FileChooserHookState::Chromium {
            connection,
            mapper_session_id,
            subscription_id,
            mapper_target_id,
        } => {
            let event = connection
                .poll_bidi_event(mapper_session_id, "input.fileDialogOpened")
                .await?
                .ok_or_else(|| "file chooser hook is still waiting for a dialog".to_string())?;
            let shared_id = required_string(&event, "/params/element/sharedId")?;
            let browsing_context_id = required_string(&event, "/params/context")?;
            let is_multiple = event
                .pointer("/params/multiple")
                .and_then(Value::as_bool)
                .unwrap_or(false);
            connection
                .send_bidi_command(
                    mapper_session_id,
                    &json!({
                        "id": 7002,
                        "method": "session.unsubscribe",
                        "params": { "subscriptions": [subscription_id] },
                    }),
                )
                .await?;
            let cdp_websocket_url = match browser_session {
                BrowserSessionHandle::Chromium { cdp_websocket_url } => cdp_websocket_url.clone(),
                _ => return Err("file chooser browser backend changed while waiting".to_string()),
            };
            (
                FileChooserState::Chromium {
                    cdp_websocket_url,
                    mapper_target_id: mapper_target_id.clone(),
                    browsing_context_id,
                    shared_id,
                },
                is_multiple,
            )
        }
        FileChooserHookState::Firefox {
            connection_id,
            subscription_id,
        } => {
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let event = bidi
                .poll_event("input.fileDialogOpened")
                .await?
                .ok_or_else(|| "file chooser hook is still waiting for a dialog".to_string())?;
            let shared_id = required_string(&event, "/params/element/sharedId")?;
            let browsing_context_id = required_string(&event, "/params/context")?;
            let is_multiple = event
                .pointer("/params/multiple")
                .and_then(Value::as_bool)
                .unwrap_or(false);
            bidi.send_command(
                "session.unsubscribe",
                json!({ "subscriptions": [subscription_id] }),
            )
            .await?;
            (
                FileChooserState::Firefox {
                    connection_id: connection_id.clone(),
                    browsing_context_id,
                    shared_id,
                },
                is_multiple,
            )
        }
    };

    hooks.remove(registration_id);
    drop(hooks);
    let file_chooser_id = next_file_chooser_id("file-chooser");
    file_chooser_registry()
        .lock()
        .await
        .insert(file_chooser_id.clone(), chooser_state);
    Ok(HookResult::FileChooser(FileChooserInfo {
        file_chooser_id,
        is_multiple,
        note: "file chooser hook completed".to_string(),
    }))
}

async fn poll_download_hook(
    browser_session: &BrowserSessionHandle,
    state: &Value,
) -> Result<HookResult, String> {
    let registration_id = state
        .get("registration_id")
        .and_then(Value::as_str)
        .ok_or_else(|| "download hook state is missing registration_id".to_string())?;
    let mut hooks = download_hook_registry().lock().await;
    let hook = hooks
        .get_mut(registration_id)
        .ok_or_else(|| "download hook registration is no longer available".to_string())?;

    let (download_state, protocol_download_id, url, suggested_filename) = match hook {
        DownloadHookState::Chromium {
            connection,
            mapper_session_id,
            subscription_id,
            target_id,
            browsing_context_id,
            download_dir,
        } => {
            let event = connection
                .poll_bidi_event(mapper_session_id, "browsingContext.downloadWillBegin")
                .await?
                .ok_or_else(|| "download hook is still waiting for a download".to_string())?;
            let event_context = required_string(&event, "/params/context")?;
            if &event_context != browsing_context_id {
                return Err("download started in a different browsing context".to_string());
            }
            let protocol_download_id = download_event_id(&event)?;
            let url = required_string(&event, "/params/url")?;
            let suggested_filename = required_string(&event, "/params/suggestedFilename")?;
            let cdp_websocket_url = match browser_session {
                BrowserSessionHandle::Chromium { cdp_websocket_url } => cdp_websocket_url.clone(),
                _ => return Err("download browser backend changed while waiting".to_string()),
            };
            (
                DownloadState::Chromium {
                    connection: std::mem::replace(
                        connection,
                        CdpConnection::connect(&cdp_websocket_url).await?,
                    ),
                    mapper_session_id: mapper_session_id.clone(),
                    subscription_id: subscription_id.clone(),
                    cdp_websocket_url,
                    target_id: target_id.clone(),
                    protocol_download_id: protocol_download_id.clone(),
                    download_dir: download_dir.clone(),
                    suggested_filename: suggested_filename.clone(),
                },
                protocol_download_id,
                url,
                suggested_filename,
            )
        }
        DownloadHookState::Firefox {
            connection_id,
            subscription_id,
            browsing_context_id,
            download_dir,
        } => {
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let event = bidi
                .poll_event("browsingContext.downloadWillBegin")
                .await?
                .ok_or_else(|| "download hook is still waiting for a download".to_string())?;
            let event_context = required_string(&event, "/params/context")?;
            if &event_context != browsing_context_id {
                return Err("download started in a different browsing context".to_string());
            }
            let protocol_download_id = download_event_id(&event)?;
            let url = required_string(&event, "/params/url")?;
            let suggested_filename = required_string(&event, "/params/suggestedFilename")?;
            (
                DownloadState::Firefox {
                    connection_id: connection_id.clone(),
                    subscription_id: subscription_id.clone(),
                    browsing_context_id: browsing_context_id.clone(),
                    protocol_download_id: protocol_download_id.clone(),
                    download_dir: download_dir.clone(),
                    suggested_filename: suggested_filename.clone(),
                },
                protocol_download_id,
                url,
                suggested_filename,
            )
        }
    };

    hooks.remove(registration_id);
    drop(hooks);
    let download_id = next_download_id("download");
    download_registry()
        .lock()
        .await
        .insert(download_id.clone(), download_state);
    Ok(HookResult::Download(DownloadInfo {
        download_id,
        url,
        suggested_filename,
        note: format!("download {protocol_download_id} started"),
    }))
}

pub async fn save_download(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    download_id: &str,
    path: &str,
) -> Result<DownloadSavedInfo, String> {
    let mut downloads = download_registry().lock().await;
    let download = downloads
        .get_mut(download_id)
        .ok_or_else(|| format!("unknown download {download_id}"))?;

    let (event, download_dir, suggested_filename) = match download {
        DownloadState::Chromium {
            connection,
            mapper_session_id,
            subscription_id,
            cdp_websocket_url,
            target_id,
            protocol_download_id,
            download_dir,
            suggested_filename,
            ..
        } => {
            match (browser_session, page_session) {
                (
                    BrowserSessionHandle::Chromium {
                        cdp_websocket_url: url,
                    },
                    PageSessionHandle::Chromium {
                        target_id: page_target_id,
                        ..
                    },
                ) if url == cdp_websocket_url && page_target_id == target_id => {}
                _ => return Err("download does not belong to this page session".to_string()),
            }
            let event = connection
                .poll_bidi_event(mapper_session_id, "browsingContext.downloadEnd")
                .await?
                .ok_or_else(|| "download is still in progress".to_string())?;
            if download_event_id(&event)? != *protocol_download_id {
                return Err("received completion for a different download".to_string());
            }
            connection
                .send_bidi_command(
                    mapper_session_id,
                    &json!({
                        "id": 7203,
                        "method": "session.unsubscribe",
                        "params": { "subscriptions": [subscription_id] },
                    }),
                )
                .await?;
            (event, download_dir.clone(), suggested_filename.clone())
        }
        DownloadState::Firefox {
            connection_id,
            subscription_id,
            browsing_context_id,
            protocol_download_id,
            download_dir,
            suggested_filename,
            ..
        } => {
            match (browser_session, page_session) {
                (
                    BrowserSessionHandle::Firefox {
                        connection_id: id, ..
                    },
                    PageSessionHandle::Firefox {
                        browsing_context_id: context,
                    },
                ) if id == connection_id && context == browsing_context_id => {}
                _ => return Err("download does not belong to this page session".to_string()),
            }
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let event = bidi
                .poll_event("browsingContext.downloadEnd")
                .await?
                .ok_or_else(|| "download is still in progress".to_string())?;
            if download_event_id(&event)? != *protocol_download_id {
                return Err("received completion for a different download".to_string());
            }
            bidi.send_command(
                "session.unsubscribe",
                json!({ "subscriptions": [subscription_id] }),
            )
            .await?;
            (event, download_dir.clone(), suggested_filename.clone())
        }
    };

    let status = required_string(&event, "/params/status")?;
    if status != "complete" {
        return Err(format!("download ended with status {status}"));
    }
    let source = event
        .pointer("/params/filepath")
        .and_then(Value::as_str)
        .map(PathBuf::from)
        .unwrap_or_else(|| download_dir.join(&suggested_filename));
    if !source.is_file() {
        return Err(format!(
            "completed download file is not available at {}",
            source.display()
        ));
    }
    let destination = PathBuf::from(path);
    let destination = if destination.is_absolute() {
        destination
    } else {
        std::env::current_dir()
            .map_err(|error| format!("resolve current directory: {error}"))?
            .join(destination)
    };
    let parent = destination.parent().ok_or_else(|| {
        format!(
            "download destination has no parent: {}",
            destination.display()
        )
    })?;
    if !parent.is_dir() {
        return Err(format!(
            "download destination directory does not exist: {}",
            parent.display()
        ));
    }
    fs::copy(&source, &destination).map_err(|error| {
        format!(
            "save download from {} to {}: {error}",
            source.display(),
            destination.display()
        )
    })?;
    let saved_path = fs::canonicalize(&destination)
        .map_err(|error| format!("resolve saved download {}: {error}", destination.display()))?;
    let size = fs::metadata(&saved_path)
        .map_err(|error| format!("read saved download metadata: {error}"))?
        .len();
    downloads.remove(download_id);
    let _ = fs::remove_dir_all(&download_dir);
    Ok(DownloadSavedInfo {
        download_id: download_id.to_string(),
        path: saved_path.to_string_lossy().to_string(),
        suggested_filename,
        size,
        note: "saved completed download".to_string(),
    })
}

pub async fn set_file_chooser_files(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    file_chooser_id: &str,
    files: &[String],
) -> Result<FileChooserFilesSetInfo, String> {
    let resolved_files = files
        .iter()
        .map(|file| {
            fs::canonicalize(file)
                .map(|path| path.to_string_lossy().to_string())
                .map_err(|error| format!("resolve upload path {file}: {error}"))
        })
        .collect::<Result<Vec<_>, _>>()?;
    let chooser = file_chooser_registry()
        .lock()
        .await
        .get(file_chooser_id)
        .cloned()
        .ok_or_else(|| format!("unknown file chooser {file_chooser_id}"))?;

    match (browser_session, page_session, chooser) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium { .. },
            FileChooserState::Chromium {
                cdp_websocket_url: chooser_url,
                mapper_target_id,
                browsing_context_id,
                shared_id,
            },
        ) if cdp_websocket_url == &chooser_url => {
            let (mut cdp, mapper, context) = chromium_bidi_session(
                cdp_websocket_url,
                Some(&mapper_target_id),
                Some(&browsing_context_id),
            )
            .await?;
            cdp.send_bidi_command(
                &mapper.mapper_session_id,
                &json!({
                    "id": 7003,
                    "method": "input.setFiles",
                    "params": {
                        "context": context,
                        "element": { "sharedId": shared_id },
                        "files": resolved_files,
                    },
                }),
            )
            .await?;
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox { .. },
            FileChooserState::Firefox {
                connection_id: chooser_connection_id,
                browsing_context_id,
                shared_id,
            },
        ) if connection_id == &chooser_connection_id => {
            let mut sessions = firefox_session_guard(connection_id).await?;
            sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection
                .send_command(
                    "input.setFiles",
                    json!({
                        "context": browsing_context_id,
                        "element": { "sharedId": shared_id },
                        "files": resolved_files,
                    }),
                )
                .await?;
        }
        _ => return Err("file chooser does not belong to this page session".to_string()),
    }

    Ok(FileChooserFilesSetInfo {
        file_chooser_id: file_chooser_id.to_string(),
        files: resolved_files,
        note: "set file chooser files".to_string(),
    })
}

pub async fn close_page(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
) -> Result<(), String> {
    dialogs::close_page(browser_session, page_session).await;
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                target_id,
                browsing_context_id,
                mapper_target_id,
            },
        ) => {
            let (mut cdp, mapper, context) = chromium_bidi_session(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                Some(browsing_context_id.as_deref().unwrap_or(target_id)),
            )
            .await?;
            cdp.send_bidi_command(
                &mapper.mapper_session_id,
                &json!({
                    "id": 2,
                    "method": "browsingContext.close",
                    "params": { "context": context },
                }),
            )
            .await?;
            Ok(())
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            bidi.send_command(
                "browsingContext.close",
                json!({
                    "context": browsing_context_id,
                }),
            )
            .await?;
            Ok(())
        }
        _ => Err("browser/page backend mismatch while closing page".to_string()),
    }
}

pub async fn close_chrome_tab(cdp_websocket_url: &str, target_id: &str) -> Result<(), String> {
    let mut cdp = CdpConnection::connect(cdp_websocket_url).await?;
    let mapper = cdp.ensure_chromium_bidi_mapper(None).await?;
    cdp.send_bidi_command(
        &mapper.mapper_session_id,
        &json!({
            "id": 2,
            "method": "browsingContext.close",
            "params": { "context": target_id },
        }),
    )
    .await?;
    Ok(())
}

pub async fn navigate_chrome_tab(
    cdp_websocket_url: &str,
    target_id: &str,
    url: &str,
) -> Result<TabNavigationInfo, String> {
    if url.trim().is_empty() {
        return Err("navigate command requires a non-empty url".to_string());
    }

    let mut cdp = CdpConnection::connect(cdp_websocket_url).await?;
    let mapper = cdp.ensure_chromium_bidi_mapper(None).await?;
    let response = cdp
        .send_bidi_command(
            &mapper.mapper_session_id,
            &json!({
                "id": 7100,
                "method": "browsingContext.navigate",
                "params": {
                    "context": target_id,
                    "url": url,
                    "wait": "complete",
                },
            }),
        )
        .await?;
    let navigated_url = required_string(&response, "/result/url")?;

    Ok(TabNavigationInfo {
        url: navigated_url,
        note: "navigated Chromium tab via injected WebDriver BiDi".to_string(),
        page_session: PageSessionHandle::Chromium {
            target_id: target_id.to_string(),
            browsing_context_id: Some(target_id.to_string()),
            mapper_target_id: Some(mapper.mapper_target_id.clone()),
        },
        automation: AutomationSessionInfo {
            bidi_session_id: format!("chromium-bidi:{target_id}"),
            note: format!(
                "navigated through the pinned chromium-bidi@{} mapper",
                mapper.package_version
            ),
            mapper_target_id: Some(mapper.mapper_target_id),
            mapper_session_id: Some(mapper.mapper_session_id),
            package_version: Some(mapper.package_version),
        },
    })
}

pub async fn navigate_page(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    url: &str,
) -> Result<TabNavigationInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                target_id,
                browsing_context_id,
                mapper_target_id,
            },
        ) => {
            if url.trim().is_empty() {
                return Err("navigate command requires a non-empty url".to_string());
            }
            let (mut cdp, mapper, resolved_browsing_context_id) = chromium_bidi_session(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
            )
            .await?;
            let response = cdp
                .send_bidi_command(
                    &mapper.mapper_session_id,
                    &json!({
                        "id": 7101,
                        "method": "browsingContext.navigate",
                        "params": {
                            "context": resolved_browsing_context_id,
                            "url": url,
                            "wait": "complete",
                        },
                    }),
                )
                .await?;
            let navigated_url = required_string(&response, "/result/url")?;
            Ok(TabNavigationInfo {
                url: navigated_url,
                note: "navigated Chromium tab via injected WebDriver BiDi".to_string(),
                page_session: PageSessionHandle::Chromium {
                    target_id: target_id.to_string(),
                    browsing_context_id: Some(resolved_browsing_context_id.clone()),
                    mapper_target_id: Some(mapper.mapper_target_id.clone()),
                },
                automation: AutomationSessionInfo {
                    bidi_session_id: format!("chromium-bidi:{resolved_browsing_context_id}"),
                    note: format!(
                        "navigated through the pinned chromium-bidi@{} mapper",
                        mapper.package_version
                    ),
                    mapper_target_id: Some(mapper.mapper_target_id),
                    mapper_session_id: Some(mapper.mapper_session_id),
                    package_version: Some(mapper.package_version),
                },
            })
        }
        (
            BrowserSessionHandle::Firefox {
                connection_id,
                bidi_session_id,
            },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let result = bidi
                .send_command(
                    "browsingContext.navigate",
                    json!({
                        "context": browsing_context_id,
                        "url": url,
                        "wait": "complete",
                    }),
                )
                .await?;
            Ok(TabNavigationInfo {
                url: json_string(&result, "/result/url")?,
                note: "navigated Firefox tab via native WebDriver BiDi".to_string(),
                page_session: page_session.clone(),
                automation: AutomationSessionInfo {
                    bidi_session_id: bidi_session_id.clone(),
                    note: "connected to native Firefox WebDriver BiDi session".to_string(),
                    mapper_target_id: None,
                    mapper_session_id: None,
                    package_version: None,
                },
            })
        }
        _ => Err("browser/page backend mismatch while navigating".to_string()),
    }
}

pub async fn inject_chromium_bidi_mapper(
    cdp_websocket_url: &str,
) -> Result<ChromiumBidiMapperInfo, String> {
    let mut cdp = CdpConnection::connect(cdp_websocket_url).await?;
    let mapper = cdp.ensure_chromium_bidi_mapper(None).await?;

    Ok(ChromiumBidiMapperInfo {
        package_version: CHROMIUM_BIDI_NPM_VERSION.to_string(),
        mapper_target_id: mapper.mapper_target_id,
        mapper_session_id: mapper.mapper_session_id,
        note: format!(
            "injected chromium-bidi mapper from pinned published chromium-bidi@{} artifact into hidden mapper target",
            CHROMIUM_BIDI_NPM_VERSION
        ),
    })
}

pub async fn click_element(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
) -> Result<ClickInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            click_element_via_bidi(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
                css_selector,
            )
            .await
        }
        (
            BrowserSessionHandle::Firefox {
                connection_id,
                bidi_session_id,
            },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let parsed = parse_selector("click_element", css_selector)?;
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let query_first_js = selector_query_first_js(parsed.kind);
            let center = firefox_evaluate_json(
                bidi,
                browsing_context_id,
                &format!(
                    "JSON.stringify((() => {{
                        const selector = {};
                        const requirements = {};
                        {}
                        const element = {query_first_js};
                        const prepared = allwrightPrepareElement(element, selector, '{}', requirements);
                        return prepared.center;
                    }})())",
                    json_string_literal(&parsed.value, "click selector")?,
                    discovery_requirements_literal(DiscoveryRequirements {
                        require_match: true,
                        require_visible: true,
                        require_pointer_interactable: true,
                        ..DiscoveryRequirements::default()
                    }),
                    actionability_helpers_js(),
                    selector_kind_label(parsed.kind),
                ),
            )
            .await?;
            let x = json_f64(&center, "/x")?;
            let y = json_f64(&center, "/y")?;
            bidi.send_command(
                "input.performActions",
                json!({
                    "context": browsing_context_id,
                    "actions": [{
                        "type": "pointer",
                        "id": "allwright-mouse",
                        "parameters": { "pointerType": "mouse" },
                        "actions": [
                            { "type": "pointerMove", "origin": "viewport", "x": x, "y": y },
                            { "type": "pointerDown", "button": 0 },
                            { "type": "pointerUp", "button": 0 }
                        ]
                    }]
                }),
            )
            .await?;
            Ok(ClickInfo {
                css_selector: css_selector.to_string(),
                note: format!(
                    "clicked element via Firefox WebDriver BiDi input.performActions using {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
                bidi_session_id: bidi_session_id.clone(),
            })
        }
        _ => Err("browser/page backend mismatch while clicking".to_string()),
    }
}

pub async fn count_elements(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
) -> Result<ElementCountInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            let parsed = parse_selector("count_elements", css_selector)?;
            let discovered = discover_elements_via_bidi(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
                css_selector,
                DiscoveryRequirements::default(),
            )
            .await?;
            Ok(ElementCountInfo {
                css_selector: css_selector.to_string(),
                count: discovered.count,
                note: format!(
                    "counted {} element(s) matching {} {}",
                    discovered.count,
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let parsed = parse_selector("count_elements", css_selector)?;
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let query_all_js = selector_query_all_js(parsed.kind);
            let count = firefox_evaluate_u32(
                bidi,
                browsing_context_id,
                &format!(
                    "(() => {{
                        const selector = {};
                        return {query_all_js}.length;
                    }})()",
                    json_string_literal(&parsed.value, "count selector")?,
                ),
            )
            .await?;
            Ok(ElementCountInfo {
                css_selector: css_selector.to_string(),
                count,
                note: format!(
                    "counted {count} element(s) matching {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        _ => Err("browser/page backend mismatch while counting elements".to_string()),
    }
}

pub async fn highlight_elements(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
    duration_ms: u64,
) -> Result<HighlightElementsInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            let parsed = parse_selector("highlight_elements", css_selector)?;
            let duration_ms = u32::try_from(duration_ms).unwrap_or(u32::MAX).max(1);
            let count = chromium_bidi_evaluate_u32(cdp_websocket_url, mapper_target_id.as_deref(), browsing_context_id.as_deref(), &format!(
                "(() => {{ const selector = {}; const durationMs = {duration_ms}; const elements = {}; for (const element of elements) {{ if (!(element instanceof HTMLElement)) continue; element.scrollIntoView({{ block: 'center', inline: 'center', behavior: 'instant' }}); const outline = element.style.outline; const outlineOffset = element.style.outlineOffset; const backgroundColor = element.style.backgroundColor; element.style.outline = '3px solid #ff5a36'; element.style.outlineOffset = '2px'; element.style.backgroundColor = 'rgba(255, 235, 59, 0.35)'; window.setTimeout(() => {{ element.style.outline = outline; element.style.outlineOffset = outlineOffset; element.style.backgroundColor = backgroundColor; }}, durationMs); }} return elements.length; }})()",
                json_string_literal(&parsed.value, "highlight selector")?, selector_query_all_js(parsed.kind),
            )).await?;
            Ok(HighlightElementsInfo {
                css_selector: css_selector.to_string(),
                count,
                note: format!(
                    "highlighted {count} element(s) matching {} {} for {duration_ms}ms",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let parsed = parse_selector("highlight_elements", css_selector)?;
            let duration_ms = u32::try_from(duration_ms).unwrap_or(u32::MAX).max(1);
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let query_all_js = selector_query_all_js(parsed.kind);
            let count = firefox_evaluate_u32(
                bidi,
                browsing_context_id,
                &format!(
                    "(() => {{
                        const selector = {};
                        const durationMs = {duration_ms};
                        const elements = {query_all_js};
                        for (const element of elements) {{
                            if (!(element instanceof HTMLElement)) {{
                                continue;
                            }}
                            element.scrollIntoView({{ block: 'center', inline: 'center', behavior: 'instant' }});
                            const priorOutline = element.style.outline;
                            const priorOutlineOffset = element.style.outlineOffset;
                            const priorBackgroundColor = element.style.backgroundColor;
                            element.style.outline = '3px solid #ff5a36';
                            element.style.outlineOffset = '2px';
                            element.style.backgroundColor = 'rgba(255, 235, 59, 0.35)';
                            window.setTimeout(() => {{
                                element.style.outline = priorOutline;
                                element.style.outlineOffset = priorOutlineOffset;
                                element.style.backgroundColor = priorBackgroundColor;
                            }}, durationMs);
                        }}
                        return elements.length;
                    }})()",
                    json_string_literal(&parsed.value, "highlight selector")?,
                ),
            )
            .await?;
            Ok(HighlightElementsInfo {
                css_selector: css_selector.to_string(),
                count,
                note: format!(
                    "highlighted {count} element(s) matching {} {} for {duration_ms}ms",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        _ => Err("browser/page backend mismatch while highlighting elements".to_string()),
    }
}

fn focus_element_expression(selector: &str) -> Result<(String, String), String> {
    let selector_literal = json_string_literal(selector, "focus selector")?;
    let segments = selector_segments_literal("focus_element", selector)?;
    let kind = selector_chain_kind_label("focus_element", selector)?;
    let query_all = selector_chain_query_all_js();
    let requirements = discovery_requirements_literal(DiscoveryRequirements {
        require_match: true,
        require_visible: true,
        focus_first_match: true,
        require_focus: true,
        ..DiscoveryRequirements::default()
    });
    Ok((
        format!(
            "(() => {{
            const selector = {selector_literal};
            const selectorSegments = {segments};
            const selectorChainQueryAll = () => {query_all};
            const elements = {query_all};
            const requirements = {requirements};
            {}
            const element = elements[0] ?? null;
            allwrightPrepareElement(element, selector, '{kind}', requirements);
        }})()",
            actionability_helpers_js(),
        ),
        kind,
    ))
}

pub async fn focus_element(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
) -> Result<FocusInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            let (expression, kind) = focus_element_expression(css_selector)?;
            chromium_bidi_evaluate_void(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
                &expression,
            )
            .await?;
            Ok(FocusInfo {
                css_selector: css_selector.to_string(),
                note: format!("focused element matching {kind}: {css_selector}"),
            })
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let (expression, kind) = focus_element_expression(css_selector)?;
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            firefox_evaluate_void(bidi, browsing_context_id, &expression).await?;
            Ok(FocusInfo {
                css_selector: css_selector.to_string(),
                note: format!("focused element matching {kind}: {css_selector}"),
            })
        }
        _ => Err("browser/page backend mismatch while focusing element".to_string()),
    }
}

pub async fn fill_element(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
    value: &str,
) -> Result<FillInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            let parsed = parse_selector("fill_element", css_selector)?;
            chromium_bidi_evaluate_void(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
                &dom_set_value_and_dispatch_events_js(
                    parsed.kind,
                    &json_string_literal(&parsed.value, "fill selector")?,
                    &json_string_literal(value, "fill value")?,
                ),
            )
            .await?;
            Ok(FillInfo {
                css_selector: css_selector.to_string(),
                value: value.to_string(),
                note: format!(
                    "filled element matching {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let parsed = parse_selector("fill_element", css_selector)?;
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            firefox_evaluate_void(
                bidi,
                browsing_context_id,
                &dom_set_value_and_dispatch_events_js(
                    parsed.kind,
                    &json_string_literal(&parsed.value, "fill selector")?,
                    &json_string_literal(value, "fill value")?,
                ),
            )
            .await?;
            Ok(FillInfo {
                css_selector: css_selector.to_string(),
                value: value.to_string(),
                note: format!(
                    "filled element matching {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        _ => Err("browser/page backend mismatch while filling element".to_string()),
    }
}

pub async fn hover_element(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
) -> Result<HoverInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            let parsed = parse_selector("hover_element", css_selector)?;
            let center = discover_elements_via_bidi(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
                css_selector,
                DiscoveryRequirements {
                    require_match: true,
                    require_visible: true,
                    require_pointer_interactable: true,
                    ..DiscoveryRequirements::default()
                },
            )
            .await?
            .first_center
            .ok_or_else(|| {
                format!(
                    "element discovery did not return a hover center for {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                )
            })?;
            let (mut cdp, mapper, context) = chromium_bidi_session(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
            )
            .await?;
            cdp.send_bidi_command(&mapper.mapper_session_id, &json!({"id": 2, "method": "input.performActions", "params": {"context": context, "actions": [{"type":"pointer", "id":"allwright-mouse", "parameters":{"pointerType":"mouse"}, "actions":[{"type":"pointerMove", "origin":"viewport", "x":center.x, "y":center.y}]}]}})).await?;
            Ok(HoverInfo {
                css_selector: css_selector.to_string(),
                note: format!(
                    "hovered element matching {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let parsed = parse_selector("hover_element", css_selector)?;
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let query_first_js = selector_query_first_js(parsed.kind);
            firefox_evaluate_void(
                bidi,
                browsing_context_id,
                &format!(
                    "(() => {{
                        const selector = {};
                        const requirements = {};
                        {}
                        const element = {query_first_js};
                        const prepared = allwrightPrepareElement(element, selector, '{}', requirements);
                        for (const type of ['pointerover', 'mouseover', 'mouseenter', 'mousemove']) {{
                            prepared.element.dispatchEvent(new MouseEvent(type, {{ bubbles: true, cancelable: true, view: window }}));
                        }}
                    }})()",
                    json_string_literal(&parsed.value, "hover selector")?,
                    discovery_requirements_literal(DiscoveryRequirements {
                        require_match: true,
                        require_visible: true,
                        require_pointer_interactable: true,
                        ..DiscoveryRequirements::default()
                    }),
                    actionability_helpers_js(),
                    selector_kind_label(parsed.kind),
                ),
            )
            .await?;
            Ok(HoverInfo {
                css_selector: css_selector.to_string(),
                note: format!(
                    "hovered element matching {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        _ => Err("browser/page backend mismatch while hovering element".to_string()),
    }
}

pub async fn press_key(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
    key: &str,
    text: Option<&str>,
) -> Result<PressKeyInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            let parsed = parse_selector("press_key", css_selector)?;
            if key.trim().is_empty() {
                return Err("press_key command requires a non-empty key".to_string());
            }
            let query_first_js = selector_query_first_js(parsed.kind);
            chromium_bidi_evaluate_void(cdp_websocket_url, mapper_target_id.as_deref(), browsing_context_id.as_deref(), &format!(
                "(() => {{ const selector = {}; const key = {}; const text = {}; const requirements = {}; {} const element = {query_first_js}; const prepared = allwrightPrepareElement(element, selector, '{}', requirements); prepared.element.dispatchEvent(new KeyboardEvent('keydown', {{ key, bubbles: true }})); if (text && 'value' in prepared.element) {{ prepared.element.value += text; prepared.element.dispatchEvent(new Event('input', {{ bubbles: true }})); prepared.element.dispatchEvent(new Event('change', {{ bubbles: true }})); }} prepared.element.dispatchEvent(new KeyboardEvent('keyup', {{ key, bubbles: true }})); }})()",
                json_string_literal(&parsed.value, "press selector")?, json_string_literal(key, "press key")?, json_string_literal(text.unwrap_or(""), "press text")?,
                discovery_requirements_literal(DiscoveryRequirements { require_match: true, require_visible: true, focus_first_match: true, require_focus: true, ..DiscoveryRequirements::default() }), actionability_helpers_js(), selector_kind_label(parsed.kind),
            )).await?;
            Ok(PressKeyInfo {
                css_selector: css_selector.to_string(),
                key: key.to_string(),
                note: format!(
                    "pressed key {key} on element matching {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let parsed = parse_selector("press_key", css_selector)?;
            if key.trim().is_empty() {
                return Err("press_key command requires a non-empty key".to_string());
            }
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let query_first_js = selector_query_first_js(parsed.kind);
            firefox_evaluate_void(
                bidi,
                browsing_context_id,
                &format!(
                    "(() => {{
                        const selector = {};
                        const key = {};
                        const text = {};
                        const requirements = {};
                        {}
                        const element = {query_first_js};
                        const prepared = allwrightPrepareElement(element, selector, '{}', requirements);
                        prepared.element.dispatchEvent(new KeyboardEvent('keydown', {{ key, bubbles: true }}));
                        if (text && 'value' in prepared.element) {{
                            prepared.element.value += text;
                            prepared.element.dispatchEvent(new Event('input', {{ bubbles: true }}));
                            prepared.element.dispatchEvent(new Event('change', {{ bubbles: true }}));
                        }}
                        prepared.element.dispatchEvent(new KeyboardEvent('keyup', {{ key, bubbles: true }}));
                    }})()",
                    json_string_literal(&parsed.value, "press selector")?,
                    json_string_literal(key, "press key")?,
                    json_string_literal(text.unwrap_or(""), "press text")?,
                    discovery_requirements_literal(DiscoveryRequirements {
                        require_match: true,
                        require_visible: true,
                        focus_first_match: true,
                        require_focus: true,
                        ..DiscoveryRequirements::default()
                    }),
                    actionability_helpers_js(),
                    selector_kind_label(parsed.kind),
                ),
            )
            .await?;
            Ok(PressKeyInfo {
                css_selector: css_selector.to_string(),
                key: key.to_string(),
                note: format!(
                    "pressed key {key} on element matching {} {}",
                    selector_kind_label(parsed.kind),
                    parsed.value
                ),
            })
        }
        _ => Err("browser/page backend mismatch while pressing a key".to_string()),
    }
}

pub async fn get_text_content(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
) -> Result<TextInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => chromium_bidi_get_text(
            cdp_websocket_url,
            mapper_target_id.as_deref(),
            browsing_context_id.as_deref(),
            css_selector,
            "textContent",
        )
        .await
        .map(|mut info| {
            info.note = format!("resolved textContent for {}", css_selector);
            info
        }),
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            firefox_get_text(
                connection_id,
                browsing_context_id,
                css_selector,
                "textContent",
            )
            .await
        }
        _ => Err("browser/page backend mismatch while resolving textContent".to_string()),
    }
}

pub async fn get_inner_text(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
) -> Result<TextInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => chromium_bidi_get_text(
            cdp_websocket_url,
            mapper_target_id.as_deref(),
            browsing_context_id.as_deref(),
            css_selector,
            "innerText",
        )
        .await
        .map(|mut info| {
            info.note = format!("resolved innerText for {}", css_selector);
            info
        }),
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            firefox_get_text(
                connection_id,
                browsing_context_id,
                css_selector,
                "innerText",
            )
            .await
        }
        _ => Err("browser/page backend mismatch while resolving innerText".to_string()),
    }
}

pub async fn wait_for_selector(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    css_selector: &str,
    visible: bool,
) -> Result<WaitForSelectorInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            let parsed = parse_selector("wait_for_selector", css_selector)?;
            let discovered = discover_elements_via_bidi(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
                css_selector,
                DiscoveryRequirements {
                    require_visible: visible,
                    ..DiscoveryRequirements::default()
                },
            )
            .await?;
            let success = if visible {
                discovered.first_center.is_some()
            } else {
                discovered.count > 0
            };
            if !success {
                return Err(if visible {
                    format!(
                        "no visible element matches {} {}",
                        selector_kind_label(parsed.kind),
                        parsed.value
                    )
                } else {
                    format!(
                        "no element matches {} {}",
                        selector_kind_label(parsed.kind),
                        parsed.value
                    )
                });
            }
            Ok(WaitForSelectorInfo {
                css_selector: css_selector.to_string(),
                visible,
                note: if visible {
                    format!(
                        "visible element matched {} {}",
                        selector_kind_label(parsed.kind),
                        parsed.value
                    )
                } else {
                    format!(
                        "element matched {} {}",
                        selector_kind_label(parsed.kind),
                        parsed.value
                    )
                },
            })
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let parsed = parse_selector("wait_for_selector", css_selector)?;
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let query_first_js = selector_query_first_js(parsed.kind);
            let success = firefox_evaluate_bool(
                bidi,
                browsing_context_id,
                &format!(
                    "(() => {{
                        const selector = {};
                        const requirements = {};
                        {}
                        const element = {query_first_js};
                        if (!element) {{
                            return false;
                        }}
                        if (!{}) {{
                            return true;
                        }}
                        try {{
                            return !!allwrightPrepareElement(element, selector, '{}', requirements);
                        }} catch (_error) {{
                            return false;
                        }}
                    }})()",
                    json_string_literal(&parsed.value, "wait selector")?,
                    discovery_requirements_literal(DiscoveryRequirements {
                        require_visible: visible,
                        ..DiscoveryRequirements::default()
                    }),
                    actionability_helpers_js(),
                    if visible { "true" } else { "false" },
                    selector_kind_label(parsed.kind),
                ),
            )
            .await?;
            if !success {
                return Err(if visible {
                    format!(
                        "no visible element matches {} {}",
                        selector_kind_label(parsed.kind),
                        parsed.value
                    )
                } else {
                    format!(
                        "no element matches {} {}",
                        selector_kind_label(parsed.kind),
                        parsed.value
                    )
                });
            }
            Ok(WaitForSelectorInfo {
                css_selector: css_selector.to_string(),
                visible,
                note: if visible {
                    format!(
                        "visible element matched {} {}",
                        selector_kind_label(parsed.kind),
                        parsed.value
                    )
                } else {
                    format!(
                        "element matched {} {}",
                        selector_kind_label(parsed.kind),
                        parsed.value
                    )
                },
            })
        }
        _ => Err("browser/page backend mismatch while waiting for selector".to_string()),
    }
}

pub async fn screenshot(
    browser_session: &BrowserSessionHandle,
    page_session: &PageSessionHandle,
    full_page: bool,
) -> Result<ScreenshotInfo, String> {
    match (browser_session, page_session) {
        (
            BrowserSessionHandle::Chromium { cdp_websocket_url },
            PageSessionHandle::Chromium {
                browsing_context_id,
                mapper_target_id,
                ..
            },
        ) => {
            chromium_bidi_screenshot(
                cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
                full_page,
            )
            .await
        }
        (
            BrowserSessionHandle::Firefox { connection_id, .. },
            PageSessionHandle::Firefox {
                browsing_context_id,
            },
        ) => {
            let mut sessions = firefox_session_guard(connection_id).await?;
            let bidi = &mut sessions
                .get_mut(connection_id)
                .expect("Firefox session guard validated connection id")
                .connection;
            let response = bidi
                .send_command(
                    "browsingContext.captureScreenshot",
                    json!({
                        "context": browsing_context_id,
                        "origin": if full_page { "document" } else { "viewport" },
                    }),
                )
                .await?;
            Ok(ScreenshotInfo {
                png_data: decode_base64_field(
                    &response,
                    "/result/data",
                    "Firefox screenshot data",
                )?,
                note: if full_page {
                    "captured full-page Firefox screenshot via native WebDriver BiDi".to_string()
                } else {
                    "captured Firefox screenshot via native WebDriver BiDi".to_string()
                },
            })
        }
        _ => Err("browser/page backend mismatch while capturing screenshot".to_string()),
    }
}

fn dom_set_value_and_dispatch_events_js(
    selector_kind: SelectorKind,
    selector_literal: &str,
    value_literal: &str,
) -> String {
    let query_first_js = selector_query_first_js(selector_kind);
    let selector_kind_label = selector_kind_label(selector_kind);
    format!(
        "(() => {{
            const selector = {selector_literal};
            const nextValue = {value_literal};
            const requirements = {requirements_literal};
            {actionability_helpers_js}
            const element = {query_first_js};
            const prepared = allwrightPrepareElement(
                element,
                selector,
                '{selector_kind_label}',
                requirements,
            );
            const editableElement = prepared.element;

            if (editableElement instanceof HTMLInputElement || editableElement instanceof HTMLTextAreaElement) {{
                const prototype = editableElement instanceof HTMLInputElement
                    ? HTMLInputElement.prototype
                    : HTMLTextAreaElement.prototype;
                const descriptor = Object.getOwnPropertyDescriptor(prototype, 'value');
                const setter = descriptor?.set;
                if (!setter) {{
                    throw new Error(`Element value setter is unavailable for {selector_kind_label}: ${{selector}}`);
                }}
                setter.call(editableElement, nextValue);
            }} else if (editableElement.isContentEditable) {{
                editableElement.textContent = nextValue;
            }} else {{
                throw new Error(`Element matched by {selector_kind_label} does not support fill: ${{selector}}`);
            }}

            const inputEvent = typeof InputEvent === 'function'
                ? new InputEvent('input', {{
                    bubbles: true,
                    composed: true,
                    data: nextValue,
                    inputType: 'insertText'
                }})
                : new Event('input', {{ bubbles: true, composed: true }});
            editableElement.dispatchEvent(inputEvent);
            editableElement.dispatchEvent(new Event('change', {{ bubbles: true, composed: true }}));
            return true;
        }})()",
        requirements_literal = discovery_requirements_literal(DiscoveryRequirements {
            require_match: true,
            require_visible: true,
            require_editable: true,
            focus_first_match: true,
            require_focus: true,
            ..DiscoveryRequirements::default()
        }),
        actionability_helpers_js = actionability_helpers_js(),
    )
}

fn discovery_requirements_literal(requirements: DiscoveryRequirements) -> String {
    format!(
        "{{ requireMatch: {}, requireVisible: {}, requirePointerInteractable: {}, requireEditable: {}, focusFirstMatch: {}, requireFocus: {} }}",
        requirements.require_match,
        requirements.require_visible,
        requirements.require_pointer_interactable,
        requirements.require_editable,
        requirements.focus_first_match,
        requirements.require_focus,
    )
}

fn actionability_helpers_js() -> &'static str {
    r#"
            const allwrightCanFocusElement = (element) => {
                if (!(element instanceof HTMLElement)) {
                    return false;
                }
                if (typeof element.matches === 'function' && element.matches(':disabled')) {
                    return false;
                }
                return !element.closest('[inert]');
            };
            const allwrightIsEditableElement = (element) => {
                if (!(element instanceof HTMLElement)) {
                    return false;
                }
                if (element instanceof HTMLInputElement || element instanceof HTMLTextAreaElement) {
                    return !element.disabled && !element.readOnly;
                }
                return element.isContentEditable && !element.closest('[inert]');
            };
            const allwrightIsVisibleElement = (rect, style) => {
                if (!rect || rect.width <= 0 || rect.height <= 0) {
                    return false;
                }
                if (!style) {
                    return false;
                }
                if (style.display === 'none' || style.visibility === 'hidden' || style.visibility === 'collapse') {
                    return false;
                }
                const opacity = Number.parseFloat(style.opacity ?? '1');
                if (!Number.isNaN(opacity) && opacity <= 0) {
                    return false;
                }
                return true;
            };
            const allwrightReceivesPointerEvents = (element, center, style) => {
                if (style.pointerEvents === 'none') {
                    return false;
                }
                const viewportWidth = window.innerWidth || document.documentElement.clientWidth || 0;
                const viewportHeight = window.innerHeight || document.documentElement.clientHeight || 0;
                if (center.x < 0 || center.y < 0 || center.x > viewportWidth || center.y > viewportHeight) {
                    return false;
                }
                const hit = document.elementFromPoint(center.x, center.y);
                return !!hit && (hit === element || element.contains(hit) || hit.contains(element));
            };
            const allwrightPrepareElement = (element, selector, selectorKindLabel, requirements) => {
                if (!element) {
                    throw new Error(`No element matches ${selectorKindLabel}: ${selector}`);
                }
                if (!(element instanceof Element)) {
                    throw new Error(`${selectorKindLabel} did not resolve to a DOM Element: ${selector}`);
                }
                if (!element.isConnected) {
                    throw new Error(`Element matched by ${selectorKindLabel} is detached from the live DOM: ${selector}`);
                }
                element.scrollIntoView({ block: 'center', inline: 'center', behavior: 'instant' });
                const rect = element.getBoundingClientRect();
                const style = window.getComputedStyle(element);
                if (requirements.requireVisible && !allwrightIsVisibleElement(rect, style)) {
                    throw new Error(`Element matched by ${selectorKindLabel} is not visible yet: ${selector}`);
                }
                const center = rect.width > 0 && rect.height > 0
                    ? { x: rect.left + (rect.width / 2), y: rect.top + (rect.height / 2) }
                    : null;
                if (requirements.requirePointerInteractable) {
                    if (!center || !allwrightReceivesPointerEvents(element, center, style)) {
                        throw new Error(`Element matched by ${selectorKindLabel} is not clickable yet: ${selector}`);
                    }
                }
                if (requirements.requireEditable && !allwrightIsEditableElement(element)) {
                    throw new Error(`Element matched by ${selectorKindLabel} is not editable yet: ${selector}`);
                }
                if (requirements.focusFirstMatch || requirements.requireFocus) {
                    if (!allwrightCanFocusElement(element)) {
                        throw new Error(`Element matched by ${selectorKindLabel} is not focusable yet: ${selector}`);
                    }
                    element.focus();
                    const activeElement = document.activeElement;
                    if (requirements.requireFocus && activeElement !== element && !element.contains(activeElement)) {
                        throw new Error(`Element matched by ${selectorKindLabel} did not receive focus yet: ${selector}`);
                    }
                }
                return { element, rect, center, style };
            };
    "#
}

pub async fn resolve_bidi_context_for_tab(
    cdp_websocket_url: &str,
    existing_mapper_target_id: Option<&str>,
    existing_context_id: Option<&str>,
    current_url: Option<&str>,
) -> Result<(String, ChromiumBidiMapperInfo), String> {
    let mut cdp = CdpConnection::connect(cdp_websocket_url).await?;
    let mapper = cdp
        .ensure_chromium_bidi_mapper(existing_mapper_target_id)
        .await?;
    let context_id = cdp
        .resolve_bidi_context_id(&mapper.mapper_session_id, existing_context_id, current_url)
        .await?;

    Ok((
        context_id,
        ChromiumBidiMapperInfo {
            package_version: mapper.package_version,
            mapper_target_id: mapper.mapper_target_id,
            mapper_session_id: mapper.mapper_session_id,
            note: "resolved BiDi browsing context for tab".to_string(),
        },
    ))
}

async fn chromium_bidi_session(
    cdp_websocket_url: &str,
    existing_mapper_target_id: Option<&str>,
    existing_context_id: Option<&str>,
) -> Result<(CdpConnection, MapperConnectionInfo, String), String> {
    let mut cdp = CdpConnection::connect(cdp_websocket_url).await?;
    let mapper = cdp
        .ensure_chromium_bidi_mapper(existing_mapper_target_id)
        .await?;
    let browsing_context_id = cdp
        .resolve_bidi_context_id(&mapper.mapper_session_id, existing_context_id, None)
        .await?;
    Ok((cdp, mapper, browsing_context_id))
}

async fn chromium_bidi_evaluate_json_on_session(
    cdp: &mut CdpConnection,
    mapper_session_id: &str,
    browsing_context_id: &str,
    expression: &str,
) -> Result<Value, String> {
    let response = cdp
        .send_bidi_command(
            mapper_session_id,
            &json!({
                "id": 2,
                "method": "script.evaluate",
                "params": {
                    "expression": expression,
                    "target": { "context": browsing_context_id },
                    "awaitPromise": true,
                },
            }),
        )
        .await?;
    let result_type = json_string(&response, "/result/type")?;
    if result_type != "success" {
        let details = response
            .pointer("/result/exceptionDetails")
            .cloned()
            .unwrap_or(Value::Null);
        return Err(format!("Chromium BiDi script.evaluate failed: {details}"));
    }
    let value = json_string(&response, "/result/result/value")?;
    serde_json::from_str::<Value>(&value)
        .map_err(|error| format!("failed to parse Chromium BiDi script result JSON: {error}"))
}

async fn chromium_bidi_evaluate_json(
    cdp_websocket_url: &str,
    existing_mapper_target_id: Option<&str>,
    existing_context_id: Option<&str>,
    expression: &str,
) -> Result<Value, String> {
    let (mut cdp, mapper, browsing_context_id) = chromium_bidi_session(
        cdp_websocket_url,
        existing_mapper_target_id,
        existing_context_id,
    )
    .await?;
    chromium_bidi_evaluate_json_on_session(
        &mut cdp,
        &mapper.mapper_session_id,
        &browsing_context_id,
        expression,
    )
    .await
}

async fn chromium_bidi_evaluate_void(
    cdp_websocket_url: &str,
    existing_mapper_target_id: Option<&str>,
    existing_context_id: Option<&str>,
    expression: &str,
) -> Result<(), String> {
    let (mut cdp, mapper, context) = chromium_bidi_session(
        cdp_websocket_url,
        existing_mapper_target_id,
        existing_context_id,
    )
    .await?;
    let response = cdp
        .send_bidi_command_with_dialogs(
            &mapper.mapper_session_id,
            &json!({
                "id": 2, "method": "script.evaluate", "params": {
                    "expression": expression, "target": {"context": context}, "awaitPromise": true,
                },
            }),
            true,
        )
        .await?;
    check_script_action_result(&response)
}

async fn chromium_bidi_evaluate_u32(
    cdp_websocket_url: &str,
    existing_mapper_target_id: Option<&str>,
    existing_context_id: Option<&str>,
    expression: &str,
) -> Result<u32, String> {
    let value = chromium_bidi_evaluate_json(
        cdp_websocket_url,
        existing_mapper_target_id,
        existing_context_id,
        &format!("JSON.stringify({expression})"),
    )
    .await?;
    json_u32(&value, "")
}

pub async fn click_element_via_bidi(
    cdp_websocket_url: &str,
    existing_mapper_target_id: Option<&str>,
    existing_context_id: Option<&str>,
    css_selector: &str,
) -> Result<ClickInfo, String> {
    let parsed = parse_selector("click_element", css_selector)?;
    let selector_literal = json_string_literal(&parsed.value, "click selector")?;
    let query_first_js = selector_query_first_js(parsed.kind);
    let expression = format!(
        "JSON.stringify((() => {{
            const selector = {selector_literal};
            const requirements = {};
            {}
            const element = {query_first_js};
            return allwrightPrepareElement(element, selector, '{}', requirements).center;
        }})())",
        discovery_requirements_literal(DiscoveryRequirements {
            require_match: true,
            require_visible: true,
            require_pointer_interactable: true,
            ..DiscoveryRequirements::default()
        }),
        actionability_helpers_js(),
        selector_kind_label(parsed.kind),
    );
    let (mut cdp, mapper, browsing_context_id) = chromium_bidi_session(
        cdp_websocket_url,
        existing_mapper_target_id,
        existing_context_id,
    )
    .await?;
    let center = chromium_bidi_evaluate_json_on_session(
        &mut cdp,
        &mapper.mapper_session_id,
        &browsing_context_id,
        &expression,
    )
    .await?;
    let x = json_f64(&center, "/x")?;
    let y = json_f64(&center, "/y")?;
    cdp.send_bidi_command(
        &mapper.mapper_session_id,
        &json!({
            "id": 3,
            "method": "input.performActions",
            "params": {
                "context": browsing_context_id,
                "actions": [{
                    "type": "pointer",
                    "id": "allwright-mouse",
                    "parameters": { "pointerType": "mouse" },
                    "actions": [
                        { "type": "pointerMove", "origin": "viewport", "x": x, "y": y },
                        { "type": "pointerDown", "button": 0 },
                        { "type": "pointerUp", "button": 0 }
                    ]
                }]
            }
        }),
    )
    .await?;
    Ok(ClickInfo {
        css_selector: css_selector.to_string(),
        note: format!(
            "clicked element via Chromium WebDriver BiDi input.performActions using {} {}",
            selector_kind_label(parsed.kind),
            parsed.value
        ),
        bidi_session_id: format!("chromium-bidi:{browsing_context_id}"),
    })
}

async fn discover_elements_via_bidi(
    cdp_websocket_url: &str,
    mapper_target_id: Option<&str>,
    browsing_context_id: Option<&str>,
    selector: &str,
    requirements: DiscoveryRequirements,
) -> Result<DiscoveredElements, String> {
    let selector_literal = json_string_literal(selector, "selector for discovery")?;
    let selector_segments_literal = selector_segments_literal("discover_elements", selector)?;
    let selector_kind_label = selector_chain_kind_label("discover_elements", selector)?;
    let query_all_js = selector_chain_query_all_js();
    let requirements_literal = discovery_requirements_literal(requirements);
    let value = chromium_bidi_evaluate_json(
        cdp_websocket_url,
        mapper_target_id,
        browsing_context_id,
        &format!(
            "JSON.stringify((() => {{
                const selector = {selector_literal};
                const selectorSegments = {selector_segments_literal};
                const selectorChainQueryAll = () => {query_all_js};
                const elements = {query_all_js};
                const first = elements[0] ?? null;
                const requirements = {requirements_literal};
                {}
                if (requirements.requireMatch && !first) {{
                    throw new Error(`No element matches {selector_kind_label}: ${{selector}}`);
                }}
                const prepared = first
                    ? allwrightPrepareElement(first, selector, '{selector_kind_label}', requirements)
                    : null;
                return {{ count: elements.length, first: prepared?.center ?? null }};
            }})())",
            actionability_helpers_js(),
        ),
    )
    .await?;
    let count = json_u32(&value, "/count")?;
    let first_center = if value
        .pointer("/first")
        .is_some_and(|center| !center.is_null())
    {
        Some(ElementCenter {
            x: json_f64(&value, "/first/x")?,
            y: json_f64(&value, "/first/y")?,
        })
    } else {
        None
    };
    Ok(DiscoveredElements {
        count,
        first_center,
    })
}

async fn chromium_bidi_get_text(
    cdp_websocket_url: &str,
    mapper_target_id: Option<&str>,
    browsing_context_id: Option<&str>,
    css_selector: &str,
    property: &str,
) -> Result<TextInfo, String> {
    let parsed = parse_selector("get_text", css_selector)?;
    discover_elements_via_bidi(
        cdp_websocket_url,
        mapper_target_id,
        browsing_context_id,
        css_selector,
        DiscoveryRequirements {
            require_match: true,
            ..DiscoveryRequirements::default()
        },
    )
    .await?;
    let value = chromium_bidi_evaluate_json(
        cdp_websocket_url,
        mapper_target_id,
        browsing_context_id,
        &format!(
            "JSON.stringify((() => {{
                const selector = {};
                const element = {};
                if (!element) {{ throw new Error(`No element matches {}: ${{selector}}`); }}
                const value = element[{}];
                return typeof value === 'string' ? value : '';
            }})())",
            json_string_literal(&parsed.value, "text selector")?,
            selector_query_first_js(parsed.kind),
            selector_kind_label(parsed.kind),
            json_string_literal(property, "text property")?,
        ),
    )
    .await?;
    let text = value.as_str().ok_or_else(|| {
        format!("expected Chromium BiDi text result to be a string, found {value}")
    })?;
    Ok(TextInfo {
        css_selector: css_selector.to_string(),
        text: text.to_string(),
        note: String::new(),
    })
}

async fn chromium_bidi_screenshot(
    cdp_websocket_url: &str,
    mapper_target_id: Option<&str>,
    browsing_context_id: Option<&str>,
    full_page: bool,
) -> Result<ScreenshotInfo, String> {
    let (mut cdp, mapper, browsing_context_id) =
        chromium_bidi_session(cdp_websocket_url, mapper_target_id, browsing_context_id).await?;
    let response = cdp
        .send_bidi_command(
            &mapper.mapper_session_id,
            &json!({
                "id": 2,
                "method": "browsingContext.captureScreenshot",
                "params": {
                    "context": browsing_context_id,
                    "origin": if full_page { "document" } else { "viewport" },
                }
            }),
        )
        .await?;
    Ok(ScreenshotInfo {
        png_data: decode_base64_field(&response, "/result/data", "Chromium BiDi screenshot data")?,
        note: if full_page {
            "captured full-page Chromium screenshot via WebDriver BiDi".to_string()
        } else {
            "captured Chromium screenshot via WebDriver BiDi".to_string()
        },
    })
}

#[cfg(target_os = "macos")]
fn launch_chrome_for_platform(chrome_binary: Option<&str>) -> Result<ChromeLaunchInfo, String> {
    let browser_binary = chrome_binary.map(ToOwned::to_owned).unwrap_or_else(|| {
        "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome".to_string()
    });
    launch_chrome_with_cdp(&browser_binary)
}

#[cfg(target_os = "linux")]
fn launch_chrome_for_platform(chrome_binary: Option<&str>) -> Result<ChromeLaunchInfo, String> {
    if let Some(chrome_binary) = chrome_binary {
        return launch_chrome_with_cdp(chrome_binary);
    }

    let candidates = ["google-chrome", "chromium", "chromium-browser"];

    for candidate in candidates {
        match launch_chrome_with_cdp(candidate) {
            Ok(launch_info) => return Ok(launch_info),
            Err(error) => {
                if error.contains("No such file or directory")
                    || error.contains("os error 2")
                    || error.contains("not found")
                {
                    continue;
                }
                return Err(format!("failed to launch {candidate}: {error}"));
            }
        }
    }

    Err("could not find a Chrome-compatible browser binary".to_string())
}

#[cfg(target_os = "windows")]
fn launch_chrome_for_platform(chrome_binary: Option<&str>) -> Result<ChromeLaunchInfo, String> {
    if let Some(chrome_binary) = chrome_binary {
        return launch_chrome_with_cdp(chrome_binary);
    }

    launch_chrome_with_cdp("chrome")
}

#[cfg(not(any(target_os = "macos", target_os = "linux", target_os = "windows")))]
fn launch_chrome_for_platform(chrome_binary: Option<&str>) -> Result<ChromeLaunchInfo, String> {
    match chrome_binary {
        Some(chrome_binary) => launch_chrome_with_cdp(chrome_binary),
        None => Err("Chrome launching is not supported on this platform yet".to_string()),
    }
}

fn launch_chrome_with_cdp(browser_binary: &str) -> Result<ChromeLaunchInfo, String> {
    let user_data_dir = create_chrome_user_data_dir()?;
    let user_data_dir_str = user_data_dir.to_string_lossy().to_string();

    let child = Command::new(browser_binary)
        .args([
            "--new-window",
            "--remote-debugging-port=0",
            "--no-first-run",
            "--no-default-browser-check",
            "--disable-sync",
            &format!("--user-data-dir={user_data_dir_str}"),
        ])
        .stdout(Stdio::null())
        .stderr(Stdio::null())
        .spawn()
        .map_err(|error| format!("failed to launch Chrome binary {browser_binary}: {error}"))?;
    let process_id = child.id();

    let cdp_websocket_url = wait_for_cdp_endpoint(&user_data_dir)?;

    Ok(ChromeLaunchInfo {
        browser: browser_binary.to_string(),
        note: "launched Chrome window with CDP enabled".to_string(),
        cdp_websocket_url,
        user_data_dir: user_data_dir_str,
        process_id,
    })
}

fn create_chrome_user_data_dir() -> Result<PathBuf, String> {
    let timestamp = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map_err(|error| format!("failed to compute timestamp: {error}"))?
        .as_millis();
    let pid = std::process::id();
    let dir = std::env::temp_dir().join(format!("allwright-chrome-{pid}-{timestamp}"));
    fs::create_dir_all(&dir).map_err(|error| {
        format!(
            "failed to create Chrome user data dir {}: {error}",
            dir.display()
        )
    })?;
    let default_profile = dir.join("Default");
    fs::create_dir_all(&default_profile).map_err(|error| {
        format!(
            "failed to create Chrome default profile {}: {error}",
            default_profile.display()
        )
    })?;
    let preferences = default_profile.join("Preferences");
    fs::write(&preferences, chromium_profile_preferences().to_string()).map_err(|error| {
        format!(
            "failed to write Chrome automation preferences {}: {error}",
            preferences.display()
        )
    })?;
    Ok(dir)
}

fn chromium_profile_preferences() -> Value {
    json!({
        "download": {
            "prompt_for_download": false,
        },
        "download_bubble": {
            "partial_view_enabled": false,
        },
        "profile": {
            "default_content_setting_values": {
                "automatic_downloads": 1,
            },
        },
    })
}

fn wait_for_cdp_endpoint(user_data_dir: &std::path::Path) -> Result<String, String> {
    let active_port_file = user_data_dir.join("DevToolsActivePort");

    for _ in 0..100 {
        if active_port_file.exists() {
            let contents = fs::read_to_string(&active_port_file).map_err(|error| {
                format!(
                    "failed to read DevToolsActivePort file {}: {error}",
                    active_port_file.display()
                )
            })?;
            return parse_devtools_active_port(&contents);
        }
        thread::sleep(StdDuration::from_millis(50));
    }

    Err(format!(
        "timed out waiting for DevToolsActivePort in {}",
        user_data_dir.display()
    ))
}

fn parse_devtools_active_port(contents: &str) -> Result<String, String> {
    let mut lines = contents.lines();
    let port = lines
        .next()
        .map(str::trim)
        .filter(|value| !value.is_empty())
        .ok_or_else(|| "DevToolsActivePort missing port line".to_string())?;
    let browser_path = lines
        .next()
        .map(str::trim)
        .filter(|value| !value.is_empty())
        .ok_or_else(|| "DevToolsActivePort missing websocket path line".to_string())?;

    Ok(format!("ws://127.0.0.1:{port}{browser_path}"))
}

#[cfg(target_os = "macos")]
fn launch_firefox_with_bidi(browser_binary: Option<&str>) -> Result<BrowserLaunchInfo, String> {
    let browser_binary = browser_binary
        .map(ToOwned::to_owned)
        .unwrap_or_else(|| "/Applications/Firefox.app/Contents/MacOS/firefox".to_string());
    launch_firefox_with_remote_agent(&browser_binary)
}

#[cfg(target_os = "linux")]
fn launch_firefox_with_bidi(browser_binary: Option<&str>) -> Result<BrowserLaunchInfo, String> {
    if let Some(browser_binary) = browser_binary {
        return launch_firefox_with_remote_agent(browser_binary);
    }

    for candidate in ["firefox", "firefox-esr"] {
        match launch_firefox_with_remote_agent(candidate) {
            Ok(launch_info) => return Ok(launch_info),
            Err(error)
                if error.contains("No such file or directory")
                    || error.contains("os error 2")
                    || error.contains("not found") => {}
            Err(error) => return Err(format!("failed to launch {candidate}: {error}")),
        }
    }

    Err("could not find a Firefox browser binary".to_string())
}

#[cfg(target_os = "windows")]
fn launch_firefox_with_bidi(browser_binary: Option<&str>) -> Result<BrowserLaunchInfo, String> {
    let browser_binary = browser_binary.unwrap_or("firefox");
    launch_firefox_with_remote_agent(browser_binary)
}

#[cfg(not(any(target_os = "macos", target_os = "linux", target_os = "windows")))]
fn launch_firefox_with_bidi(browser_binary: Option<&str>) -> Result<BrowserLaunchInfo, String> {
    match browser_binary {
        Some(browser_binary) => launch_firefox_with_remote_agent(browser_binary),
        None => Err("Firefox launching is not supported on this platform yet".to_string()),
    }
}

fn launch_firefox_with_remote_agent(browser_binary: &str) -> Result<BrowserLaunchInfo, String> {
    let user_data_dir = create_firefox_user_data_dir()?;
    let user_data_dir_str = user_data_dir.to_string_lossy().to_string();
    let remote_port = choose_loopback_port()?;

    let child = Command::new(browser_binary)
        .args([
            "--new-instance",
            "--no-remote",
            "--remote-debugging-port",
            &remote_port.to_string(),
            "--profile",
            &user_data_dir_str,
            "about:blank",
        ])
        .stdout(Stdio::null())
        .stderr(Stdio::null())
        .spawn()
        .map_err(|error| format!("failed to launch Firefox binary {browser_binary}: {error}"))?;
    let process_id = child.id();

    let bootstrap_ws_url = format!("ws://127.0.0.1:{remote_port}/session");
    let (connection_id, bidi_session_id, initial_context_id) = block_on_plugin_future(async {
        let mut bidi = wait_for_bidi_endpoint(&bootstrap_ws_url).await?;
        let session = bidi
            .send_command(
                "session.new",
                json!({
                    "capabilities": {
                        "alwaysMatch": {
                            "acceptInsecureCerts": true,
                            "unhandledPromptBehavior": suppressed_file_dialog_behavior()
                        }
                    }
                }),
            )
            .await?;
        let session_id = required_string(&session, "/result/sessionId")?;
        let initial_context_id = bidi.wait_for_top_level_context().await?;
        let connection_id = format!("firefox-bidi-{process_id}-{session_id}");
        firefox_session_registry().lock().await.insert(
            connection_id.clone(),
            FirefoxSessionState {
                process_id,
                connection: bidi,
            },
        );
        Ok((connection_id, session_id, initial_context_id))
    })?;

    Ok(BrowserLaunchInfo {
        browser_kind: BrowserKind::Firefox,
        browser: browser_binary.to_string(),
        note: "launched Firefox window with native WebDriver BiDi enabled".to_string(),
        user_data_dir: user_data_dir_str,
        process_id,
        browser_session: BrowserSessionHandle::Firefox {
            connection_id,
            bidi_session_id: bidi_session_id.clone(),
        },
        initial_page: PageInfo {
            note: "tracked initial Firefox tab via native WebDriver BiDi".to_string(),
            page_session: PageSessionHandle::Firefox {
                browsing_context_id: initial_context_id,
            },
        },
    })
}

fn create_firefox_user_data_dir() -> Result<PathBuf, String> {
    let timestamp = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map_err(|error| format!("failed to compute timestamp: {error}"))?
        .as_millis();
    let pid = std::process::id();
    let dir = std::env::temp_dir().join(format!("allwright-firefox-{pid}-{timestamp}"));
    fs::create_dir_all(&dir).map_err(|error| {
        format!(
            "failed to create Firefox profile dir {}: {error}",
            dir.display()
        )
    })?;
    let user_js = dir.join("user.js");
    fs::write(&user_js, firefox_download_preferences()).map_err(|error| {
        format!(
            "failed to write Firefox automation preferences {}: {error}",
            user_js.display()
        )
    })?;
    Ok(dir)
}

fn firefox_download_preferences() -> &'static str {
    r#"user_pref("browser.download.useDownloadDir", true);
user_pref("browser.download.always_ask_before_handling_new_types", false);
user_pref("browser.download.alwaysOpenPanel", false);
"#
}

fn choose_loopback_port() -> Result<u16, String> {
    let listener = TcpListener::bind("127.0.0.1:0")
        .map_err(|error| format!("failed to reserve a loopback port for Firefox: {error}"))?;
    let port = listener
        .local_addr()
        .map_err(|error| format!("failed to read reserved loopback port: {error}"))?
        .port();
    drop(listener);
    Ok(port)
}

async fn wait_for_bidi_endpoint(ws_url: &str) -> Result<BidiConnection, String> {
    let mut last_error = None;
    for _ in 0..100 {
        match BidiConnection::connect(ws_url).await {
            Ok(connection) => return Ok(connection),
            Err(error) => {
                last_error = Some(error);
                sleep(Duration::from_millis(50)).await;
            }
        }
    }

    Err(format!(
        "timed out waiting for WebDriver BiDi endpoint {ws_url}: {}",
        last_error.unwrap_or_else(|| "unknown error".to_string())
    ))
}

fn parse_selector<'a>(command_name: &str, selector: &'a str) -> Result<ParsedSelector<'a>, String> {
    let trimmed = selector.trim();
    if trimmed.is_empty() {
        return Err(format!(
            "{command_name} command requires a non-empty selector"
        ));
    }

    if parse_explicit_selector_prefix(trimmed).is_some()
        && parse_selector_chain(command_name, trimmed)?.len() > 1
    {
        return Ok(ParsedSelector {
            value: Cow::Owned(format!(
                "{{\"chain\":{}}}",
                selector_segments_literal(command_name, trimmed)?
            )),
            kind: SelectorKind::Structured,
        });
    }
    if trimmed.to_ascii_lowercase().starts_with("aw=") {
        return Ok(ParsedSelector {
            value: parse_selector_value(&trimmed[3..])?,
            kind: SelectorKind::Structured,
        });
    }
    let lowercase = trimmed.to_ascii_lowercase();
    let parsed = if lowercase.starts_with("xpath=") {
        let value = parse_selector_value(&trimmed[6..])?;
        ParsedSelector {
            value,
            kind: SelectorKind::XPath,
        }
    } else if lowercase.starts_with("xpath:") {
        let value = parse_selector_value(&trimmed[6..])?;
        ParsedSelector {
            value,
            kind: SelectorKind::XPath,
        }
    } else if lowercase.starts_with("css=") {
        let value = parse_selector_value(&trimmed[4..])?;
        ParsedSelector {
            value,
            kind: SelectorKind::Css,
        }
    } else if lowercase.starts_with("css:") {
        let value = parse_selector_value(&trimmed[4..])?;
        ParsedSelector {
            value,
            kind: SelectorKind::Css,
        }
    } else if looks_like_xpath(trimmed) {
        ParsedSelector {
            value: Cow::Borrowed(trimmed),
            kind: SelectorKind::XPath,
        }
    } else {
        ParsedSelector {
            value: Cow::Borrowed(trimmed),
            kind: SelectorKind::Css,
        }
    };

    if parsed.value.is_empty() {
        return Err(format!(
            "{command_name} command requires a non-empty selector"
        ));
    }

    Ok(parsed)
}

fn parse_selector_value(raw: &str) -> Result<Cow<'_, str>, String> {
    let trimmed = raw.trim();
    if trimmed.is_empty() {
        return Err("selector value must not be empty".to_string());
    }
    if trimmed.starts_with('"') {
        let end = find_json_string_end(trimmed)
            .ok_or_else(|| "selector value contains an unterminated JSON string".to_string())?;
        if trimmed[end..].trim().is_empty() {
            let decoded = serde_json::from_str::<String>(&trimmed[..end])
                .map_err(|error| format!("failed to decode selector JSON string: {error}"))?;
            return Ok(Cow::Owned(decoded));
        }
    }
    Ok(Cow::Borrowed(trimmed))
}

fn find_json_string_end(value: &str) -> Option<usize> {
    let bytes = value.as_bytes();
    if bytes.first().copied()? != b'"' {
        return None;
    }

    let mut index = 1usize;
    let mut escaped = false;
    while index < bytes.len() {
        let byte = bytes[index];
        if escaped {
            escaped = false;
            index += 1;
            continue;
        }
        match byte {
            b'\\' => escaped = true,
            b'"' => return Some(index + 1),
            _ => {}
        }
        index += 1;
    }
    None
}

fn looks_like_xpath(selector: &str) -> bool {
    selector.starts_with("//")
        || selector.starts_with(".//")
        || selector.starts_with("../")
        || selector.starts_with('/')
        || selector.starts_with('(')
}

fn parse_explicit_selector_prefix(selector: &str) -> Option<ExplicitSelectorPrefix> {
    let lower = selector.to_ascii_lowercase();
    if lower.starts_with("aw=") {
        return Some(ExplicitSelectorPrefix {
            kind: SelectorKind::Structured,
            prefix_len: 3,
        });
    }
    if lower.starts_with("xpath=") || lower.starts_with("xpath:") {
        return Some(ExplicitSelectorPrefix {
            kind: SelectorKind::XPath,
            prefix_len: 6,
        });
    }
    if lower.starts_with("css=") || lower.starts_with("css:") {
        return Some(ExplicitSelectorPrefix {
            kind: SelectorKind::Css,
            prefix_len: 4,
        });
    }
    None
}

fn parse_selector_chain<'a>(
    command_name: &str,
    selector: &'a str,
) -> Result<Vec<ParsedSelector<'a>>, String> {
    let trimmed = selector.trim();
    if trimmed.is_empty() {
        return Err(format!(
            "{command_name} command requires a non-empty selector"
        ));
    }

    let Some(first_prefix) = parse_explicit_selector_prefix(trimmed) else {
        return Ok(vec![parse_selector(command_name, selector)?]);
    };

    let mut segments = Vec::new();
    let mut current_start = 0usize;
    let mut current_prefix = first_prefix;
    let bytes = trimmed.as_bytes();
    let mut scan = current_prefix.prefix_len;
    let mut in_json_string = false;
    let mut escaped = false;

    while scan < bytes.len() {
        let byte = bytes[scan];
        if in_json_string {
            if escaped {
                escaped = false;
            } else if byte == b'\\' {
                escaped = true;
            } else if byte == b'"' {
                in_json_string = false;
            }
            scan += 1;
            continue;
        }

        if byte == b'"' {
            in_json_string = true;
            scan += 1;
            continue;
        }

        if byte.is_ascii_whitespace() {
            let next = scan + 1;
            if next < bytes.len()
                && let Some(prefix) = parse_explicit_selector_prefix(&trimmed[next..])
            {
                let value = parse_selector_value(
                    &trimmed[current_start + current_prefix.prefix_len..scan],
                )?;
                if value.is_empty() {
                    return Err(format!(
                        "{command_name} command requires a non-empty selector"
                    ));
                }
                segments.push(ParsedSelector {
                    value,
                    kind: current_prefix.kind,
                });
                current_start = next;
                current_prefix = prefix;
                scan = current_start + current_prefix.prefix_len;
                continue;
            }
        }
        scan += 1;
    }

    let value = parse_selector_value(&trimmed[current_start + current_prefix.prefix_len..])?;
    if value.is_empty() {
        return Err(format!(
            "{command_name} command requires a non-empty selector"
        ));
    }
    segments.push(ParsedSelector {
        value,
        kind: current_prefix.kind,
    });
    Ok(segments)
}

fn selector_kind_label(kind: SelectorKind) -> &'static str {
    match kind {
        SelectorKind::Structured => "semantic selector",
        SelectorKind::Css => "css selector",
        SelectorKind::XPath => "xpath selector",
    }
}

fn decode_base64_field(value: &Value, pointer: &str, label: &str) -> Result<Vec<u8>, String> {
    let encoded = value
        .pointer(pointer)
        .and_then(Value::as_str)
        .ok_or_else(|| format!("{label} is missing from protocol response: {value}"))?;
    base64::engine::general_purpose::STANDARD
        .decode(encoded)
        .map_err(|error| format!("failed to decode {label}: {error}"))
}

fn semantic_query_js(expression: &str) -> String {
    format!(
        "(() => {{\n{}\n{}\nreturn allwrightQuery({expression});\n}})()",
        include_str!("accessibility_semantics.js"),
        include_str!("selectors.js")
    )
}

fn selector_query_all_js(kind: SelectorKind) -> String {
    let kind = match kind {
        SelectorKind::Css => "css",
        SelectorKind::XPath => "xpath",
        SelectorKind::Structured => "aw",
    };
    semantic_query_js(&format!("[{{kind: '{kind}', value: selector}}]"))
}

fn selector_query_first_js(kind: SelectorKind) -> String {
    format!("({})[0]", selector_query_all_js(kind))
}

fn selector_segments_literal(command_name: &str, selector: &str) -> Result<String, String> {
    let segments = parse_selector_chain(command_name, selector)?;
    let serialized = segments
        .into_iter()
        .enumerate()
        .map(|(index, segment)| {
            let value = if index > 0 && segment.kind == SelectorKind::XPath {
                if segment.value.starts_with("//") {
                    format!(".{}", segment.value)
                } else if segment.value.starts_with('/') {
                    format!(".{}", segment.value)
                } else if segment.value.starts_with('.') {
                    segment.value.to_string()
                } else {
                    format!(".//{}", segment.value)
                }
            } else {
                segment.value.to_string()
            };
            json!({
                "kind": match segment.kind {
                    SelectorKind::Structured => "aw",
                    SelectorKind::Css => "css",
                    SelectorKind::XPath => "xpath",
                },
                "value": value,
            })
        })
        .collect::<Vec<_>>();
    serde_json::to_string(&serialized)
        .map_err(|error| format!("failed to serialize selector chain: {error}"))
}

fn selector_chain_kind_label(command_name: &str, selector: &str) -> Result<String, String> {
    let segments = parse_selector_chain(command_name, selector)?;
    if segments.len() == 1 {
        Ok(selector_kind_label(segments[0].kind).to_string())
    } else {
        Ok("selector chain".to_string())
    }
}

fn selector_chain_query_all_js() -> String {
    semantic_query_js("selectorSegments")
}

fn firefox_session_registry() -> &'static Mutex<HashMap<String, FirefoxSessionState>> {
    FIREFOX_SESSION_REGISTRY.get_or_init(|| Mutex::new(HashMap::new()))
}

fn file_chooser_hook_registry() -> &'static Mutex<HashMap<String, FileChooserHookState>> {
    FILE_CHOOSER_HOOK_REGISTRY.get_or_init(|| Mutex::new(HashMap::new()))
}

fn file_chooser_registry() -> &'static Mutex<HashMap<String, FileChooserState>> {
    FILE_CHOOSER_REGISTRY.get_or_init(|| Mutex::new(HashMap::new()))
}

fn download_hook_registry() -> &'static Mutex<HashMap<String, DownloadHookState>> {
    DOWNLOAD_HOOK_REGISTRY.get_or_init(|| Mutex::new(HashMap::new()))
}

fn download_registry() -> &'static Mutex<HashMap<String, DownloadState>> {
    DOWNLOAD_REGISTRY.get_or_init(|| Mutex::new(HashMap::new()))
}

fn next_file_chooser_id(prefix: &str) -> String {
    format!(
        "{prefix}-{}",
        FILE_CHOOSER_COUNTER.fetch_add(1, Ordering::Relaxed)
    )
}

fn next_download_id(prefix: &str) -> String {
    format!(
        "{prefix}-{}",
        DOWNLOAD_COUNTER.fetch_add(1, Ordering::Relaxed)
    )
}

fn create_download_dir() -> Result<PathBuf, String> {
    let dir = std::env::temp_dir().join(next_download_id("allwright-downloads"));
    fs::create_dir_all(&dir).map_err(|error| {
        format!(
            "failed to create download directory {}: {error}",
            dir.display()
        )
    })?;
    fs::canonicalize(&dir).map_err(|error| {
        format!(
            "failed to resolve download directory {}: {error}",
            dir.display()
        )
    })
}

async fn firefox_session_guard(
    connection_id: &str,
) -> Result<MutexGuard<'static, HashMap<String, FirefoxSessionState>>, String> {
    let registry = firefox_session_registry();
    let sessions = registry.lock().await;
    if sessions.contains_key(connection_id) {
        Ok(sessions)
    } else {
        Err(format!(
            "Firefox automation session {connection_id} is not available"
        ))
    }
}

async fn firefox_evaluate_json(
    bidi: &mut BidiConnection,
    browsing_context_id: &str,
    expression: &str,
) -> Result<Value, String> {
    let result = bidi
        .send_command(
            "script.evaluate",
            json!({
                "expression": expression,
                "target": {
                    "context": browsing_context_id,
                },
                "awaitPromise": true,
            }),
        )
        .await?;
    let result_type = json_string(&result, "/result/type")?;
    if result_type != "success" {
        let details = result
            .pointer("/result/exceptionDetails")
            .cloned()
            .unwrap_or(Value::Null);
        return Err(format!("Firefox script.evaluate failed: {details}"));
    }
    let value = json_string(&result, "/result/result/value")?;
    serde_json::from_str::<Value>(&value)
        .map_err(|error| format!("failed to parse Firefox script result JSON: {error}"))
}

async fn firefox_evaluate_void(
    bidi: &mut BidiConnection,
    browsing_context_id: &str,
    expression: &str,
) -> Result<(), String> {
    let response = bidi.send_command_with_dialogs("script.evaluate", json!({
        "expression": expression, "target": {"context": browsing_context_id}, "awaitPromise": true,
    }), true).await?;
    check_script_action_result(&response)
}

fn check_script_action_result(response: &Value) -> Result<(), String> {
    if json_string(response, "/result/type")? != "success" {
        return Err(format!(
            "BiDi script action failed: {}",
            response
                .pointer("/result/exceptionDetails")
                .unwrap_or(&Value::Null)
        ));
    }
    Ok(())
}

async fn firefox_evaluate_u32(
    bidi: &mut BidiConnection,
    browsing_context_id: &str,
    expression: &str,
) -> Result<u32, String> {
    let value = firefox_evaluate_json(
        bidi,
        browsing_context_id,
        &format!("JSON.stringify({expression})"),
    )
    .await?;
    if let Some(number) = value.as_u64() {
        return u32::try_from(number)
            .map_err(|_| format!("Firefox script result exceeds u32: {number}"));
    }
    if let Some(number) = value.as_i64() {
        return u32::try_from(number)
            .map_err(|_| format!("Firefox script result is out of range for u32: {number}"));
    }
    Err(format!(
        "expected numeric Firefox script result, found {value}"
    ))
}

async fn firefox_evaluate_bool(
    bidi: &mut BidiConnection,
    browsing_context_id: &str,
    expression: &str,
) -> Result<bool, String> {
    let value = firefox_evaluate_json(
        bidi,
        browsing_context_id,
        &format!("JSON.stringify({expression})"),
    )
    .await?;
    value
        .as_bool()
        .ok_or_else(|| format!("expected boolean Firefox script result, found {value}"))
}

async fn firefox_get_text(
    connection_id: &str,
    browsing_context_id: &str,
    css_selector: &str,
    property: &str,
) -> Result<TextInfo, String> {
    let parsed = parse_selector("get_text", css_selector)?;
    let mut sessions = firefox_session_guard(connection_id).await?;
    let bidi = &mut sessions
        .get_mut(connection_id)
        .expect("Firefox session guard validated connection id")
        .connection;
    let query_first_js = selector_query_first_js(parsed.kind);
    let value = firefox_evaluate_json(
        bidi,
        browsing_context_id,
        &format!(
            "JSON.stringify((() => {{
                const selector = {};
                const property = {};
                const element = {query_first_js};
                if (!element) {{
                    throw new Error(`No element matches {}: ${{selector}}`);
                }}
                const value = element[property];
                return typeof value === 'string' ? value : '';
            }})())",
            json_string_literal(&parsed.value, "text selector")?,
            json_string_literal(property, "text property")?,
            selector_kind_label(parsed.kind),
        ),
    )
    .await?;

    Ok(TextInfo {
        css_selector: css_selector.to_string(),
        text: value.as_str().map(str::to_string).unwrap_or_default(),
        note: format!(
            "resolved {property} for {} {}",
            selector_kind_label(parsed.kind),
            parsed.value
        ),
    })
}

pub async fn boot() -> String {
    sleep(Duration::from_millis(15)).await;
    "web ready".to_string()
}

#[cfg(unix)]
fn close_browser_process_for_platform(process_id: u32) -> Result<(), String> {
    let status = Command::new("kill")
        .args(["-TERM", &process_id.to_string()])
        .status()
        .map_err(|error| format!("failed to send TERM to process {process_id}: {error}"))?;

    if status.success() {
        Ok(())
    } else {
        Err(format!(
            "kill -TERM returned non-zero exit status for process {process_id}: {status}"
        ))
    }
}

#[cfg(windows)]
fn close_browser_process_for_platform(process_id: u32) -> Result<(), String> {
    let status = Command::new("taskkill")
        .args(["/PID", &process_id.to_string(), "/T", "/F"])
        .status()
        .map_err(|error| format!("failed to terminate process {process_id}: {error}"))?;

    if status.success() {
        Ok(())
    } else {
        Err(format!(
            "taskkill returned non-zero exit status for process {process_id}: {status}"
        ))
    }
}

#[cfg(not(any(unix, windows)))]
fn close_browser_process_for_platform(process_id: u32) -> Result<(), String> {
    Err(format!(
        "browser process termination is not implemented for process {process_id} on this platform"
    ))
}

type CdpSocket = WebSocketStream<MaybeTlsStream<TcpStream>>;
type BidiSocket = WebSocketStream<MaybeTlsStream<TcpStream>>;

struct MapperConnectionInfo {
    package_version: String,
    mapper_target_id: String,
    mapper_session_id: String,
}

struct CdpConnection {
    websocket_url: String,
    socket: CdpSocket,
    next_id: u64,
    pending_bidi_events: VecDeque<Value>,
}

impl CdpConnection {
    async fn connect(cdp_websocket_url: &str) -> Result<Self, String> {
        let (socket, _) = connect_async(cdp_websocket_url).await.map_err(|error| {
            format!("failed to connect to CDP websocket {cdp_websocket_url}: {error}")
        })?;
        Ok(Self {
            websocket_url: cdp_websocket_url.to_string(),
            socket,
            next_id: 1,
            pending_bidi_events: VecDeque::new(),
        })
    }

    async fn send_command(
        &mut self,
        method: &str,
        params: Value,
        session_id: Option<&str>,
    ) -> Result<Value, String> {
        let id = self.next_id;
        self.next_id += 1;

        let mut payload = json!({
            "id": id,
            "method": method,
            "params": params,
        });
        if let Some(session_id) = session_id {
            payload["sessionId"] = Value::String(session_id.to_string());
        }

        self.socket
            .send(Message::Text(payload.to_string().into()))
            .await
            .map_err(|error| format!("failed to send CDP command {method}: {error}"))?;

        loop {
            let message = self.next_json_message().await?;
            if message.get("id").and_then(Value::as_u64) != Some(id) {
                continue;
            }

            if let Some(error) = message.get("error") {
                return Err(format!("CDP command {method} failed: {error}"));
            }

            return Ok(message.get("result").cloned().unwrap_or(Value::Null));
        }
    }

    async fn evaluate_expression(
        &mut self,
        session_id: &str,
        expression: &str,
        await_promise: bool,
    ) -> Result<Value, String> {
        let result = self
            .send_command(
                "Runtime.evaluate",
                json!({
                    "expression": expression,
                    "awaitPromise": await_promise,
                    "returnByValue": true,
                }),
                Some(session_id),
            )
            .await?;
        if let Some(exception_details) = result.get("exceptionDetails") {
            return Err(format_runtime_exception_message(exception_details));
        }
        Ok(result)
    }

    async fn ensure_chromium_bidi_mapper(
        &mut self,
        existing_mapper_target_id: Option<&str>,
    ) -> Result<MapperConnectionInfo, String> {
        let (mut mapper_target_id, mut created_target) = match existing_mapper_target_id {
            Some(existing_mapper_target_id) if !existing_mapper_target_id.trim().is_empty() => {
                (existing_mapper_target_id.to_string(), false)
            }
            _ => (self.create_bidi_mapper_target().await?, true),
        };

        let attached = match self
            .send_command(
                "Target.attachToTarget",
                json!({
                    "targetId": mapper_target_id,
                    "flatten": true,
                }),
                None,
            )
            .await
        {
            Ok(attached) => attached,
            Err(error)
                if !created_target
                    && error.contains("Target.attachToTarget failed")
                    && error.contains("No target with given id found") =>
            {
                mapper_target_id = self.create_bidi_mapper_target().await?;
                created_target = true;
                self.send_command(
                    "Target.attachToTarget",
                    json!({
                        "targetId": mapper_target_id,
                        "flatten": true,
                    }),
                    None,
                )
                .await?
            }
            Err(error) => return Err(error),
        };
        let mapper_session_id = required_string(&attached, "/sessionId")?;

        self.send_command("Runtime.enable", json!({}), Some(&mapper_session_id))
            .await?;
        self.ensure_runtime_binding(&mapper_session_id, "sendBidiResponse")
            .await?;

        if created_target
            || !self
                .mapper_runtime_is_initialized(&mapper_session_id)
                .await?
        {
            self.send_command(
                "Target.exposeDevToolsProtocol",
                json!({
                    "bindingName": "cdp",
                    "targetId": mapper_target_id,
                    "inheritPermissions": true,
                }),
                None,
            )
            .await?;
            self.evaluate_expression(&mapper_session_id, CHROMIUM_BIDI_MAPPER_BUNDLE, false)
                .await?;
            self.evaluate_expression(
                &mapper_session_id,
                &format!(
                    "window.runMapperInstance('{}')",
                    js_single_quote(&mapper_target_id)
                ),
                true,
            )
            .await?;
            self.send_bidi_command(
                &mapper_session_id,
                &json!({
                    "id": 1,
                    "method": "session.new",
                    "params": {
                        "capabilities": {
                            "alwaysMatch": {
                                "unhandledPromptBehavior": suppressed_file_dialog_behavior()
                            }
                        }
                    }
                }),
            )
            .await?;
        }

        Ok(MapperConnectionInfo {
            package_version: CHROMIUM_BIDI_NPM_VERSION.to_string(),
            mapper_target_id,
            mapper_session_id,
        })
    }

    async fn create_bidi_mapper_target(&mut self) -> Result<String, String> {
        // Chromium destroys hidden targets when their creating CDP connection closes.
        // Keep that connection alive for the mapper's lifetime, independently of commands.
        let mut owner = CdpConnection::connect(&self.websocket_url).await?;
        let created = owner
            .send_command(
                "Target.createTarget",
                json!({
                    "url": "about:blank#MAPPER_TARGET", "hidden": true, "background": true,
                }),
                None,
            )
            .await?;
        let target = required_string(&created, "/targetId")?;
        tokio::spawn(async move { while owner.next_json_message().await.is_ok() {} });
        Ok(target)
    }

    async fn ensure_runtime_binding(&mut self, session_id: &str, name: &str) -> Result<(), String> {
        match self
            .send_command(
                "Runtime.addBinding",
                json!({
                    "name": name,
                }),
                Some(session_id),
            )
            .await
        {
            Ok(_) => Ok(()),
            Err(error)
                if error.contains("already exists") || error.contains("Cannot add binding") =>
            {
                Ok(())
            }
            Err(error) => Err(error),
        }
    }

    async fn mapper_runtime_is_initialized(&mut self, session_id: &str) -> Result<bool, String> {
        let result = self
            .evaluate_expression(
                session_id,
                "typeof window.runMapperInstance === 'function' && typeof window.onBidiMessage === 'function'",
                true,
            )
            .await?;
        Ok(result
            .pointer("/result/value")
            .and_then(Value::as_bool)
            .unwrap_or(false))
    }

    async fn send_bidi_command(
        &mut self,
        mapper_session_id: &str,
        command: &Value,
    ) -> Result<Value, String> {
        self.send_bidi_command_with_dialogs(mapper_session_id, command, false)
            .await
    }

    async fn send_bidi_command_with_dialogs(
        &mut self,
        mapper_session_id: &str,
        command: &Value,
        script_action: bool,
    ) -> Result<Value, String> {
        let command_id = command
            .get("id")
            .and_then(Value::as_u64)
            .ok_or_else(|| "BiDi command is missing numeric id".to_string())?;
        let eval_id = self.next_id;
        self.next_id += 1;
        let command_json = command.to_string();
        let expression = format!(
            "onBidiMessage({})",
            serde_json::to_string(&command_json)
                .map_err(|error| format!("failed to encode BiDi command payload: {error}"))?
        );

        let payload = json!({
            "id": eval_id,
            "method": "Runtime.evaluate",
            "params": {
                "expression": expression,
                "awaitPromise": true,
            },
            "sessionId": mapper_session_id,
        });
        self.socket
            .send(Message::Text(payload.to_string().into()))
            .await
            .map_err(|error| format!("failed to send mapper Runtime.evaluate call: {error}"))?;

        loop {
            let message = self.next_json_message().await?;

            if message.get("id").and_then(Value::as_u64) == Some(eval_id) {
                if let Some(error) = message.get("error") {
                    return Err(format!("mapper Runtime.evaluate failed: {error}"));
                }
                if let Some(exception_details) = message.pointer("/result/exceptionDetails") {
                    return Err(format_mapper_runtime_exception_message(exception_details));
                }
                continue;
            }

            if message.get("sessionId").and_then(Value::as_str) != Some(mapper_session_id) {
                continue;
            }

            if message.get("method").and_then(Value::as_str) != Some("Runtime.bindingCalled") {
                continue;
            }

            let params = message
                .get("params")
                .ok_or_else(|| "Runtime.bindingCalled message is missing params".to_string())?;
            if params.get("name").and_then(Value::as_str) != Some("sendBidiResponse") {
                continue;
            }

            let payload = params
                .get("payload")
                .and_then(Value::as_str)
                .ok_or_else(|| "Runtime.bindingCalled payload is missing".to_string())?;
            let response = serde_json::from_str::<Value>(payload)
                .map_err(|error| format!("failed to parse BiDi mapper response JSON: {error}"))?;
            if response.get("id").and_then(Value::as_u64) != Some(command_id) {
                // Input dispatch can remain pending while a modal prompt is open.
                // Yield at that boundary so the sequential context stream can handle it.
                let interrupted =
                    dialogs::interrupted_action_result(command, &response, script_action);
                if response.get("method").is_some() {
                    self.pending_bidi_events.push_back(response);
                }
                if let Some(result) = interrupted {
                    return Ok(result);
                }
                continue;
            }

            match response.get("type").and_then(Value::as_str) {
                Some("success") => return Ok(response),
                Some("error") => {
                    let message = response
                        .get("message")
                        .and_then(Value::as_str)
                        .unwrap_or("unknown BiDi mapper error");
                    let error = response
                        .get("error")
                        .and_then(Value::as_str)
                        .unwrap_or("unknown error");
                    return Err(format!("BiDi mapper returned {error}: {message}"));
                }
                Some(other) => {
                    return Err(format!("unexpected BiDi response type {other}: {response}"));
                }
                None => {
                    return Err(format!("BiDi mapper response is missing type: {response}"));
                }
            }
        }
    }

    async fn poll_bidi_event(
        &mut self,
        mapper_session_id: &str,
        method: &str,
    ) -> Result<Option<Value>, String> {
        self.poll_bidi_event_for_context(mapper_session_id, method, None)
            .await
    }

    async fn poll_bidi_event_for_context(
        &mut self,
        mapper_session_id: &str,
        method: &str,
        context: Option<&str>,
    ) -> Result<Option<Value>, String> {
        let matches = |event: &Value| {
            event.get("method").and_then(Value::as_str) == Some(method)
                && context.is_none_or(|id| {
                    event.pointer("/params/context").and_then(Value::as_str) == Some(id)
                })
        };
        if let Some(index) = self.pending_bidi_events.iter().position(matches) {
            return Ok(self.pending_bidi_events.remove(index));
        }

        let message = match timeout(Duration::from_millis(10), self.next_json_message()).await {
            Ok(result) => result?,
            Err(_) => return Ok(None),
        };
        if message.get("sessionId").and_then(Value::as_str) != Some(mapper_session_id)
            || message.get("method").and_then(Value::as_str) != Some("Runtime.bindingCalled")
            || message.pointer("/params/name").and_then(Value::as_str) != Some("sendBidiResponse")
        {
            return Ok(None);
        }
        let payload = message
            .pointer("/params/payload")
            .and_then(Value::as_str)
            .ok_or_else(|| "Runtime.bindingCalled payload is missing".to_string())?;
        let event = serde_json::from_str::<Value>(payload)
            .map_err(|error| format!("failed to parse BiDi mapper event JSON: {error}"))?;
        if matches(&event) {
            Ok(Some(event))
        } else {
            if event.get("method").is_some() {
                self.pending_bidi_events.push_back(event);
            }
            Ok(None)
        }
    }

    async fn resolve_bidi_context_id(
        &mut self,
        mapper_session_id: &str,
        existing_context_id: Option<&str>,
        current_url: Option<&str>,
    ) -> Result<String, String> {
        if let Some(existing_context_id) = existing_context_id {
            if !existing_context_id.trim().is_empty() {
                let probe = json!({
                    "id": 1,
                    "method": "browsingContext.getTree",
                    "params": {
                        "root": existing_context_id,
                        "maxDepth": 0,
                    }
                });
                match self.send_bidi_command(mapper_session_id, &probe).await {
                    Ok(_) => return Ok(existing_context_id.to_string()),
                    Err(error) => return Err(error),
                }
            }
        }

        let target_url = current_url
            .map(str::trim)
            .filter(|value| !value.is_empty())
            .map(str::to_string);

        for attempt in 0..20 {
            let tree = self
                .send_bidi_command(
                    mapper_session_id,
                    &json!({
                        "id": 2,
                        "method": "browsingContext.getTree",
                        "params": {
                            "maxDepth": 0,
                        }
                    }),
                )
                .await?;
            let contexts = tree
                .pointer("/result/contexts")
                .and_then(Value::as_array)
                .ok_or_else(|| {
                    format!("BiDi browsingContext.getTree returned no contexts: {tree}")
                })?;

            if let Some(target_url) = target_url.as_deref() {
                if let Some(context_id) = contexts.iter().find_map(|context| {
                    let url = context.get("url").and_then(Value::as_str)?;
                    if url == target_url {
                        context
                            .get("context")
                            .and_then(Value::as_str)
                            .map(str::to_string)
                    } else {
                        None
                    }
                }) {
                    return Ok(context_id);
                }
            }

            if let Some(context_id) = contexts
                .first()
                .and_then(|context| context.get("context"))
                .and_then(Value::as_str)
                .filter(|value| !value.trim().is_empty())
                .map(str::to_string)
            {
                return Ok(context_id);
            }

            if attempt < 19 {
                sleep(Duration::from_millis(150)).await;
            } else {
                return Err(format!(
                    "BiDi browsingContext.getTree returned no usable context ids after retries: {tree}"
                ));
            }
        }

        Err("BiDi browsing context resolution exhausted retries unexpectedly".to_string())
    }

    async fn next_json_message(&mut self) -> Result<Value, String> {
        loop {
            let message = self
                .socket
                .next()
                .await
                .ok_or_else(|| "CDP websocket closed unexpectedly".to_string())?
                .map_err(|error| format!("failed to read CDP websocket message: {error}"))?;

            match message {
                Message::Text(text) => {
                    let value = serde_json::from_str::<Value>(&text)
                        .map_err(|error| format!("failed to parse CDP JSON message: {error}"))?;
                    return Ok(value);
                }
                Message::Binary(bytes) => {
                    let value = serde_json::from_slice::<Value>(&bytes)
                        .map_err(|error| format!("failed to parse CDP JSON message: {error}"))?;
                    return Ok(value);
                }
                Message::Ping(payload) => {
                    self.socket
                        .send(Message::Pong(payload))
                        .await
                        .map_err(|error| format!("failed to reply to CDP ping: {error}"))?;
                }
                Message::Pong(_) => {}
                Message::Frame(_) => {}
                Message::Close(frame) => {
                    return Err(format!("CDP websocket closed: {frame:?}"));
                }
            }
        }
    }
}

struct BidiConnection {
    frame_contexts: std::collections::HashSet<String>,
    socket: BidiSocket,
    next_id: u64,
    pending_events: VecDeque<Value>,
}

impl BidiConnection {
    async fn connect(bidi_websocket_url: &str) -> Result<Self, String> {
        let (socket, _) = connect_async(bidi_websocket_url).await.map_err(|error| {
            format!("failed to connect to WebDriver BiDi websocket {bidi_websocket_url}: {error}")
        })?;
        Ok(Self {
            frame_contexts: std::collections::HashSet::new(),
            socket,
            next_id: 1,
            pending_events: VecDeque::new(),
        })
    }

    async fn send_command(&mut self, method: &str, params: Value) -> Result<Value, String> {
        self.send_command_with_dialogs(method, params, false).await
    }

    async fn send_command_with_dialogs(
        &mut self,
        method: &str,
        params: Value,
        script_action: bool,
    ) -> Result<Value, String> {
        if let Some(context) = params
            .pointer("/target/context")
            .or_else(|| params.get("context"))
            .and_then(Value::as_str)
        {
            if self.frame_contexts.contains(context) {
                // Firefox can retain script realms in the back/forward cache after a
                // frame leaves the active document. Only active tree members are pages.
                let tree = self
                    .send_command_raw("browsingContext.getTree", json!({}))
                    .await?;
                if !frames::tree_contains_context(&tree, context) {
                    return Err(format!("frame browsing context {context} is detached"));
                }
            }
        }
        self.send_command_raw_with_dialogs(method, params, script_action)
            .await
    }

    async fn send_command_raw(&mut self, method: &str, params: Value) -> Result<Value, String> {
        self.send_command_raw_with_dialogs(method, params, false)
            .await
    }

    async fn send_command_raw_with_dialogs(
        &mut self,
        method: &str,
        params: Value,
        script_action: bool,
    ) -> Result<Value, String> {
        let id = self.next_id;
        self.next_id += 1;
        let payload = json!({
            "id": id,
            "method": method,
            "params": params,
        });
        self.socket
            .send(Message::Text(payload.to_string().into()))
            .await
            .map_err(|error| format!("failed to send WebDriver BiDi command {method}: {error}"))?;

        loop {
            let message = self.next_json_message().await?;
            if message.get("id").and_then(Value::as_u64) != Some(id) {
                let interrupted =
                    dialogs::interrupted_action_result(&payload, &message, script_action);
                if message.get("method").is_some() {
                    self.pending_events.push_back(message);
                }
                if let Some(result) = interrupted {
                    return Ok(result);
                }
                continue;
            }

            if let Some(error) = message.get("error").and_then(Value::as_str) {
                let message_text = message
                    .get("message")
                    .and_then(Value::as_str)
                    .unwrap_or("unknown WebDriver BiDi error");
                return Err(format!(
                    "WebDriver BiDi command {method} returned {error}: {message_text}"
                ));
            }

            return Ok(message);
        }
    }

    async fn poll_event(&mut self, method: &str) -> Result<Option<Value>, String> {
        self.poll_event_for_context(method, None).await
    }

    async fn poll_event_for_context(
        &mut self,
        method: &str,
        context: Option<&str>,
    ) -> Result<Option<Value>, String> {
        let matches = |event: &Value| {
            event.get("method").and_then(Value::as_str) == Some(method)
                && context.is_none_or(|id| {
                    event.pointer("/params/context").and_then(Value::as_str) == Some(id)
                })
        };
        if let Some(index) = self.pending_events.iter().position(matches) {
            return Ok(self.pending_events.remove(index));
        }
        match timeout(Duration::from_millis(10), self.next_json_message()).await {
            Ok(Ok(event)) if matches(&event) => Ok(Some(event)),
            Ok(Ok(event)) => {
                self.pending_events.push_back(event);
                Ok(None)
            }
            Ok(Err(error)) => Err(error),
            Err(_) => Ok(None),
        }
    }

    async fn wait_for_top_level_context(&mut self) -> Result<String, String> {
        for _ in 0..100 {
            let tree = self
                .send_command(
                    "browsingContext.getTree",
                    json!({
                        "maxDepth": 0,
                    }),
                )
                .await?;
            if let Some(context_id) = tree
                .pointer("/result/contexts")
                .and_then(Value::as_array)
                .and_then(|contexts| contexts.first())
                .and_then(|context| context.get("context"))
                .and_then(Value::as_str)
                .filter(|value| !value.trim().is_empty())
            {
                return Ok(context_id.to_string());
            }
            sleep(Duration::from_millis(50)).await;
        }

        Err("timed out waiting for an initial Firefox browsing context".to_string())
    }

    async fn next_json_message(&mut self) -> Result<Value, String> {
        loop {
            let message = self
                .socket
                .next()
                .await
                .ok_or_else(|| "WebDriver BiDi websocket closed unexpectedly".to_string())?
                .map_err(|error| {
                    format!("failed to read WebDriver BiDi websocket message: {error}")
                })?;

            match message {
                Message::Text(text) => {
                    let value = serde_json::from_str::<Value>(&text).map_err(|error| {
                        format!("failed to parse WebDriver BiDi JSON message: {error}")
                    })?;
                    return Ok(value);
                }
                Message::Binary(bytes) => {
                    let value = serde_json::from_slice::<Value>(&bytes).map_err(|error| {
                        format!("failed to parse WebDriver BiDi JSON message: {error}")
                    })?;
                    return Ok(value);
                }
                Message::Ping(payload) => {
                    self.socket
                        .send(Message::Pong(payload))
                        .await
                        .map_err(|error| {
                            format!("failed to reply to WebDriver BiDi ping: {error}")
                        })?;
                }
                Message::Pong(_) => {}
                Message::Frame(_) => {}
                Message::Close(frame) => {
                    return Err(format!("WebDriver BiDi websocket closed: {frame:?}"));
                }
            }
        }
    }
}

fn required_string(value: &Value, pointer: &str) -> Result<String, String> {
    value
        .pointer(pointer)
        .and_then(Value::as_str)
        .map(str::to_string)
        .filter(|value| !value.is_empty())
        .ok_or_else(|| format!("missing string field at JSON pointer {pointer}"))
}

fn download_event_id(event: &Value) -> Result<String, String> {
    event
        .pointer("/params/download")
        .or_else(|| event.pointer("/params/navigation"))
        .and_then(Value::as_str)
        .filter(|value| !value.is_empty())
        .map(str::to_string)
        .ok_or_else(|| "download event is missing its download identifier".to_string())
}

fn js_single_quote(value: &str) -> String {
    value.replace('\\', "\\\\").replace('\'', "\\'")
}

fn json_u32(value: &Value, pointer: &str) -> Result<u32, String> {
    let raw = value
        .pointer(pointer)
        .ok_or_else(|| format!("missing JSON value at {pointer}: {value}"))?;

    if let Some(number) = raw.as_u64() {
        return u32::try_from(number)
            .map_err(|_| format!("numeric JSON value at {pointer} exceeds u32: {number}"));
    }

    if let Some(number) = raw.as_i64() {
        return u32::try_from(number).map_err(|_| {
            format!("numeric JSON value at {pointer} is out of range for u32: {number}")
        });
    }

    Err(format!(
        "expected numeric JSON value at {pointer}, found {raw}"
    ))
}

fn json_f64(value: &Value, pointer: &str) -> Result<f64, String> {
    let raw = value
        .pointer(pointer)
        .ok_or_else(|| format!("missing JSON value at {pointer}: {value}"))?;

    raw.as_f64()
        .ok_or_else(|| format!("expected numeric JSON value at {pointer}, found {raw}"))
}

fn json_string(value: &Value, pointer: &str) -> Result<String, String> {
    value
        .pointer(pointer)
        .and_then(Value::as_str)
        .map(str::to_string)
        .ok_or_else(|| format!("expected string JSON value at {pointer}"))
}

fn json_string_literal(value: &str, context: &str) -> Result<String, String> {
    serde_json::to_string(value).map_err(|error| format!("failed to serialize {context}: {error}"))
}

fn format_runtime_exception_message(exception_details: &Value) -> String {
    format_exception_message(
        "Runtime.evaluate failed",
        exception_details,
        "Runtime.evaluate failed with exception details",
    )
}

fn format_mapper_runtime_exception_message(exception_details: &Value) -> String {
    format_exception_message(
        "mapper Runtime.evaluate failed",
        exception_details,
        "mapper Runtime.evaluate raised exception details",
    )
}

fn format_exception_message(
    prefix: &str,
    exception_details: &Value,
    fallback_prefix: &str,
) -> String {
    let message = exception_details
        .pointer("/exception/preview/properties")
        .and_then(Value::as_array)
        .and_then(|properties| {
            properties.iter().find_map(|property| {
                let name = property.get("name").and_then(Value::as_str)?;
                let value = property.get("value").and_then(Value::as_str)?;
                (name == "message").then_some(value)
            })
        })
        .or_else(|| {
            exception_details
                .pointer("/exception/message")
                .and_then(Value::as_str)
        })
        .or_else(|| {
            exception_details
                .pointer("/exception/description")
                .and_then(Value::as_str)
        })
        .unwrap_or_default();

    let compact = message.split_whitespace().collect::<Vec<_>>().join(" ");
    if let Some(selector) = extract_invalid_selector(&compact) {
        return format!("{prefix}: invalid selector {selector}");
    }
    if !compact.is_empty() {
        return format!(
            "{prefix}: {}",
            compact
                .trim_start_matches("SyntaxError: ")
                .trim_start_matches("DOMException: ")
        );
    }
    format!("{fallback_prefix}: {exception_details}")
}

fn extract_invalid_selector(message: &str) -> Option<String> {
    let prefix = "Failed to execute 'querySelectorAll' on 'Document': '";
    let suffix = "' is not a valid selector.";
    let start = message.find(prefix)? + prefix.len();
    let end = message[start..].find(suffix)? + start;
    Some(message[start..end].to_string())
}

fn block_on_plugin_future<T, F>(future: F) -> Result<T, String>
where
    F: Future<Output = Result<T, String>>,
{
    let runtime = plugin_runtime()?;
    runtime.block_on(future)
}

fn plugin_runtime() -> Result<&'static tokio::runtime::Runtime, String> {
    if let Some(runtime) = PLUGIN_RUNTIME.get() {
        return Ok(runtime);
    }

    let runtime = tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .worker_threads(2)
        .thread_name("allwright-web-plugin")
        .build()
        .map_err(|error| format!("failed to create plugin runtime: {error}"))?;
    let _ = PLUGIN_RUNTIME.set(runtime);
    PLUGIN_RUNTIME
        .get()
        .ok_or_else(|| "failed to initialize plugin runtime".to_string())
}

fn plugin_response(result: Result<PluginResult, String>) -> *mut c_char {
    let envelope = match result {
        Ok(result) => PluginEnvelope {
            ok: true,
            result: Some(result),
            error: None,
        },
        Err(error) => PluginEnvelope {
            ok: false,
            result: None,
            error: Some(error),
        },
    };

    let json = match serde_json::to_string(&envelope) {
        Ok(json) => json,
        Err(error) => {
            let fallback = PluginEnvelope {
                ok: false,
                result: None,
                error: Some(format!("failed to serialize plugin response: {error}")),
            };
            serde_json::to_string(&fallback).unwrap_or_else(|_| {
                "{\"ok\":false,\"result\":null,\"error\":\"failed to serialize plugin response\"}"
                    .to_string()
            })
        }
    };

    CString::new(json).unwrap().into_raw()
}

fn handle_plugin_command(command: PluginCommand) -> Result<PluginResult, String> {
    match command {
        PluginCommand::LaunchBrowser {
            browser_kind,
            browser_binary,
        } => {
            launch_browser(browser_kind, browser_binary.as_deref()).map(PluginResult::LaunchBrowser)
        }
        PluginCommand::ResolveFrame {
            browser_session,
            page_session,
            css_selector,
            timeout_ms,
        } => block_on_plugin_future(async move {
            timeout(
                Duration::from_millis(timeout_ms),
                resolve_frame(&browser_session, &page_session, &css_selector),
            )
            .await
            .map_err(|_| "timed out waiting for frame to load and become stable".to_string())?
            .map(PluginResult::ResolveFrame)
        }),
        PluginCommand::OpenPage { browser_session } => block_on_plugin_future(async move {
            open_page(&browser_session)
                .await
                .map(PluginResult::OpenPage)
        }),
        PluginCommand::RegisterHook {
            browser_session,
            page_session,
            hook_type,
        } => block_on_plugin_future(async move {
            register_hook(&browser_session, &page_session, hook_type)
                .await
                .map(PluginResult::RegisterHook)
        }),
        PluginCommand::HandleDialog {
            browser_session,
            page_session,
            dialog_id,
            accept,
            prompt_text,
        } => block_on_plugin_future(async move {
            dialogs::handle(
                &browser_session,
                &page_session,
                &dialog_id,
                accept,
                prompt_text,
            )
            .await?;
            Ok(PluginResult::HandleDialog)
        }),
        PluginCommand::PollHook {
            browser_session,
            registration,
        } => block_on_plugin_future(async move {
            poll_hook(&browser_session, &registration)
                .await
                .map(PluginResult::PollHook)
        }),
        PluginCommand::SetFileChooserFiles {
            browser_session,
            page_session,
            file_chooser_id,
            files,
        } => block_on_plugin_future(async move {
            set_file_chooser_files(&browser_session, &page_session, &file_chooser_id, &files)
                .await
                .map(PluginResult::SetFileChooserFiles)
        }),
        PluginCommand::SaveDownload {
            browser_session,
            page_session,
            download_id,
            path,
        } => block_on_plugin_future(async move {
            save_download(&browser_session, &page_session, &download_id, &path)
                .await
                .map(PluginResult::SaveDownload)
        }),
        PluginCommand::ClosePage {
            browser_session,
            page_session,
        } => block_on_plugin_future(async move {
            close_page(&browser_session, &page_session).await?;
            Ok(PluginResult::ClosePage)
        }),
        PluginCommand::NavigatePage {
            browser_session,
            page_session,
            url,
        } => block_on_plugin_future(async move {
            navigate_page(&browser_session, &page_session, &url)
                .await
                .map(PluginResult::NavigatePage)
        }),
        PluginCommand::ClickElement {
            browser_session,
            page_session,
            css_selector,
        } => block_on_plugin_future(async move {
            click_element(&browser_session, &page_session, &css_selector)
                .await
                .map(PluginResult::ClickElement)
        }),
        PluginCommand::CountElements {
            browser_session,
            page_session,
            css_selector,
        } => block_on_plugin_future(async move {
            count_elements(&browser_session, &page_session, &css_selector)
                .await
                .map(PluginResult::CountElements)
        }),
        PluginCommand::HighlightElements {
            browser_session,
            page_session,
            css_selector,
            duration_ms,
        } => block_on_plugin_future(async move {
            highlight_elements(&browser_session, &page_session, &css_selector, duration_ms)
                .await
                .map(PluginResult::HighlightElements)
        }),
        PluginCommand::FocusElement {
            browser_session,
            page_session,
            css_selector,
        } => block_on_plugin_future(async move {
            focus_element(&browser_session, &page_session, &css_selector)
                .await
                .map(PluginResult::FocusElement)
        }),
        PluginCommand::FillElement {
            browser_session,
            page_session,
            css_selector,
            value,
        } => block_on_plugin_future(async move {
            fill_element(&browser_session, &page_session, &css_selector, &value)
                .await
                .map(PluginResult::FillElement)
        }),
        PluginCommand::HoverElement {
            browser_session,
            page_session,
            css_selector,
        } => block_on_plugin_future(async move {
            hover_element(&browser_session, &page_session, &css_selector)
                .await
                .map(PluginResult::HoverElement)
        }),
        PluginCommand::PressKey {
            browser_session,
            page_session,
            css_selector,
            key,
            text,
        } => block_on_plugin_future(async move {
            press_key(
                &browser_session,
                &page_session,
                &css_selector,
                &key,
                text.as_deref(),
            )
            .await
            .map(PluginResult::PressKey)
        }),
        PluginCommand::Capture {
            browser_session,
            page_session,
            kind,
            css_selector,
            attribute_name,
        } => block_on_plugin_future(async move {
            capture(
                &browser_session,
                &page_session,
                &kind,
                &css_selector,
                &attribute_name,
            )
            .await
            .map(PluginResult::Capture)
        }),
        PluginCommand::GetTextContent {
            browser_session,
            page_session,
            css_selector,
        } => block_on_plugin_future(async move {
            get_text_content(&browser_session, &page_session, &css_selector)
                .await
                .map(PluginResult::GetTextContent)
        }),
        PluginCommand::GetInnerText {
            browser_session,
            page_session,
            css_selector,
        } => block_on_plugin_future(async move {
            get_inner_text(&browser_session, &page_session, &css_selector)
                .await
                .map(PluginResult::GetInnerText)
        }),
        PluginCommand::WaitForSelector {
            browser_session,
            page_session,
            css_selector,
            visible,
        } => block_on_plugin_future(async move {
            wait_for_selector(&browser_session, &page_session, &css_selector, visible)
                .await
                .map(PluginResult::WaitForSelector)
        }),
        PluginCommand::AccessibilitySnapshot {
            browser_session,
            page_session,
            format,
            mode,
        } => block_on_plugin_future(async move {
            accessibility::accessibility_snapshot_with_mode(
                &browser_session,
                &page_session,
                &format,
                &mode,
            )
            .await
            .map(PluginResult::AccessibilitySnapshot)
        }),
        PluginCommand::Screenshot {
            browser_session,
            page_session,
            full_page,
        } => block_on_plugin_future(async move {
            screenshot(&browser_session, &page_session, full_page)
                .await
                .map(PluginResult::Screenshot)
        }),
        PluginCommand::OpenChromeWindow { chrome_binary } => {
            open_chrome_window(chrome_binary.as_deref()).map(PluginResult::OpenChromeWindow)
        }
        PluginCommand::DiscoverInitialTab { cdp_websocket_url } => {
            block_on_plugin_future(async move {
                let result = discover_initial_tab(&cdp_websocket_url).await?;
                Ok(PluginResult::DiscoverInitialTab(result))
            })
        }
        PluginCommand::OpenChromeTab { cdp_websocket_url } => block_on_plugin_future(async move {
            let result = open_chrome_tab(&cdp_websocket_url).await?;
            Ok(PluginResult::OpenChromeTab(result))
        }),
        PluginCommand::CloseBrowserProcess { process_id } => {
            close_browser_process(process_id)?;
            Ok(PluginResult::CloseBrowserProcess)
        }
        PluginCommand::CloseChromeTab {
            cdp_websocket_url,
            target_id,
        } => block_on_plugin_future(async move {
            close_chrome_tab(&cdp_websocket_url, &target_id).await?;
            Ok(PluginResult::CloseChromeTab)
        }),
        PluginCommand::NavigateChromeTab {
            cdp_websocket_url,
            target_id,
            url,
        } => block_on_plugin_future(async move {
            let result = navigate_chrome_tab(&cdp_websocket_url, &target_id, &url).await?;
            Ok(PluginResult::NavigateChromeTab(result))
        }),
        PluginCommand::InjectChromiumBidiMapper { cdp_websocket_url } => {
            block_on_plugin_future(async move {
                let result = inject_chromium_bidi_mapper(&cdp_websocket_url).await?;
                Ok(PluginResult::InjectChromiumBidiMapper(result))
            })
        }
        PluginCommand::ResolveBidiContextForTab {
            cdp_websocket_url,
            mapper_target_id,
            browsing_context_id,
            url,
        } => block_on_plugin_future(async move {
            let (resolved_browsing_context_id, mapper) = resolve_bidi_context_for_tab(
                &cdp_websocket_url,
                mapper_target_id.as_deref(),
                browsing_context_id.as_deref(),
                url.as_deref(),
            )
            .await?;
            Ok(PluginResult::ResolveBidiContextForTab {
                browsing_context_id: resolved_browsing_context_id,
                mapper,
            })
        }),
    }
}

#[unsafe(no_mangle)]
pub extern "C" fn allwright_plugin_api_version() -> u32 {
    ALLWRIGHT_PLUGIN_API_VERSION
}

#[unsafe(no_mangle)]
pub extern "C" fn allwright_plugin_id() -> *const c_char {
    c"web".as_ptr()
}

#[unsafe(no_mangle)]
pub unsafe extern "C" fn allwright_plugin_invoke(request_json: *const c_char) -> *mut c_char {
    if request_json.is_null() {
        return plugin_response(Err("plugin request pointer is null".to_string()));
    }

    let request = match unsafe { CStr::from_ptr(request_json) }.to_str() {
        Ok(request) => request,
        Err(error) => {
            return plugin_response(Err(format!("plugin request is not valid UTF-8: {error}")));
        }
    };

    let command: PluginCommand = match serde_json::from_str(request) {
        Ok(command) => command,
        Err(error) => {
            return plugin_response(Err(format!("failed to parse plugin request JSON: {error}")));
        }
    };

    plugin_response(handle_plugin_command(command))
}

#[unsafe(no_mangle)]
pub unsafe extern "C" fn allwright_plugin_free_string(value: *mut c_char) {
    if !value.is_null() {
        unsafe {
            let _ = CString::from_raw(value);
        }
    }
}

#[cfg(test)]
mod tests {
    use super::{
        SelectorKind, chromium_profile_preferences, firefox_download_preferences, parse_selector,
        suppressed_file_dialog_behavior,
    };

    #[test]
    fn bidi_sessions_dismiss_native_file_dialogs() {
        assert_eq!(
            suppressed_file_dialog_behavior().pointer("/file"),
            Some(&serde_json::Value::String("dismiss".to_string()))
        );
    }

    #[test]
    fn chromium_automation_has_no_direct_cdp_fallbacks() {
        let source = include_str!("lib.rs");
        for (prefix, suffix) in [
            ("Via", "Cdp"),
            ("_via_", "cdp"),
            ("Target.", "getTargets"),
            ("Target.", "closeTarget"),
            ("Page.", "getFrameTree"),
            ("Page.", "captureScreenshot"),
            ("Input.", "dispatchMouseEvent"),
            ("Input.", "dispatchKeyEvent"),
        ] {
            let forbidden = format!("{prefix}{suffix}");
            assert!(
                !source.contains(&forbidden),
                "direct CDP automation token must not appear in the web surface: {forbidden}"
            );
        }
    }

    #[test]
    fn browser_profiles_suppress_download_prompts_and_panels() {
        let chromium = chromium_profile_preferences();
        assert_eq!(
            chromium.pointer("/download/prompt_for_download"),
            Some(&serde_json::Value::Bool(false))
        );
        assert_eq!(
            chromium.pointer("/download_bubble/partial_view_enabled"),
            Some(&serde_json::Value::Bool(false))
        );
        assert_eq!(
            chromium.pointer("/profile/default_content_setting_values/automatic_downloads"),
            Some(&serde_json::Value::Number(1.into()))
        );

        let firefox = firefox_download_preferences();
        assert!(firefox.contains(r#""browser.download.useDownloadDir", true"#));
        assert!(
            firefox.contains(r#""browser.download.always_ask_before_handling_new_types", false"#)
        );
        assert!(firefox.contains(r#""browser.download.alwaysOpenPanel", false"#));
    }

    #[test]
    fn parse_selector_detects_explicit_xpath_prefix() {
        let parsed = parse_selector("click_element", "xpath=//button[normalize-space()='Save']")
            .expect("selector should parse");
        assert_eq!(parsed.kind, SelectorKind::XPath);
        assert_eq!(parsed.value, "//button[normalize-space()='Save']");
    }

    #[test]
    fn parse_selector_detects_xpath_heuristically() {
        let parsed = parse_selector("click_element", "//div[@role='dialog']")
            .expect("selector should parse");
        assert_eq!(parsed.kind, SelectorKind::XPath);
    }

    #[test]
    fn parse_selector_preserves_css_by_default() {
        let parsed = parse_selector("click_element", "button[data-testid='save']")
            .expect("selector should parse");
        assert_eq!(parsed.kind, SelectorKind::Css);
        assert_eq!(parsed.value, "button[data-testid='save']");
    }

    #[test]
    fn parse_selector_decodes_json_quoted_xpath_payload() {
        let parsed = parse_selector(
            "click_element",
            r#"xpath="//button[normalize-space()='Save Changes']""#,
        )
        .expect("selector should parse");
        assert_eq!(parsed.kind, SelectorKind::XPath);
        assert_eq!(parsed.value, "//button[normalize-space()='Save Changes']");
    }

    #[test]
    fn parse_selector_chain_handles_quoted_hybrid_segments() {
        let segments = super::parse_selector_chain(
            "click_element",
            r#"xpath="//div[@role='dialog']" css="button.primary""#,
        )
        .expect("selector chain should parse");
        assert_eq!(segments.len(), 2);
        assert_eq!(segments[0].kind, SelectorKind::XPath);
        assert_eq!(segments[0].value, "//div[@role='dialog']");
        assert_eq!(segments[1].kind, SelectorKind::Css);
        assert_eq!(segments[1].value, "button.primary");
    }
}