Skip to main content

freya_devtools_app/
node.rs

1use freya::prelude::*;
2use freya_core::integration::NodeId;
3
4use crate::hooks::use_node_info;
5
6#[derive(PartialEq)]
7pub struct NodeElement {
8    pub node_id: NodeId,
9    pub window_id: u64,
10    pub is_selected: bool,
11    pub is_open: Option<bool>,
12    pub on_selected: EventHandler<()>,
13    pub on_toggle: EventHandler<()>,
14    pub on_expand_all: EventHandler<()>,
15    pub on_collapse_all: EventHandler<()>,
16    pub on_hover: EventHandler<Option<NodeId>>,
17}
18
19impl Component for NodeElement {
20    fn render_key(&self) -> DiffKey
21    where
22        Self: ComponentKey,
23    {
24        DiffKey::from(&(self.node_id, self.window_id))
25    }
26
27    fn render(&self) -> impl IntoElement {
28        let Some(node) = use_node_info(self.node_id, self.window_id) else {
29            return rect().into_element();
30        };
31
32        let margin_left = ((node.height + 1) * 10) as f32 - 18.;
33        let id = self.node_id.0;
34
35        let role = node.state.accessibility.builder.role();
36
37        let on_select = {
38            let on_selected = self.on_selected.clone();
39            move |_| on_selected.call(())
40        };
41
42        let on_open = {
43            let handler = self.on_toggle.clone();
44            let is_open = self.is_open;
45            move |e: Event<PressEventData>| {
46                if is_open.is_some() {
47                    handler.call(());
48                    e.stop_propagation();
49                }
50            }
51        };
52
53        let on_hover_enter = {
54            let on_hover = self.on_hover.clone();
55            let node_id = self.node_id;
56            move |_| on_hover.call(Some(node_id))
57        };
58
59        let on_hover_leave = {
60            let on_hover = self.on_hover.clone();
61            move |_| on_hover.call(None)
62        };
63
64        let arrow_button = self.is_open.map(|is_open| {
65            let arrow_degrees = if is_open { 0. } else { 270. };
66            Button::new()
67                .corner_radius(99.)
68                .border_fill(Color::TRANSPARENT)
69                .padding(Gaps::new_all(6.))
70                .background(Color::TRANSPARENT)
71                .on_press(on_open)
72                .child(ArrowIcon::new().rotate(arrow_degrees).fill(Color::WHITE))
73        });
74
75        let on_secondary_press = {
76            let on_toggle = self.on_toggle.clone();
77            let on_expand_all = self.on_expand_all.clone();
78            let on_collapse_all = self.on_collapse_all.clone();
79            let is_open = self.is_open;
80            move |_| {
81                let on_toggle = on_toggle.clone();
82                let on_expand_all = on_expand_all.clone();
83                let on_collapse_all = on_collapse_all.clone();
84                ContextMenu::open(
85                    Menu::new()
86                        .child(MenuItem::new().on_press(move |_| on_toggle.call(())).child(
87                            if Some(true) == is_open {
88                                "Collapse"
89                            } else {
90                                "Expand"
91                            },
92                        ))
93                        .child(
94                            MenuItem::new()
95                                .on_press(move |_| on_expand_all.call(()))
96                                .child("Expand All"),
97                        )
98                        .child(
99                            MenuItem::new()
100                                .on_press(move |_| on_collapse_all.call(()))
101                                .child("Collapse All"),
102                        ),
103                );
104            }
105        };
106
107        let button = Button::new()
108            .corner_radius(99.)
109            .width(Size::fill())
110            .height(Size::px(27.))
111            .border_fill(Color::TRANSPARENT)
112            .background(if self.is_selected {
113                (40, 40, 40).into()
114            } else {
115                Color::TRANSPARENT
116            })
117            .hover_background(if self.is_selected {
118                (40, 40, 40).into()
119            } else {
120                Color::from((45, 45, 45))
121            })
122            .on_press(on_select)
123            .on_secondary_press(on_secondary_press)
124            .child(
125                rect()
126                    .offset_x(margin_left)
127                    .direction(Direction::Horizontal)
128                    .width(Size::fill())
129                    .cross_align(Alignment::center())
130                    .child(rect().width(Size::px(25.)).maybe_child(arrow_button))
131                    .child(
132                        paragraph()
133                            .max_lines(1)
134                            .font_size(14.)
135                            .text_overflow(TextOverflow::Ellipsis)
136                            .span(
137                                Span::new(if node.is_window {
138                                    "Window".to_string()
139                                } else if role == AccessibilityRole::GenericContainer {
140                                    "rect".to_string()
141                                } else {
142                                    format!("{role:?}")
143                                })
144                                .color(Color::WHITE),
145                            )
146                            .span(
147                                Span::new(if node.is_window {
148                                    format!(", id: {}", self.window_id)
149                                } else {
150                                    format!(", id: {}", id)
151                                })
152                                .color(Color::from_rgb(200, 200, 200)),
153                            ),
154                    ),
155            );
156
157        rect()
158            .on_pointer_enter(on_hover_enter)
159            .on_pointer_leave(on_hover_leave)
160            .child(button)
161            .into()
162    }
163}