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
94

pub mod accessibility_manager;
pub use accessibility_manager::*;
use freya_native_core::{
    node::NodeType,
    real_dom::NodeImmutable,
    tags::TagName,
    NodeId,
};
use freya_node_state::AccessibilityNodeState;
use torin::torin::Torin;

use crate::{
    dom::{
        DioxusDOM,
        DioxusNode,
    },
    types::AccessibilityId,
};

/// Direction for the next Accessibility Node to be focused.
#[derive(PartialEq)]
pub enum AccessibilityFocusDirection {
    Forward,
    Backward,
}

/// Shortcut functions to retrieve Acessibility info from a Dioxus Node
trait NodeAccessibility {
    /// Return the first text node from this Node
    fn get_inner_texts(&self) -> Option<String>;

    /// Collect all the AccessibilityIDs from a Node's children
    fn get_accessibility_children(&self) -> Vec<AccessibilityId>;
}

impl NodeAccessibility for DioxusNode<'_> {
    /// Return the first text node from this Node
    fn get_inner_texts(&self) -> Option<String> {
        let children = self.children();
        let first_child = children.first()?;
        let node_type = first_child.node_type();
        if let NodeType::Text(text) = &*node_type {
            Some(text.to_owned())
        } else {
            None
        }
    }

    /// Collect all the AccessibilityIDs from a Node's children
    fn get_accessibility_children(&self) -> Vec<AccessibilityId> {
        self.children()
            .iter()
            .filter_map(|child| {
                if child.node_type().is_visible_element() {
                    let node_accessibility = &*child.get::<AccessibilityNodeState>().unwrap();
                    node_accessibility.accessibility_id
                } else {
                    None
                }
            })
            .collect::<Vec<AccessibilityId>>()
    }
}

pub fn process_accessibility(
    layout: &Torin<NodeId>,
    rdom: &DioxusDOM,
    accessibility_manager: &mut AccessibilityManager,
) {
    rdom.traverse_depth_first_advanced(|node| {
        if !node.node_type().is_element() {
            return false;
        }

        let layout_node = layout.get(node.id()).unwrap();
        let node_accessibility = &*node.get::<AccessibilityNodeState>().unwrap();
        if let Some(accessibility_id) = node_accessibility.accessibility_id {
            accessibility_manager.add_node(
                &node,
                layout_node,
                accessibility_id,
                node_accessibility,
            );
        }

        if let Some(tag) = node.node_type().tag() {
            if *tag == TagName::Paragraph || *tag == TagName::Label {
                return false;
            }
        }

        true
    });
}