137 lines
3.2 KiB
C++
137 lines
3.2 KiB
C++
|
/*
|
|||
|
* ap.cpp
|
|||
|
*
|
|||
|
* Created on: 05 сен. 2022 г.
|
|||
|
* Author: alexander
|
|||
|
*/
|
|||
|
|
|||
|
#include <ap/ap.hpp>
|
|||
|
|
|||
|
namespace ap
|
|||
|
{
|
|||
|
|
|||
|
ConfigOption::ConfigOption(const std::string &longParameter, const char shortParameter, const hasArg ha) :
|
|||
|
_longParameter(nullptr), _shortParameter(shortParameter), _ha(ha)
|
|||
|
{
|
|||
|
_longParameter = new char[longParameter.length() + 1];
|
|||
|
strcpy(_longParameter, longParameter.c_str());
|
|||
|
}
|
|||
|
|
|||
|
const char* ConfigOption::getLongParameter() const
|
|||
|
{
|
|||
|
return _longParameter;
|
|||
|
}
|
|||
|
|
|||
|
const char& ConfigOption::getShortParameter() const
|
|||
|
{
|
|||
|
return _shortParameter;
|
|||
|
}
|
|||
|
|
|||
|
const hasArg& ConfigOption::getPresenceArgument() const
|
|||
|
{
|
|||
|
return _ha;
|
|||
|
}
|
|||
|
|
|||
|
Option::Option() : _set(false)
|
|||
|
{
|
|||
|
}
|
|||
|
|
|||
|
void Option::set()
|
|||
|
{
|
|||
|
_set = true;
|
|||
|
}
|
|||
|
|
|||
|
void Option::push(const std::string &value)
|
|||
|
{
|
|||
|
_values.push_back(value);
|
|||
|
}
|
|||
|
|
|||
|
std::vector<std::string>& Option::getValues()
|
|||
|
{
|
|||
|
return _values;
|
|||
|
}
|
|||
|
|
|||
|
bool Option::isSet() const
|
|||
|
{
|
|||
|
return _set;
|
|||
|
}
|
|||
|
|
|||
|
void Hub::_createArguments(const std::vector<ConfigOption> &options, bool silence)
|
|||
|
{
|
|||
|
_longOptions = new struct option[options.size() + 1];
|
|||
|
_sizeOptions = options.size();
|
|||
|
std::string temp;
|
|||
|
if (silence)
|
|||
|
temp.push_back(':');
|
|||
|
for (auto const &opt : options | boost::adaptors::indexed(0))
|
|||
|
{
|
|||
|
_longOptions[opt.index()].name = opt.value().getLongParameter();
|
|||
|
_longOptions[opt.index()].has_arg = opt.value().getPresenceArgument();
|
|||
|
_longOptions[opt.index()].flag = nullptr;
|
|||
|
_longOptions[opt.index()].val = opt.value().getShortParameter();
|
|||
|
|
|||
|
temp.push_back(opt.value().getShortParameter());
|
|||
|
switch (opt.value().getPresenceArgument())
|
|||
|
{
|
|||
|
case hasArg::OPTIONAL:
|
|||
|
temp.push_back(':');
|
|||
|
/* no break */
|
|||
|
case hasArg::REQUIRED:
|
|||
|
temp.push_back(':');
|
|||
|
break;
|
|||
|
case hasArg::NO:
|
|||
|
break;
|
|||
|
}
|
|||
|
|
|||
|
_arguments[opt.value().getShortParameter()].first = false;
|
|||
|
}
|
|||
|
|
|||
|
_longOptions[options.size()].name = nullptr;
|
|||
|
_longOptions[options.size()].has_arg = 0;
|
|||
|
_longOptions[options.size()].flag = nullptr;
|
|||
|
_longOptions[options.size()].val = 0;
|
|||
|
|
|||
|
_shortOptions = new char[temp.size() + 1];
|
|||
|
strcpy(_shortOptions, temp.c_str());
|
|||
|
}
|
|||
|
|
|||
|
Hub::Hub(const std::vector<ConfigOption> &options, bool silence) :
|
|||
|
_longOptions(nullptr), _shortOptions(nullptr)
|
|||
|
{
|
|||
|
_createArguments(options, silence);
|
|||
|
}
|
|||
|
|
|||
|
void Hub::readArguments(int argc, char *argv[], void (*_callback)())
|
|||
|
{
|
|||
|
int next_option;
|
|||
|
while ((next_option = getopt_long(argc, argv, _shortOptions, _longOptions, nullptr)) != -1)
|
|||
|
{
|
|||
|
if (_arguments.count(next_option))
|
|||
|
{
|
|||
|
_arguments[next_option].first = true;
|
|||
|
_arguments[next_option].second.set();
|
|||
|
if (optarg)
|
|||
|
_arguments[next_option].second.push(std::string(optarg));
|
|||
|
}
|
|||
|
if (next_option == '?' && _callback)
|
|||
|
_callback();
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
Option Hub::getOption(char key) const
|
|||
|
{
|
|||
|
return _arguments.count(key) && _arguments.at(key).first ? _arguments.at(key).second : Option();
|
|||
|
}
|
|||
|
|
|||
|
Hub::~Hub()
|
|||
|
{
|
|||
|
delete[] _shortOptions;
|
|||
|
for (size_t i = 0; i < _sizeOptions; ++i)
|
|||
|
{
|
|||
|
delete[] _longOptions[i].name;
|
|||
|
}
|
|||
|
delete[] _longOptions;
|
|||
|
}
|
|||
|
|
|||
|
}
|