#include <engine/scene/components.hpp>
#include <entt/entt.hpp>
#include <imgui.h>
#include <mirror/panel/properties.hpp>
#include <mirror/panel/scene_hierarchy.hpp>

namespace Light {

SceneHierarchyPanel::SceneHierarchyPanel()
    : m_Context(nullptr)
    , m_PropertiesPanelContext(nullptr)
    , m_SelectionContext()
{
}

SceneHierarchyPanel::
    SceneHierarchyPanel(Ref<Scene> context, Ref<PropertiesPanel> propertiesPanel /* = nullptr */)
    : m_Context(context)
    , m_PropertiesPanelContext(propertiesPanel)
{
}

void SceneHierarchyPanel::OnUserInterfaceUpdate()
{
	if (m_Context)
	{
		ImGui::Begin("Hierarchy");

		for (auto entityID : m_Context->m_Registry.view<TagComponent>())
		{
			Entity entity(static_cast<entt::entity>(entityID), m_Context.get());
			const std::string &tag = entity.GetComponent<TagComponent>();

			DrawNode(entity, tag);
		};
	}

	ImGui::End();
}

void SceneHierarchyPanel::
    SetContext(Ref<Scene> context, Ref<PropertiesPanel> propertiesPanel /* = nullptr */)
{
	if (propertiesPanel)
		m_PropertiesPanelContext = propertiesPanel;

	m_Context = context;
}

void SceneHierarchyPanel::DrawNode(Entity entity, const std::string &label)
{
	ImGuiTreeNodeFlags flags = (m_SelectionContext == entity ? ImGuiTreeNodeFlags_Selected : NULL)
	                           | ImGuiTreeNodeFlags_OpenOnArrow | ImGuiTreeNodeFlags_SpanFullWidth;

	bool expanded = ImGui::TreeNodeEx((void *)(uint64_t)(uint32_t)(entity), flags, label.c_str());

	if (ImGui::IsItemClicked())
	{
		m_SelectionContext = entity;
		m_PropertiesPanelContext->SetEntityContext(entity);
	}

	if (expanded)
	{
		ImGui::Text("TEST_OPENED_TREE!");
		ImGui::TreePop();
	}
}

} // namespace Light