use itertools::Itertools;
use nohash_hasher::IntMap;

use re_data_ui::item_ui;
use re_entity_db::{EntityPath, EntityTree, InstancePath};
use re_log_types::{ResolvedEntityPathFilter, ResolvedEntityPathRule};
use re_ui::{list_item, UiExt as _};
use re_viewer_context::{DataQueryResult, ViewId, ViewerContext};
use re_viewport_blueprint::{
    create_entity_add_info, CanAddToView, EntityAddInfo, ViewBlueprint, ViewportBlueprint,
};

/// Window for adding/removing entities from a view.
///
/// Delegates to [`re_ui::modal::ModalHandler`]
#[derive(Default)]
pub(crate) struct ViewEntityPicker {
    view_id: Option<ViewId>,
    modal_handler: re_ui::modal::ModalHandler,
}

impl ViewEntityPicker {
    pub fn open(&mut self, view_id: ViewId) {
        self.view_id = Some(view_id);
        self.modal_handler.open();
    }

    #[allow(clippy::unused_self)]
    pub fn ui(
        &mut self,
        egui_ctx: &egui::Context,
        ctx: &ViewerContext<'_>,
        viewport_blueprint: &ViewportBlueprint,
    ) {
        self.modal_handler.ui(
            egui_ctx,
            || {
                re_ui::modal::ModalWrapper::new("Add/remove Entities")
                    .default_height(640.0)
                    .full_span_content(true)
                    .scrollable([false, true])
            },
            |ui, open| {
                let Some(view_id) = &self.view_id else {
                    *open = false;
                    return;
                };

                let Some(view) = viewport_blueprint.view(view_id) else {
                    *open = false;
                    return;
                };

                add_entities_ui(ctx, ui, view);
            },
        );
    }
}

fn add_entities_ui(ctx: &ViewerContext<'_>, ui: &mut egui::Ui, view: &ViewBlueprint) {
    re_tracing::profile_function!();

    let tree = &ctx.recording().tree();
    let query_result = ctx.lookup_query_result(view.id);
    let entity_path_filter = view.contents.entity_path_filter();
    let entities_add_info = create_entity_add_info(ctx, tree, view, query_result);

    list_item::list_item_scope(ui, "view_entity_picker", |ui| {
        add_entities_tree_ui(
            ctx,
            ui,
            &tree.path.to_string(),
            tree,
            view,
            query_result,
            entity_path_filter,
            &entities_add_info,
        );
    });
}

#[allow(clippy::too_many_arguments)]
fn add_entities_tree_ui(
    ctx: &ViewerContext<'_>,
    ui: &mut egui::Ui,
    name: &str,
    tree: &EntityTree,
    view: &ViewBlueprint,
    query_result: &DataQueryResult,
    entity_path_filter: &ResolvedEntityPathFilter,
    entities_add_info: &IntMap<EntityPath, EntityAddInfo>,
) {
    let item_content = list_item::CustomContent::new(|ui, _| {
        add_entities_line_ui(
            ctx,
            ui,
            name,
            tree,
            view,
            query_result,
            entity_path_filter,
            entities_add_info,
        );
    });

    let list_item = ui.list_item().interactive(false);
    if tree.is_leaf() {
        list_item.show_hierarchical(ui, item_content);
    } else {
        let level = tree.path.len();
        let default_open =
            view.space_origin.is_descendant_of(&tree.path) || tree.children.len() <= 3 || level < 2;

        list_item.show_hierarchical_with_children(
            ui,
            ui.id().with(name),
            default_open,
            item_content,
            |ui| {
                for (path_comp, child_tree) in
                    tree.children.iter().sorted_by_key(|(_, child_tree)| {
                        // Put descendants of the space path always first
                        let put_first = child_tree.path.starts_with(&view.space_origin);
                        !put_first
                    })
                {
                    add_entities_tree_ui(
                        ctx,
                        ui,
                        &path_comp.ui_string(),
                        child_tree,
                        view,
                        query_result,
                        entity_path_filter,
                        entities_add_info,
                    );
                }
            },
        );
    };
}

#[allow(clippy::too_many_arguments)]
fn add_entities_line_ui(
    ctx: &ViewerContext<'_>,
    ui: &mut egui::Ui,
    name: &str,
    entity_tree: &EntityTree,
    view: &ViewBlueprint,
    query_result: &DataQueryResult,
    entity_path_filter: &ResolvedEntityPathFilter,
    entities_add_info: &IntMap<EntityPath, EntityAddInfo>,
) {
    re_tracing::profile_function!();

    let query = ctx.current_query();
    let entity_path = &entity_tree.path;

    #[allow(clippy::unwrap_used)]
    let add_info = entities_add_info.get(entity_path).unwrap();

    let is_explicitly_excluded = entity_path_filter.is_explicitly_excluded(entity_path);
    let is_explicitly_included = entity_path_filter.is_explicitly_included(entity_path);
    let is_included = entity_path_filter.matches(entity_path);

    ui.add_enabled_ui(add_info.can_add_self_or_descendant.is_compatible(), |ui| {
        let widget_text = if is_explicitly_excluded {
            // TODO(jleibs): Better design-language for excluded.
            egui::RichText::new(name).italics()
        } else if entity_path == &view.space_origin {
            egui::RichText::new(name).strong()
        } else {
            egui::RichText::new(name)
        };
        let response = item_ui::instance_path_button_to(
            ctx,
            &query,
            ctx.recording(),
            ui,
            Some(view.id),
            &InstancePath::entity_all(entity_path.clone()),
            widget_text,
        );
        if query_result.contains_entity(entity_path) {
            response.highlight();
        }
    });

    //TODO(ab): use `CustomContent` support for action button to implement this.
    ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
        if entity_path_filter.contains_rule_for_exactly(entity_path) {
            // Reset-button
            // Shows when an entity is explicitly excluded or included
            let response = ui.small_icon_button(&re_ui::icons::RESET);

            if response.clicked() {
                view.contents.remove_filter_rule_for(ctx, &entity_tree.path);
            }

            if is_explicitly_excluded {
                response.on_hover_text("Stop excluding this entity path.");
            } else if is_explicitly_included {
                response.on_hover_text("Stop including this entity path.");
            }
        } else if is_included {
            // Remove-button
            // Shows when an entity is already included (but not explicitly)
            let response = ui.small_icon_button(&re_ui::icons::REMOVE);

            if response.clicked() {
                view.contents.raw_add_entity_exclusion(
                    ctx,
                    ResolvedEntityPathRule::including_subtree(&entity_tree.path),
                );
            }

            response.on_hover_text("Exclude this entity and all its descendants from the view");
        } else {
            // Add-button:
            // Shows when an entity is not included
            // Only enabled if the entity is compatible.
            let enabled = add_info.can_add_self_or_descendant.is_compatible();

            ui.add_enabled_ui(enabled, |ui| {
                let response = ui.small_icon_button(&re_ui::icons::ADD);

                if response.clicked() {
                    view.contents.raw_add_entity_inclusion(
                        ctx,
                        ResolvedEntityPathRule::including_subtree(&entity_tree.path),
                    );
                }

                if enabled {
                    if add_info.can_add.is_compatible_and_missing() {
                        response.on_hover_text(
                            "Include this entity and all its descendants in the view",
                        );
                    } else {
                        response.on_hover_text("Add descendants of this entity to the view");
                    }
                } else if let CanAddToView::No { reason } = &add_info.can_add {
                    response.on_disabled_hover_text(reason);
                }
            });
        }
    });
}