jellyflow-egui 0.3.0

use eframe::egui::{ComboBox, Ui};

use crate::bridge::JellyflowEguiBridge;
use crate::samples::SampleGraphKind;
use crate::state::{CanvasTool, JellyflowEguiState, LayoutPresetChoice};

pub fn show_toolbar(ui: &mut Ui, bridge: &mut JellyflowEguiBridge, state: &mut JellyflowEguiState) {
    ui.horizontal(|ui| {
        ui.selectable_value(&mut state.canvas_tool, CanvasTool::Select, "Select")
            .on_hover_text("Select, drag, and marquee nodes");
        ui.selectable_value(&mut state.canvas_tool, CanvasTool::Pan, "Pan")
            .on_hover_text("Drag the canvas viewport");
        ui.selectable_value(&mut state.canvas_tool, CanvasTool::Connect, "Connect")
            .on_hover_text("Drag from one handle to another");
        ui.selectable_value(&mut state.canvas_tool, CanvasTool::Resize, "Resize")
            .on_hover_text("Drag selected node corners and edges");
        ui.separator();

        if ui
            .button("Fit")
            .on_hover_text("Fit graph to view")
            .clicked()
            && bridge.fit_view(state.canvas.snapshot.viewport_size)
        {
            state.set_status("Fit view");
        }

        if ui
            .add_enabled(bridge.store().can_undo(), eframe::egui::Button::new("Undo"))
            .clicked()
        {
            match bridge.undo() {
                Ok(Some(_)) => state.set_status("Undo"),
                Ok(None) => state.set_status("Nothing to undo"),
                Err(err) => state.set_status(err.to_string()),
            }
        }

        if ui
            .add_enabled(bridge.store().can_redo(), eframe::egui::Button::new("Redo"))
            .clicked()
        {
            match bridge.redo() {
                Ok(Some(_)) => state.set_status("Redo"),
                Ok(None) => state.set_status("Nothing to redo"),
                Err(err) => state.set_status(err.to_string()),
            }
        }

        ui.separator();

        ComboBox::from_id_salt("jellyflow_sample_graph")
            .selected_text(state.selected_sample.label())
            .show_ui(ui, |ui| {
                for sample in SampleGraphKind::ALL {
                    ui.selectable_value(&mut state.selected_sample, sample, sample.label());
                }
            });

        if ui.button("Load").clicked() {
            match JellyflowEguiBridge::sample(state.selected_sample) {
                Ok((next_bridge, default_layout)) => {
                    *bridge = next_bridge;
                    state.selected_layout_preset = default_layout;
                    state.canvas.clear_active();
                    state.canvas.hovered = None;
                    state.canvas.snapshot = crate::state::CanvasSnapshot::empty();
                    state.canvas.request_fit_view();
                    state.pending_create_kind = None;
                    state.set_status(format!("Loaded {}", state.selected_sample.label()));
                }
                Err(err) => state.set_status(err.to_string()),
            }
        }

        ui.separator();

        ComboBox::from_id_salt("jellyflow_layout_preset")
            .selected_text(state.selected_layout_preset.label())
            .show_ui(ui, |ui| {
                for preset in LayoutPresetChoice::ALL {
                    ui.selectable_value(&mut state.selected_layout_preset, preset, preset.label());
                }
            });

        if ui.button("Layout").clicked() {
            match bridge.apply_layout(state.selected_layout_preset) {
                Ok(Some(_)) => state.set_status("Layout applied"),
                Ok(None) => state.set_status("Layout unchanged"),
                Err(err) => state.set_status(err.to_string()),
            }
        }

        if let Some(message) = &state.status_message {
            ui.separator();
            ui.label(message);
        }
    });
}