Menu.cpp

#include "Menu.h"

#undef max
#undef min

ImVec4 Menu::ARGBToImVec4(const uint32_t& color)
{
	const float
		a = (color >> 24 & 0xFF) / 255.0f,
		r = (color >> 16 & 0xFF) / 255.0f,
		g = (color >> 8 & 0xFF) / 255.0f,
		b = (color & 0xFF) / 255.0f;

	return ImVec4(r, g, b, a);
}

uint32_t Menu::ImVec4ToARGB(const ImVec4& color)
{
	const uint32_t
		a = static_cast<uint32_t>(color.w * 255.0f) & 0xFF,
		r = static_cast<uint32_t>(color.x * 255.0f) & 0xFF,
		g = static_cast<uint32_t>(color.y * 255.0f) & 0xFF,
		b = static_cast<uint32_t>(color.z * 255.0f) & 0xFF;

	return (a << 24) | (r << 16) | (g << 8) | b;
}

void Menu::Render()
{
	const auto& mSelectedLine = Chat::getInstance().mSelectedEntry;

	const auto& io = ImGui::GetIO();
	
	if (IsEditLineActive()) Chat::setSampCursorMode(2);
	const int cursorMode = Chat::getSampCursorMode();
	const bool cursorEnabled = cursorMode >= 2 && cursorMode <= 3;

	// Stop render
	if (Chat::isGTAMenuActive()) return;
	if (!cursorEnabled && !IsPopupActive()) return;

	if (mSelectedLine > -1)
	{
		if (ImGui::Begin("##SelectedLine", nullptr, ImGuiWindowFlags_NoResize | ImGuiWindowFlags_NoCollapse | ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoScrollbar | ImGuiWindowFlags_NoSavedSettings | ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoBackground))
		{
			auto* position = Chat::getInstance().getChatEntryManager().getByEntryId(mSelectedLine);

			if (!position)
				return;

			const auto& rect = position->rect;
			const auto& charHeight = static_cast<size_t>(Chat::getInstance().pChat->m_nCharHeight);
			
			const auto
				xPos = static_cast<float>(rect.x1) - ImGui::CalcTextSize("> ").x,
				yPos = static_cast<float>(rect.y1) + std::ceil(static_cast<float>(rect.y2 - rect.y1) / 2.0f) - Chat::getSampFontSize() + (Chat::getSampFontSizeParam() + static_cast<float>(charHeight) / 6);
			
			ImGui::SetWindowPos(ImVec2(xPos, yPos));
			ImGui::SetCursorPosX(0.0f);
			ImGui::Text(">");

			ImGui::End();
		}
	}

	const auto& chat = Chat::getInstance();
	if (ImGui::Begin("##PopupWrapper", nullptr, ImGuiWindowFlags_NoResize | ImGuiWindowFlags_NoCollapse | ImGuiWindowFlags_NoScrollbar | ImGuiWindowFlags_NoSavedSettings | ImGuiWindowFlags_NoBackground | ImGuiWindowFlags_NoTitleBar))
	{
		if (this->openPopup && !ImGui::IsPopupOpen("TextContextMenu"))
		{
			ImGui::OpenPopup("TextContextMenu");
			this->openPopup = false;
			this->popupActive = true;
		} else if (this->popupActive && (!ImGui::IsPopupOpen("TextContextMenu") || mSelectedLine < 0) )
		{
			ClosePopup();
		}

		if (ImGui::BeginPopupContextItem("TextContextMenu"))
		{
			if (chat.mSelectedEntry != -1)
			{
				ImGui::PushItemWidth(350.0f);
				ImGui::PushStyleVar(ImGuiStyleVar_ItemSpacing, ImVec2(12.0f, 12.0f));
				ImGui::PushStyleColor(ImGuiCol_PopupBg, ImVec4(0.12f, 0.12f, 0.12f, 0.9f));
				ImGui::PushStyleColor(ImGuiCol_HeaderHovered, ImVec4(0.5f, 0.5f, 0.5f, 1.0f));
				if (ImGui::MenuItem("Copy"))
				{
					const bool with_colors = (GetKeyState(VK_SHIFT) & 0x8000) != 0; // Hold shift

					const auto& entry = chat.pChat->m_entry[chat.mSelectedEntry];
					
					std::string text;
					if (entry.m_prefixColor != 0 && entry.m_szPrefix[0] != '\0')
					{
						text += entry.m_szPrefix;
						text += " ";
					}
					text += entry.m_szText;

					if (!with_colors)
					{
						for (size_t pos = 0; pos + 7 < text.size();)
						{
							if (text[pos] == '{' && text[pos + 7] == '}')
							{
								bool is_hex = true;
								for (size_t i = 1; i < 7; ++i)
								{
									if (!std::isxdigit(static_cast<unsigned char>(text[pos + i])))
									{
										is_hex = false;
										break;
									}
								}
								if (is_hex)
								{
									text.erase(pos, 8);
									continue;
								}
							}
							++pos;
						}
					}
					ImGui::SetClipboardText(Chat::convertToUTF8(text).c_str());
				}
				if (ImGui::MenuItem(u8"Delete"))
				{
					Chat::sampDeleteChatLine(chat.mSelectedEntry);
				}
				if (ImGui::MenuItem("Edit"))
				{
					const auto& pChat = Chat::getInstance().pChat;

					auto& chatEntry = pChat->m_entry[chat.mSelectedEntry];

					// text
					editLineBuffer[0] = '\0';
					strcat(editLineBuffer, chat.convertToUTF8(chatEntry.m_szText).c_str());
					editLineColor = ARGBToImVec4(chatEntry.m_textColor);

					// prefix
					editPrefixBuffer[0] = '\0';
					strcat(editPrefixBuffer, chat.convertToUTF8(chatEntry.m_szPrefix).c_str());
					editPrefixColor = ARGBToImVec4(chatEntry.m_prefixColor);

					editLineActive = true;
				}
				if (ImGui::MenuItem("Clear"))
				{
					Chat::sampDeleteChatLineAll();
				}
				ImGui::PopItemWidth();
				ImGui::PopStyleVar();
				ImGui::PopStyleColor(2);
			}

			ImGui::EndPopup();
			ImGui::End();
		}
	}

	if (IsEditLineActive() && chat.mSelectedEntry > -1)
	{
		ImGui::PushStyleColor(ImGuiCol_FrameBg, ImVec4(0.12f, 0.12f, 0.12f, 0.9f));
		if (ImGui::Begin("##EditLine", &editLineActive, ImGuiWindowFlags_NoResize | ImGuiWindowFlags_NoCollapse | ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoScrollbar | ImGuiWindowFlags_NoSavedSettings | ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoBackground))
		{
			ImGui::SetWindowPos(ImVec2((io.DisplaySize.x - ImGui::GetWindowWidth()) * 0.5f,
				(io.DisplaySize.y - ImGui::GetWindowHeight()) * 0.5f));

			ImGui::PushStyleVar(ImGuiStyleVar_ItemSpacing, ImVec2(8.0f, 4.0f));
			ImGui::PushStyleVar(ImGuiStyleVar_FramePadding, ImVec2(4.0f, 12.0f));
			ImGui::PushStyleColor(ImGuiCol_TextSelectedBg, ImVec4(0.5f, 0.5f, 0.5f, 1.0f));
			ImGui::PushItemWidth(io.DisplaySize.x * (750.0f / 1920));

			const bool hasPrefix = (chat.pChat->m_entry[mSelectedLine].m_prefixColor != 0);
			
			if (hasPrefix)
			{
				// ----- Prefix color button -----
				if (ImGui::ColorButton("##PrefixColorBtn", editPrefixColor, ImGuiColorEditFlags_NoTooltip))
				{
					ImGui::OpenPopup("##PrefixColorPopup");
				}

				if (ImGui::BeginPopup("##PrefixColorPopup"))
				{
					if (ImGui::ColorPicker3("##PrefixPicker",
						(float*)&editPrefixColor,
						ImGuiColorEditFlags_NoSidePreview | ImGuiColorEditFlags_NoSmallPreview))
					{
						auto& entry = chat.pChat->m_entry[mSelectedLine];
						entry.m_prefixColor = ImVec4ToARGB(editPrefixColor);
						chat.chatUpdate();
					}
					ImGui::EndPopup();
				}

				ImGui::SameLine(0.0f, 5.0f);

				// ----- Prefix text -----
				if (ImGui::InputText("##PrefixText", editPrefixBuffer, IM_ARRAYSIZE(editPrefixBuffer)))
				{
					auto& entry = chat.pChat->m_entry[mSelectedLine];
					entry.m_szPrefix[0] = '\0';
					const std::string prefix(Chat::convertFromUTF8(editPrefixBuffer));
					strcat(entry.m_szPrefix, prefix.c_str());
					chat.chatUpdate();
				}

				// ----- Outline -----
				ImVec2 pos = ImGui::GetItemRectMin();
				ImVec2 size = ImGui::GetItemRectSize();
				ImGui::GetWindowDrawList()->AddRect(
					pos,
					ImVec2(pos.x + size.x, pos.y + size.y),
					IM_COL32(255, 255, 255, 255)
				);
			}

			if (ImGui::ColorButton("ColorButton", editLineColor, ImGuiColorEditFlags_NoTooltip))
			{
				ImGui::OpenPopup("ColorPickerPopup");
			}

			if (ImGui::BeginPopup("ColorPickerPopup"))
			{
				this->colorPopupActive = true;
				if (ImGui::ColorPicker3("##picker", (float*)&editLineColor, ImGuiColorEditFlags_NoSidePreview | ImGuiColorEditFlags_NoSmallPreview))
				{
					auto& pChat = chat.pChat;
					pChat->m_entry[mSelectedLine].m_textColor = ImVec4ToARGB(editLineColor);
					chat.chatUpdate();
				}
				ImGui::EndPopup();
			} else this->colorPopupActive = false;

			ImGui::SameLine(0.0f, 5.0f);

			if (ImGui::InputText("##ChatLine", editLineBuffer, IM_ARRAYSIZE(editLineBuffer)))
			{
				if (chat.mSelectedEntry > -1)
				{
					const auto& pChat = chat.pChat;

					pChat->m_entry[chat.mSelectedEntry].m_szText[0] = '\0';
					const std::string text(Chat::convertFromUTF8(editLineBuffer));
					strcat(pChat->m_entry[chat.mSelectedEntry].m_szText, text.c_str());
					chat.chatUpdate();
				}
			}

			// Input outline
			ImVec2 inputPos = ImGui::GetItemRectMin();
			ImVec2 inputSize = ImGui::GetItemRectSize();
			ImDrawList* drawList = ImGui::GetWindowDrawList();
			drawList->AddRect(inputPos, ImVec2(inputPos.x + inputSize.x, inputPos.y + inputSize.y), IM_COL32(255, 255, 255, 255));

			ImGui::PopItemWidth();
			ImGui::PopStyleVar(2);
			ImGui::PopStyleColor();
			ImGui::End();
		}
		ImGui::PopStyleColor();
	}
}

void Menu::RebuildFonts()
{
	std::string fontPath(256, '\0');

	if (SHGetSpecialFolderPathA(Chat::getGameHWND(), fontPath.data(), CSIDL_FONTS, false))
	{
		ImGui_ImplDX9_InvalidateDeviceObjects();

		fontPath.resize(fontPath.find('\0'));
		fontPath += "\\";

		std::string fontFileName{ Chat::getFontRelativePathByName(Chat::getSampFontName()) };
		if (fontFileName.empty() || fontFileName.find("arial") != std::string::npos) fontFileName = "ArialBd.ttf";
		fontPath += fontFileName;

		auto& io = ImGui::GetIO();
		ImVector<ImWchar> ranges;
		ImFontGlyphRangesBuilder builder;
		io.ConfigFlags |= ImGuiConfigFlags_NoMouseCursorChange;
		builder.AddRanges(io.Fonts->GetGlyphRangesCyrillic());
		builder.AddText(u8"‚„…†‡ˆ‰‹‘’“”•–—™›¹");
		builder.BuildRanges(&ranges);

		ImFontConfig cfg;
		cfg.FontBuilderFlags |= ImGuiFreeTypeBuilderFlags_LightHinting | ImGuiFreeTypeBuilderFlags_ForceAutoHint;
		io.Fonts->Clear();
		io.Fonts->AddFontFromFileTTF(fontPath.c_str(), Chat::getSampFontSize() - 2, &cfg, ranges.Data);
		io.Fonts->Build();

		ImGui::GetStyle().ItemSpacing.y = 2;

		ImGui_ImplDX9_CreateDeviceObjects();
	}
}

void Menu::ShowPopup()
{
	this->openPopup = true;
}

void Menu::ClosePopup()
{
	ImGui::CloseCurrentPopup();
	this->popupActive = false;
}

bool Menu::IsPopupActive() const
{
	return popupActive;
}

bool Menu::IsEditLineActive() const
{
	return editLineActive;
}

bool Menu::IsColorPopupActive() const
{
	return this->colorPopupActive;
}

void Menu::CloseEditLine()
{
	this->editLineActive = false;
	Chat::setSampCursorMode(0);
	Chat::getInstance().mSelectedEntry = -1;
}