script/dom/
customstateset.rs

1/* This Source Code Form is subject to the terms of the Mozilla Public
2 * License, v. 2.0. If a copy of the MPL was not distributed with this
3 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
4
5use dom_struct::dom_struct;
6use indexmap::IndexSet;
7use script_bindings::codegen::GenericBindings::ElementInternalsBinding::CustomStateSetMethods;
8use script_bindings::like::Setlike;
9use script_bindings::root::{Dom, DomRoot};
10use script_bindings::script_runtime::CanGc;
11use script_bindings::str::DOMString;
12use script_bindings::trace::CustomTraceable;
13
14use crate::dom::bindings::cell::DomRefCell;
15use crate::dom::bindings::inheritance::Castable;
16use crate::dom::bindings::reflector::{Reflector, reflect_dom_object};
17use crate::dom::html::htmlelement::HTMLElement;
18use crate::dom::node::{Node, NodeDamage};
19use crate::dom::window::Window;
20
21/// <https://html.spec.whatwg.org/multipage/#customstateset>
22#[dom_struct]
23pub(crate) struct CustomStateSet {
24    reflector: Reflector,
25    internal: DomRefCell<IndexSet<DOMString>>,
26    owner_element: Dom<HTMLElement>,
27}
28
29impl CustomStateSet {
30    fn new_inherited(element: &HTMLElement) -> Self {
31        Self {
32            reflector: Reflector::new(),
33            internal: DomRefCell::new(Default::default()),
34            owner_element: Dom::from_ref(element),
35        }
36    }
37
38    pub(crate) fn new(window: &Window, element: &HTMLElement, can_gc: CanGc) -> DomRoot<Self> {
39        reflect_dom_object(Box::new(Self::new_inherited(element)), window, can_gc)
40    }
41
42    /// Returns a borrowed version of the set without the usual Ref wrapper.
43    /// Mutating the underlying refcell while this value is active is
44    /// undefined behaviour.
45    #[expect(unsafe_code)]
46    pub(crate) unsafe fn set_for_layout(&self) -> &IndexSet<DOMString> {
47        unsafe { self.internal.borrow_for_layout() }
48    }
49
50    fn states_did_change(&self) {
51        self.owner_element.upcast::<Node>().dirty(NodeDamage::Other);
52    }
53}
54
55impl Setlike for CustomStateSet {
56    type Key = DOMString;
57
58    #[inline(always)]
59    fn get_index(&self, index: u32) -> Option<Self::Key> {
60        self.internal.get_index(index)
61    }
62
63    #[inline(always)]
64    fn size(&self) -> u32 {
65        self.internal.size()
66    }
67
68    #[inline(always)]
69    fn add(&self, key: Self::Key) {
70        self.internal.add(key);
71        self.states_did_change();
72    }
73
74    #[inline(always)]
75    fn has(&self, key: Self::Key) -> bool {
76        self.internal.has(key)
77    }
78
79    #[inline(always)]
80    fn clear(&self) {
81        let old_size = self.internal.size();
82        self.internal.clear();
83        if old_size != 0 {
84            self.states_did_change();
85        }
86    }
87
88    #[inline(always)]
89    fn delete(&self, key: Self::Key) -> bool {
90        if self.internal.delete(key) {
91            self.states_did_change();
92            true
93        } else {
94            false
95        }
96    }
97}
98
99impl CustomStateSetMethods<crate::DomTypeHolder> for CustomStateSet {
100    /// <https://html.spec.whatwg.org/multipage/#customstateset>
101    fn Size(&self) -> u32 {
102        self.internal.size()
103    }
104}
script/dom/customstateset.rs

script/dom/
customstateset.rs
