117 lines
3.1 KiB
C++
117 lines
3.1 KiB
C++
/*
|
||
* argpars.cpp
|
||
*
|
||
* Created on: 30 авг. 2022 г.
|
||
* Author: alexander
|
||
*/
|
||
|
||
#include <argpars.hpp>
|
||
|
||
ArgPars::ArgPars(cPtr callback, bool silencemode) :
|
||
longOptions(nullptr), sizeLongOption(0), shortOptions(nullptr), sizeShortOption(0), _callback(callback)
|
||
{
|
||
if (silencemode)
|
||
{
|
||
shortOptions = new char[++sizeShortOption];
|
||
shortOptions[0] = ':';
|
||
}
|
||
}
|
||
|
||
void ArgPars::addShortOption(const char shortKey, hasArg argument)
|
||
{
|
||
size_t sizeTemp = sizeShortOption;
|
||
if (argument == hasArg::REQUIRED)
|
||
++sizeShortOption;
|
||
else if (argument == hasArg::OPTIONAL)
|
||
sizeShortOption += 2;
|
||
char *temp = new char[++sizeShortOption];
|
||
if (sizeTemp)
|
||
strcpy(temp, shortOptions);
|
||
temp[sizeTemp++] = shortKey;
|
||
if (argument == hasArg::REQUIRED)
|
||
temp[sizeTemp] = ':';
|
||
else if (argument == hasArg::OPTIONAL)
|
||
strcpy(&temp[sizeTemp], "::");
|
||
delete[] shortOptions;
|
||
shortOptions = temp;
|
||
cfg[shortKey].first = false;
|
||
}
|
||
|
||
struct option ArgPars::createNewOption(const char shortKey, const std::string &longKey, hasArg argument)
|
||
{
|
||
struct option newOption;
|
||
char *name = new char[longKey.length() + 1];
|
||
strcpy(name, longKey.c_str());
|
||
newOption.name = name;
|
||
newOption.has_arg = argument;
|
||
newOption.flag = nullptr;
|
||
newOption.val = shortKey;
|
||
return newOption;
|
||
}
|
||
|
||
void ArgPars::addLongOption(const char shortKey, const std::string &longKey, hasArg argument)
|
||
{
|
||
struct option *temp = new struct option[++sizeLongOption];
|
||
for (size_t i = 0; i < sizeLongOption - 1; ++i)
|
||
temp[i] = longOptions[i];
|
||
temp[sizeLongOption - 1] = createNewOption(shortKey, longKey, argument);
|
||
delete[] longOptions;
|
||
longOptions = temp;
|
||
}
|
||
|
||
void ArgPars::readArguments(int argc, char *argv[])
|
||
{
|
||
int option_index;
|
||
int next_option;
|
||
while ((next_option = getopt_long(argc, argv, shortOptions, longOptions, &option_index)) != -1)
|
||
{
|
||
if (cfg.count(next_option))
|
||
{
|
||
cfg[next_option].first = true;
|
||
if (optarg)
|
||
cfg[next_option].second.push_back(std::string(optarg));
|
||
}
|
||
if (next_option == '?' && _callback)
|
||
_callback();
|
||
}
|
||
}
|
||
|
||
void ArgPars::addKey(const char shortKey, const std::string &longKey, hasArg argument)
|
||
{
|
||
addShortOption(shortKey, argument);
|
||
addLongOption(shortKey, longKey, argument);
|
||
}
|
||
|
||
void ArgPars::addNullKey()
|
||
{
|
||
struct option *temp = new struct option[++sizeLongOption];
|
||
for (size_t i = 0; i < sizeLongOption - 1; ++i)
|
||
temp[i] = longOptions[i];
|
||
temp[sizeLongOption - 1] = option
|
||
{ nullptr, 0, nullptr, 0 };
|
||
delete[] longOptions;
|
||
longOptions = temp;
|
||
}
|
||
|
||
bool ArgPars::checkKey(const char shortKey)
|
||
{
|
||
return cfg.count(shortKey) > 0 && cfg[shortKey].first;
|
||
}
|
||
|
||
std::string ArgPars::getValue(const char shortKey)
|
||
{
|
||
if (cfg.count(shortKey) && cfg[shortKey].first)
|
||
{
|
||
if (cfg[shortKey].second.size())
|
||
return cfg[shortKey].second.front();
|
||
}
|
||
return {};
|
||
}
|
||
|
||
std::vector<std::string> ArgPars::getValues(const char shortKey)
|
||
{
|
||
if (cfg.count(shortKey) && cfg[shortKey].first)
|
||
return cfg[shortKey].second;
|
||
return {};
|
||
}
|