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_down = {
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 |e: Event<PressEventData>| {
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_from_event(
85                    &e,
86                    Menu::new()
87                        .child(MenuItem::new().on_press(move |_| on_toggle.call(())).child(
88                            if Some(true) == is_open {
89                                "Collapse"
90                            } else {
91                                "Expand"
92                            },
93                        ))
94                        .child(
95                            MenuItem::new()
96                                .on_press(move |_| on_expand_all.call(()))
97                                .child("Expand All"),
98                        )
99                        .child(
100                            MenuItem::new()
101                                .on_press(move |_| on_collapse_all.call(()))
102                                .child("Collapse All"),
103                        ),
104                );
105            }
106        };
107
108        let button = Button::new()
109            .corner_radius(99.)
110            .width(Size::fill())
111            .height(Size::px(27.))
112            .border_fill(Color::TRANSPARENT)
113            .background(if self.is_selected {
114                (40, 40, 40).into()
115            } else {
116                Color::TRANSPARENT
117            })
118            .hover_background(if self.is_selected {
119                (40, 40, 40).into()
120            } else {
121                Color::from((45, 45, 45))
122            })
123            .on_press(on_select)
124            .on_secondary_down(on_secondary_down)
125            .child(
126                rect()
127                    .offset_x(margin_left)
128                    .direction(Direction::Horizontal)
129                    .width(Size::fill())
130                    .cross_align(Alignment::center())
131                    .child(rect().width(Size::px(25.)).maybe_child(arrow_button))
132                    .child(
133                        paragraph()
134                            .max_lines(1)
135                            .font_size(14.)
136                            .text_overflow(TextOverflow::Ellipsis)
137                            .span(
138                                Span::new(if node.is_window {
139                                    "Window".to_string()
140                                } else if role == AccessibilityRole::GenericContainer {
141                                    "rect".to_string()
142                                } else {
143                                    format!("{role:?}")
144                                })
145                                .color(Color::WHITE),
146                            )
147                            .span(
148                                Span::new(if node.is_window {
149                                    format!(", id: {}", self.window_id)
150                                } else {
151                                    format!(", id: {}", id)
152                                })
153                                .color(Color::from_rgb(200, 200, 200)),
154                            ),
155                    ),
156            );
157
158        rect()
159            .on_pointer_enter(on_hover_enter)
160            .on_pointer_leave(on_hover_leave)
161            .child(button)
162            .into()
163    }
164}