script/dom/node/

focus.rs

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use std::collections::VecDeque;

use html5ever::local_name;
use js::context::JSContext;
use script_bindings::codegen::GenericBindings::ShadowRootBinding::ShadowRootMethods;
use script_bindings::inheritance::Castable;
use script_bindings::root::{Dom, DomRoot};

use crate::dom::document::focus::{FocusableArea, FocusableAreaKind};
use crate::dom::iterators::ShadowIncluding;
use crate::dom::node::iterators::TreeIterator;
use crate::dom::types::{
    Element, HTMLDialogElement, HTMLIFrameElement, HTMLSlotElement, ShadowRoot,
};
use crate::dom::{Document, Node, NodeTraits};

/// <https://html.spec.whatwg.org/multipage/#focus-navigation-scope-owner>
///
/// This enum represents the "focus navigation scope owner" in Servo.
pub(crate) enum FocusNavigationScopeOwner {
    Document(DomRoot<Document>),
    ShadowHost {
        shadow_host: DomRoot<Element>,
        shadow_root: DomRoot<ShadowRoot>,
    },
    Slot(DomRoot<HTMLSlotElement>),
}

impl FocusNavigationScopeOwner {
    pub(crate) fn iterator(&self) -> FocusNavigationScopeIterator {
        let iterators = match self {
            Self::Document(document) => VecDeque::from([document
                .upcast::<Node>()
                .traverse_preorder(ShadowIncluding::No)]),
            Self::ShadowHost { shadow_root, .. } => VecDeque::from([shadow_root
                .upcast::<Node>()
                .traverse_preorder(ShadowIncluding::No)]),
            Self::Slot(html_slot_element) => html_slot_element
                .assigned_nodes()
                .iter()
                .map(|slottable| slottable.node().traverse_preorder(ShadowIncluding::No))
                .collect(),
        };

        FocusNavigationScopeIterator { iterators }
    }

    /// Returns the `Node` that backs this [`FocusNavigationScopeOwner`]. This is a node that
    /// can be found in the containing focus navigation scope, so a traversal on it will
    /// traverse the nodes in the scope.
    pub(crate) fn node(&self) -> &Node {
        match self {
            FocusNavigationScopeOwner::Document(document) => document.upcast(),
            FocusNavigationScopeOwner::ShadowHost { shadow_host, .. } => shadow_host.upcast(),
            FocusNavigationScopeOwner::Slot(html_slot_element) => html_slot_element.upcast(),
        }
    }
}

pub(crate) struct FocusNavigationScopeIterator {
    iterators: VecDeque<TreeIterator>,
}

impl Iterator for FocusNavigationScopeIterator {
    type Item = DomRoot<Node>;

    fn next(&mut self) -> Option<Self::Item> {
        let should_skip_element_children = |element: &Element| {
            element.is_shadow_host() ||
                element
                    .downcast::<HTMLSlotElement>()
                    .is_some_and(|html_slot_element| html_slot_element.has_assigned_nodes())
        };

        while !self.iterators.is_empty() {
            if let Some(next) = self.iterators.front_mut().and_then(|front| {
                let should_skip_children_on_next_iteration = front
                    .peek()
                    .and_then(|node| node.downcast::<Element>())
                    .is_some_and(should_skip_element_children);
                if should_skip_children_on_next_iteration {
                    front.next_skipping_children()
                } else {
                    front.next()
                }
            }) {
                return Some(next);
            }

            self.iterators.pop_front();
        }
        None
    }
}

#[derive(Clone, Copy, PartialEq)]
pub(crate) enum FocusTrigger {
    Click,
    Other,
}

impl Node {
    /// Returns the appropriate [`FocusableArea`] when this [`Node`] is clicked on according to
    /// <https://www.w3.org/TR/pointerevents4/#handle-native-mouse-down>.
    ///
    /// Note that this is doing more than the specification which says to only take into account
    /// the node from the hit test. This isn't exactly how browsers work though, as they seem
    /// to look for the first inclusive ancestor node that has a focusable area associated with it.
    /// Note also that this may return [`FocusableArea::Viewport`].
    pub(crate) fn find_click_focusable_area(&self, cx: &JSContext) -> FocusableArea {
        self.inclusive_ancestors(ShadowIncluding::Yes)
            .find_map(|node| {
                node.get_the_focusable_area(cx, FocusTrigger::Click)
                    .filter(|focusable_area| {
                        focusable_area.kind().contains(FocusableAreaKind::Click)
                    })
            })
            .unwrap_or(FocusableArea::Viewport)
    }

    /// <https://html.spec.whatwg.org/multipage/#get-the-focusable-area>
    ///
    /// There seems to be hole in the specification here. It describes how to get the focusable
    /// area for a focus target that isn't a focuable area, but is ambiguous about how to do
    /// this for a focus target that actually *is* a focusable area. The obvious thing is to
    /// just return the focus target, but it's still odd that this isn't mentioned in the
    /// specification.
    pub(crate) fn get_the_focusable_area(
        &self,
        cx: &JSContext,
        focus_trigger: FocusTrigger,
    ) -> Option<FocusableArea> {
        let kind = self
            .downcast::<Element>()
            .map(|element| Element::focusable_area_kind(element, cx.no_gc()))
            .unwrap_or_default();
        if !kind.is_empty() {
            if let Some(iframe_element) = self.downcast::<HTMLIFrameElement>() {
                return Some(FocusableArea::IFrameViewport {
                    iframe_element: Dom::from_ref(iframe_element),
                    kind,
                });
            }

            return Some(FocusableArea::Node {
                node: Dom::from_ref(self),
                kind,
            });
        }

        self.get_the_focusable_area_if_not_a_focusable_area(cx, focus_trigger)
    }

    /// <https://html.spec.whatwg.org/multipage/#get-the-focusable-area>
    ///
    /// In addition to returning the DOM anchor of the focusable area for this [`Node`], this
    /// method also returns the [`FocusableAreaKind`] for efficiency reasons. Note that `None`
    /// is returned if this [`Node`] does not have a focusable area or if its focusable area
    /// is the `Document`'s viewport.
    ///
    /// TODO: It might be better to distinguish these two cases in the future.
    fn get_the_focusable_area_if_not_a_focusable_area(
        &self,
        cx: &JSContext,
        focus_trigger: FocusTrigger,
    ) -> Option<FocusableArea> {
        // > To get the focusable area for a focus target that is either an element that is not a
        // > focusable area, or is a navigable, given an optional string focus trigger (default
        // > "other"), run the first matching set of steps from the following list:
        //
        // > ↪ If focus target is an area element with one or more shapes that are focusable areas
        // >     Return the shape corresponding to the first img element in tree order that uses the image
        // >     map to which the area element belongs.
        // TODO: Implement this.

        // > ↪ If focus target is an element with one or more scrollable regions that are focusable areas
        // >     Return the element's first scrollable region, according to a pre-order, depth-first
        // >     traversal of the flat tree. [CSSSCOPING]
        // TODO: Implement this.

        // > ↪ If focus target is the document element of its Document
        // >     Return the Document's viewport.
        if self == self.owner_document().upcast::<Node>() {
            return Some(FocusableArea::Viewport);
        }

        // > ↪ If focus target is a navigable
        // >     Return the navigable's active document.
        // TODO: Implement this.

        // > ↪ If focus target is a navigable container with a non-null content navigable
        // >     Return the navigable container's content navigable's active document.
        // TODO: Implement this.

        // > ↪ If focus target is a shadow host whose shadow root's delegates focus is true
        if self
            .downcast::<Element>()
            .and_then(Element::shadow_root)
            .is_some_and(|shadow_root| shadow_root.DelegatesFocus())
        {
            // >   Step 1. Let focusedElement be the currently focused area of a top-level
            // >           traversable's DOM anchor.
            //
            // Note: This is a bit of a misnomer, because it might be a Node and not an Element.
            let document = self.owner_document();
            let focused_area = document.focus_handler().focused_area();
            let focused_element = focused_area.dom_anchor(&document);

            // >   Step 2. If focus target is a shadow-including inclusive ancestor of
            // >           focusedElement, then return focusedElement.
            if self
                .upcast::<Node>()
                .is_shadow_including_inclusive_ancestor_of(&focused_element)
            {
                return Some(focused_area.clone());
            }

            // >   Step 3. Return the focus delegate for focus target given focus trigger.
            return self.focus_delegate(cx, Some(focus_trigger));
        }

        None
    }

    /// <https://html.spec.whatwg.org/multipage/#focus-delegate>
    ///
    /// In addition to returning the focus delegate for this [`Node`], this method also returns
    /// the [`FocusableAreaKind`] for efficiency reasons.
    pub(crate) fn focus_delegate(
        &self,
        cx: &JSContext,
        focus_trigger: Option<FocusTrigger>,
    ) -> Option<FocusableArea> {
        let focus_trigger = focus_trigger.unwrap_or(FocusTrigger::Other);
        // > 1. If focusTarget is a shadow host and its shadow root's delegates focus is false, then
        // >    return null.
        let shadow_root = self.downcast::<Element>().and_then(Element::shadow_root);
        if shadow_root
            .as_ref()
            .is_some_and(|shadow_root| !shadow_root.DelegatesFocus())
        {
            return None;
        }

        // > 2. Let whereToLook be focusTarget.
        let mut where_to_look = self.upcast::<Node>();

        // > 3. If whereToLook is a shadow host, then set whereToLook to whereToLook's shadow root.
        if let Some(shadow_root) = shadow_root.as_ref() {
            where_to_look = shadow_root.upcast();
        }

        // > 4. Let autofocusDelegate be the autofocus delegate for whereToLook given focusTrigger.
        rooted!(&in(cx) let autofocus_delegate = where_to_look.autofocus_delegate(cx, focus_trigger));

        // > 5. If autofocusDelegate is not null, then return autofocusDelegate.
        if autofocus_delegate.is_some() {
            return autofocus_delegate.clone();
        }

        // > 6. For each descendant of whereToLook's descendants, in tree order:
        let is_dialog_element = self.is::<HTMLDialogElement>();
        for descendant in where_to_look.traverse_preorder(ShadowIncluding::No).skip(1) {
            // > 6.1. Let focusableArea be null.
            // Handled via early return.

            // > 6.2. If focusTarget is a dialog element and descendant is sequentially focusable, then
            // >      set focusableArea to descendant.
            let kind = descendant
                .downcast::<Element>()
                .map(|node| Element::focusable_area_kind(node, cx.no_gc()))
                .unwrap_or_default();
            if is_dialog_element && kind.contains(FocusableAreaKind::Sequential) {
                return Some(FocusableArea::Node {
                    node: descendant.as_traced(),
                    kind,
                });
            }

            // > 6.3. Otherwise, if focusTarget is not a dialog and descendant is a focusable area, set
            // >      focusableArea to descendant.
            if !kind.is_empty() {
                return Some(FocusableArea::Node {
                    node: descendant.as_traced(),
                    kind,
                });
            }

            // > 6.4. Otherwise, set focusableArea to the result of getting the focusable area for
            //        descendant given focusTrigger.
            rooted!(&in(cx) let focusable_area = descendant.get_the_focusable_area_if_not_a_focusable_area(
                cx,
                focus_trigger)
            );
            if let Some(focusable_area) = &*focusable_area {
                // > 6.5. If focusableArea is not null, then return focusableArea.
                return Some(focusable_area.clone());
            }
        }

        // > 7. Return null.
        None
    }

    /// <https://html.spec.whatwg.org/multipage/#autofocus-delegate>
    fn autofocus_delegate(
        &self,
        cx: &JSContext,
        focus_trigger: FocusTrigger,
    ) -> Option<FocusableArea> {
        // > 1. For each descendant descendant of focus target, in tree order:
        for descendant in self.traverse_preorder(ShadowIncluding::No).skip(1) {
            // > 1.1. If descendant does not have an autofocus content attribute, then continue.
            if !descendant
                .downcast::<Element>()
                .is_some_and(|element| element.has_attribute(&local_name!("autofocus")))
            {
                continue;
            }

            // > 1.2. Let focusable area be descendant, if descendant is a focusable area; otherwise
            // >      let focusable area be the result of getting the focusable area for descendant
            // >      given focus trigger.
            rooted!(&in(cx) let focusable_area = descendant.get_the_focusable_area(cx, focus_trigger));
            let Some(focusable_area) = &*focusable_area else {
                // > 1.3. If focusable area is null, then continue.
                continue;
            };

            // > 1.4. If focusable area is not click focusable and focus trigger is "click", then continue.
            if !focusable_area.kind().contains(FocusableAreaKind::Click) &&
                focus_trigger == FocusTrigger::Click
            {
                continue;
            }

            // > 1.5. Return focusable area.
            return Some(focusable_area.clone());
        }

        // > 2. Return null.
        None
    }

    /// <https://html.spec.whatwg.org/multipage/#focusing-steps>
    ///
    /// This is an initial implementation of the "focusing steps" from the HTML specification. Note
    /// that this is currently in a state of transition from Servo's old internal focus APIs to ones
    /// that match the specification. That is why the arguments to this method do not match the
    /// specification yet.
    ///
    /// Return `true` if anything was focused or `false` otherwise.
    #[cfg_attr(crown, expect(crown::unrooted_must_root))]
    pub(crate) fn run_the_focusing_steps(
        &self,
        cx: &mut JSContext,
        fallback_target: Option<FocusableArea>,
        focus_trigger: FocusTrigger,
    ) -> bool {
        // > 1. If new focus target is not a focusable area, then set new focus target to the result
        // >    of getting the focusable area for new focus target, given focus trigger if it was
        // >    passed.
        // > 2. If new focus target is null, then:
        // > 2.1 If no fallback target was specified, then return.
        // > 2.2 Otherwise, set new focus target to the fallback target.
        rooted!(&in(cx) let mut focusable_area = self
            .get_the_focusable_area(cx, focus_trigger)
            .or(fallback_target)
        );
        if focusable_area.is_none() {
            return false;
        }

        // > 3. If new focus target is a navigable container with non-null content navigable, then
        // >    set new focus target to the content navigable's active document.
        // > 4. If new focus target is a focusable area and its DOM anchor is inert, then return.
        // > 5. If new focus target is the currently focused area of a top-level traversable, then
        // >    return.
        // > 6. Let old chain be the current focus chain of the top-level traversable in which new
        // >    focus target finds itself.
        // > 6.1. Let new chain be the focus chain of new focus target.
        // > 6.2. Run the focus update steps with old chain, new chain, and new focus target
        // >      respectively.
        //
        // TODO: Handle all of these steps by converting the focus transaction code to follow
        // the HTML focus specification.
        let document = self.owner_document();
        document
            .focus_handler()
            .focus(cx, &focusable_area.take().unwrap());
        true
    }

    /// If this node is a focus navigation scope owner, return the corresponding
    /// [`FocusNavigationScopeOwner`] that describes it or `None` if this node is not
    /// a focus navigation scope owner.
    ///
    /// <https://html.spec.whatwg.org/multipage/#focus-navigation-scope-owner>
    pub(crate) fn as_focus_navigation_scope_owner(&self) -> Option<FocusNavigationScopeOwner> {
        if let Some(element) = self.downcast::<Element>() {
            if let Some(shadow_root) = element.shadow_root() {
                return Some(FocusNavigationScopeOwner::ShadowHost {
                    shadow_host: DomRoot::from_ref(element),
                    shadow_root,
                });
            }

            if let Some(html_slot_element) = self.downcast::<HTMLSlotElement>() {
                // Only consider this `<slot>` element a focus navigation scope owner if
                // it has assigned slottables and isn't displaying fallback content.
                if html_slot_element.has_assigned_nodes() {
                    return Some(FocusNavigationScopeOwner::Slot(DomRoot::from_ref(
                        html_slot_element,
                    )));
                }
            }
        }

        Some(FocusNavigationScopeOwner::Document(DomRoot::from_ref(
            self.downcast::<Document>()?,
        )))
    }

    /// Find the focus navigation scope owner for this node. If this node is itself
    /// a focus navigation scope owner, this will return its containing focus navigation
    /// scope owner.
    ///
    /// This will return `None` if this node is the `Document` element.
    ///
    /// <https://html.spec.whatwg.org/multipage/#focus-navigation-scope-owner>
    pub(crate) fn containing_focus_navigation_scope_owner(
        &self,
    ) -> Option<FocusNavigationScopeOwner> {
        for ancestor in self.inclusive_ancestors(ShadowIncluding::No) {
            // When a slot has an attached shadow DOM it takes precedence so this comes before
            // the check for slot elements with assigned slots.
            if let Some(shadow_root) = ancestor.downcast::<ShadowRoot>() {
                return Some(FocusNavigationScopeOwner::ShadowHost {
                    shadow_host: shadow_root.Host(),
                    shadow_root: DomRoot::from_ref(shadow_root),
                });
            }

            if let Some(html_slot_element) = ancestor.assigned_slot() {
                return Some(FocusNavigationScopeOwner::Slot(html_slot_element));
            }
        }

        if self.is::<Document>() {
            return None;
        }
        Some(FocusNavigationScopeOwner::Document(self.owner_document()))
    }
}