#include "UI.hpp"

#include "Model/GlobalStates.hpp"
#include "Model/Project.hpp"
#include "Model/Template/TableTemplate.hpp"
#include "Model/Template/Template.hpp"
#include "UI/Localization.hpp"

#include <imgui.h>
#include <imgui_extra_math.h>
#include <imgui_stdlib.h>
#include <fstream>
#include <iostream>
#include <utility>

namespace {
class TemplateUI
{
public:
	static std::unique_ptr<TemplateUI> CreateByKind(Template::Kind kind, std::unique_ptr<Template> tmpl);
	static std::unique_ptr<TemplateUI> CreateByKind(Template::Kind kind);

	virtual ~TemplateUI() = default;
	virtual void Draw() = 0;
};

class TableTemplateUI : public TemplateUI
{
private:
	std::unique_ptr<TableTemplate> mTable;

	TableCell* mSelectedCell = nullptr;

public:
	TableTemplateUI(std::unique_ptr<TableTemplate> table)
		: mTable{ std::move(table) }
	{
	}

	virtual void Draw() override
	{
		ImGui::Columns(2);

		DrawInspector();
		ImGui::NextColumn();

		DrawTable();
		ImGui::NextColumn();

		ImGui::Columns(1);
	}

private:
	void DrawInspector()
	{
		if (ImGui::BeginTabBar("Inspector")) {
			if (ImGui::BeginTabItem("Table")) {
				DrawTableInspector();
				ImGui::EndTabItem();
			}
			if (ImGui::BeginTabItem("Cell")) {
				DrawCellInspector();
				ImGui::EndTabItem();
			}
			ImGui::EndTabBar();
		}
	}

	void DrawTableInspector()
	{
		// TODO
	}

	void DrawCellInspector()
	{
		if (mSelectedCell) {

		} else {
			ImGui::Text("Select a cell to edit");
		}
	}

	void DrawTable()
	{
		constexpr int kCellSpacing = 20;

		int colCount = mTable->GetTableWidth();
		int rowCount = mTable->GetTableHeight();
		float x = 0.0f;
		float y = 0.0f;
		for (int rowIdx = 0; rowIdx < rowCount; ++rowIdx) {
			int rowHeight = mTable->GetRowHeight(rowIdx);

			for (int colIdx = 0; colIdx < colCount; ++colIdx) {
				int colWidth = mTable->GetColumnWidth(colIdx);

				ImRect rect{
					ImVec2(x, y),
					ImVec2(colWidth, rowHeight),
				};

				// TODO

				x += colWidth;
				x += kCellSpacing;
			}

			y += rowHeight;
			y += kCellSpacing;
		}
	}
};

std::unique_ptr<TemplateUI> TemplateUI::CreateByKind(Template::Kind kind, std::unique_ptr<Template> tmpl)
{
#pragma push_macro("UNIQUE_CAST")
#undef UNIQUE_CAST
#define UNIQUE_CAST(TargetType, input) std::unique_ptr<TargetType>(static_cast<TargetType*>(input.release()))
	switch (kind) {
		case Template::KD_Table: return std::make_unique<TableTemplateUI>(UNIQUE_CAST(TableTemplate, tmpl));
		case Template::InvalidKind: return nullptr;
	}
#pragma pop_macro("UNIQUE_CAST")
}

std::unique_ptr<TemplateUI> TemplateUI::CreateByKind(Template::Kind kind)
{
	switch (kind) {
		case Template::KD_Table: return std::make_unique<TableTemplateUI>(std::make_unique<TableTemplate>());
		case Template::InvalidKind: return nullptr;
	}
}
} // namespace

void UI::TemplatesTab()
{
	auto ls = LocaleStrings::Instance.get();
	auto& gs = GlobalStates::GetInstance();
	auto& project = *gs.GetCurrentProject();

	bool openedDummy = true;
	static std::unique_ptr<TemplateUI> openTemplate;
	static AssetList::ListState state;

	// Toolbar item: close
	if (ImGui::Button(ls->Close.Get(), openTemplate == nullptr)) {
		openTemplate = nullptr;
	}

	// Toolbar item: open...
	ImGui::SameLine();
	if (ImGui::Button("Open...")) {
		ImGui::OpenPopup("Open template");
	}
	if (ImGui::BeginPopupModal("Open template", &openedDummy, ImGuiWindowFlags_AlwaysAutoResize)) {
		project.Templates.DrawDetails(state);

		if (state.SelectedAsset) {
			auto kind = static_cast<Template::Kind>(state.SelectedAsset->Payload);
			auto tmpl = project.Templates.Load(*state.SelectedAsset);
			openTemplate = TemplateUI::CreateByKind(kind, std::move(tmpl));
		}

		ImGui::EndPopup();
	}

	// Toolbar item: manage...
	ImGui::SameLine();
	if (ImGui::Button("Manage...")) {
		ImGui::OpenPopup("Manage templates");
	}
	if (ImGui::BeginPopupModal("Manage templates", &openedDummy, ImGuiWindowFlags_AlwaysAutoResize)) {
		project.Templates.DrawDetails(state);

		enum class NameSelectionError
		{
			None,
			Duplicated,
			Empty,
		};

		static std::string newName;
		static NameSelectionError newNameError;
		auto ValidateNewName = [&]() -> void {
			if (newName.empty()) {
				newNameError = NameSelectionError::Empty;
			}

			if (project.Templates.FindByName(newName)) {
				newNameError = NameSelectionError::Duplicated;
			}
		};
		auto ShowNewNameErrors = [&]() -> void {
			switch (newNameError) {
				case NameSelectionError::None: break;
				case NameSelectionError::Duplicated:
					ImGui::ErrorMessage("Duplicate template name");
					break;
				case NameSelectionError::Empty:
					ImGui::ErrorMessage("Template name cannot be empty");
					break;
			}
		};

		static Template::Kind newKind = Template::InvalidKind;
		auto ShowNewKindErrors = [&]() -> void {
			if (newKind == Template::InvalidKind) {
				ImGui::ErrorMessage("Must select a valid template type");
			}
		};

		auto IsInputValid = [&]() -> bool {
			return newNameError == NameSelectionError::None && newKind != Template::InvalidKind;
		};

		if (ImGui::Button(ls->Add.Get())) {
			ImGui::OpenPopup("Create template");
		}
		if (ImGui::BeginPopupModal("Create template")) {
			if (ImGui::InputText("Name", &newName)) {
				ValidateNewName();
			}

			if (ImGui::BeginCombo("Type", Template::FormatKind(newKind))) {
				for (int i = 0; i < Template::KindCount; ++i) {
					auto kind = static_cast<Template::Kind>(i);
					if (ImGui::Selectable(Template::FormatKind(kind), newKind == kind)) {
						newKind = kind;
					}
				}
				ImGui::EndCombo();
			}

			if (ImGui::Button(ls->DialogConfirm.Get(), IsInputValid())) {
				project.Templates.Create(SavedAsset{
					.Name = newName,
					.Payload = static_cast<uint64_t>(newKind),
				});

				openTemplate = TemplateUI::CreateByKind(newKind);
			}
			ImGui::SameLine();
			if (ImGui::Button(ls->DialogCancel.Get())) {
				ImGui::CloseCurrentPopup();
			}

			ShowNewNameErrors();
			ShowNewKindErrors();

			ImGui::EndPopup();
		}

		ImGui::SameLine();
		if (ImGui::Button(ls->Rename.Get(), state.SelectedAsset == nullptr)) {
			ImGui::OpenPopup("Rename template");
			newName.clear();
		}
		if (ImGui::BeginPopupModal("Rename template")) {
			if (ImGui::InputText("New name", &newName)) {
				ValidateNewName();
			}

			if (ImGui::Button(ls->DialogConfirm.Get(), IsInputValid())) {
				auto tmpl = project.Templates.Rename(
					state.SelectedAsset->Name,
					newName);

				// Update the selected pointer to the new location (we mutated the map, the pointer may be invalid now)
				state.SelectedAsset = tmpl;
			}
			ImGui::SameLine();
			if (ImGui::Button(ls->DialogCancel.Get())) {
				ImGui::CloseCurrentPopup();
			}

			ShowNewNameErrors();

			ImGui::EndPopup();
		}

		ImGui::SameLine();
		if (ImGui::Button(ls->Delete.Get(), state.SelectedAsset == nullptr)) {
			ImGui::OpenPopup("Delete confirmation");
		}
		if (ImGui::BeginPopupModal("Delete confirmation")) {
			if (ImGui::Button(ls->DialogConfirm.Get())) {
				auto& assetName = state.SelectedAsset->Name;
				project.Templates.Remove(assetName);

				state.SelectedAsset = nullptr;
			}
			ImGui::SameLine();
			if (ImGui::Button(ls->DialogCancel.Get())) {
				ImGui::CloseCurrentPopup();
			}
			ImGui::EndPopup();
		}

		ImGui::EndPopup();
	}

	if (openTemplate) {
		openTemplate->Draw();
	}
}