#pragma once #include #include #include #include #include #include #include #include #include #include #include #include using ImGuiDataType = int; using ImGuiInputTextFlags = int; struct ImColor; namespace hex { class View; class LanguageDefinition; class Shortcut; namespace dp { class Node; } namespace prv { class Provider; } /* The Content Registry is the heart of all features in ImHex that are in some way extendable by Plugins. It allows you to add/register new content that will be picked up and used by the ImHex core or by other plugins when needed. */ namespace ContentRegistry { /* Settings Registry. Allows adding of new entries into the ImHex preferences window. */ namespace Settings { namespace Widgets { class Widget { public: virtual ~Widget() = default; virtual bool draw(const std::string &name) = 0; virtual void load(const nlohmann::json &data) = 0; virtual nlohmann::json store() = 0; class Interface { public: friend class Widget; Interface& requiresRestart() { this->m_requiresRestart = true; return *this; } Interface& setEnabledCallback(std::function callback) { this->m_enabledCallback = std::move(callback); return *this; } Interface& setChangedCallback(std::function callback) { this->m_changedCallback = std::move(callback); return *this; } Interface& setTooltip(const std::string &tooltip) { this->m_tooltip = tooltip; return *this; } [[nodiscard]] Widget& getWidget() const { return *this->m_widget; } private: explicit Interface(Widget *widget) : m_widget(widget) {} Widget *m_widget; bool m_requiresRestart = false; std::function m_enabledCallback; std::function m_changedCallback; std::optional m_tooltip; }; [[nodiscard]] bool doesRequireRestart() const { return this->m_interface.m_requiresRestart; } [[nodiscard]] bool isEnabled() const { return !this->m_interface.m_enabledCallback || this->m_interface.m_enabledCallback(); } [[nodiscard]] const std::optional& getTooltip() const { return this->m_interface.m_tooltip; } void onChanged() { if (this->m_interface.m_changedCallback) this->m_interface.m_changedCallback(*this); } [[nodiscard]] Interface& getInterface() { return this->m_interface; } private: Interface m_interface = Interface(this); }; class Checkbox : public Widget { public: explicit Checkbox(bool defaultValue) : m_value(defaultValue) { } bool draw(const std::string &name) override; void load(const nlohmann::json &data) override; nlohmann::json store() override; [[nodiscard]] bool isChecked() const { return this->m_value; } private: bool m_value; }; class SliderInteger : public Widget { public: SliderInteger(i32 defaultValue, i32 min, i32 max) : m_value(defaultValue), m_min(min), m_max(max) { } bool draw(const std::string &name) override; void load(const nlohmann::json &data) override; nlohmann::json store() override; [[nodiscard]] i32 getValue() const { return this->m_value; } private: int m_value; i32 m_min, m_max; }; class SliderFloat : public Widget { public: SliderFloat(float defaultValue, float min, float max) : m_value(defaultValue), m_min(min), m_max(max) { } bool draw(const std::string &name) override; void load(const nlohmann::json &data) override; nlohmann::json store() override; [[nodiscard]] float getValue() const { return this->m_value; } private: float m_value; float m_min, m_max; }; class ColorPicker : public Widget { public: explicit ColorPicker(ImColor defaultColor); bool draw(const std::string &name) override; void load(const nlohmann::json &data) override; nlohmann::json store() override; [[nodiscard]] ImColor getColor() const; private: std::array m_value{}; }; class DropDown : public Widget { public: explicit DropDown(const std::vector &items, const std::vector &settingsValues, const nlohmann::json &defaultItem) : m_items(items), m_settingsValues(settingsValues), m_defaultItem(defaultItem) { } bool draw(const std::string &name) override; void load(const nlohmann::json &data) override; nlohmann::json store() override; [[nodiscard]] const nlohmann::json& getValue() const; private: std::vector m_items; std::vector m_settingsValues; nlohmann::json m_defaultItem; int m_value = -1; }; class TextBox : public Widget { public: explicit TextBox(std::string defaultValue) : m_value(std::move(defaultValue)) { } bool draw(const std::string &name) override; void load(const nlohmann::json &data) override; nlohmann::json store() override; [[nodiscard]] const std::string& getValue() const { return this->m_value; } private: std::string m_value; }; class FilePicker : public Widget { public: bool draw(const std::string &name) override; void load(const nlohmann::json &data) override; nlohmann::json store() override; [[nodiscard]] std::fs::path getPath() const { return this->m_value; } private: std::string m_value; }; } namespace impl { struct Entry { std::string unlocalizedName; std::unique_ptr widget; }; struct SubCategory { std::string unlocalizedName; std::vector entries; }; struct Category { std::string unlocalizedName; std::string unlocalizedDescription; std::vector subCategories; }; void load(); void store(); void clear(); std::vector &getSettings(); nlohmann::json& getSetting(const std::string &unlocalizedCategory, const std::string &unlocalizedName, const nlohmann::json &defaultValue); nlohmann::json &getSettingsData(); Widgets::Widget* add(const std::string &unlocalizedCategory, const std::string &unlocalizedSubCategory, const std::string &unlocalizedName, std::unique_ptr &&widget); } template T> Widgets::Widget::Interface& add(const std::string &unlocalizedCategory, const std::string &unlocalizedSubCategory, const std::string &unlocalizedName, auto && ... args) { return impl::add( unlocalizedCategory, unlocalizedSubCategory, unlocalizedName, std::make_unique(std::forward(args)...) )->getInterface(); } void setCategoryDescription(const std::string &unlocalizedCategory, const std::string &unlocalizedDescription); nlohmann::json read(const std::string &unlocalizedCategory, const std::string &unlocalizedName, const nlohmann::json &defaultValue); void write(const std::string &unlocalizedCategory, const std::string &unlocalizedName, const nlohmann::json &value); } /* Command Palette Command Registry. Allows adding of new commands to the command palette */ namespace CommandPaletteCommands { enum class Type : u32 { SymbolCommand, KeywordCommand }; namespace impl { struct QueryResult { std::string name; std::function callback; }; using DisplayCallback = std::function; using ExecuteCallback = std::function; using QueryCallback = std::function(std::string)>; struct Entry { Type type; std::string command; std::string unlocalizedDescription; DisplayCallback displayCallback; ExecuteCallback executeCallback; }; struct Handler { Type type; std::string command; QueryCallback queryCallback; DisplayCallback displayCallback; }; std::vector &getEntries(); std::vector &getHandlers(); } /** * @brief Adds a new command to the command palette * @param type The type of the command * @param command The command to add * @param unlocalizedDescription The description of the command * @param displayCallback The callback that will be called when the command is displayed in the command palette * @param executeCallback The callback that will be called when the command is executed */ void add( Type type, const std::string &command, const std::string &unlocalizedDescription, const impl::DisplayCallback &displayCallback, const impl::ExecuteCallback &executeCallback = [](auto) {}); /** * @brief Adds a new command handler to the command palette * @param type The type of the command * @param command The command to add * @param queryCallback The callback that will be called when the command palette wants to load the name and callback items * @param displayCallback The callback that will be called when the command is displayed in the command palette */ void addHandler( Type type, const std::string &command, const impl::QueryCallback &queryCallback, const impl::DisplayCallback &displayCallback); } /* Pattern Language Function Registry. Allows adding of new functions that may be used inside the pattern language */ namespace PatternLanguage { namespace impl { using VisualizerFunctionCallback = std::function)>; struct FunctionDefinition { pl::api::Namespace ns; std::string name; pl::api::FunctionParameterCount parameterCount; pl::api::FunctionCallback callback; bool dangerous; }; struct Visualizer { pl::api::FunctionParameterCount parameterCount; VisualizerFunctionCallback callback; }; std::map &getVisualizers(); std::map &getInlineVisualizers(); std::map &getPragmas(); std::vector &getFunctions(); } /** * @brief Provides access to the current provider's pattern language runtime * @return Runtime */ pl::PatternLanguage& getRuntime(); /** * @brief Provides access to the current provider's pattern language runtime's lock * @return Lock */ std::mutex& getRuntimeLock(); /** * @brief Configures the pattern language runtime using ImHex's default settings * @param runtime The pattern language runtime to configure * @param provider The provider to use for data access */ void configureRuntime(pl::PatternLanguage &runtime, prv::Provider *provider); /** * @brief Adds a new pragma to the pattern language * @param name The name of the pragma * @param handler The handler that will be called when the pragma is encountered */ void addPragma(const std::string &name, const pl::api::PragmaHandler &handler); /** * @brief Adds a new function to the pattern language * @param ns The namespace of the function * @param name The name of the function * @param parameterCount The amount of parameters the function takes * @param func The function callback */ void addFunction(const pl::api::Namespace &ns, const std::string &name, pl::api::FunctionParameterCount parameterCount, const pl::api::FunctionCallback &func); /** * @brief Adds a new dangerous function to the pattern language * @note Dangerous functions are functions that require the user to explicitly allow them to be used * @param ns The namespace of the function * @param name The name of the function * @param parameterCount The amount of parameters the function takes * @param func The function callback */ void addDangerousFunction(const pl::api::Namespace &ns, const std::string &name, pl::api::FunctionParameterCount parameterCount, const pl::api::FunctionCallback &func); /** * @brief Adds a new visualizer to the pattern language * @note Visualizers are extensions to the [[hex::visualize]] attribute, used to visualize data * @param name The name of the visualizer * @param function The function callback * @param parameterCount The amount of parameters the function takes */ void addVisualizer(const std::string &name, const impl::VisualizerFunctionCallback &function, pl::api::FunctionParameterCount parameterCount); /** * @brief Adds a new inline visualizer to the pattern language * @note Inline visualizers are extensions to the [[hex::inline_visualize]] attribute, used to visualize data * @param name The name of the visualizer * @param function The function callback * @param parameterCount The amount of parameters the function takes */ void addInlineVisualizer(const std::string &name, const impl::VisualizerFunctionCallback &function, pl::api::FunctionParameterCount parameterCount); } /* View Registry. Allows adding of new windows */ namespace Views { namespace impl { void add(std::unique_ptr &&view); std::map> &getEntries(); } /** * @brief Adds a new view to ImHex * @tparam T The custom view class that extends View * @tparam Args Arguments types * @param args Arguments passed to the constructor of the view */ template T, typename... Args> void add(Args &&...args) { return impl::add(std::make_unique(std::forward(args)...)); } /** * @brief Gets a view by its unlocalized name * @param unlocalizedName The unlocalized name of the view * @return The view if it exists, nullptr otherwise */ View* getViewByName(const std::string &unlocalizedName); } /* Tools Registry. Allows adding new entries to the tools window */ namespace Tools { namespace impl { using Callback = std::function; struct Entry { std::string name; Callback function; bool detached; }; std::vector &getEntries(); } /** * @brief Adds a new tool to the tools window * @param unlocalizedName The unlocalized name of the tool * @param function The function that will be called to draw the tool */ void add(const std::string &unlocalizedName, const impl::Callback &function); } /* Data Inspector Registry. Allows adding of new types to the data inspector */ namespace DataInspector { enum class NumberDisplayStyle { Decimal, Hexadecimal, Octal }; namespace impl { using DisplayFunction = std::function; using EditingFunction = std::function(std::string, std::endian)>; using GeneratorFunction = std::function &, std::endian, NumberDisplayStyle)>; struct Entry { std::string unlocalizedName; size_t requiredSize; size_t maxSize; GeneratorFunction generatorFunction; std::optional editingFunction; }; std::vector &getEntries(); } /** * @brief Adds a new entry to the data inspector * @param unlocalizedName The unlocalized name of the entry * @param requiredSize The minimum required number of bytes available for the entry to appear * @param displayGeneratorFunction The function that will be called to generate the display function * @param editingFunction The function that will be called to edit the data */ void add(const std::string &unlocalizedName, size_t requiredSize, impl::GeneratorFunction displayGeneratorFunction, std::optional editingFunction = std::nullopt); /** * @brief Adds a new entry to the data inspector * @param unlocalizedName The unlocalized name of the entry * @param requiredSize The minimum required number of bytes available for the entry to appear * @param maxSize The maximum number of bytes to read from the data * @param displayGeneratorFunction The function that will be called to generate the display function * @param editingFunction The function that will be called to edit the data */ void add(const std::string &unlocalizedName, size_t requiredSize, size_t maxSize, impl::GeneratorFunction displayGeneratorFunction, std::optional editingFunction = std::nullopt); } /* Data Processor Node Registry. Allows adding new processor nodes to be used in the data processor */ namespace DataProcessorNode { namespace impl { using CreatorFunction = std::function()>; struct Entry { std::string category; std::string name; CreatorFunction creatorFunction; }; void add(const Entry &entry); std::vector &getEntries(); } /** * @brief Adds a new node to the data processor * @tparam T The custom node class that extends dp::Node * @tparam Args Arguments types * @param unlocalizedCategory The unlocalized category name of the node * @param unlocalizedName The unlocalized name of the node * @param args Arguments passed to the constructor of the node */ template T, typename... Args> void add(const std::string &unlocalizedCategory, const std::string &unlocalizedName, Args &&...args) { add(impl::Entry { unlocalizedCategory.c_str(), unlocalizedName.c_str(), [=, ...args = std::forward(args)] mutable { auto node = std::make_unique(std::forward(args)...); node->setUnlocalizedName(unlocalizedName); return node; } }); } /** * @brief Adds a separator to the data processor right click menu */ void addSeparator(); } /* Language Registry. Allows together with the LangEntry class and the _lang user defined literal to add new languages */ namespace Language { /** * @brief Loads localization information from json data * @param data The language data */ void addLocalization(const nlohmann::json &data); namespace impl { std::map &getLanguages(); std::map> &getLanguageDefinitions(); } } /* Interface Registry. Allows adding new items to various interfaces */ namespace Interface { namespace impl { using DrawCallback = std::function; using MenuCallback = std::function; using EnabledCallback = std::function; using ClickCallback = std::function; struct MainMenuItem { std::string unlocalizedName; }; struct MenuItem { std::vector unlocalizedNames; std::unique_ptr shortcut; View *view; MenuCallback callback; EnabledCallback enabledCallback; }; struct SidebarItem { std::string icon; DrawCallback callback; EnabledCallback enabledCallback; }; struct TitleBarButton { std::string icon; std::string unlocalizedTooltip; ClickCallback callback; }; constexpr static auto SeparatorValue = "$SEPARATOR$"; constexpr static auto SubMenuValue = "$SUBMENU$"; std::multimap &getMainMenuItems(); std::multimap &getMenuItems(); std::vector &getWelcomeScreenEntries(); std::vector &getFooterItems(); std::vector &getToolbarItems(); std::vector &getSidebarItems(); std::vector &getTitleBarButtons(); } /** * @brief Adds a new top-level main menu entry * @param unlocalizedName The unlocalized name of the entry * @param priority The priority of the entry. Lower values are displayed first */ void registerMainMenuItem(const std::string &unlocalizedName, u32 priority); /** * @brief Adds a new main menu entry * @param unlocalizedMainMenuNames The unlocalized names of the main menu entries * @param priority The priority of the entry. Lower values are displayed first * @param shortcut The shortcut to use for the entry * @param function The function to call when the entry is clicked * @param enabledCallback The function to call to determine if the entry is enabled * @param view The view to use for the entry. If nullptr, the shortcut will work globally */ void addMenuItem(const std::vector &unlocalizedMainMenuNames, u32 priority, const Shortcut &shortcut, const impl::MenuCallback &function, const impl::EnabledCallback& enabledCallback = []{ return true; }, View *view = nullptr); /** * @brief Adds a new main menu sub-menu entry * @param unlocalizedMainMenuNames The unlocalized names of the main menu entries * @param priority The priority of the entry. Lower values are displayed first * @param function The function to call when the entry is clicked * @param enabledCallback The function to call to determine if the entry is enabled */ void addMenuItemSubMenu(std::vector unlocalizedMainMenuNames, u32 priority, const impl::MenuCallback &function, const impl::EnabledCallback& enabledCallback = []{ return true; }); /** * @brief Adds a new main menu separator * @param unlocalizedMainMenuNames The unlocalized names of the main menu entries * @param priority The priority of the entry. Lower values are displayed first */ void addMenuItemSeparator(std::vector unlocalizedMainMenuNames, u32 priority); /** * @brief Adds a new welcome screen entry * @param function The function to call to draw the entry */ void addWelcomeScreenEntry(const impl::DrawCallback &function); /** * @brief Adds a new footer item * @param function The function to call to draw the item */ void addFooterItem(const impl::DrawCallback &function); /** * @brief Adds a new toolbar item * @param function The function to call to draw the item */ void addToolbarItem(const impl::DrawCallback &function); /** * @brief Adds a new sidebar item * @param icon The icon to use for the item * @param function The function to call to draw the item * @param enabledCallback The function */ void addSidebarItem(const std::string &icon, const impl::DrawCallback &function, const impl::EnabledCallback &enabledCallback = []{ return true; }); /** * @brief Adds a new title bar button * @param icon The icon to use for the button * @param unlocalizedTooltip The unlocalized tooltip to use for the button * @param function The function to call when the button is clicked */ void addTitleBarButton(const std::string &icon, const std::string &unlocalizedTooltip, const impl::ClickCallback &function); } /* Provider Registry. Allows adding new data providers to be created from the UI */ namespace Provider { namespace impl { void addProviderName(const std::string &unlocalizedName); using ProviderCreationFunction = prv::Provider*(*)(); void add(const std::string &typeName, ProviderCreationFunction creationFunction); std::vector &getEntries(); } /** * @brief Adds a new provider to the list of providers * @tparam T The provider type that extends hex::prv::Provider * @param addToList Whether to display the provider in the Other Providers list in the welcome screen and File menu */ template T> void add(bool addToList = true) { auto typeName = T().getTypeName(); impl::add(typeName, [] -> prv::Provider* { return new T(); }); if (addToList) impl::addProviderName(typeName); } } /* Data Formatter Registry. Allows adding formatters that are used in the Copy-As menu for example */ namespace DataFormatter { namespace impl { using Callback = std::function; struct Entry { std::string unlocalizedName; Callback callback; }; std::vector &getEntries(); } /** * @brief Adds a new data formatter * @param unlocalizedName The unlocalized name of the formatter * @param callback The function to call to format the data */ void add(const std::string &unlocalizedName, const impl::Callback &callback); } /* File Handler Registry. Allows adding handlers for opening files specific file types */ namespace FileHandler { namespace impl { using Callback = std::function; struct Entry { std::vector extensions; Callback callback; }; std::vector &getEntries(); } /** * @brief Adds a new file handler * @param extensions The file extensions to handle * @param callback The function to call to handle the file */ void add(const std::vector &extensions, const impl::Callback &callback); } /* Hex Editor Registry. Allows adding new functionality to the hex editor */ namespace HexEditor { class DataVisualizer { public: DataVisualizer(std::string unlocalizedName, u16 bytesPerCell, u16 maxCharsPerCell) : m_unlocalizedName(std::move(unlocalizedName)), m_bytesPerCell(bytesPerCell), m_maxCharsPerCell(maxCharsPerCell) { } virtual ~DataVisualizer() = default; virtual void draw(u64 address, const u8 *data, size_t size, bool upperCase) = 0; virtual bool drawEditing(u64 address, u8 *data, size_t size, bool upperCase, bool startedEditing) = 0; [[nodiscard]] u16 getBytesPerCell() const { return this->m_bytesPerCell; } [[nodiscard]] u16 getMaxCharsPerCell() const { return this->m_maxCharsPerCell; } [[nodiscard]] const std::string& getUnlocalizedName() const { return this->m_unlocalizedName; } protected: const static int TextInputFlags; bool drawDefaultScalarEditingTextBox(u64 address, const char *format, ImGuiDataType dataType, u8 *data, ImGuiInputTextFlags flags) const; bool drawDefaultTextEditingTextBox(u64 address, std::string &data, ImGuiInputTextFlags flags) const; private: std::string m_unlocalizedName; u16 m_bytesPerCell; u16 m_maxCharsPerCell; }; namespace impl { void addDataVisualizer(std::shared_ptr &&visualizer); std::vector> &getVisualizers(); } /** * @brief Adds a new cell data visualizer * @tparam T The data visualizer type that extends hex::DataVisualizer * @param args The arguments to pass to the constructor of the data visualizer */ template T, typename... Args> void addDataVisualizer(Args &&...args) { return impl::addDataVisualizer(std::make_shared(std::forward(args)...)); } /** * @brief Gets a data visualizer by its unlocalized name * @param unlocalizedName Unlocalized name of the data visualizer * @return The data visualizer, or nullptr if it doesn't exist */ std::shared_ptr getVisualizerByName(const std::string &unlocalizedName); } /* Hash Registry. Allows adding new hashes to the Hash view */ namespace Hashes { class Hash { public: explicit Hash(std::string unlocalizedName) : m_unlocalizedName(std::move(unlocalizedName)) {} virtual ~Hash() = default; class Function { public: using Callback = std::function(const Region&, prv::Provider *)>; Function(Hash *type, std::string name, Callback callback) : m_type(type), m_name(std::move(name)), m_callback(std::move(callback)) { } [[nodiscard]] Hash *getType() { return this->m_type; } [[nodiscard]] const Hash *getType() const { return this->m_type; } [[nodiscard]] const std::string &getName() const { return this->m_name; } const std::vector& get(const Region& region, prv::Provider *provider) { if (this->m_cache.empty()) { this->m_cache = this->m_callback(region, provider); } return this->m_cache; } void reset() { this->m_cache.clear(); } private: Hash *m_type; std::string m_name; Callback m_callback; std::vector m_cache; }; virtual void draw() { } [[nodiscard]] virtual Function create(std::string name) = 0; [[nodiscard]] virtual nlohmann::json store() const = 0; virtual void load(const nlohmann::json &json) = 0; [[nodiscard]] const std::string &getUnlocalizedName() const { return this->m_unlocalizedName; } protected: [[nodiscard]] Function create(const std::string &name, const Function::Callback &callback) { return { this, name, callback }; } private: std::string m_unlocalizedName; }; namespace impl { std::vector> &getHashes(); void add(std::unique_ptr &&hash); } /** * @brief Adds a new hash * @tparam T The hash type that extends hex::Hash * @param args The arguments to pass to the constructor of the hash */ template void add(Args && ... args) { impl::add(std::make_unique(std::forward(args)...)); } } /* Background Service Registry. Allows adding new background services */ namespace BackgroundServices { namespace impl { using Callback = std::function; struct Service { std::string name; std::jthread thread; }; std::vector &getServices(); void stopServices(); } void registerService(const std::string &unlocalizedName, const impl::Callback &callback); } /* Network Communication Interface Registry. Allows adding new communication interface endpoints */ namespace CommunicationInterface { namespace impl { using NetworkCallback = std::function; std::map &getNetworkEndpoints(); } void registerNetworkEndpoint(const std::string &endpoint, const impl::NetworkCallback &callback); } /* Experiments Registry. Allows adding new experiments */ namespace Experiments { namespace impl { struct Experiment { std::string unlocalizedName, unlocalizedDescription; bool enabled; }; std::map &getExperiments(); } void addExperiment(const std::string &experimentName, const std::string &unlocalizedName, const std::string &unlocalizedDescription = ""); void enableExperiement(const std::string &experimentName, bool enabled); [[nodiscard]] bool isExperimentEnabled(const std::string &experimentName); } } }