re_view 0.31.0-rc.2

use re_sdk_types::{ComponentDescriptor, ComponentIdentifier};
use re_types_core::reflection::ArchetypeFieldReflection;
use re_types_core::{Archetype, ArchetypeReflectionMarker};
use re_ui::list_item::ListItemContentButtonsExt as _;
use re_ui::{UiExt as _, list_item};
use re_viewer_context::{
    BlueprintContext as _, ComponentUiTypes, QueryContext, ViewContext, ViewerContext,
};
use re_viewport_blueprint::ViewProperty;

/// Display the UI for editing all components of a blueprint archetype.
///
/// Note that this will show default values for components that are null.
pub fn view_property_ui<A: Archetype + ArchetypeReflectionMarker>(
    ctx: &ViewContext<'_>,
    ui: &mut egui::Ui,
) {
    let view_property =
        ViewProperty::from_archetype::<A>(ctx.blueprint_db(), ctx.blueprint_query(), ctx.view_id);
    view_property_ui_impl(ctx, ui, &view_property, None);
}

/// See [`view_property_ui`].
///
/// This will also redirect the given component to instead use the given view id as the
/// data source.
pub fn view_property_ui_with_redirect<A: Archetype + ArchetypeReflectionMarker>(
    ctx: &ViewContext<'_>,
    ui: &mut egui::Ui,
    redirect_component: ComponentIdentifier,
    redirect_with_view_id: re_viewer_context::ViewId,
) {
    let view_property =
        ViewProperty::from_archetype::<A>(ctx.blueprint_db(), ctx.blueprint_query(), ctx.view_id);
    view_property_ui_impl(
        ctx,
        ui,
        &view_property,
        Some(&RedirectComponentView {
            component: redirect_component,
            ctx: ViewContext {
                viewer_ctx: ctx.viewer_ctx,
                view_id: redirect_with_view_id,
                view_class_identifier: ctx.view_class_identifier,
                space_origin: ctx.space_origin,
                view_state: ctx.view_state,
                query_result: &re_viewer_context::DataQueryResult::default(),
            },
            view_property: ViewProperty::from_archetype::<A>(
                ctx.blueprint_db(),
                ctx.blueprint_query(),
                redirect_with_view_id,
            ),
        }),
    );
}

struct RedirectComponentView<'a> {
    component: ComponentIdentifier,
    ctx: ViewContext<'a>,
    view_property: ViewProperty,
}

fn view_property_ui_impl(
    ctx: &ViewContext<'_>,
    ui: &mut egui::Ui,
    property: &ViewProperty,
    override_component_view: Option<&RedirectComponentView<'_>>,
) {
    let reflection = ctx.viewer_ctx.reflection();
    let Some(archetype) = reflection.archetypes.get(&property.archetype_name) else {
        // The `ArchetypeReflectionMarker` bound should make this impossible.
        re_log::warn_once!(
            "Missing reflection data for archetype {:?}.",
            property.archetype_name
        );
        return;
    };

    let archetype_display_name = archetype.display_name;

    let query_ctx = property.query_context(ctx);

    // If the property archetype only has a single component,
    // and it has the same name as the archetype, then combine them.
    // Happens in some cases, like for the `NearClipPlane` archetype that
    // only has one component which is also called `NearClipPlane`.
    if archetype.fields.len() == 1 && archetype_display_name == archetype.fields[0].display_name {
        view_property_component_ui(
            &query_ctx,
            ui,
            property,
            archetype_display_name,
            &archetype.fields[0],
        );
    } else {
        let sub_prop_ui = |ui: &mut egui::Ui| {
            for field in &archetype.fields {
                let component = field
                    .component_descriptor(property.archetype_name)
                    .component;

                let (query_ctx, property) = if let Some(override_component_view) =
                    &override_component_view
                    && component == override_component_view.component
                {
                    (
                        &override_component_view
                            .view_property
                            .query_context(&override_component_view.ctx),
                        &override_component_view.view_property,
                    )
                } else {
                    (&query_ctx, property)
                };

                view_property_component_ui(query_ctx, ui, property, field.display_name, field);
            }
        };

        ui.list_item()
            .interactive(false)
            .show_hierarchical_with_children(
                ui,
                ui.make_persistent_id(property.archetype_name.full_name()),
                true,
                list_item::LabelContent::new(archetype_display_name),
                sub_prop_ui,
            );
    }
}

/// Draw view property ui for a single component of a view property archetype.
///
/// Use [`view_property_ui`] whenever possible to show the ui for all components of a view property archetype.
/// This function is only useful if you want to show custom ui for some of the components.
pub fn view_property_component_ui(
    query_ctx: &QueryContext<'_>,
    ui: &mut egui::Ui,
    property: &ViewProperty,
    display_name: &str,
    field: &ArchetypeFieldReflection,
) {
    let component_descr = field.component_descriptor(property.archetype_name);
    let component = component_descr.component;

    let component_array = property.component_raw(component);
    let row_id = property.component_row_id(component);

    let viewer_ctx = query_ctx.viewer_ctx();
    let ui_types = viewer_ctx
        .component_ui_registry()
        .registered_ui_types(field.component_type);

    let singleline_ui: &dyn Fn(&mut egui::Ui) = &|ui| {
        viewer_ctx.component_ui_registry().singleline_edit_ui(
            query_ctx,
            &viewer_ctx.blueprint_store_view_ctx(),
            ui,
            query_ctx.target_entity_path.clone(),
            &component_descr,
            row_id,
            component_array.as_deref(),
        );
    };

    let multiline_ui: &dyn Fn(&mut egui::Ui) = &|ui| {
        viewer_ctx.component_ui_registry().multiline_edit_ui(
            query_ctx,
            &viewer_ctx.blueprint_store_view_ctx(),
            ui,
            query_ctx.target_entity_path.clone(),
            &component_descr,
            row_id,
            component_array.as_deref(),
        );
    };
    // Do this as a separate step to avoid borrowing issues.
    let multiline_ui_ref: Option<&dyn Fn(&mut egui::Ui)> =
        if ui_types.contains(ComponentUiTypes::MultiLineEditor) {
            Some(multiline_ui)
        } else {
            None
        };

    view_property_component_ui_custom(
        query_ctx,
        ui,
        property,
        display_name,
        field,
        singleline_ui,
        multiline_ui_ref,
    );
}

/// Draw view property ui for a single component of a view property archetype with custom ui for singleline & multiline.
///
/// Use [`view_property_ui`] whenever possible to show the ui for all components of a view property archetype.
/// This function is only useful if you want to show custom ui for some of the components.
pub fn view_property_component_ui_custom(
    ctx: &QueryContext<'_>,
    ui: &mut egui::Ui,
    property: &ViewProperty,
    display_name: &str,
    field: &ArchetypeFieldReflection,
    singleline_ui: &dyn Fn(&mut egui::Ui),
    multiline_ui: Option<&dyn Fn(&mut egui::Ui)>,
) {
    let component_descr = field.component_descriptor(property.archetype_name);

    let singleline_list_item_content = list_item::PropertyContent::new(display_name)
        .with_menu_button(&re_ui::icons::MORE, "More options", |ui| {
            menu_more(ctx.viewer_ctx(), ui, property, &component_descr);
        })
        .with_always_show_buttons(true)
        .value_fn(move |ui, _| singleline_ui(ui));

    let list_item_response = if let Some(multiline_ui) = multiline_ui {
        let default_open = false;
        let id = egui::Id::new((ctx.target_entity_path.hash(), &component_descr));
        ui.list_item()
            .interactive(false)
            .show_hierarchical_with_children(
                ui,
                id,
                default_open,
                singleline_list_item_content,
                |ui| {
                    multiline_ui(ui);
                },
            )
            .item_response
    } else {
        ui.list_item()
            .interactive(false)
            // It might have siblings that have a hierarchy.
            .show_hierarchical(ui, singleline_list_item_content)
    };

    list_item_response.on_hover_ui(|ui| {
        ui.markdown_ui(field.docstring_md);
    });
}

fn menu_more(
    ctx: &ViewerContext<'_>,
    ui: &mut egui::Ui,
    property: &ViewProperty,
    component_descr: &ComponentDescriptor,
) {
    let component = component_descr.component;
    let component_array = property.component_raw(component);

    let property_differs_from_default = component_array
        != ctx.raw_latest_at_in_default_blueprint(&property.blueprint_store_path, component);

    let response = ui
        .add_enabled(
            property_differs_from_default,
            egui::Button::new("Reset to default blueprint"),
        )
        .on_hover_text(
"Resets this property to the value in the default blueprint.
If no default blueprint was set or it didn't set any value for this field, this is the same as resetting to empty."
        )
        .on_disabled_hover_text(
            "The property is already set to the same value it has in the default blueprint",
        );
    if response.clicked() {
        ctx.reset_blueprint_component(
            property.blueprint_store_path.clone(),
            component_descr.clone(),
        );
        ui.close();
    }

    let response = ui
        .add_enabled(
            component_array.is_some(),
            egui::Button::new("Unset"),
        )
        .on_hover_text(
"Resets this property to an unset value, meaning that a heuristically determined value will be used instead.
This has the same effect as not setting the value in the blueprint at all."
        )
        .on_disabled_hover_text("The property is already unset.");
    if response.clicked() {
        ctx.clear_blueprint_component(
            property.blueprint_store_path.clone(),
            component_descr.clone(),
        );
        ui.close();
    }

    // TODO(andreas): The next logical thing here is now to save it to the default blueprint!
    // This should be fairly straight forward except that we need to make sure that a default blueprint exists in the first place.
}