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;
13use style::values::AtomIdent;
14
15use crate::dom::bindings::cell::DomRefCell;
16use crate::dom::bindings::inheritance::Castable;
17use crate::dom::bindings::reflector::{Reflector, reflect_dom_object};
18use crate::dom::html::htmlelement::HTMLElement;
19use crate::dom::node::{Node, NodeDamage};
20use crate::dom::window::Window;
21
22/// <https://html.spec.whatwg.org/multipage/#customstateset>
23#[dom_struct]
24pub(crate) struct CustomStateSet {
25    reflector: Reflector,
26    internal: DomRefCell<IndexSet<DOMString>>,
27    owner_element: Dom<HTMLElement>,
28}
29
30impl CustomStateSet {
31    fn new_inherited(element: &HTMLElement) -> Self {
32        Self {
33            reflector: Reflector::new(),
34            internal: DomRefCell::new(Default::default()),
35            owner_element: Dom::from_ref(element),
36        }
37    }
38
39    pub(crate) fn new(window: &Window, element: &HTMLElement, can_gc: CanGc) -> DomRoot<Self> {
40        reflect_dom_object(Box::new(Self::new_inherited(element)), window, can_gc)
41    }
42
43    pub(crate) fn for_each_state<F>(&self, mut callback: F)
44    where
45        F: FnMut(&AtomIdent),
46    {
47        // FIXME: This creates new atoms whenever it is called, which is not optimal.
48        for state in self.internal.borrow().iter() {
49            callback(&AtomIdent::from(state.str()));
50        }
51    }
52
53    fn states_did_change(&self) {
54        self.owner_element.upcast::<Node>().dirty(NodeDamage::Other);
55    }
56}
57
58impl Setlike for CustomStateSet {
59    type Key = DOMString;
60
61    #[inline(always)]
62    fn get_index(&self, index: u32) -> Option<Self::Key> {
63        self.internal.get_index(index)
64    }
65
66    #[inline(always)]
67    fn size(&self) -> u32 {
68        self.internal.size()
69    }
70
71    #[inline(always)]
72    fn add(&self, key: Self::Key) {
73        self.internal.add(key);
74        self.states_did_change();
75    }
76
77    #[inline(always)]
78    fn has(&self, key: Self::Key) -> bool {
79        self.internal.has(key)
80    }
81
82    #[inline(always)]
83    fn clear(&self) {
84        let old_size = self.internal.size();
85        self.internal.clear();
86        if old_size != 0 {
87            self.states_did_change();
88        }
89    }
90
91    #[inline(always)]
92    fn delete(&self, key: Self::Key) -> bool {
93        if self.internal.delete(key) {
94            self.states_did_change();
95            true
96        } else {
97            false
98        }
99    }
100}
101
102impl CustomStateSetMethods<crate::DomTypeHolder> for CustomStateSet {
103    /// <https://html.spec.whatwg.org/multipage/#customstateset>
104    fn Size(&self) -> u32 {
105        self.internal.size()
106    }
107}
script/dom/customstateset.rs

script/dom/
customstateset.rs
