windows: Added context menu entry option

Closes #333
2026-03-30 13:05:25 -05:00 · 2021-10-31 16:28:10 +01:00
parent 1f250e87a3
commit c88053a575
7 changed files with 109 additions and 8 deletions
--- a/plugins/windows/source/content/settings_entries.cpp
+++ b/plugins/windows/source/content/settings_entries.cpp
@@ -0,0 +1,76 @@
+#include <hex/api/content_registry.hpp>
+#include <hex/api/imhex_api.hpp>
+
+#include <hex/helpers/fmt.hpp>
+#include <hex/helpers/lang.hpp>
+using namespace hex::lang_literals;
+
+#include <imgui.h>
+
+#include <nlohmann/json.hpp>
+#include <windows.h>
+
+namespace hex::plugin::windows {
+
+    namespace {
+
+        constexpr auto ImHexContextMenuKey = R"(Software\Classes\*\shell\ImHex)";
+
+        void addImHexContextMenuEntry() {
+            // Create ImHex Root Key
+            HKEY imHexRootKey;
+            RegCreateKeyExA(HKEY_CURRENT_USER, ImHexContextMenuKey, 0x00, nullptr, REG_OPTION_NON_VOLATILE, KEY_SET_VALUE , nullptr, &imHexRootKey, nullptr);
+            RegSetValueA(imHexRootKey, nullptr, REG_SZ, "Open with ImHex", 0x00);
+
+            // Add Icon key to use first icon embedded in exe
+            std::array<char, MAX_PATH> imHexPath = { 0 };
+            GetModuleFileNameA(nullptr, imHexPath.data(), imHexPath.size());
+            auto iconValue = hex::format(R"("{}",0)", imHexPath.data());
+            RegSetKeyValueA(imHexRootKey, nullptr, "Icon", REG_SZ, iconValue.c_str(), iconValue.size() + 1);
+
+            // Add command key to pass file path as first argument to ImHex
+            auto commandValue = hex::format(R"("{}" "%1")", imHexPath.data());
+            RegSetValueA(imHexRootKey, "command", REG_SZ, commandValue.c_str(), commandValue.size() + 1);
+            RegCloseKey(imHexRootKey);
+        }
+
+        void removeImHexContextMenuEntry() {
+            RegDeleteTreeA(HKEY_CURRENT_USER, ImHexContextMenuKey);
+        }
+
+        bool hasImHexContextMenuEntry() {
+            HKEY key;
+            bool keyExists = (RegOpenKeyExA(HKEY_CURRENT_USER, ImHexContextMenuKey, 0x00, KEY_SET_VALUE, &key) == ERROR_SUCCESS);
+            RegCloseKey(key);
+
+            return keyExists;
+        }
+
+    }
+
+    void registerSettings() {
+
+        /* General */
+
+        ContentRegistry::Settings::add("hex.builtin.setting.general", "hex.builtin.setting.general.context_menu_entry", 0, [](auto name, nlohmann::json &setting) {
+            static bool enabled = hasImHexContextMenuEntry();
+
+            if (ImGui::Checkbox(name.data(), &enabled)) {
+
+                if (enabled)
+                    addImHexContextMenuEntry();
+                else
+                    removeImHexContextMenuEntry();
+
+                enabled = hasImHexContextMenuEntry();
+                setting = enabled;
+
+                return true;
+            }
+
+            return false;
+        });
+
+    }
+
+}