#include "content/views/view_command_palette.hpp" #include #include namespace hex::plugin::builtin { ViewCommandPalette::ViewCommandPalette() : View("hex.builtin.view.command_palette.name") { // Add global shortcut to open the command palette ShortcutManager::addGlobalShortcut(CTRLCMD + SHIFT + Keys::P, "hex.builtin.view.command_palette.name", [this] { EventManager::post("hex.builtin.view.command_palette.name"_lang); this->m_commandPaletteOpen = true; this->m_justOpened = true; }); EventManager::subscribe([this] { EventManager::post("hex.builtin.view.command_palette.name"_lang); this->m_commandPaletteOpen = true; this->m_justOpened = true; }); } void ViewCommandPalette::drawContent() { // If the command palette is hidden, don't draw it if (!this->m_commandPaletteOpen) return; auto windowPos = ImHexApi::System::getMainWindowPosition(); auto windowSize = ImHexApi::System::getMainWindowSize(); ImGui::SetNextWindowPos(ImVec2(windowPos.x + windowSize.x * 0.5F, windowPos.y), ImGuiCond_Always, ImVec2(0.5F, 0.0F)); if (ImGui::BeginPopup("hex.builtin.view.command_palette.name"_lang)) { // Close the popup if the user presses ESC if (ImGui::IsKeyDown(ImGui::GetKeyIndex(ImGuiKey_Escape))) ImGui::CloseCurrentPopup(); const auto buttonColor = [](float alpha) { return ImU32(ImColor(ImGui::GetStyleColorVec4(ImGuiCol_ModalWindowDimBg) * ImVec4(1, 1, 1, alpha))); }; // Draw the main input text box ImGui::PushItemWidth(-1); ImGui::PushStyleColor(ImGuiCol_FrameBg, buttonColor(0.5F)); ImGui::PushStyleColor(ImGuiCol_FrameBgHovered, buttonColor(0.7F)); ImGui::PushStyleColor(ImGuiCol_FrameBgActive, buttonColor(0.9F)); ImGui::PushStyleVar(ImGuiStyleVar_FrameBorderSize, 1.0_scaled); ImGui::PushStyleVar(ImGuiStyleVar_FrameRounding, 4_scaled); if (ImGui::InputText("##command_input", this->m_commandBuffer)) { this->m_lastResults = this->getCommandResults(this->m_commandBuffer); } ImGui::PopStyleVar(2); ImGui::PopStyleColor(3); ImGui::PopItemWidth(); ImGui::SetItemDefaultFocus(); // Handle giving back focus to the input text box if (this->m_focusInputTextBox) { ImGui::SetKeyboardFocusHere(-1); ImGui::ActivateItemByID(ImGui::GetID("##command_input")); auto textState = ImGui::GetInputTextState(ImGui::GetID("##command_input")); if (textState != nullptr) { textState->Stb.cursor = textState->Stb.select_start = textState->Stb.select_end = this->m_commandBuffer.size(); } this->m_focusInputTextBox = false; } // Execute the currently selected command when pressing enter if (ImGui::IsItemFocused() && (ImGui::IsKeyPressed(ImGuiKey_Enter, false) || ImGui::IsKeyPressed(ImGuiKey_KeypadEnter, false))) { if (!this->m_lastResults.empty()) { auto &[displayResult, matchedCommand, callback] = this->m_lastResults.front(); callback(matchedCommand); } ImGui::CloseCurrentPopup(); } // Focus the input text box when the popup is opened if (this->m_justOpened) { focusInputTextBox(); this->m_lastResults = this->getCommandResults(""); this->m_commandBuffer.clear(); this->m_justOpened = false; } ImGui::SetCursorPosY(ImGui::GetCursorPosY() + ImGui::GetStyle().FramePadding.y); ImGui::Separator(); // Draw the results if (ImGui::BeginChild("##results", ImGui::GetContentRegionAvail(), false, ImGuiWindowFlags_AlwaysVerticalScrollbar | ImGuiWindowFlags_NavFlattened)) { for (const auto &[displayResult, matchedCommand, callback] : this->m_lastResults) { ImGui::PushTabStop(true); ON_SCOPE_EXIT { ImGui::PopTabStop(); }; // Allow executing a command by clicking on it or selecting it with the keyboard and pressing enter if (ImGui::Selectable(displayResult.c_str(), false, ImGuiSelectableFlags_DontClosePopups)) { callback(matchedCommand); break; } if (ImGui::IsItemFocused() && (ImGui::IsKeyDown(ImGuiKey_Enter) || ImGui::IsKeyDown(ImGuiKey_KeypadEnter))) { callback(matchedCommand); break; } } ImGui::EndChild(); } ImGui::EndPopup(); } else { this->m_commandPaletteOpen = false; } } std::vector ViewCommandPalette::getCommandResults(const std::string &input) { constexpr static auto MatchCommand = [](const std::string &currCommand, const std::string &commandToMatch) -> std::pair { // Check if the current input matches a command // NoMatch means that the input doesn't match the command // PartialMatch means that the input partially matches the command but is not a perfect match // PerfectMatch means that the input perfectly matches the command if (currCommand.empty()) { return { MatchType::InfoMatch, "" }; } else if (currCommand.size() <= commandToMatch.size()) { if (commandToMatch.starts_with(currCommand)) return { MatchType::PartialMatch, currCommand }; else return { MatchType::NoMatch, {} }; } else { if (currCommand.starts_with(commandToMatch)) return { MatchType::PerfectMatch, currCommand.substr(commandToMatch.length()) }; else return { MatchType::NoMatch, {} }; } }; std::vector results; // Loop over every registered command and check if the input matches it for (const auto &[type, command, unlocalizedDescription, displayCallback, executeCallback] : ContentRegistry::CommandPaletteCommands::impl::getEntries()) { auto AutoComplete = [this, currCommand = command](auto) { this->focusInputTextBox(); this->m_commandBuffer = currCommand; this->m_lastResults = this->getCommandResults(currCommand); }; if (type == ContentRegistry::CommandPaletteCommands::Type::SymbolCommand) { // Handle symbol commands // These commands are used by entering a single symbol and then any input if (auto [match, value] = MatchCommand(input, command); match != MatchType::NoMatch) { if (match != MatchType::PerfectMatch) results.push_back({ command + " (" + LangEntry(unlocalizedDescription) + ")", "", AutoComplete }); else { auto matchedCommand = input.substr(command.length()); results.push_back({ displayCallback(matchedCommand), matchedCommand, executeCallback }); } } } else if (type == ContentRegistry::CommandPaletteCommands::Type::KeywordCommand) { // Handle keyword commands // These commands are used by entering a keyword followed by a space and then any input if (auto [match, value] = MatchCommand(input, command + " "); match != MatchType::NoMatch) { if (match != MatchType::PerfectMatch) results.push_back({ command + " (" + LangEntry(unlocalizedDescription) + ")", "", AutoComplete }); else { auto matchedCommand = input.substr(command.length() + 1); results.push_back({ displayCallback(matchedCommand), matchedCommand, executeCallback }); } } } } // WHen a command has been identified, show the query results for that command for (const auto &handler : ContentRegistry::CommandPaletteCommands::impl::getHandlers()) { const auto &[type, command, queryCallback, displayCallback] = handler; auto processedInput = input; if (processedInput.starts_with(command)) processedInput = processedInput.substr(command.length()); for (const auto &[description, callback] : queryCallback(processedInput)) { if (type == ContentRegistry::CommandPaletteCommands::Type::SymbolCommand) { if (auto [match, value] = MatchCommand(input, command); match != MatchType::NoMatch) { results.push_back({ hex::format("{} ({})", command, description), "", callback }); } } else if (type == ContentRegistry::CommandPaletteCommands::Type::KeywordCommand) { if (auto [match, value] = MatchCommand(input, command + " "); match != MatchType::NoMatch) { results.push_back({ hex::format("{} ({})", command, description), "", callback }); } } } } return results; } }