playwright-cdp 0.3.0

//! `Locator` — a Playwright-style element reference resolved lazily against
//! the page's selector engine.

use crate::error::{Error, Result};
use crate::options::{
    CheckOptions, ClickOptions, DragToOptions, FillOptions, FilterOptions, HoverOptions,
    PressOptions, PressSequentiallyOptions, ScreenshotOptions, SelectOption, SelectOptions,
    WaitForOptions,
};
use crate::page::Page;
use crate::selectors;
use crate::types::BoundingBox;
use base64::Engine;
use serde_json::{json, Value};
use std::time::Duration;

/// A lazy element locator.
#[derive(Clone)]
pub struct Locator {
    page: Page,
    selector: String,
    strict: bool,
    forced_index: Option<usize>,
    /// Ordered list of same-origin iframe selectors to descend through before
    /// querying `selector`. Each entry is resolved in the previous frame's
    /// `contentDocument` (the first in the top document). Empty means query
    /// the top document. Populated by [`FrameLocator`](crate::FrameLocator).
    frame_chain: Vec<String>,
    /// The execution-context id of the frame this locator is scoped to, when it
    /// was produced by a [`Frame`](crate::Frame) (rather than a
    /// [`FrameLocator`](crate::FrameLocator) frame chain). When set and
    /// `frame_chain` is empty, resolution runs in this context directly, so
    /// `document` is the child frame's document.
    frame_ctx: Option<i64>,
}

impl Locator {
    pub(crate) fn new(page: Page, selector: String, strict: bool, forced_index: Option<usize>) -> Self {
        Self {
            page,
            selector,
            strict,
            forced_index,
            frame_chain: Vec::new(),
            frame_ctx: None,
        }
    }

    /// Construct a locator scoped to a specific frame's execution context.
    /// Resolution runs via `Runtime.evaluate { contextId }` in that context,
    /// so `document` resolves to the child frame's document.
    pub(crate) fn new_in_frame_ctx(
        page: Page,
        frame_ctx: Option<i64>,
        selector: String,
        strict: bool,
        forced_index: Option<usize>,
    ) -> Self {
        Self {
            page,
            selector,
            strict,
            forced_index,
            frame_chain: Vec::new(),
            frame_ctx,
        }
    }

    /// Construct a locator scoped to a chain of same-origin iframes' content
    /// documents. `frame_chain` is the ordered list of iframe selectors
    /// (outermost first); `selector` is resolved in the deepest frame.
    pub(crate) fn new_in_frame(
        page: Page,
        frame_chain: Vec<String>,
        selector: String,
        strict: bool,
        forced_index: Option<usize>,
    ) -> Self {
        Self {
            page,
            selector,
            strict,
            forced_index,
            frame_chain,
            frame_ctx: None,
        }
    }

    pub fn selector(&self) -> &str {
        &self.selector
    }

    /// Pick the element at a specific index (disables strict mode).
    pub fn nth(&self, index: i32) -> Locator {
        let mut l = self.clone();
        l.strict = false;
        l.forced_index = Some(if index < 0 { 0 } else { index as usize });
        l
    }

    pub fn first(&self) -> Locator {
        self.nth(0)
    }

    pub fn last(&self) -> Locator {
        let mut l = self.clone();
        l.strict = false;
        l.forced_index = Some(usize::MAX); // resolved at query time
        l
    }

    /// Number of matching elements.
    pub async fn count(&self) -> Result<usize> {
        self.count_query().await
    }

    /// Resolve the count, descending this locator's iframe chain.
    async fn count_query(&self) -> Result<usize> {
        if self.frame_chain.is_empty() {
            let ctx = self.resolve_ctx().await;
            selectors::count(self.page.session(), ctx, &self.selector).await
        } else {
            let ctx = self.page.ctx().await;
            selectors::count_in(self.page.session(), ctx, &self.frame_chain, &self.selector).await
        }
    }

    /// Resolve a single matching element to its `RemoteObjectId`, descending
    /// this locator's iframe chain.
    async fn element_query(&self, index: usize) -> Result<Option<String>> {
        if self.frame_chain.is_empty() {
            let ctx = self.resolve_ctx().await;
            selectors::element_at(self.page.session(), ctx, &self.selector, index).await
        } else {
            let ctx = self.page.ctx().await;
            selectors::element_at_in(self.page.session(), ctx, &self.frame_chain, &self.selector, index).await
        }
    }

    /// The execution context to run a frame-scoped query in: the locator's
    /// `frame_ctx` if it was produced by a [`Frame`](crate::Frame), otherwise
    /// the page's main-world context.
    async fn resolve_ctx(&self) -> Option<i64> {
        match self.frame_ctx {
            // A pinned child-frame context may go stale across navigations; if
            // it is gone from the page's frame map, fall back to the page ctx.
            Some(id) => Some(id),
            None => self.page.ctx().await,
        }
    }

    /// Resolve to a single element RemoteObjectId, enforcing strict mode.
    async fn resolve(&self) -> Result<String> {
        let n = self.count_query().await?;
        if n == 0 {
            return Err(Error::ElementNotFound(format!(
                "no element matched '{}'",
                self.selector
            )));
        }
        if self.strict && self.forced_index.is_none() && n > 1 {
            return Err(Error::ElementNotFound(format!(
                "strict mode violation: {n} elements matched '{}'",
                self.selector
            )));
        }
        let index = match self.forced_index {
            Some(usize::MAX) => n - 1,
            Some(i) => i,
            None => 0,
        };
        self.element_query(index)
            .await?
            .ok_or_else(|| Error::ElementNotFound(format!("no element at index {index} for '{}'", self.selector)))
    }

    async fn release(&self, object_id: &str) {
        let _ = self
            .page
            .session()
            .send("Runtime.releaseObject", json!({ "objectId": object_id }))
            .await;
    }

    // --- actions ---

    pub async fn click(&self, opts: Option<ClickOptions>) -> Result<()> {
        let opts = opts.unwrap_or_default();
        let oid = self.resolve().await?;
        let result = click_element(&self.page, &oid, &opts).await;
        self.release(&oid).await;
        result
    }

    pub async fn fill(&self, text: &str, _opts: Option<FillOptions>) -> Result<()> {
        let oid = self.resolve().await?;
        let result = selectors::eval_object(
            self.page.session(),
            &oid,
            "(el, text) => { el.focus(); el.value = text; el.dispatchEvent(new Event('input', { bubbles: true })); el.dispatchEvent(new Event('change', { bubbles: true })); }",
            json!(text),
        )
        .await
        .map(|_| ());
        self.release(&oid).await;
        result
    }

    pub async fn type_text(&self, text: &str) -> Result<()> {
        self.fill(text, None).await
    }

    /// Type each character of `text` sequentially, optionally pausing between
    /// keystrokes. Unlike [`Self::fill`](crate::Locator::fill) (which sets the
    /// value atomically), this dispatches one `keyDown`/`keyUp` per character,
    /// so it triggers per-keystroke event handlers (e.g. typeahead search).
    ///
    /// `options.delay` is in milliseconds, matching Playwright.
    pub async fn press_sequentially(&self, text: &str, opts: Option<PressSequentiallyOptions>) -> Result<()> {
        let delay = opts
            .and_then(|o| o.delay)
            .map(|ms| Duration::from_millis(ms as u64));
        let oid = self.resolve().await?;
        // Focus so the keystrokes land on this element.
        let _ = self
            .page
            .session()
            .send("DOM.focus", json!({ "objectId": oid }))
            .await;
        let result = press_text(&self.page, &oid, text, delay).await;
        self.release(&oid).await;
        result
    }

    /// Playwright-style alias: type each character sequentially. Mirrors the
    /// reserved keyword with a trailing underscore.
    pub async fn type_(&self, text: &str, opts: Option<PressSequentiallyOptions>) -> Result<()> {
        self.press_sequentially(text, opts).await
    }

    pub async fn press(&self, key: &str, options: Option<PressOptions>) -> Result<()> {
        let oid = self.resolve().await?;
        let delay = options.and_then(|o| o.delay).map(|ms| Duration::from_millis(ms as u64));
        let result = press_key(&self.page, &oid, key, delay).await;
        self.release(&oid).await;
        result
    }

    pub async fn hover(&self, opts: Option<HoverOptions>) -> Result<()> {
        let opts = opts.unwrap_or_default();
        let oid = self.resolve().await?;
        let result = hover_element(&self.page, &oid, &opts).await;
        self.release(&oid).await;
        result
    }

    pub async fn focus(&self) -> Result<()> {
        let oid = self.resolve().await?;
        let result = self
            .page
            .session()
            .send("DOM.focus", json!({ "objectId": oid }))
            .await
            .map(|_: Value| ());
        self.release(&oid).await;
        result
    }

    /// Remove focus from the element.
    pub async fn blur(&self) -> Result<()> {
        let oid = self.resolve().await?;
        let r = selectors::eval_object(self.page.session(), &oid, "(el) => { el.blur(); }", Value::Null)
            .await
            .map(|_| ());
        self.release(&oid).await;
        r
    }

    /// The current value of an `<input>`/`<textarea>`/`<select>`.
    pub async fn input_value(&self, _options: Option<()>) -> Result<String> {
        let oid = self.resolve().await?;
        let v: Value = {
            let res = selectors::eval_object(
                self.page.session(),
                &oid,
                "(el) => el.value",
                Value::Null,
            )
            .await;
            self.release(&oid).await;
            res?
        };
        Ok(v.as_str().unwrap_or("").to_string())
    }

    /// Scroll the element into view.
    pub async fn scroll_into_view_if_needed(&self) -> Result<()> {
        let oid = self.resolve().await?;
        let r = self
            .page
            .session()
            .send("DOM.scrollIntoViewIfNeeded", json!({ "objectId": oid }))
            .await
            .map(|_: Value| ());
        self.release(&oid).await;
        r
    }

    /// Dispatch a synthetic DOM event on the element.
    pub async fn dispatch_event(&self, type_: &str, init: Option<Value>) -> Result<()> {
        let oid = self.resolve().await?;
        let init = init.unwrap_or(Value::Null);
        let r = selectors::eval_object(
            self.page.session(),
            &oid,
            "(el, args) => { el.dispatchEvent(new Event(args.type, args.init || undefined)); }",
            json!({ "type": type_, "init": init }),
        )
        .await
        .map(|_| ());
        self.release(&oid).await;
        r
    }

    /// Evaluate `expression` with the resolved element bound as the first arg.
    pub async fn evaluate<R: serde::de::DeserializeOwned>(
        &self,
        expression: &str,
        arg: Option<Value>,
    ) -> Result<R> {
        let oid = self.resolve().await?;
        let function = format!("(el, arg) => {{ return ({expression}); }}");
        let v = selectors::eval_object(self.page.session(), &oid, &function, arg.unwrap_or(Value::Null))
            .await?;
        self.release(&oid).await;
        serde_json::from_value::<R>(v).map_err(Error::from)
    }

    /// Evaluate `page_function` with the resolved element bound as the first
    /// argument, returning a [`JSHandle`](crate::js_handle::JSHandle) to the
    /// resulting remote object (by reference, not by value).
    ///
    /// `page_function` is wrapped as `(el, arg) => { return (<page_function>); }`
    /// and run via `Runtime.callFunctionOn` with the element's `objectId` as
    /// the receiver (`returnByValue: false`). The resolved element is released
    /// afterward, mirroring [`Locator::evaluate`].
    pub async fn evaluate_handle(
        &self,
        page_function: &str,
        arg: Option<Value>,
    ) -> Result<crate::js_handle::JSHandle> {
        let oid = self.resolve().await?;
        let function = format!("(el, arg) => {{ return ({page_function}); }}");
        let params = json!({
            "objectId": oid,
            "functionDeclaration": function,
            "arguments": [{ "value": arg.unwrap_or(Value::Null) }],
            "returnByValue": false,
            "awaitPromise": true,
        });
        let resp = self.page.session().send("Runtime.callFunctionOn", params).await;
        // Release the resolved element regardless of the call outcome.
        self.release(&oid).await;
        let resp = resp?;
        if let Some(exc) = resp.get("exceptionDetails") {
            let msg = exc
                .get("exception")
                .and_then(|e| e.get("description"))
                .and_then(|v| v.as_str())
                .unwrap_or("evaluation threw");
            return Err(Error::ProtocolError(format!("eval error: {msg}")));
        }
        let result_oid = resp
            .get("result")
            .and_then(|r| r.get("objectId"))
            .and_then(|v| v.as_str())
            .ok_or_else(|| {
                Error::ProtocolError(
                    "callFunctionOn did not return a remote object (no objectId)".into(),
                )
            })?
            .to_string();
        Ok(crate::js_handle::JSHandle::new(self.page.session_arc(), result_oid))
    }

    /// Evaluate `expression` against all matching elements (bound as an array).
    pub async fn evaluate_all<R: serde::de::DeserializeOwned>(
        &self,
        expression: &str,
        arg: Option<Value>,
    ) -> Result<R> {
        let function = if self.frame_chain.is_empty() {
            format!(
                "(arg) => {{ const els = self.__pwcdpInjected.querySelectorAll(document, {selector:?}); return ({expression}); }}",
                selector = self.selector,
                expression = expression
            )
        } else {
            // Descend the same-origin iframe chain to the deepest contentDocument.
            format!(
                "(arg) => {{ let root = document; for (const fs of {chain:?}) {{ const f = self.__pwcdpInjected.querySelectorAll(root, fs)[0]; root = f && f.contentDocument; if (!root) {{ const els = []; return ({expression}); }} }} const els = self.__pwcdpInjected.querySelectorAll(root, {selector:?}); return ({expression}); }}",
                chain = self.frame_chain,
                selector = self.selector,
                expression = expression
            )
        };
        let ctx = if self.frame_chain.is_empty() {
            self.resolve_ctx().await
        } else {
            self.page.ctx().await
        };
        let v = selectors::eval_context(
            self.page.session(),
            ctx,
            &function,
            arg.unwrap_or(Value::Null),
        )
        .await?;
        serde_json::from_value::<R>(v).map_err(Error::from)
    }

    // --- info ---

    pub async fn text_content(&self) -> Result<Option<String>> {
        self.read("(el) => el.textContent").await
    }

    pub async fn inner_text(&self) -> Result<String> {
        self.read("(el) => el.innerText")
            .await
            .map(|v: Option<String>| v.unwrap_or_default())
    }

    pub async fn inner_html(&self) -> Result<String> {
        self.read("(el) => el.innerHTML")
            .await
            .map(|v: Option<String>| v.unwrap_or_default())
    }

    pub async fn get_attribute(&self, name: &str) -> Result<Option<String>> {
        let oid = self.resolve().await?;
        let v = selectors::eval_object(
            self.page.session(),
            &oid,
            "(el, name) => el.getAttribute(name)",
            json!(name),
        )
        .await?;
        self.release(&oid).await;
        Ok(v.as_str().map(String::from))
    }

    async fn read<R: serde::de::DeserializeOwned>(&self, function: &str) -> Result<R> {
        let oid = self.resolve().await?;
        let v = selectors::eval_object(self.page.session(), &oid, function, Value::Null).await?;
        self.release(&oid).await;
        serde_json::from_value::<R>(v).map_err(Error::from)
    }

    pub async fn is_visible(&self) -> Result<bool> {
        self.state("visible").await
    }
    pub async fn is_enabled(&self) -> Result<bool> {
        self.state("enabled").await
    }
    pub async fn is_checked(&self) -> Result<bool> {
        self.state("checked").await
    }
    pub async fn is_editable(&self) -> Result<bool> {
        self.state("editable").await
    }

    async fn state(&self, field: &str) -> Result<bool> {
        let oid = self.resolve().await?;
        let function = format!(
            "(el) => self.__pwcdpInjected && self.__pwcdpInjected.elementState(el).{field}"
        );
        let v = selectors::eval_object(self.page.session(), &oid, &function, Value::Null).await?;
        self.release(&oid).await;
        Ok(v.as_bool().unwrap_or(false))
    }

    pub async fn bounding_box(&self) -> Result<Option<BoundingBox>> {
        let oid = self.resolve().await?;
        let r = element_box(&self.page, &oid).await;
        self.release(&oid).await;
        match r {
            Ok((x, y, w, h)) => Ok(Some(BoundingBox { x, y, width: w, height: h })),
            Err(_) => Ok(None),
        }
    }

    /// Wait until the element resolves (and, for actions, is visible).
    pub async fn wait_for(&self, opts: Option<WaitForOptions>) -> Result<()> {
        let timeout = opts
            .and_then(|o| o.timeout)
            .map(|ms| Duration::from_millis(ms as u64))
            .unwrap_or_else(|| Duration::from_secs(30));
        let deadline = tokio::time::Instant::now() + timeout;
        loop {
            if self.count().await? > 0 {
                return Ok(());
            }
            if tokio::time::Instant::now() >= deadline {
                return Err(Error::Timeout(format!(
                    "wait_for '{}' timed out after {}ms",
                    self.selector,
                    timeout.as_millis()
                )));
            }
            tokio::time::sleep(Duration::from_millis(100)).await;
        }
    }

    // --- additional actions ---

    /// Clear an input/textarea (fill with empty string).
    pub async fn clear(&self, options: Option<FillOptions>) -> Result<()> {
        self.fill("", options).await
    }

    /// Double-click (two press/release cycles with clickCount 2).
    pub async fn dblclick(&self, options: Option<ClickOptions>) -> Result<()> {
        let mut opts = options.unwrap_or_default();
        opts.click_count = Some(opts.click_count.unwrap_or(2));
        self.click(Some(opts)).await
    }

    /// Check a checkbox/radio if not already checked.
    pub async fn check(&self, options: Option<CheckOptions>) -> Result<()> {
        self.set_checked(true, options).await
    }

    /// Uncheck a checkbox if checked.
    pub async fn uncheck(&self, options: Option<CheckOptions>) -> Result<()> {
        self.set_checked(false, options).await
    }

    /// Set the checked state of a checkbox/radio, clicking only if needed.
    pub async fn set_checked(&self, checked: bool, options: Option<CheckOptions>) -> Result<()> {
        let opts = options.unwrap_or_default();
        let oid = self.resolve().await?;
        let result = set_checked_element(&self.page, &oid, checked, &opts).await;
        self.release(&oid).await;
        result
    }

    /// Select an `<option>` by value/label/index. Returns the now-selected values.
    pub async fn select_option(
        &self,
        value: impl Into<SelectOption>,
        _options: Option<SelectOptions>,
    ) -> Result<Vec<String>> {
        let value = value.into();
        let arg: Value = match value {
            SelectOption::Value(v) => json!({ "Value": v }),
            SelectOption::Label(v) => json!({ "Label": v }),
            SelectOption::Index(i) => json!({ "Index": i }),
        };
        let oid = self.resolve().await?;
        let result = select_option_element(&self.page, &oid, arg).await;
        self.release(&oid).await;
        result
    }

    /// Set files on an `<input type=file>` by server-side path(s).
    pub async fn set_input_files(&self, files: &[&str]) -> Result<()> {
        let oid = self.resolve().await?;
        // Resolve the remote object to a DOM nodeId via DOM.describeNode.
        let desc = self
            .page
            .session()
            .send("DOM.requestNode", json!({ "objectId": oid }))
            .await?;
        let node_id = desc
            .get("nodeId")
            .and_then(|v| v.as_i64())
            .ok_or_else(|| Error::ProtocolError("requestNode missing nodeId".into()))?;
        let r = self
            .page
            .session()
            .send(
                "DOM.setFileInputFiles",
                json!({ "nodeId": node_id, "files": files }),
            )
            .await
            .map(|_: Value| ());
        self.release(&oid).await;
        r
    }

    /// Tap (touch) the element. Approximated as a click on desktop CDP.
    pub async fn tap(&self, options: Option<ClickOptions>) -> Result<()> {
        self.click(options).await
    }

    /// Drag this element onto `target` via mouse motion.
    pub async fn drag_to(&self, target: &Locator, _options: Option<DragToOptions>) -> Result<()> {
        let oid = self.resolve().await?;
        let target_oid = target.resolve().await?;
        let (sx, sy) = element_center(&self.page, &oid).await?;
        let (tx, ty) = element_center(&target.page, &target_oid).await?;
        let r = drag_mouse(&self.page, sx, sy, tx, ty).await;
        target.release(&target_oid).await;
        self.release(&oid).await;
        r
    }

    /// Highlight this element in the page using CDP's `Overlay` domain.
    ///
    /// Resolves the element, maps it to a DOM `nodeId` via `DOM.requestNode`,
    /// then asks the browser to render a translucent box over it
    /// (`Overlay.highlightNode`). Highlighting is purely cosmetic for
    /// debugging: it should never fail an action, so any non-fatal CDP error
    /// is tolerated and logged away.
    pub async fn highlight(&self) -> Result<()> {
        let oid = self.resolve().await?;
        // Best-effort: the Overlay domain may not be enabled. Ignore failures.
        let _ = self
            .page
            .session()
            .send("Overlay.enable", json!({}))
            .await;
        let r = async {
            // Resolve the remote object to a DOM nodeId.
            let desc = self
                .page
                .session()
                .send("DOM.requestNode", json!({ "objectId": oid }))
                .await?;
            let node_id = desc
                .get("nodeId")
                .and_then(|v| v.as_i64())
                .ok_or_else(|| Error::ProtocolError("requestNode missing nodeId".into()))?;
            // A reasonable default highlight config: a translucent orange fill
            // with a contrasting border, matching DevTools' inspector style.
            self.page
                .session()
                .send(
                    "Overlay.highlightNode",
                    json!({
                        "highlightConfig": {
                            "showInfo": true,
                            "showStyles": false,
                            "contentColor": { "r": 111, "g": 168, "b": 220, "a": 0.66 },
                            "paddingColor": { "r": 147, "g": 196, "b": 125, "a": 0.55 },
                            "borderColor": { "r": 255, "g": 229, "b": 76, "a": 0.66 },
                            "marginColor": { "r": 246, "g": 178, "b": 107, "a": 0.66 }
                        },
                        "nodeId": node_id
                    }),
                )
                .await
                .map(|_: Value| ())
        }
        .await;
        self.release(&oid).await;
        // Highlight is cosmetic — swallow non-fatal errors but surface real ones
        // (e.g. element-not-found) only via the resolve above.
        let _ = r;
        Ok(())
    }

    /// If this element is an `<iframe>`, return a [`FrameLocator`] scoped to
    /// its content document.
    ///
    /// The iframe's content document is reached through its `contentDocument`,
    /// so this is **same-origin only** (a cross-origin iframe's
    /// `contentDocument` is `null` — use the frame's own target for those).
    /// The returned locator is built from the resolved iframe element rather
    /// than a selector, so it pins the specific element even if the page's DOM
    /// later shifts.
    pub async fn frame_locator(&self) -> Result<crate::frame_locator::FrameLocator> {
        let oid = self.resolve().await?;
        let fid = content_frame_id(&self.page, &oid).await;
        self.release(&oid).await;
        let fid = fid?;
        if fid.is_none() {
            return Err(Error::InvalidArgument(format!(
                "element '{}' is not an iframe",
                self.selector
            )));
        }
        // FrameLocator descends through a chain of iframe selectors resolved in
        // each frame's contentDocument. We resolved a concrete element, so
        // build a FrameLocator scoped to this iframe via a selector matching
        // it by the resolved frame id. Best-effort: if the frame id cannot be
        // expressed as a selector the engine understands, queries inside the
        // returned locator will report zero matches.
        let frame_selector = match fid {
            Some(fid) => format!("iframe >> internal:frame={fid}"),
            None => "iframe".to_string(),
        };
        Ok(crate::frame_locator::FrameLocator::new(
            self.page.clone(),
            frame_selector,
        ))
    }

    /// Drag this element and drop it onto `target` using CDP's drag events.
    ///
    /// Dispatches the HTML5 drag-and-drop event sequence
    /// (`dragEnter` → `dragOver` → `drop`) on the target, preceded by a
    /// `dragStart` on the source. This is a best-effort implementation: the
    /// page must have a drop handler that honors synthetic `DataTransfer`
    /// events. Apps relying on mouse-motion drag (rather than the DnD event
    /// API) should use [`Self::drag_to`] instead.
    pub async fn drop(&self, target: &Locator) -> Result<()> {
        let src_oid = self.resolve().await?;
        let _ = self
            .page
            .session()
            .send("DOM.scrollIntoViewIfNeeded", json!({ "objectId": src_oid }))
            .await;
        let (sx, sy) = element_center(&self.page, &src_oid).await?;

        let tgt_oid = target.resolve().await?;
        let _ = self
            .page
            .session()
            .send("DOM.scrollIntoViewIfNeeded", json!({ "objectId": tgt_oid }))
            .await;
        let (tx, ty) = element_center(&target.page, &tgt_oid).await?;

        // CDP's Input.dispatchDragEvent accepts {type, x, y, data: {items,...}}
        // and a drag event type. We synthesize a minimal empty data transfer.
        let data = json!({
            "items": [],
            "files": [],
        });
        let mut out = Ok(());
        // dragStart on the source.
        if let Err(e) = self
            .page
            .session()
            .send(
                "Input.dispatchDragEvent",
                json!({
                    "type": "dragStart",
                    "x": sx,
                    "y": sy,
                    "data": data
                }),
            )
            .await
        {
            out = Err(e);
        }
        // dragEnter / dragOver / drop on the target (all at the target center).
        for ev in ["dragEnter", "dragOver", "drop"] {
            if out.is_err() {
                break;
            }
            if let Err(e) = self
                .page
                .session()
                .send(
                    "Input.dispatchDragEvent",
                    json!({
                        "type": ev,
                        "x": tx,
                        "y": ty,
                        "data": data
                    }),
                )
                .await
            {
                out = Err(e);
            }
        }
        target.release(&tgt_oid).await;
        self.release(&src_oid).await;
        out
    }

    // --- more info ---

    pub async fn is_hidden(&self) -> Result<bool> {
        Ok(!self.is_visible().await?)
    }

    pub async fn is_disabled(&self) -> Result<bool> {
        let oid = self.resolve().await?;
        let v = selectors::eval_object(
            self.page.session(),
            &oid,
            "(el) => el.disabled === true",
            Value::Null,
        )
        .await?;
        self.release(&oid).await;
        Ok(v.as_bool().unwrap_or(false))
    }

    pub async fn is_focused(&self) -> Result<bool> {
        let oid = self.resolve().await?;
        let v = selectors::eval_object(
            self.page.session(),
            &oid,
            "(el) => el === document.activeElement",
            Value::Null,
        )
        .await?;
        self.release(&oid).await;
        Ok(v.as_bool().unwrap_or(false))
    }

    /// All matching locators (one per resolved element).
    pub async fn all(&self) -> Result<Vec<Locator>> {
        let n = self.count().await?;
        let mut out = Vec::with_capacity(n);
        for i in 0..n {
            let mut l = self.clone();
            l.strict = false;
            l.forced_index = Some(i);
            out.push(l);
        }
        Ok(out)
    }

    pub async fn all_inner_texts(&self) -> Result<Vec<String>> {
        let n = self.count().await?;
        let mut out = Vec::with_capacity(n);
        for i in 0..n {
            if let Some(oid) = self.element_query(i).await? {
                let v = selectors::eval_object(
                    self.page.session(),
                    &oid,
                    "(el) => el.innerText",
                    Value::Null,
                )
                .await
                .ok();
                let _ = self
                    .page
                    .session()
                    .send("Runtime.releaseObject", json!({ "objectId": oid }))
                    .await;
                out.push(v.and_then(|x| x.as_str().map(String::from)).unwrap_or_default());
            }
        }
        Ok(out)
    }

    pub async fn all_text_contents(&self) -> Result<Vec<String>> {
        let n = self.count().await?;
        let mut out = Vec::with_capacity(n);
        for i in 0..n {
            if let Some(oid) = self.element_query(i).await? {
                let v = selectors::eval_object(
                    self.page.session(),
                    &oid,
                    "(el) => el.textContent",
                    Value::Null,
                )
                .await
                .ok();
                let _ = self
                    .page
                    .session()
                    .send("Runtime.releaseObject", json!({ "objectId": oid }))
                    .await;
                out.push(v.and_then(|x| x.as_str().map(String::from)).unwrap_or_default());
            }
        }
        Ok(out)
    }

    /// Capture a screenshot clipped to this element.
    pub async fn screenshot(&self, opts: Option<ScreenshotOptions>) -> Result<Vec<u8>> {
        let oid = self.resolve().await?;
        let (x, y, w, h) = element_box(&self.page, &oid).await?;
        self.release(&oid).await;

        let opts = opts.unwrap_or_default();
        let format = match opts.r#type.unwrap_or_default() {
            crate::types::ScreenshotType::Png => "png",
            crate::types::ScreenshotType::Jpeg => "jpeg",
            crate::types::ScreenshotType::Webp => "webp",
        };
        let mut params = json!({
            "format": format,
            "clip": { "x": x, "y": y, "width": w, "height": h, "scale": 1 },
        });
        if opts.omit_background.unwrap_or(false) && format == "png" {
            params["omitBackground"] = json!(true);
        }
        if let Some(q) = opts.quality {
            if format == "jpeg" {
                params["quality"] = json!(q);
            }
        }
        let resp = self
            .page
            .session()
            .send("Page.captureScreenshot", params)
            .await?;
        let data = resp
            .get("data")
            .and_then(|v| v.as_str())
            .ok_or_else(|| Error::ProtocolError("screenshot missing data".into()))?;
        let bytes = base64::engine::general_purpose::STANDARD
            .decode(data)
            .map_err(|e| Error::ProtocolError(format!("base64 decode: {e}")))?;
        if let Some(path) = &opts.path {
            tokio::fs::write(path, &bytes).await?;
        }
        Ok(bytes)
    }

    // --- composition ---

    /// Filter matches by text presence (minimal).
    pub fn filter(&self, options: FilterOptions) -> Locator {
        let mut sel = self.selector.clone();
        if let Some(t) = options.has_text {
            sel.push_str(&format!(" >> text={t}"));
        }
        let mut l = self.clone();
        l.selector = sel;
        l
    }

    /// Chain this locator with another (both must match, in order).
    pub fn and_(&self, other: &Locator) -> Locator {
        let mut l = self.clone();
        l.selector = format!("{} >> {}", self.selector, other.selector);
        l
    }

    /// Union with another locator (either matches).
    pub fn or_(&self, other: &Locator) -> Locator {
        let mut l = self.clone();
        l.selector = format!("{}, {}", self.selector, other.selector);
        l
    }

    // --- locator-relative chaining ---

    fn chain(&self, sub: impl Into<String>) -> Locator {
        let mut l = self.clone();
        l.selector = format!("{} >> {}", self.selector, sub.into());
        l.forced_index = None;
        l.strict = true;
        l
    }

    /// Find a descendant matching `selector`.
    pub fn locator(&self, selector: impl Into<String>) -> Locator {
        self.chain(selector)
    }

    pub fn get_by_text(&self, text: &str, _exact: bool) -> Locator {
        self.chain(format!("text={text}"))
    }

    pub fn get_by_label(&self, text: &str) -> Locator {
        self.chain(format!("[aria-label=\"{}\"]", esc(text)))
    }

    pub fn get_by_placeholder(&self, text: &str) -> Locator {
        self.chain(format!("[placeholder=\"{}\"]", esc(text)))
    }

    pub fn get_by_alt_text(&self, text: &str) -> Locator {
        self.chain(format!("[alt=\"{}\"]", esc(text)))
    }

    pub fn get_by_title(&self, text: &str) -> Locator {
        self.chain(format!("[title=\"{}\"]", esc(text)))
    }

    pub fn get_by_test_id(&self, test_id: &str) -> Locator {
        self.chain(format!("[data-testid=\"{}\"]", esc(test_id)))
    }

    pub fn get_by_role(
        &self,
        role: crate::types::AriaRole,
        opts: Option<crate::options::GetByRoleOptions>,
    ) -> Locator {
        let opts = opts.unwrap_or_default();
        let mut sel = format!("role={}", role.as_str());
        if let Some(name) = &opts.name {
            sel.push_str(&format!("[name=\"{name}\"]"));
        }
        if opts.exact == Some(true) {
            sel.push_str("[exact=\"true\"]");
        }
        self.chain(sel)
    }

    // --- aria snapshot ---

    /// Capture an aria-snapshot (Playwright's YAML-ish accessibility-tree
    /// format) rooted at this locator's element.
    ///
    /// Resolves the element to its `RemoteObjectId`, then asks CDP for the
    /// partial accessibility tree rooted there (`fetchRelatives: false`) and
    /// serializes it via [`crate::aria_snapshot`].
    pub async fn aria_snapshot(&self) -> Result<String> {
        let oid = self.resolve().await?;

        // Resolve the element's backend DOM node id, which is how AX nodes link
        // back to the DOM. We then fetch the full AX tree and render the
        // subtree rooted at the matching node. (CDP's `getPartialAXTree` with
        // `fetchRelatives: false` returns only the single node, with no
        // descendants — insufficient for a useful snapshot.)
        let desc = self
            .page
            .session()
            .send("DOM.describeNode", json!({ "objectId": oid }))
            .await;
        self.release(&oid).await;
        let desc = desc?;
        let backend_node_id = desc
            .get("node")
            .and_then(|n| n.get("backendNodeId"))
            .and_then(|v| v.as_i64())
            .ok_or_else(|| {
                Error::ProtocolError("DOM.describeNode missing backendNodeId".into())
            })?;

        // Best-effort: some Chrome builds need the Accessibility domain enabled.
        let _ = self
            .page
            .session()
            .send("Accessibility.enable", json!({}))
            .await;
        let resp = self
            .page
            .session()
            .send("Accessibility.getFullAXTree", json!({}))
            .await?;
        let nodes = resp
            .get("nodes")
            .and_then(|v| v.as_array())
            .cloned()
            .unwrap_or_default();
        Ok(crate::aria_snapshot::serialize(&nodes, Some(backend_node_id)))
    }
}

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

// --- element-level action helpers ---

/// Compute an element's content-box as `(x, y, width, height)` where `(x, y)`
/// is the top-left corner.
pub(crate) async fn element_box(page: &Page, object_id: &str) -> Result<(f64, f64, f64, f64)> {
    let resp = page
        .session()
        .send("DOM.getBoxModel", json!({ "objectId": object_id }))
        .await?;
    let model = resp
        .get("model")
        .ok_or_else(|| Error::ProtocolError("getBoxModel missing model".into()))?;
    let (x, y, w, h) = quad_box(model.get("content"))
        .ok_or_else(|| Error::ProtocolError("getBoxModel content quad malformed".into()))?;
    // Prefer the model's own dimensions when available.
    let mw = model.get("width").and_then(|v| v.as_f64());
    let mh = model.get("height").and_then(|v| v.as_f64());
    Ok((x, y, mw.unwrap_or(w), mh.unwrap_or(h)))
}

/// Center of an element's content box.
pub(crate) async fn element_center(page: &Page, object_id: &str) -> Result<(f64, f64)> {
    let (x, y, w, h) = element_box(page, object_id).await?;
    Ok((x + w / 2.0, y + h / 2.0))
}

/// Resolve the CDP content-frame id of an `<iframe>`/`<frame>` element.
/// Returns `Ok(Some(frame_id))` if the element is a frame with a content
/// document, `Ok(None)` if it is a frame but has no resolvable content frame,
/// and `Err` only if the node description itself fails. Callers decide whether
/// a missing frame id is an error.
pub(crate) async fn content_frame_id(page: &Page, object_id: &str) -> Result<Option<String>> {
    let desc = page
        .session()
        .send("DOM.describeNode", json!({ "objectId": object_id }))
        .await?;
    let node = desc
        .get("node")
        .ok_or_else(|| Error::ProtocolError("DOM.describeNode missing node".into()))?;
    let is_frame = node
        .get("localName")
        .and_then(|v| v.as_str())
        .map(|name| name.eq_ignore_ascii_case("iframe") || name.eq_ignore_ascii_case("frame"))
        .unwrap_or(false);
    if !is_frame {
        return Ok(None);
    }
    let fid = node
        .get("contentDocument")
        .and_then(|c| c.get("frameId"))
        .and_then(|v| v.as_str())
        .map(String::from);
    Ok(fid)
}

/// Drag the mouse from `(sx, sy)` to `(tx, ty)`: press, move in 10 steps,
/// release. Shared by `Locator::drag_to` and `ElementHandle::drag_and_drop`.
/// Callers handle scrolling the source/target into view first.
pub(crate) async fn drag_mouse(
    page: &Page,
    sx: f64,
    sy: f64,
    tx: f64,
    ty: f64,
) -> Result<()> {
    page.session()
        .send(
            "Input.dispatchMouseEvent",
            json!({ "type": "mouseMoved", "x": sx, "y": sy, "button": "none", "buttons": 0 }),
        )
        .await?;
    page.session()
        .send(
            "Input.dispatchMouseEvent",
            json!({ "type": "mousePressed", "x": sx, "y": sy, "button": "left", "buttons": 1, "clickCount": 1 }),
        )
        .await?;
    let steps = 10;
    for i in 1..=steps {
        let x = sx + (tx - sx) * (i as f64) / (steps as f64);
        let y = sy + (ty - sy) * (i as f64) / (steps as f64);
        page.session()
            .send(
                "Input.dispatchMouseEvent",
                json!({ "type": "mouseMoved", "x": x, "y": y, "button": "none", "buttons": 1 }),
            )
            .await?;
    }
    page.session()
        .send(
            "Input.dispatchMouseEvent",
            json!({ "type": "mouseReleased", "x": tx, "y": ty, "button": "left", "buttons": 0, "clickCount": 1 }),
        )
        .await
        .map(|_: Value| ())
}

/// Type each character of `text` as a `keyDown`/`keyUp` pair, optionally
/// sleeping `delay` between keystrokes. Reuses the same `Input.dispatchKeyEvent`
/// path as `Locator::press`. The caller is responsible for focusing the element
/// first.
pub(crate) async fn press_text(
    page: &Page,
    _object_id: &str,
    text: &str,
    delay: Option<Duration>,
) -> Result<()> {
    for ch in text.chars() {
        // A single logical key may be multiple UTF-16 units; CDP wants the
        // text payload in `text` and, for printable chars, the char as `key`.
        let key = ch.to_string();
        page.session()
            .send(
                "Input.dispatchKeyEvent",
                json!({
                    "type": "char",
                    "text": key,
                    "key": key,
                }),
            )
            .await
            .map(|_: Value| ())?;
        if let Some(d) = delay {
            tokio::time::sleep(d).await;
        }
    }
    Ok(())
}

/// Set a checkbox/radio to `checked`, clicking only when the current state
/// differs. Shared by `Locator::set_checked` and `ElementHandle::set_checked`.
pub(crate) async fn set_checked_element(
    page: &Page,
    object_id: &str,
    checked: bool,
    opts: &CheckOptions,
) -> Result<()> {
    // Read current checked state from the element directly.
    let v = selectors::eval_object(
        page.session(),
        object_id,
        "(el) => el.checked === true",
        Value::Null,
    )
    .await?;
    let current = v.as_bool().unwrap_or(false);
    if current == checked {
        return Ok(());
    }
    let click_opts = ClickOptions {
        force: opts.force,
        position: opts.position,
        timeout: opts.timeout,
        trial: opts.trial,
        ..Default::default()
    };
    click_element(page, object_id, &click_opts).await
}

/// Hover over an element: scroll it in, then move the mouse to its center
/// (or the configured offset). Shared by `Locator::hover` and `ElementHandle::hover`.
pub(crate) async fn hover_element(page: &Page, object_id: &str, opts: &HoverOptions) -> Result<()> {
    let _ = page
        .session()
        .send("DOM.scrollIntoViewIfNeeded", json!({ "objectId": object_id }))
        .await;
    let (bx, by, bw, bh) = element_box(page, object_id).await?;
    let (x, y) = match &opts.position {
        Some(p) => (bx + p.x, by + p.y),
        None => (bx + bw / 2.0, by + bh / 2.0),
    };
    page.session()
        .send(
            "Input.dispatchMouseEvent",
            json!({ "type": "mouseMoved", "x": x, "y": y, "button": "none", "buttons": 0 }),
        )
        .await
        .map(|_: Value| ())
}

/// Press a single key against the currently-focused element. Shared by
/// `Locator::press` and `ElementHandle::press`.
pub(crate) async fn press_key(
    page: &Page,
    object_id: &str,
    key: &str,
    delay: Option<Duration>,
) -> Result<()> {
    let _ = page
        .session()
        .send("DOM.focus", json!({ "objectId": object_id }))
        .await;
    page.session()
        .send("Input.dispatchKeyEvent", json!({ "type": "keyDown", "key": key }))
        .await
        .map(|_: Value| ())?;
    if let Some(d) = delay {
        tokio::time::sleep(d).await;
    }
    page.session()
        .send("Input.dispatchKeyEvent", json!({ "type": "keyUp", "key": key }))
        .await
        .map(|_: Value| ())
}

/// Select an `<option>` on a `<select>`. `arg` is one of `{Value}`/`{Label}`/
/// `{Index}`. Returns the now-selected option values. Shared by
/// `Locator::select_option` and `ElementHandle::select_option`.
pub(crate) async fn select_option_element(
    page: &Page,
    object_id: &str,
    arg: Value,
) -> Result<Vec<String>> {
    let v: Value = selectors::eval_object(
        page.session(),
        object_id,
        "(el, sel) => {
            const opts = Array.from(el.options || []);
            const matched = [];
            for (const o of opts) {
                const hit = sel.Value != null ? o.value === sel.Value
                        : sel.Label != null ? o.label === sel.Label
                        : sel.Index != null ? o.index === sel.Index : false;
                if (hit) { o.selected = true; matched.push(o.value); }
            }
            el.dispatchEvent(new Event('input', { bubbles: true }));
            el.dispatchEvent(new Event('change', { bubbles: true }));
            return matched;
        }",
        arg,
    )
    .await?;
    Ok(v.as_array()
        .map(|a| a.iter().filter_map(|x| x.as_str().map(String::from)).collect())
        .unwrap_or_default())
}

/// Wait for an element to be actionable (visible) unless skipping via `force`.
async fn wait_actionable(
    page: &Page,
    object_id: &str,
    timeout: Duration,
) -> Result<()> {
    let function = "(el) => self.__pwcdpInjected && self.__pwcdpInjected.elementState(el).visible";
    let deadline = tokio::time::Instant::now() + timeout;
    loop {
        let v = selectors::eval_object(page.session(), object_id, function, Value::Null).await?;
        if v.as_bool() == Some(true) {
            return Ok(());
        }
        if tokio::time::Instant::now() >= deadline {
            return Err(Error::Timeout(format!(
                "element not visible within {}ms",
                timeout.as_millis()
            )));
        }
        tokio::time::sleep(Duration::from_millis(100)).await;
    }
}

fn modifier_key(m: &crate::types::KeyboardModifier) -> &'static str {
    use crate::types::KeyboardModifier::*;
    match m {
        Alt => "Alt",
        Control => "Control",
        ControlOrMeta => "Control",
        Meta => "Meta",
        Shift => "Shift",
    }
}

/// Dispatch a mouse click (press + release) honoring the full `ClickOptions`.
/// Fixes the historical `buttons`-bitmask bug: press uses the button's bitmask,
/// release uses `0`.
pub(crate) async fn click_element(
    page: &Page,
    object_id: &str,
    opts: &ClickOptions,
) -> Result<()> {
    let _ = page
        .session()
        .send("DOM.scrollIntoViewIfNeeded", json!({ "objectId": object_id }))
        .await;

    let force = opts.force.unwrap_or(false);
    let trial = opts.trial.unwrap_or(false);
    if !force {
        let timeout = opts
            .timeout
            .map(|ms| Duration::from_millis(ms as u64))
            .unwrap_or_else(|| page.default_timeout());
        wait_actionable(page, object_id, timeout).await?;
    }

    let (bx, by, bw, bh) = element_box(page, object_id).await?;
    let (x, y) = match &opts.position {
        Some(p) => (bx + p.x, by + p.y),
        None => (bx + bw / 2.0, by + bh / 2.0),
    };
    let button = opts.button.unwrap_or_default();
    let count = opts.click_count.unwrap_or(1);
    let delay = opts.delay.map(|ms| Duration::from_millis(ms as u64));

    // Hold modifiers.
    if let Some(mods) = &opts.modifiers {
        for m in mods {
            let _ = page
                .session()
                .send(
                    "Input.dispatchKeyEvent",
                    json!({ "type": "keyDown", "key": modifier_key(m) }),
                )
                .await;
        }
    }

    let result = if trial {
        Ok(())
    } else {
        let mut out = Ok(());
        page.session()
            .send(
                "Input.dispatchMouseEvent",
                json!({ "type": "mouseMoved", "x": x, "y": y, "button": "none", "buttons": 0 }),
            )
            .await?;
        for i in 0..count {
            if let Err(e) = page
                .session()
                .send(
                    "Input.dispatchMouseEvent",
                    json!({ "type": "mousePressed", "x": x, "y": y, "button": button.as_str(), "buttons": button.bitmask(), "clickCount": i + 1 }),
                )
                .await
            {
                out = Err(e);
                break;
            }
            if let Err(e) = page
                .session()
                .send(
                    "Input.dispatchMouseEvent",
                    json!({ "type": "mouseReleased", "x": x, "y": y, "button": button.as_str(), "buttons": 0, "clickCount": i + 1 }),
                )
                .await
            {
                out = Err(e);
                break;
            }
            if let Some(d) = delay {
                if i + 1 < count {
                    tokio::time::sleep(d).await;
                }
            }
        }
        out
    };

    // Release modifiers.
    if let Some(mods) = &opts.modifiers {
        for m in mods {
            let _ = page
                .session()
                .send(
                    "Input.dispatchKeyEvent",
                    json!({ "type": "keyUp", "key": modifier_key(m) }),
                )
                .await;
        }
    }

    result
}

fn quad_box(quad: Option<&Value>) -> Option<(f64, f64, f64, f64)> {
    let arr = quad.and_then(|v| v.as_array())?;
    let nums: Vec<f64> = arr.iter().filter_map(|v| v.as_f64()).collect();
    if nums.len() < 8 {
        return None;
    }
    let xs = [nums[0], nums[2], nums[4], nums[6]];
    let ys = [nums[1], nums[3], nums[5], nums[7]];
    let xmin = xs.iter().cloned().fold(f64::INFINITY, f64::min);
    let xmax = xs.iter().cloned().fold(f64::NEG_INFINITY, f64::max);
    let ymin = ys.iter().cloned().fold(f64::INFINITY, f64::min);
    let ymax = ys.iter().cloned().fold(f64::NEG_INFINITY, f64::max);
    Some((xmin, ymin, xmax - xmin, ymax - ymin))
}