#ifndef SCENE_HIERARCHY_HPP
#define SCENE_HIERARCHY_HPP

#include "imgui.h"
#include <open_engine.hpp>

namespace OpenEngine {
    class SceneHierarchy
    {
        public:
            SceneHierarchy() = default;
            SceneHierarchy(const Ref<Scene>& scene);

            void SetContext(const Ref<Scene>& scene);

            void OnImGuiRender();

        private:
            void DrawEntityNode(Entity& entity);
            void DrawComponents(Entity& entity);

            template <typename T>
                void DrawComponentDrawer(Entity& entity, const char* header)
                {
                    bool component_marked_deletion = false;
                    ImVec2 region_available = ImGui::GetContentRegionAvail();
                    ImGuiTreeNodeFlags tree_node_flags = ImGuiTreeNodeFlags_DefaultOpen
                        | ImGuiTreeNodeFlags_Framed
                        | ImGuiTreeNodeFlags_AllowOverlap;
                    if (entity.HasComponent<T>()) {
                        ImGui::PushStyleVar(ImGuiStyleVar_FramePadding, ImVec2{ 4, 4 });
                        float line_height = ImGui::GetFontSize() + ImGui::GetStyle().FramePadding.y * 2.0f;

                        bool opened = ImGui::TreeNodeEx((void*)typeid(T).hash_code(), tree_node_flags, "%s", header);
                        ImGui::SameLine(region_available.x - line_height * 0.5f);

                        ImGui::PushStyleColor(ImGuiCol_Button, { 0.290f, 0.301f, 0.388f, 1.0f });
                        if (ImGui::Button("...", ImVec2{ line_height, line_height }))
                            ImGui::OpenPopup("component_settings");
                        ImGui::PopStyleColor();

                        ImGui::PopStyleVar();

                        if (ImGui::BeginPopup("component_settings")) {
                            if (ImGui::MenuItem("Remove component"))
                                component_marked_deletion = true;

                            ImGui::EndPopup();
                        }

                        if (opened) {
                            entity.GetComponents<T>().OnImGuiRender(entity);
                            ImGui::TreePop();
                        }

                        ImGui::Separator();

                    }
                    if (component_marked_deletion)
                        entity.RemoveComponents<T>();
                }

        private:
            Ref<Scene> scene;
            Entity selected_context, renamed_entity;
            double last_selected_time = 10.0;
    };
}

#endif // SCENE_HIERARCHY_HPP