1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use accesskit::{NodeId as AccessibilityId, Role};
use dioxus_native_core::node::OwnedAttributeValue;
use dioxus_native_core::{attributes::AttributeName, exports::shipyard::Component};
use dioxus_native_core::{
    node_ref::NodeView,
    prelude::{AttributeMaskBuilder, Dependancy, NodeMaskBuilder, State},
    SendAnyMap,
};
use dioxus_native_core_macro::partial_derive_state;

use crate::CustomAttributeValues;

#[derive(Clone, Debug, PartialEq, Eq, Default, Component)]
pub struct AccessibilityNodeState {
    pub accessibility_id: Option<AccessibilityId>,
    pub role: Option<Role>,
    pub alt: Option<String>,
    pub name: Option<String>,
    pub focusable: bool,
}

#[partial_derive_state]
impl State<CustomAttributeValues> for AccessibilityNodeState {
    type ParentDependencies = ();

    type ChildDependencies = ();

    type NodeDependencies = ();

    const NODE_MASK: NodeMaskBuilder<'static> =
        NodeMaskBuilder::new().with_attrs(AttributeMaskBuilder::Some(&[
            AttributeName::FocusId,
            AttributeName::Role,
            AttributeName::Alt,
            AttributeName::Name,
            AttributeName::Focusable,
        ]));

    fn update<'a>(
        &mut self,
        node_view: NodeView<CustomAttributeValues>,
        _node: <Self::NodeDependencies as Dependancy>::ElementBorrowed<'a>,
        _parent: Option<<Self::ParentDependencies as Dependancy>::ElementBorrowed<'a>>,
        _children: Vec<<Self::ChildDependencies as Dependancy>::ElementBorrowed<'a>>,
        _context: &SendAnyMap,
    ) -> bool {
        let mut accessibility = AccessibilityNodeState::default();

        if let Some(attributes) = node_view.attributes() {
            for attr in attributes {
                match attr.attribute {
                    AttributeName::FocusId => {
                        if let OwnedAttributeValue::Custom(
                            CustomAttributeValues::AccessibilityId(id),
                        ) = attr.value
                        {
                            accessibility.accessibility_id = Some(*id);
                        }
                    }
                    AttributeName::Role => {
                        if let OwnedAttributeValue::Text(attr) = attr.value {
                            if let Ok(new_role) =
                                serde_json::from_str::<Role>(&format!("\"{attr}\""))
                            {
                                accessibility.role = Some(new_role)
                            }
                        }
                    }
                    AttributeName::Alt => {
                        if let OwnedAttributeValue::Text(attr) = attr.value {
                            accessibility.alt = Some(attr.to_owned())
                        }
                    }
                    AttributeName::Name => {
                        if let OwnedAttributeValue::Text(attr) = attr.value {
                            accessibility.name = Some(attr.to_owned())
                        }
                    }
                    AttributeName::Focusable => {
                        if let OwnedAttributeValue::Text(attr) = attr.value {
                            accessibility.focusable = attr.parse().unwrap_or_default()
                        }
                    }
                    _ => {}
                }
            }
        }
        let changed = &accessibility != self;

        *self = accessibility;
        changed
    }
}