129 lines
4.4 KiB
C++
129 lines
4.4 KiB
C++
#include <numeric>
|
|
#include <string_view>
|
|
#include <ranges>
|
|
|
|
#include "hex/subcommands/subcommands.hpp"
|
|
|
|
#include <hex/api/event_manager.hpp>
|
|
#include <hex/api/plugin_manager.hpp>
|
|
#include <hex/api/imhex_api.hpp>
|
|
#include <hex/helpers/logger.hpp>
|
|
#include <hex/helpers/fmt.hpp>
|
|
|
|
namespace hex::subcommands {
|
|
|
|
std::optional<SubCommand> findSubCommand(const std::string &arg) {
|
|
for (auto &plugin : PluginManager::getPlugins()) {
|
|
for (auto &subCommand : plugin.getSubCommands()) {
|
|
if (hex::format("--{}", subCommand.commandLong) == arg || hex::format("-{}", subCommand.commandShort) == arg) {
|
|
return subCommand;
|
|
}
|
|
}
|
|
}
|
|
|
|
return std::nullopt;
|
|
}
|
|
|
|
void processArguments(const std::vector<std::string> &args) {
|
|
// If no arguments, do not even try to process arguments
|
|
// (important because this function will exit ImHex if an instance is already opened,
|
|
// and we don't want that if no arguments were provided)
|
|
if (args.empty())
|
|
return;
|
|
|
|
std::vector<std::pair<SubCommand, std::vector<std::string>>> subCommands;
|
|
|
|
auto argsIter = args.begin();
|
|
|
|
// Get subcommand associated with the first argument
|
|
std::optional<SubCommand> currentSubCommand = findSubCommand(*argsIter);
|
|
|
|
if (currentSubCommand) {
|
|
argsIter += 1;
|
|
// If it is a valid subcommand, remove it from the argument list
|
|
} else {
|
|
// If no (valid) subcommand was provided, the default one is --open
|
|
currentSubCommand = findSubCommand("--open");
|
|
}
|
|
|
|
// Arguments of the current subcommand
|
|
std::vector<std::string> currentSubCommandArgs;
|
|
|
|
// Compute all subcommands to run
|
|
while (argsIter != args.end()) {
|
|
const std::string &arg = *argsIter;
|
|
|
|
if (arg == "--othercmd") {
|
|
// Save command to run
|
|
if (currentSubCommand) {
|
|
subCommands.emplace_back(*currentSubCommand, currentSubCommandArgs);
|
|
}
|
|
|
|
currentSubCommand = std::nullopt;
|
|
currentSubCommandArgs = { };
|
|
|
|
} else if (currentSubCommand) {
|
|
// Add current argument to the current command
|
|
currentSubCommandArgs.push_back(arg);
|
|
} else {
|
|
// Get next subcommand from current argument
|
|
currentSubCommand = findSubCommand(arg);
|
|
if (!currentSubCommand) {
|
|
log::error("No subcommand named '{}' found", arg);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
argsIter += 1;
|
|
}
|
|
|
|
// Save last command to run
|
|
if (currentSubCommand.has_value()) {
|
|
subCommands.emplace_back(*currentSubCommand, currentSubCommandArgs);
|
|
}
|
|
|
|
// Run the subcommands
|
|
for (auto &[subcommand, args] : subCommands) {
|
|
subcommand.callback(args);
|
|
}
|
|
|
|
// Exit the process if it's not the main instance (the commands have been forwarded to another instance)
|
|
if (!ImHexApi::System::isMainInstance()) {
|
|
std::exit(0);
|
|
}
|
|
}
|
|
|
|
void forwardSubCommand(const std::string &cmdName, const std::vector<std::string> &args) {
|
|
log::debug("Forwarding subcommand {} (maybe to us)", cmdName);
|
|
|
|
std::vector<u8> data;
|
|
if (!args.empty()) {
|
|
for (const auto &arg: args) {
|
|
data.insert(data.end(), arg.begin(), arg.end());
|
|
data.push_back('\0');
|
|
}
|
|
|
|
data.pop_back();
|
|
}
|
|
|
|
SendMessageToMainInstance::post(hex::format("command/{}", cmdName), data);
|
|
}
|
|
|
|
void registerSubCommand(const std::string &cmdName, const ForwardCommandHandler &handler) {
|
|
log::debug("Registered new forward command handler: {}", cmdName);
|
|
|
|
ImHexApi::Messaging::registerHandler(hex::format("command/{}", cmdName), [handler](const std::vector<u8> &eventData){
|
|
std::string string(reinterpret_cast<const char *>(eventData.data()), eventData.size());
|
|
|
|
std::vector<std::string> args;
|
|
|
|
for (const auto &argument : std::views::split(string, char(0x00))) {
|
|
std::string arg(argument.data(), argument.size());
|
|
args.push_back(arg);
|
|
}
|
|
|
|
handler(args);
|
|
});
|
|
}
|
|
}
|