d398a6ced2
- CitrineOS core extracted (CSMS OCPP 2.0.1) - OpenOCPP extracted (firmware OCPP 1.6J/2.0.1) - ShapeShifter library installed (pip install -e) - ShapeShifter specification extracted - EVerest extracted TODO updated with progress
85 lines
2.9 KiB
C++
85 lines
2.9 KiB
C++
// SPDX-License-Identifier: Apache-2.0
|
|
// Copyright Pionix GmbH and Contributors to EVerest
|
|
|
|
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <stdexcept>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
class RegisteredCommandBase {
|
|
public:
|
|
virtual ~RegisteredCommandBase() = default;
|
|
virtual bool operator()(const std::vector<std::string>& /*arguments*/) const = 0;
|
|
[[nodiscard]] virtual std::size_t get_argument_count() const = 0;
|
|
};
|
|
|
|
class RegisteredCommand : public RegisteredCommandBase {
|
|
public:
|
|
RegisteredCommand(std::string command_, std::size_t argument_count_,
|
|
std::function<bool(std::vector<std::string>)> function_) :
|
|
command_name{std::move(command_)}, argument_count(argument_count_), function{std::move(function_)} {
|
|
}
|
|
|
|
~RegisteredCommand() override = default;
|
|
|
|
bool operator()(const std::vector<std::string>& arguments) const override {
|
|
if (arguments.size() != argument_count) {
|
|
throw std::invalid_argument{"Invalid number of arguments for: " + command_name + " expected " +
|
|
std::to_string(argument_count) + " got " + std::to_string(arguments.size())};
|
|
}
|
|
return function(arguments);
|
|
}
|
|
|
|
[[nodiscard]] std::size_t get_argument_count() const override {
|
|
return argument_count;
|
|
}
|
|
|
|
private:
|
|
std::string command_name;
|
|
std::size_t argument_count;
|
|
std::function<bool(std::vector<std::string>)> function;
|
|
};
|
|
|
|
class CommandRegistry {
|
|
public:
|
|
CommandRegistry() = default;
|
|
|
|
void register_command(std::string command_name, size_t argument_count,
|
|
const std::function<bool(std::vector<std::string>)>& function) {
|
|
registered_commands.emplace(command_name,
|
|
std::make_unique<RegisteredCommand>(command_name, argument_count, function));
|
|
}
|
|
|
|
const RegisteredCommandBase& get_registered_command(const std::string& command_name,
|
|
std::size_t arguments_count) const {
|
|
|
|
const auto cmd_count = registered_commands.count(command_name);
|
|
|
|
if (cmd_count == 1) {
|
|
if (auto iter = registered_commands.find(command_name); iter != registered_commands.end()) {
|
|
return *iter->second;
|
|
}
|
|
} else if (cmd_count >= 2) {
|
|
auto range = registered_commands.equal_range(command_name);
|
|
for (auto iter = range.first; iter != range.second; ++iter) {
|
|
if (iter->second->get_argument_count() == arguments_count) {
|
|
return *iter->second;
|
|
}
|
|
}
|
|
}
|
|
|
|
throw std::invalid_argument{"Command not found: " + command_name};
|
|
}
|
|
|
|
private:
|
|
using RegisteredCommands = std::unordered_multimap<std::string, std::unique_ptr<RegisteredCommandBase>>;
|
|
|
|
RegisteredCommands registered_commands;
|
|
};
|