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 atomic_refcell::{AtomicRef, AtomicRefCell};
6use dom_struct::dom_struct;
7use indexmap::IndexSet;
8use js::context::{JSContext, NoGC};
9use script_bindings::codegen::GenericBindings::ElementInternalsBinding::CustomStateSetMethods;
10use script_bindings::like::Setlike;
11use script_bindings::reflector::{Reflector, reflect_dom_object_with_cx};
12use script_bindings::root::{Dom, DomRoot};
13use script_bindings::str::DOMString;
14
15use crate::dom::bindings::inheritance::Castable;
16use crate::dom::html::htmlelement::HTMLElement;
17use crate::dom::node::{Node, NodeDamage};
18use crate::dom::window::Window;
19
20/// <https://html.spec.whatwg.org/multipage/#customstateset>
21#[dom_struct]
22pub(crate) struct CustomStateSet {
23    reflector: Reflector,
24    #[no_trace]
25    internal: AtomicRefCell<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: Default::default(),
34            owner_element: Dom::from_ref(element),
35        }
36    }
37
38    pub(crate) fn new(cx: &mut JSContext, window: &Window, element: &HTMLElement) -> DomRoot<Self> {
39        reflect_dom_object_with_cx(Box::new(Self::new_inherited(element)), window, cx)
40    }
41
42    pub(crate) fn set_for_layout(&self) -> AtomicRef<'_, IndexSet<DOMString>> {
43        self.internal.borrow()
44    }
45
46    fn states_did_change(&self, no_gc: &NoGC) {
47        self.owner_element
48            .upcast::<Node>()
49            .dirty(no_gc, NodeDamage::Other);
50    }
51}
52
53impl Setlike for CustomStateSet {
54    type Key = DOMString;
55
56    #[inline(always)]
57    fn get_index(&self, cx: &mut JSContext, index: u32) -> Option<Self::Key> {
58        self.internal.get_index(cx, index)
59    }
60
61    #[inline(always)]
62    fn size(&self, cx: &mut JSContext) -> u32 {
63        self.internal.size(cx)
64    }
65
66    #[inline(always)]
67    fn add(&self, cx: &mut JSContext, key: Self::Key) {
68        self.internal.add(cx, key);
69        self.states_did_change(cx.no_gc());
70    }
71
72    #[inline(always)]
73    fn has(&self, cx: &mut JSContext, key: Self::Key) -> bool {
74        self.internal.has(cx, key)
75    }
76
77    #[inline(always)]
78    fn clear(&self, cx: &mut JSContext) {
79        let old_size = self.internal.size(cx);
80        self.internal.clear(cx);
81        if old_size != 0 {
82            self.states_did_change(cx.no_gc());
83        }
84    }
85
86    #[inline(always)]
87    fn delete(&self, cx: &mut JSContext, key: Self::Key) -> bool {
88        if self.internal.delete(cx, key) {
89            self.states_did_change(cx.no_gc());
90            true
91        } else {
92            false
93        }
94    }
95}
96
97impl CustomStateSetMethods<crate::DomTypeHolder> for CustomStateSet {
98    /// <https://html.spec.whatwg.org/multipage/#customstateset>
99    fn Size(&self) -> u32 {
100        self.internal.borrow().len() as u32
101    }
102}
script/dom/customstateset.rs

script/dom/
customstateset.rs
