parent
215220f98b
commit
9e4bfad087
|
@ -77,7 +77,7 @@ namespace blt
|
|||
}
|
||||
|
||||
// returns the first flag that starts with '--' otherwise return the first '-' flag
|
||||
[[nodiscard]] std::string getFirstFullFlag();
|
||||
[[nodiscard]] std::string getFirstFullFlag() const;
|
||||
};
|
||||
|
||||
class arg_nargs_t
|
||||
|
@ -127,13 +127,22 @@ namespace blt
|
|||
std::string a_version{};
|
||||
std::string a_metavar{};
|
||||
bool a_required = true;
|
||||
bool a_disable_help = false;
|
||||
};
|
||||
|
||||
class arg_builder
|
||||
{
|
||||
private:
|
||||
arg_properties_t properties;
|
||||
public:
|
||||
arg_builder(const arg_vector_t& flags): properties(flags)
|
||||
explicit arg_builder(const arg_vector_t& flags): properties(flags)
|
||||
{}
|
||||
|
||||
arg_builder(const std::initializer_list<std::string>& flags): properties(flags)
|
||||
{}
|
||||
|
||||
template<typename... string_args>
|
||||
explicit arg_builder(string_args... flags): properties({flags...})
|
||||
{}
|
||||
|
||||
inline arg_properties_t build()
|
||||
|
@ -171,6 +180,12 @@ namespace blt
|
|||
return *this;
|
||||
}
|
||||
|
||||
inline arg_builder& disableHelp()
|
||||
{
|
||||
properties.a_disable_help = true;
|
||||
return *this;
|
||||
}
|
||||
|
||||
inline arg_builder& setHelp(const std::string& help)
|
||||
{
|
||||
properties.a_help = help;
|
||||
|
@ -203,7 +218,7 @@ namespace blt
|
|||
std::vector<std::string> args;
|
||||
size_t currentIndex = 0;
|
||||
public:
|
||||
arg_tokenizer(std::vector<std::string> args): args(std::move(args))
|
||||
explicit arg_tokenizer(std::vector<std::string> args): args(std::move(args))
|
||||
{}
|
||||
|
||||
// returns the current arg
|
||||
|
@ -296,8 +311,11 @@ namespace blt
|
|||
}
|
||||
} loaded_args;
|
||||
|
||||
bool help_disabled = false;
|
||||
std::string help_inclusion;
|
||||
private:
|
||||
static std::string getMetavar(const arg_properties_t* const& arg);
|
||||
|
||||
static std::string getFlagHelp(const arg_properties_t* const& arg);
|
||||
|
||||
static bool takesArgs(const arg_properties_t* const& arg);
|
||||
|
@ -319,6 +337,8 @@ namespace blt
|
|||
|
||||
void processFlag(arg_tokenizer& tokenizer, const std::string& flag);
|
||||
|
||||
void handleFlag(arg_tokenizer& tokenizer, const std::string& flag, const arg_properties_t* properties);
|
||||
|
||||
public:
|
||||
|
||||
template<typename T>
|
||||
|
@ -398,6 +418,11 @@ namespace blt
|
|||
user_args.max_line_length = size;
|
||||
}
|
||||
|
||||
inline void setHelpExtras(std::string str)
|
||||
{
|
||||
help_inclusion = std::move(str);
|
||||
}
|
||||
|
||||
static std::string filename(const std::string& path);
|
||||
|
||||
~arg_parse()
|
||||
|
|
|
@ -1 +1 @@
|
|||
Subproject commit 93201da2ba5a6aba0a6e57ada64973555629b3e3
|
||||
Subproject commit 77cab8192a879e5d27188f97e8f2080dd7e36ca8
|
|
@ -68,7 +68,7 @@ namespace blt
|
|||
name = str;
|
||||
}
|
||||
|
||||
std::string arg_vector_t::getFirstFullFlag()
|
||||
std::string arg_vector_t::getFirstFullFlag() const
|
||||
{
|
||||
// assign flag so it always exists, will be first '-' flag if we fail to find a '--' flag
|
||||
std::string flag = flags[0];
|
||||
|
@ -231,8 +231,26 @@ namespace blt
|
|||
loaded_args.unrecognized_args.push_back(tokenizer.get());
|
||||
else
|
||||
{
|
||||
loaded_args.data[user_args.name_associations[index]->a_dest] = tokenizer.get();
|
||||
loaded_args.found_args.insert(user_args.name_associations[index]->a_dest);
|
||||
bool found = false;
|
||||
for (const auto& pos_properties : user_args.name_associations)
|
||||
{
|
||||
const auto& flag = tokenizer.get();
|
||||
if (pos_properties->a_flags.contains(flag))
|
||||
{
|
||||
tokenizer.advance();
|
||||
handleFlag(tokenizer, flag, pos_properties);
|
||||
found = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!found)
|
||||
{
|
||||
const auto& arg_property = user_args.name_associations[index];
|
||||
if (arg_property->a_disable_help)
|
||||
help_disabled = true;
|
||||
loaded_args.data[arg_property->a_dest] = tokenizer.get();
|
||||
loaded_args.found_args.insert(arg_property->a_dest);
|
||||
}
|
||||
}
|
||||
tokenizer.advance();
|
||||
}
|
||||
|
@ -282,6 +300,11 @@ namespace blt
|
|||
|
||||
const auto* const properties = user_args.flag_associations.at(flag);
|
||||
|
||||
handleFlag(tokenizer, flag, properties);
|
||||
}
|
||||
|
||||
void arg_parse::handleFlag(arg_tokenizer& tokenizer, const std::string& flag, const arg_properties_t* properties)
|
||||
{
|
||||
if (properties->a_dest.empty())
|
||||
{
|
||||
loaded_args.unrecognized_args.push_back(flag);
|
||||
|
@ -291,6 +314,9 @@ namespace blt
|
|||
|
||||
loaded_args.found_args.insert(dest);
|
||||
|
||||
if (properties->a_disable_help)
|
||||
help_disabled = true;
|
||||
|
||||
switch (properties->a_action)
|
||||
{
|
||||
case arg_action_t::HELP:
|
||||
|
@ -438,6 +464,8 @@ namespace blt
|
|||
|
||||
void arg_parse::printHelp() const
|
||||
{
|
||||
if (help_disabled)
|
||||
return;
|
||||
if (!user_args.prefix.empty())
|
||||
{
|
||||
std::cout << "\n";
|
||||
|
@ -452,7 +480,8 @@ namespace blt
|
|||
{
|
||||
if (!arg->a_flags.isFlag())
|
||||
max_length = std::max(arg->a_flags.name.size(), max_length);
|
||||
else {
|
||||
else
|
||||
{
|
||||
auto tmp = getFlagHelp(arg);
|
||||
max_length = std::max(tmp.size(), max_length);
|
||||
}
|
||||
|
@ -495,7 +524,9 @@ namespace blt
|
|||
|
||||
void arg_parse::printUsage() const
|
||||
{
|
||||
std::string usage = "Usage: " + filename(loaded_args.program_name) + " ";
|
||||
if (help_disabled)
|
||||
return;
|
||||
std::string usage = "Usage: " + filename(loaded_args.program_name) + " " + help_inclusion + " ";
|
||||
std::cout << usage;
|
||||
size_t current_line_length = 0;
|
||||
|
||||
|
|
Loading…
Reference in New Issue