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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

use std::rc::Rc;

use dioxus_core::ScopeState;
use dioxus_hooks::{
    to_owned, use_context_provider, use_effect, use_memo, use_shared_state,
    use_shared_state_provider, RefCell,
};
use freya_common::EventMessage;
use freya_core::{
    accessibility::ACCESSIBILITY_ROOT_ID,
    types::{AccessibilityId, FocusReceiver},
};

use crate::use_platform;

pub type AccessibilityIdCounter = Rc<RefCell<u64>>;

/// Sync both the Focus shared state and the platform accessibility focus
pub fn use_init_accessibility(cx: &ScopeState) {
    let platform = use_platform(cx);
    use_context_provider(cx, || Rc::new(RefCell::new(0u64)));
    use_shared_state_provider::<AccessibilityId>(cx, || ACCESSIBILITY_ROOT_ID);
    let focused_id = use_shared_state::<AccessibilityId>(cx).unwrap();

    let current_focused_id = *focused_id.read();

    // Notify the platform that a new Node has been focused manually
    let _ = use_memo(cx, &(current_focused_id,), move |(focused_id,)| {
        platform
            .send(EventMessage::FocusAccessibilityNode(focused_id))
            .unwrap();
    });

    // Notify the app that a new Node has been focusd by the platform
    use_effect(cx, (), {
        to_owned![focused_id];
        move |_| {
            let focus_id_listener = cx.consume_context::<FocusReceiver>();
            async move {
                let focus_id_listener = focus_id_listener.clone();
                if let Some(mut focus_id_listener) = focus_id_listener {
                    while focus_id_listener.changed().await.is_ok() {
                        *focused_id.write() = *focus_id_listener.borrow();
                    }
                }
            }
        }
    });
}

#[cfg(test)]
mod test {
    use freya::prelude::*;
    use freya_core::accessibility::ACCESSIBILITY_ROOT_ID;
    use freya_testing::{
        events::pointer::MouseButton, launch_test_with_config, FreyaEvent, TestingConfig,
    };

    #[tokio::test]
    pub async fn focus_accessibility() {
        #[allow(non_snake_case)]
        fn OherChild(cx: Scope) -> Element {
            let focus_manager = use_focus(cx);

            render!(rect {
                width: "100%",
                height: "50%",
                onclick: move |_| focus_manager.focus(),
            })
        }

        fn use_focus_app(cx: Scope) -> Element {
            render!(
                rect {
                    width: "100%",
                    height: "100%",
                    OherChild {},
                    OherChild {}
                }
            )
        }

        let mut utils = launch_test_with_config(
            use_focus_app,
            *TestingConfig::default().with_size((100.0, 100.0).into()),
        );

        // Initial state
        utils.wait_for_update().await;
        assert_eq!(utils.focus_id(), ACCESSIBILITY_ROOT_ID);

        // Click on the first rect
        utils.push_event(FreyaEvent::Mouse {
            name: "click".to_string(),
            cursor: (5.0, 5.0).into(),
            button: Some(MouseButton::Left),
        });

        // First rect is now focused
        utils.wait_for_update().await;
        utils.wait_for_update().await;
        let first_focus_id = utils.focus_id();
        assert_ne!(first_focus_id, ACCESSIBILITY_ROOT_ID);

        // Click on the second rect
        utils.push_event(FreyaEvent::Mouse {
            name: "click".to_string(),
            cursor: (5.0, 75.0).into(),
            button: Some(MouseButton::Left),
        });

        // Second rect is now focused
        utils.wait_for_update().await;
        utils.wait_for_update().await;
        let second_focus_id = utils.focus_id();
        assert_ne!(first_focus_id, second_focus_id);
        assert_ne!(second_focus_id, ACCESSIBILITY_ROOT_ID);
    }
}