137 lines
3.1 KiB
C++
137 lines
3.1 KiB
C++
/*
|
||
* ap.cpp
|
||
*
|
||
* Created on: 05 сен. 2022 г.
|
||
* Author: alexander
|
||
*/
|
||
|
||
#include <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;
|
||
}
|
||
|
||
}
|