use re_types::{components::MarkerShape, reflection::Enum as _};
use re_ui::UiExt as _;
use re_viewer_context::{MaybeMutRef, ViewerContext};

use crate::response_utils::response_with_changes_of_inner;

pub(crate) fn edit_marker_shape_ui(
    _ctx: &ViewerContext<'_>,
    ui: &mut egui::Ui,
    marker: &mut MaybeMutRef<'_, MarkerShape>,
) -> egui::Response {
    let item_width = 100.0;

    if let Some(edit_marker) = marker.as_mut() {
        let marker_text = edit_marker.to_string(); // TODO(emilk): Show marker shape in the selected text

        response_with_changes_of_inner(
            egui::ComboBox::from_id_source("marker_shape")
                .selected_text(marker_text)
                .height(320.0)
                .show_ui(ui, |ui| {
                    let list_ui = |ui: &mut egui::Ui| {
                        let mut combined_response: Option<egui::Response> = None;
                        for &marker in MarkerShape::variants() {
                            let mut response =
                                ui.list_item().selected(*edit_marker == marker).show_flat(
                                    ui,
                                    re_ui::list_item::LabelContent::new(marker.to_string())
                                        .min_desired_width(item_width)
                                        .with_icon_fn(|ui, rect, visuals| {
                                            paint_marker(
                                                ui,
                                                marker.into(),
                                                rect,
                                                visuals.text_color(),
                                            );
                                        }),
                                );

                            if response.clicked() {
                                *edit_marker = marker;
                                response.changed = true;
                            }

                            combined_response = Some(match combined_response {
                                Some(combined_response) => combined_response | response,
                                None => response,
                            });
                        }
                        combined_response.expect("At least one marker shape should be available")
                    };

                    re_ui::list_item::list_item_scope(ui, "marker_shape", list_ui)
                }),
        )
    } else {
        let marker: MarkerShape = *marker.as_ref();
        ui.list_item_flat_noninteractive(
            re_ui::list_item::LabelContent::new(marker.to_string())
                .min_desired_width(item_width)
                .with_icon_fn(|ui, rect, visuals| {
                    paint_marker(ui, marker.into(), rect, visuals.text_color());
                }),
        )
    }
}

pub(crate) fn paint_marker(
    ui: &egui::Ui,
    marker: egui_plot::MarkerShape,
    rect: egui::Rect,
    color: egui::Color32,
) {
    use egui_plot::PlotItem as _;

    let points = egui_plot::Points::new([0.0, 0.0])
        .shape(marker)
        .color(color)
        .radius(rect.size().min_elem() / 2.0)
        .filled(true);

    let bounds = egui_plot::PlotBounds::new_symmetrical(0.5);
    let transform = egui_plot::PlotTransform::new(rect, bounds, true, true);

    let mut shapes = vec![];
    points.shapes(ui, &transform, &mut shapes);
    ui.painter().extend(shapes);
}