#include "Panels/AssetBrowser.hpp"

#include <LightEngine.hpp>
#include <Utility/Serializer.hpp>
#include <imgui.h>

namespace Light {

AssetBrowserPanel::AssetBrowserPanel(Ref<Scene> activeScene)
    : m_CurrentDirectory("Assets"), m_AssetsPath("Assets"), m_ActiveScene(activeScene)
{
	ResourceManager::LoadTexture("_Assets_Directory", "EngineResources/Icons/Asset_Directory.png");
	ResourceManager::LoadTexture("_Assets_Scene", "EngineResources/Icons/Asset_Scene.png");
	ResourceManager::LoadTexture("_Assets_Image", "EngineResources/Icons/Asset_Image.png");
	ResourceManager::LoadTexture("_Assets_Text", "EngineResources/Icons/Asset_Text.png");

	m_DirectoryTexture = ResourceManager::GetTexture("_Assets_Directory");
	m_SceneTexture     = ResourceManager::GetTexture("_Assets_Scene");
	m_ImageTexture     = ResourceManager::GetTexture("_Assets_Image");
	m_TextTexture      = ResourceManager::GetTexture("_Assets_Text");
}

void AssetBrowserPanel::OnUserInterfaceUpdate()
{
	ImGui::Begin("Content Browser");

	// Parent directory button
	if (m_CurrentDirectory != std::filesystem::path("Assets"))
	{
		if (ImGui::Button(" <--  "))
		{
			m_CurrentDirectory = m_CurrentDirectory.parent_path();
		}
	}

	ImVec2 regionAvail   = ImGui::GetContentRegionAvail();
	uint32_t cellSize    = m_FileSize + m_FilePadding;
	uint32_t columnCount = std::clamp(static_cast<uint32_t>(std::floor(regionAvail.x / cellSize)), 1u, 64u);

	if (ImGui::BeginTable("ContentBrowser", columnCount))
	{
		m_DirectoryTexture->Bind(0u);
		for (auto& dirEntry : std::filesystem::directory_iterator(m_CurrentDirectory))
		{
			const auto& path      = dirEntry.path();
			std::string extension = dirEntry.path().extension().string();

			// TODO: Tidy up
			AssetType assetType;
			assetType = extension.empty() ? AssetType::Directory :

			            extension == ".txt"  ? AssetType::Text :
			            extension == ".glsl" ? AssetType::Text :

			            extension == ".png" ? AssetType::Image :

			            extension == ".scene" ? AssetType::Scene :

                                                AssetType::None;

			// Extension not supported
			if (assetType == AssetType::None)
			{
				continue;
			}

			// Button
			ImGui::TableNextColumn();
			ImGui::PushID(path.c_str());
			switch (assetType)
			{
			// Directory
			case AssetType::Directory:
				if (ImGui::ImageButton(m_DirectoryTexture->GetTexture(), ImVec2(m_FileSize, m_FileSize), ImVec2 { 0.0f, 0.0f }, ImVec2 { 1.0f, 1.0f }, 0, ImVec4 { 0.0f, 0.0f, 0.0f, 0.0f }, ImVec4 { 1.0f, 1.0f, 1.0f, 1.0f }))
				{
					m_CurrentDirectory /= path.filename();
				}
				break;

			// Scene
			case AssetType::Scene:
				if (ImGui::ImageButton(m_SceneTexture->GetTexture(), ImVec2(m_FileSize, m_FileSize), ImVec2 { 0.0f, 0.0f }, ImVec2 { 1.0f, 1.0f }, 0, ImVec4 { 0.0f, 0.0f, 0.0f, 0.0f }, ImVec4 { 1.0f, 1.0f, 1.0f, 1.0f }))
				{
					SceneSerializer serializer(m_ActiveScene);
					LOG(info, "Attempting to deserialize: {}", path.string());
					serializer.Deserialize(path.string());
				}
				break;

			// Image
			case AssetType::Image:
				if (ImGui::ImageButton(m_ImageTexture->GetTexture(), ImVec2(m_FileSize, m_FileSize), ImVec2 { 0.0f, 0.0f }, ImVec2 { 1.0f, 1.0f }, 0, ImVec4 { 0.0f, 0.0f, 0.0f, 0.0f }, ImVec4 { 1.0f, 1.0f, 1.0f, 1.0f }))
				{
				}
				break;

			// Text
			case AssetType::Text:
				if (ImGui::ImageButton(m_TextTexture->GetTexture(), ImVec2(m_FileSize, m_FileSize), ImVec2 { 0.0f, 0.0f }, ImVec2 { 1.0f, 1.0f }, 0, ImVec4 { 0.0f, 0.0f, 0.0f, 0.0f }, ImVec4 { 1.0f, 1.0f, 1.0f, 1.0f }))
				{
				}
				break;


			default:
				break;
			}
			// Label
			ImGui::Text("%s", path.filename().c_str());
			ImGui::PopID();
		}

		ImGui::EndTable();
	}
	ImGui::End();
}

} // namespace Light