freya-devtools-app 0.4.0-rc.15

use freya::prelude::*;
use freya_core::integration::NodeId;

use crate::hooks::use_node_info;

#[derive(PartialEq)]
pub struct NodeElement {
    pub node_id: NodeId,
    pub window_id: u64,
    pub is_selected: bool,
    pub is_open: Option<bool>,
    pub on_selected: EventHandler<()>,
    pub on_toggle: EventHandler<()>,
    pub on_expand_all: EventHandler<()>,
    pub on_collapse_all: EventHandler<()>,
    pub on_hover: EventHandler<Option<NodeId>>,
}

impl Component for NodeElement {
    fn render_key(&self) -> DiffKey
    where
        Self: ComponentKey,
    {
        DiffKey::from(&(self.node_id, self.window_id))
    }

    fn render(&self) -> impl IntoElement {
        let Some(node) = use_node_info(self.node_id, self.window_id) else {
            return rect().into_element();
        };

        let margin_left = ((node.height + 1) * 10) as f32 - 18.;
        let id = self.node_id.0;

        let role = node.state.accessibility.builder.role();

        let on_select = {
            let on_selected = self.on_selected.clone();
            move |_| on_selected.call(())
        };

        let on_open = {
            let handler = self.on_toggle.clone();
            let is_open = self.is_open;
            move |e: Event<PressEventData>| {
                if is_open.is_some() {
                    handler.call(());
                    e.stop_propagation();
                }
            }
        };

        let on_hover_enter = {
            let on_hover = self.on_hover.clone();
            let node_id = self.node_id;
            move |_| on_hover.call(Some(node_id))
        };

        let on_hover_leave = {
            let on_hover = self.on_hover.clone();
            move |_| on_hover.call(None)
        };

        let arrow_button = self.is_open.map(|is_open| {
            let arrow_degrees = if is_open { 0. } else { 270. };
            Button::new()
                .corner_radius(99.)
                .border_fill(Color::TRANSPARENT)
                .padding(Gaps::new_all(6.))
                .background(Color::TRANSPARENT)
                .on_press(on_open)
                .child(ArrowIcon::new().rotate(arrow_degrees).fill(Color::WHITE))
        });

        let on_secondary_press = {
            let on_toggle = self.on_toggle.clone();
            let on_expand_all = self.on_expand_all.clone();
            let on_collapse_all = self.on_collapse_all.clone();
            let is_open = self.is_open;
            move |_| {
                let on_toggle = on_toggle.clone();
                let on_expand_all = on_expand_all.clone();
                let on_collapse_all = on_collapse_all.clone();
                ContextMenu::open(
                    Menu::new()
                        .child(MenuItem::new().on_press(move |_| on_toggle.call(())).child(
                            if Some(true) == is_open {
                                "Collapse"
                            } else {
                                "Expand"
                            },
                        ))
                        .child(
                            MenuItem::new()
                                .on_press(move |_| on_expand_all.call(()))
                                .child("Expand All"),
                        )
                        .child(
                            MenuItem::new()
                                .on_press(move |_| on_collapse_all.call(()))
                                .child("Collapse All"),
                        ),
                );
            }
        };

        let button = Button::new()
            .corner_radius(99.)
            .width(Size::fill())
            .height(Size::px(27.))
            .border_fill(Color::TRANSPARENT)
            .background(if self.is_selected {
                (40, 40, 40).into()
            } else {
                Color::TRANSPARENT
            })
            .hover_background(if self.is_selected {
                (40, 40, 40).into()
            } else {
                Color::from((45, 45, 45))
            })
            .on_press(on_select)
            .on_secondary_press(on_secondary_press)
            .child(
                rect()
                    .offset_x(margin_left)
                    .direction(Direction::Horizontal)
                    .width(Size::fill())
                    .cross_align(Alignment::center())
                    .child(rect().width(Size::px(25.)).maybe_child(arrow_button))
                    .child(
                        paragraph()
                            .max_lines(1)
                            .font_size(14.)
                            .text_overflow(TextOverflow::Ellipsis)
                            .span(
                                Span::new(if node.is_window {
                                    "Window".to_string()
                                } else if role == AccessibilityRole::GenericContainer {
                                    "rect".to_string()
                                } else {
                                    format!("{role:?}")
                                })
                                .color(Color::WHITE),
                            )
                            .span(
                                Span::new(if node.is_window {
                                    format!(", id: {}", self.window_id)
                                } else {
                                    format!(", id: {}", id)
                                })
                                .color(Color::from_rgb(200, 200, 200)),
                            ),
                    ),
            );

        rect()
            .on_pointer_enter(on_hover_enter)
            .on_pointer_leave(on_hover_leave)
            .child(button)
            .into()
    }
}