mirror of
https://github.com/nlohmann/json.git
synced 2024-12-13 04:36:45 +08:00
1312 lines
25 KiB
C++
1312 lines
25 KiB
C++
|
/*
|
|||
|
|
|||
|
Copyright (c) 2014 Jarryd Beck
|
|||
|
|
|||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|||
|
of this software and associated documentation files (the "Software"), to deal
|
|||
|
in the Software without restriction, including without limitation the rights
|
|||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|||
|
copies of the Software, and to permit persons to whom the Software is
|
|||
|
furnished to do so, subject to the following conditions:
|
|||
|
|
|||
|
The above copyright notice and this permission notice shall be included in
|
|||
|
all copies or substantial portions of the Software.
|
|||
|
|
|||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|||
|
THE SOFTWARE.
|
|||
|
|
|||
|
*/
|
|||
|
|
|||
|
#ifndef CXX_OPTS_HPP
|
|||
|
#define CXX_OPTS_HPP
|
|||
|
|
|||
|
#if defined(__GNUC__)
|
|||
|
#pragma GCC diagnostic push
|
|||
|
#pragma GCC diagnostic ignored "-Wnon-virtual-dtor"
|
|||
|
#endif
|
|||
|
|
|||
|
#include <exception>
|
|||
|
#include <iostream>
|
|||
|
#include <map>
|
|||
|
#include <memory>
|
|||
|
#include <regex>
|
|||
|
#include <sstream>
|
|||
|
#include <string>
|
|||
|
#include <vector>
|
|||
|
|
|||
|
//when we ask cxxopts to use Unicode, help strings are processed using ICU,
|
|||
|
//which results in the correct lengths being computed for strings when they
|
|||
|
//are formatted for the help output
|
|||
|
//it is necessary to make sure that <unicode/unistr.h> can be found by the
|
|||
|
//compiler, and that icu-uc is linked in to the binary.
|
|||
|
|
|||
|
#ifdef CXXOPTS_USE_UNICODE
|
|||
|
#include <unicode/unistr.h>
|
|||
|
|
|||
|
namespace cxxopts
|
|||
|
{
|
|||
|
typedef icu::UnicodeString String;
|
|||
|
|
|||
|
inline
|
|||
|
String
|
|||
|
toLocalString(std::string s)
|
|||
|
{
|
|||
|
return icu::UnicodeString::fromUTF8(s);
|
|||
|
}
|
|||
|
|
|||
|
class UnicodeStringIterator : public
|
|||
|
std::iterator<std::forward_iterator_tag, int32_t>
|
|||
|
{
|
|||
|
public:
|
|||
|
|
|||
|
UnicodeStringIterator(const icu::UnicodeString* s, int32_t pos)
|
|||
|
: s(s)
|
|||
|
, i(pos)
|
|||
|
{
|
|||
|
}
|
|||
|
|
|||
|
value_type
|
|||
|
operator*() const
|
|||
|
{
|
|||
|
return s->char32At(i);
|
|||
|
}
|
|||
|
|
|||
|
bool
|
|||
|
operator==(const UnicodeStringIterator& rhs) const
|
|||
|
{
|
|||
|
return s == rhs.s && i == rhs.i;
|
|||
|
}
|
|||
|
|
|||
|
bool
|
|||
|
operator!=(const UnicodeStringIterator& rhs) const
|
|||
|
{
|
|||
|
return !(*this == rhs);
|
|||
|
}
|
|||
|
|
|||
|
UnicodeStringIterator&
|
|||
|
operator++()
|
|||
|
{
|
|||
|
++i;
|
|||
|
return *this;
|
|||
|
}
|
|||
|
|
|||
|
UnicodeStringIterator
|
|||
|
operator+(int32_t v)
|
|||
|
{
|
|||
|
return UnicodeStringIterator(s, i + v);
|
|||
|
}
|
|||
|
|
|||
|
private:
|
|||
|
const icu::UnicodeString* s;
|
|||
|
int32_t i;
|
|||
|
};
|
|||
|
|
|||
|
inline
|
|||
|
String&
|
|||
|
stringAppend(String&s, String a)
|
|||
|
{
|
|||
|
return s.append(std::move(a));
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
String&
|
|||
|
stringAppend(String& s, int n, UChar32 c)
|
|||
|
{
|
|||
|
for (int i = 0; i != n; ++i)
|
|||
|
{
|
|||
|
s.append(c);
|
|||
|
}
|
|||
|
|
|||
|
return s;
|
|||
|
}
|
|||
|
|
|||
|
template <typename Iterator>
|
|||
|
String&
|
|||
|
stringAppend(String& s, Iterator begin, Iterator end)
|
|||
|
{
|
|||
|
while (begin != end)
|
|||
|
{
|
|||
|
s.append(*begin);
|
|||
|
++begin;
|
|||
|
}
|
|||
|
|
|||
|
return s;
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
size_t
|
|||
|
stringLength(const String& s)
|
|||
|
{
|
|||
|
return s.length();
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
std::string
|
|||
|
toUTF8String(const String& s)
|
|||
|
{
|
|||
|
std::string result;
|
|||
|
s.toUTF8String(result);
|
|||
|
|
|||
|
return result;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
namespace std
|
|||
|
{
|
|||
|
cxxopts::UnicodeStringIterator
|
|||
|
begin(const icu::UnicodeString& s)
|
|||
|
{
|
|||
|
return cxxopts::UnicodeStringIterator(&s, 0);
|
|||
|
}
|
|||
|
|
|||
|
cxxopts::UnicodeStringIterator
|
|||
|
end(const icu::UnicodeString& s)
|
|||
|
{
|
|||
|
return cxxopts::UnicodeStringIterator(&s, s.length());
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
//ifdef CXXOPTS_USE_UNICODE
|
|||
|
#else
|
|||
|
|
|||
|
namespace cxxopts
|
|||
|
{
|
|||
|
typedef std::string String;
|
|||
|
|
|||
|
template <typename T>
|
|||
|
T
|
|||
|
toLocalString(T&& t)
|
|||
|
{
|
|||
|
return t;
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
size_t
|
|||
|
stringLength(const String& s)
|
|||
|
{
|
|||
|
return s.length();
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
String&
|
|||
|
stringAppend(String&s, String a)
|
|||
|
{
|
|||
|
return s.append(std::move(a));
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
String&
|
|||
|
stringAppend(String& s, size_t n, char c)
|
|||
|
{
|
|||
|
return s.append(n, c);
|
|||
|
}
|
|||
|
|
|||
|
template <typename Iterator>
|
|||
|
String&
|
|||
|
stringAppend(String& s, Iterator begin, Iterator end)
|
|||
|
{
|
|||
|
return s.append(begin, end);
|
|||
|
}
|
|||
|
|
|||
|
template <typename T>
|
|||
|
std::string
|
|||
|
toUTF8String(T&& t)
|
|||
|
{
|
|||
|
return std::forward<T>(t);
|
|||
|
}
|
|||
|
|
|||
|
}
|
|||
|
|
|||
|
//ifdef CXXOPTS_USE_UNICODE
|
|||
|
#endif
|
|||
|
|
|||
|
namespace cxxopts
|
|||
|
{
|
|||
|
class Value : public std::enable_shared_from_this<Value>
|
|||
|
{
|
|||
|
public:
|
|||
|
|
|||
|
virtual void
|
|||
|
parse(const std::string& text) const = 0;
|
|||
|
|
|||
|
virtual void
|
|||
|
parse() const = 0;
|
|||
|
|
|||
|
virtual bool
|
|||
|
has_arg() const = 0;
|
|||
|
|
|||
|
virtual bool
|
|||
|
has_default() const = 0;
|
|||
|
|
|||
|
virtual bool
|
|||
|
has_implicit() const = 0;
|
|||
|
|
|||
|
virtual std::string
|
|||
|
get_default_value() const = 0;
|
|||
|
|
|||
|
virtual std::string
|
|||
|
get_implicit_value() const = 0;
|
|||
|
|
|||
|
virtual std::shared_ptr<Value>
|
|||
|
default_value(const std::string& value) = 0;
|
|||
|
|
|||
|
virtual std::shared_ptr<Value>
|
|||
|
implicit_value(const std::string& value) = 0;
|
|||
|
};
|
|||
|
|
|||
|
class OptionException : public std::exception
|
|||
|
{
|
|||
|
public:
|
|||
|
OptionException(const std::string& message)
|
|||
|
: m_message(message)
|
|||
|
{
|
|||
|
}
|
|||
|
|
|||
|
virtual const char*
|
|||
|
what() const noexcept
|
|||
|
{
|
|||
|
return m_message.c_str();
|
|||
|
}
|
|||
|
|
|||
|
private:
|
|||
|
std::string m_message;
|
|||
|
};
|
|||
|
|
|||
|
class OptionSpecException : public OptionException
|
|||
|
{
|
|||
|
public:
|
|||
|
|
|||
|
OptionSpecException(const std::string& message)
|
|||
|
: OptionException(message)
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class OptionParseException : public OptionException
|
|||
|
{
|
|||
|
public:
|
|||
|
OptionParseException(const std::string& message)
|
|||
|
: OptionException(message)
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class option_exists_error : public OptionSpecException
|
|||
|
{
|
|||
|
public:
|
|||
|
option_exists_error(const std::string& option)
|
|||
|
: OptionSpecException(u8"Option ‘" + option + u8"’ already exists")
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class invalid_option_format_error : public OptionSpecException
|
|||
|
{
|
|||
|
public:
|
|||
|
invalid_option_format_error(const std::string& format)
|
|||
|
: OptionSpecException(u8"Invalid option format ‘" + format + u8"’")
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class option_not_exists_exception : public OptionParseException
|
|||
|
{
|
|||
|
public:
|
|||
|
option_not_exists_exception(const std::string& option)
|
|||
|
: OptionParseException(u8"Option ‘" + option + u8"’ does not exist")
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class missing_argument_exception : public OptionParseException
|
|||
|
{
|
|||
|
public:
|
|||
|
missing_argument_exception(const std::string& option)
|
|||
|
: OptionParseException(u8"Option ‘" + option + u8"’ is missing an argument")
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class option_requires_argument_exception : public OptionParseException
|
|||
|
{
|
|||
|
public:
|
|||
|
option_requires_argument_exception(const std::string& option)
|
|||
|
: OptionParseException(u8"Option ‘" + option + u8"’ requires an argument")
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class option_not_has_argument_exception : public OptionParseException
|
|||
|
{
|
|||
|
public:
|
|||
|
option_not_has_argument_exception
|
|||
|
(
|
|||
|
const std::string& option,
|
|||
|
const std::string& arg
|
|||
|
)
|
|||
|
: OptionParseException(
|
|||
|
u8"Option ‘" + option + u8"’ does not take an argument, but argument‘"
|
|||
|
+ arg + "’ given")
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class option_not_present_exception : public OptionParseException
|
|||
|
{
|
|||
|
public:
|
|||
|
option_not_present_exception(const std::string& option)
|
|||
|
: OptionParseException(u8"Option ‘" + option + u8"’ not present")
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
class argument_incorrect_type : public OptionParseException
|
|||
|
{
|
|||
|
public:
|
|||
|
argument_incorrect_type
|
|||
|
(
|
|||
|
const std::string& arg
|
|||
|
)
|
|||
|
: OptionParseException(
|
|||
|
u8"Argument ‘" + arg + u8"’ failed to parse"
|
|||
|
)
|
|||
|
{
|
|||
|
}
|
|||
|
};
|
|||
|
|
|||
|
namespace values
|
|||
|
{
|
|||
|
template <typename T>
|
|||
|
void
|
|||
|
parse_value(const std::string& text, T& value)
|
|||
|
{
|
|||
|
std::istringstream is(text);
|
|||
|
if (!(is >> value))
|
|||
|
{
|
|||
|
std::cerr << "cannot parse empty value" << std::endl;
|
|||
|
throw argument_incorrect_type(text);
|
|||
|
}
|
|||
|
|
|||
|
if (is.rdbuf()->in_avail() != 0)
|
|||
|
{
|
|||
|
throw argument_incorrect_type(text);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
template <typename T>
|
|||
|
void
|
|||
|
parse_value(const std::string& text, std::vector<T>& value)
|
|||
|
{
|
|||
|
T v;
|
|||
|
parse_value(text, v);
|
|||
|
value.push_back(v);
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
void
|
|||
|
parse_value(const std::string& /*text*/, bool& value)
|
|||
|
{
|
|||
|
//TODO recognise on, off, yes, no, enable, disable
|
|||
|
//so that we can write --long=yes explicitly
|
|||
|
value = true;
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
void
|
|||
|
parse_value(const std::string& text, std::string& value)
|
|||
|
{
|
|||
|
value = text;
|
|||
|
}
|
|||
|
|
|||
|
template <typename T>
|
|||
|
struct value_has_arg
|
|||
|
{
|
|||
|
static constexpr bool value = true;
|
|||
|
};
|
|||
|
|
|||
|
template <>
|
|||
|
struct value_has_arg<bool>
|
|||
|
{
|
|||
|
static constexpr bool value = false;
|
|||
|
};
|
|||
|
|
|||
|
template <typename T>
|
|||
|
class standard_value : public Value
|
|||
|
{
|
|||
|
public:
|
|||
|
standard_value()
|
|||
|
: m_result(std::make_shared<T>())
|
|||
|
, m_store(m_result.get())
|
|||
|
{
|
|||
|
}
|
|||
|
|
|||
|
standard_value(T* t)
|
|||
|
: m_store(t)
|
|||
|
{
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
parse(const std::string& text) const
|
|||
|
{
|
|||
|
if (m_implicit && text.empty())
|
|||
|
{
|
|||
|
parse_value(m_implicit_value, *m_store);
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
parse_value(text, *m_store);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
parse() const
|
|||
|
{
|
|||
|
parse_value(m_default_value, *m_store);
|
|||
|
}
|
|||
|
|
|||
|
bool
|
|||
|
has_arg() const
|
|||
|
{
|
|||
|
return value_has_arg<T>::value;
|
|||
|
}
|
|||
|
|
|||
|
bool
|
|||
|
has_default() const
|
|||
|
{
|
|||
|
return m_default;
|
|||
|
}
|
|||
|
|
|||
|
bool
|
|||
|
has_implicit() const
|
|||
|
{
|
|||
|
return m_implicit;
|
|||
|
}
|
|||
|
|
|||
|
virtual std::shared_ptr<Value>
|
|||
|
default_value(const std::string& value){
|
|||
|
m_default = true;
|
|||
|
m_default_value = value;
|
|||
|
return shared_from_this();
|
|||
|
}
|
|||
|
|
|||
|
virtual std::shared_ptr<Value>
|
|||
|
implicit_value(const std::string& value){
|
|||
|
m_implicit = true;
|
|||
|
m_implicit_value = value;
|
|||
|
return shared_from_this();
|
|||
|
}
|
|||
|
|
|||
|
std::string
|
|||
|
get_default_value() const
|
|||
|
{
|
|||
|
return m_default_value;
|
|||
|
}
|
|||
|
|
|||
|
std::string
|
|||
|
get_implicit_value() const
|
|||
|
{
|
|||
|
return m_implicit_value;
|
|||
|
}
|
|||
|
|
|||
|
const T&
|
|||
|
get() const
|
|||
|
{
|
|||
|
if (m_store == nullptr)
|
|||
|
{
|
|||
|
return *m_result;
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
return *m_store;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
protected:
|
|||
|
std::shared_ptr<T> m_result;
|
|||
|
T* m_store;
|
|||
|
bool m_default = false;
|
|||
|
std::string m_default_value;
|
|||
|
bool m_implicit = false;
|
|||
|
std::string m_implicit_value;
|
|||
|
};
|
|||
|
}
|
|||
|
|
|||
|
template <typename T>
|
|||
|
std::shared_ptr<Value>
|
|||
|
value()
|
|||
|
{
|
|||
|
return std::make_shared<values::standard_value<T>>();
|
|||
|
}
|
|||
|
|
|||
|
template <typename T>
|
|||
|
std::shared_ptr<Value>
|
|||
|
value(T& t)
|
|||
|
{
|
|||
|
return std::make_shared<values::standard_value<T>>(&t);
|
|||
|
}
|
|||
|
|
|||
|
class OptionAdder;
|
|||
|
|
|||
|
class OptionDetails
|
|||
|
{
|
|||
|
public:
|
|||
|
OptionDetails
|
|||
|
(
|
|||
|
const String& description,
|
|||
|
std::shared_ptr<const Value> value
|
|||
|
)
|
|||
|
: m_desc(description)
|
|||
|
, m_value(value)
|
|||
|
, m_count(0)
|
|||
|
{
|
|||
|
}
|
|||
|
|
|||
|
const String&
|
|||
|
description() const
|
|||
|
{
|
|||
|
return m_desc;
|
|||
|
}
|
|||
|
|
|||
|
bool
|
|||
|
has_arg() const
|
|||
|
{
|
|||
|
return m_value->has_arg();
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
parse(const std::string& text)
|
|||
|
{
|
|||
|
m_value->parse(text);
|
|||
|
++m_count;
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
parse_default()
|
|||
|
{
|
|||
|
m_value->parse();
|
|||
|
++m_count;
|
|||
|
}
|
|||
|
|
|||
|
int
|
|||
|
count() const
|
|||
|
{
|
|||
|
return m_count;
|
|||
|
}
|
|||
|
|
|||
|
const Value& value() const {
|
|||
|
return *m_value;
|
|||
|
}
|
|||
|
|
|||
|
template <typename T>
|
|||
|
const T&
|
|||
|
as() const
|
|||
|
{
|
|||
|
#ifdef CXXOPTS_NO_RTTI
|
|||
|
return static_cast<const values::standard_value<T>&>(*m_value).get();
|
|||
|
#else
|
|||
|
return dynamic_cast<const values::standard_value<T>&>(*m_value).get();
|
|||
|
#endif
|
|||
|
}
|
|||
|
|
|||
|
private:
|
|||
|
String m_desc;
|
|||
|
std::shared_ptr<const Value> m_value;
|
|||
|
int m_count;
|
|||
|
};
|
|||
|
|
|||
|
struct HelpOptionDetails
|
|||
|
{
|
|||
|
std::string s;
|
|||
|
std::string l;
|
|||
|
String desc;
|
|||
|
bool has_arg;
|
|||
|
bool has_default;
|
|||
|
std::string default_value;
|
|||
|
bool has_implicit;
|
|||
|
std::string implicit_value;
|
|||
|
std::string arg_help;
|
|||
|
};
|
|||
|
|
|||
|
struct HelpGroupDetails
|
|||
|
{
|
|||
|
std::string name;
|
|||
|
std::string description;
|
|||
|
std::vector<HelpOptionDetails> options;
|
|||
|
};
|
|||
|
|
|||
|
class Options
|
|||
|
{
|
|||
|
public:
|
|||
|
|
|||
|
Options(std::string program, std::string help_string = "")
|
|||
|
: m_program(std::move(program))
|
|||
|
, m_help_string(toLocalString(std::move(help_string)))
|
|||
|
{
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
void
|
|||
|
parse(int& argc, char**& argv);
|
|||
|
|
|||
|
inline
|
|||
|
OptionAdder
|
|||
|
add_options(std::string group = "");
|
|||
|
|
|||
|
inline
|
|||
|
void
|
|||
|
add_option
|
|||
|
(
|
|||
|
const std::string& group,
|
|||
|
const std::string& s,
|
|||
|
const std::string& l,
|
|||
|
std::string desc,
|
|||
|
std::shared_ptr<const Value> value,
|
|||
|
std::string arg_help
|
|||
|
);
|
|||
|
|
|||
|
int
|
|||
|
count(const std::string& o) const
|
|||
|
{
|
|||
|
auto iter = m_options.find(o);
|
|||
|
if (iter == m_options.end())
|
|||
|
{
|
|||
|
return 0;
|
|||
|
}
|
|||
|
|
|||
|
return iter->second->count();
|
|||
|
}
|
|||
|
|
|||
|
const OptionDetails&
|
|||
|
operator[](const std::string& option) const
|
|||
|
{
|
|||
|
auto iter = m_options.find(option);
|
|||
|
|
|||
|
if (iter == m_options.end())
|
|||
|
{
|
|||
|
throw option_not_present_exception(option);
|
|||
|
}
|
|||
|
|
|||
|
return *iter->second;
|
|||
|
}
|
|||
|
|
|||
|
//parse positional arguments into the given option
|
|||
|
inline
|
|||
|
void
|
|||
|
parse_positional(std::string option);
|
|||
|
|
|||
|
inline
|
|||
|
std::string
|
|||
|
help(const std::vector<std::string>& groups = {""}) const;
|
|||
|
|
|||
|
inline
|
|||
|
const std::vector<std::string>
|
|||
|
groups() const;
|
|||
|
|
|||
|
inline
|
|||
|
const HelpGroupDetails&
|
|||
|
group_help(const std::string& group) const;
|
|||
|
|
|||
|
private:
|
|||
|
|
|||
|
inline
|
|||
|
void
|
|||
|
add_one_option
|
|||
|
(
|
|||
|
const std::string& option,
|
|||
|
std::shared_ptr<OptionDetails> details
|
|||
|
);
|
|||
|
|
|||
|
inline
|
|||
|
bool
|
|||
|
consume_positional(std::string a);
|
|||
|
|
|||
|
inline
|
|||
|
void
|
|||
|
add_to_option(const std::string& option, const std::string& arg);
|
|||
|
|
|||
|
inline
|
|||
|
void
|
|||
|
parse_option
|
|||
|
(
|
|||
|
std::shared_ptr<OptionDetails> value,
|
|||
|
const std::string& name,
|
|||
|
const std::string& arg = ""
|
|||
|
);
|
|||
|
|
|||
|
inline
|
|||
|
void
|
|||
|
checked_parse_arg
|
|||
|
(
|
|||
|
int argc,
|
|||
|
char* argv[],
|
|||
|
int& current,
|
|||
|
std::shared_ptr<OptionDetails> value,
|
|||
|
const std::string& name
|
|||
|
);
|
|||
|
|
|||
|
inline
|
|||
|
String
|
|||
|
help_one_group(const std::string& group) const;
|
|||
|
|
|||
|
std::string m_program;
|
|||
|
String m_help_string;
|
|||
|
|
|||
|
std::map<std::string, std::shared_ptr<OptionDetails>> m_options;
|
|||
|
std::string m_positional;
|
|||
|
|
|||
|
//mapping from groups to help options
|
|||
|
std::map<std::string, HelpGroupDetails> m_help;
|
|||
|
};
|
|||
|
|
|||
|
class OptionAdder
|
|||
|
{
|
|||
|
public:
|
|||
|
|
|||
|
OptionAdder(Options& options, std::string group)
|
|||
|
: m_options(options), m_group(std::move(group))
|
|||
|
{
|
|||
|
}
|
|||
|
|
|||
|
inline
|
|||
|
OptionAdder&
|
|||
|
operator()
|
|||
|
(
|
|||
|
const std::string& opts,
|
|||
|
const std::string& desc,
|
|||
|
std::shared_ptr<const Value> value
|
|||
|
= ::cxxopts::value<bool>(),
|
|||
|
std::string arg_help = ""
|
|||
|
);
|
|||
|
|
|||
|
private:
|
|||
|
Options& m_options;
|
|||
|
std::string m_group;
|
|||
|
};
|
|||
|
|
|||
|
}
|
|||
|
|
|||
|
namespace cxxopts
|
|||
|
{
|
|||
|
|
|||
|
namespace
|
|||
|
{
|
|||
|
|
|||
|
constexpr int OPTION_LONGEST = 30;
|
|||
|
constexpr int OPTION_DESC_GAP = 2;
|
|||
|
|
|||
|
std::basic_regex<char> option_matcher
|
|||
|
("--([[:alnum:]][-_[:alnum:]]+)(=(.*))?|-([a-zA-Z]+)");
|
|||
|
|
|||
|
std::basic_regex<char> option_specifier
|
|||
|
("(([a-zA-Z]),)?([a-zA-Z0-9][-_a-zA-Z0-9]+)");
|
|||
|
|
|||
|
String
|
|||
|
format_option
|
|||
|
(
|
|||
|
const HelpOptionDetails& o
|
|||
|
)
|
|||
|
{
|
|||
|
auto& s = o.s;
|
|||
|
auto& l = o.l;
|
|||
|
|
|||
|
String result = " ";
|
|||
|
|
|||
|
if (s.size() > 0)
|
|||
|
{
|
|||
|
result += "-" + toLocalString(s) + ",";
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
result += " ";
|
|||
|
}
|
|||
|
|
|||
|
if (l.size() > 0)
|
|||
|
{
|
|||
|
result += " --" + toLocalString(l);
|
|||
|
}
|
|||
|
|
|||
|
if (o.has_arg)
|
|||
|
{
|
|||
|
auto arg = o.arg_help.size() > 0 ? toLocalString(o.arg_help) : "arg";
|
|||
|
|
|||
|
if (o.has_implicit)
|
|||
|
{
|
|||
|
result += " [=" + arg + "(=" + toLocalString(o.implicit_value) + ")]";
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
result += " " + arg;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
return result;
|
|||
|
}
|
|||
|
|
|||
|
String
|
|||
|
format_description
|
|||
|
(
|
|||
|
const HelpOptionDetails& o,
|
|||
|
size_t start,
|
|||
|
size_t width
|
|||
|
)
|
|||
|
{
|
|||
|
auto desc = o.desc;
|
|||
|
|
|||
|
if (o.has_default)
|
|||
|
{
|
|||
|
desc += toLocalString(" (default:" + o.default_value + ")");
|
|||
|
}
|
|||
|
|
|||
|
String result;
|
|||
|
|
|||
|
auto current = std::begin(desc);
|
|||
|
auto startLine = current;
|
|||
|
auto lastSpace = current;
|
|||
|
|
|||
|
auto size = size_t{};
|
|||
|
|
|||
|
while (current != std::end(desc))
|
|||
|
{
|
|||
|
if (*current == ' ')
|
|||
|
{
|
|||
|
lastSpace = current;
|
|||
|
}
|
|||
|
|
|||
|
if (size > width)
|
|||
|
{
|
|||
|
if (lastSpace == startLine)
|
|||
|
{
|
|||
|
stringAppend(result, startLine, current + 1);
|
|||
|
stringAppend(result, "\n");
|
|||
|
stringAppend(result, start, ' ');
|
|||
|
startLine = current + 1;
|
|||
|
lastSpace = startLine;
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
stringAppend(result, startLine, lastSpace);
|
|||
|
stringAppend(result, "\n");
|
|||
|
stringAppend(result, start, ' ');
|
|||
|
startLine = lastSpace + 1;
|
|||
|
}
|
|||
|
size = 0;
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
++size;
|
|||
|
}
|
|||
|
|
|||
|
++current;
|
|||
|
}
|
|||
|
|
|||
|
//append whatever is left
|
|||
|
stringAppend(result, startLine, current);
|
|||
|
|
|||
|
return result;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
OptionAdder
|
|||
|
Options::add_options(std::string group)
|
|||
|
{
|
|||
|
return OptionAdder(*this, std::move(group));
|
|||
|
}
|
|||
|
|
|||
|
OptionAdder&
|
|||
|
OptionAdder::operator()
|
|||
|
(
|
|||
|
const std::string& opts,
|
|||
|
const std::string& desc,
|
|||
|
std::shared_ptr<const Value> value,
|
|||
|
std::string arg_help
|
|||
|
)
|
|||
|
{
|
|||
|
std::match_results<const char*> result;
|
|||
|
std::regex_match(opts.c_str(), result, option_specifier);
|
|||
|
|
|||
|
if (result.empty())
|
|||
|
{
|
|||
|
throw invalid_option_format_error(opts);
|
|||
|
}
|
|||
|
|
|||
|
const auto& s = result[2];
|
|||
|
const auto& l = result[3];
|
|||
|
|
|||
|
m_options.add_option(m_group, s.str(), l.str(), desc, value,
|
|||
|
std::move(arg_help));
|
|||
|
|
|||
|
return *this;
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
Options::parse_option
|
|||
|
(
|
|||
|
std::shared_ptr<OptionDetails> value,
|
|||
|
const std::string& /*name*/,
|
|||
|
const std::string& arg
|
|||
|
)
|
|||
|
{
|
|||
|
value->parse(arg);
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
Options::checked_parse_arg
|
|||
|
(
|
|||
|
int argc,
|
|||
|
char* argv[],
|
|||
|
int& current,
|
|||
|
std::shared_ptr<OptionDetails> value,
|
|||
|
const std::string& name
|
|||
|
)
|
|||
|
{
|
|||
|
if (current + 1 >= argc)
|
|||
|
{
|
|||
|
if (value->value().has_implicit())
|
|||
|
{
|
|||
|
parse_option(value, name, "");
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
throw missing_argument_exception(name);
|
|||
|
}
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
if (argv[current + 1][0] == '-' && value->value().has_implicit())
|
|||
|
{
|
|||
|
parse_option(value, name, "");
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
parse_option(value, name, argv[current + 1]);
|
|||
|
++current;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
Options::add_to_option(const std::string& option, const std::string& arg)
|
|||
|
{
|
|||
|
auto iter = m_options.find(option);
|
|||
|
|
|||
|
if (iter == m_options.end())
|
|||
|
{
|
|||
|
throw option_not_exists_exception(option);
|
|||
|
}
|
|||
|
|
|||
|
parse_option(iter->second, option, arg);
|
|||
|
}
|
|||
|
|
|||
|
bool
|
|||
|
Options::consume_positional(std::string a)
|
|||
|
{
|
|||
|
if (m_positional.size() > 0)
|
|||
|
{
|
|||
|
add_to_option(m_positional, a);
|
|||
|
return true;
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
return false;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
Options::parse_positional(std::string option)
|
|||
|
{
|
|||
|
m_positional = std::move(option);
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
Options::parse(int& argc, char**& argv)
|
|||
|
{
|
|||
|
int current = 1;
|
|||
|
|
|||
|
int nextKeep = 1;
|
|||
|
|
|||
|
while (current != argc)
|
|||
|
{
|
|||
|
std::match_results<const char*> result;
|
|||
|
std::regex_match(argv[current], result, option_matcher);
|
|||
|
|
|||
|
if (result.empty())
|
|||
|
{
|
|||
|
//not a flag
|
|||
|
|
|||
|
//if true is returned here then it was consumed, otherwise it is
|
|||
|
//ignored
|
|||
|
if (consume_positional(argv[current]))
|
|||
|
{
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
argv[nextKeep] = argv[current];
|
|||
|
++nextKeep;
|
|||
|
}
|
|||
|
//if we return from here then it was parsed successfully, so continue
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
//short or long option?
|
|||
|
if (result[4].length() != 0)
|
|||
|
{
|
|||
|
const std::string& s = result[4];
|
|||
|
|
|||
|
for (std::size_t i = 0; i != s.size(); ++i)
|
|||
|
{
|
|||
|
std::string name(1, s[i]);
|
|||
|
auto iter = m_options.find(name);
|
|||
|
|
|||
|
if (iter == m_options.end())
|
|||
|
{
|
|||
|
throw option_not_exists_exception(name);
|
|||
|
}
|
|||
|
|
|||
|
auto value = iter->second;
|
|||
|
|
|||
|
//if no argument then just add it
|
|||
|
if (!value->has_arg())
|
|||
|
{
|
|||
|
parse_option(value, name);
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
//it must be the last argument
|
|||
|
if (i + 1 == s.size())
|
|||
|
{
|
|||
|
checked_parse_arg(argc, argv, current, value, name);
|
|||
|
}
|
|||
|
else if (value->value().has_implicit())
|
|||
|
{
|
|||
|
parse_option(value, name, "");
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
//error
|
|||
|
throw option_requires_argument_exception(name);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
else if (result[1].length() != 0)
|
|||
|
{
|
|||
|
const std::string& name = result[1];
|
|||
|
|
|||
|
auto iter = m_options.find(name);
|
|||
|
|
|||
|
if (iter == m_options.end())
|
|||
|
{
|
|||
|
throw option_not_exists_exception(name);
|
|||
|
}
|
|||
|
|
|||
|
auto opt = iter->second;
|
|||
|
|
|||
|
//equals provided for long option?
|
|||
|
if (result[3].length() != 0)
|
|||
|
{
|
|||
|
//parse the option given
|
|||
|
|
|||
|
//but if it doesn't take an argument, this is an error
|
|||
|
if (!opt->has_arg())
|
|||
|
{
|
|||
|
throw option_not_has_argument_exception(name, result[3]);
|
|||
|
}
|
|||
|
|
|||
|
parse_option(opt, name, result[3]);
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
if (opt->has_arg())
|
|||
|
{
|
|||
|
//parse the next argument
|
|||
|
checked_parse_arg(argc, argv, current, opt, name);
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
//parse with empty argument
|
|||
|
parse_option(opt, name);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
}
|
|||
|
|
|||
|
++current;
|
|||
|
}
|
|||
|
|
|||
|
for (auto& opt : m_options)
|
|||
|
{
|
|||
|
auto& detail = opt.second;
|
|||
|
auto& value = detail->value();
|
|||
|
|
|||
|
if(!detail->count() && value.has_default()){
|
|||
|
detail->parse_default();
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
argc = nextKeep;
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
Options::add_option
|
|||
|
(
|
|||
|
const std::string& group,
|
|||
|
const std::string& s,
|
|||
|
const std::string& l,
|
|||
|
std::string desc,
|
|||
|
std::shared_ptr<const Value> value,
|
|||
|
std::string arg_help
|
|||
|
)
|
|||
|
{
|
|||
|
auto stringDesc = toLocalString(std::move(desc));
|
|||
|
auto option = std::make_shared<OptionDetails>(stringDesc, value);
|
|||
|
|
|||
|
if (s.size() > 0)
|
|||
|
{
|
|||
|
add_one_option(s, option);
|
|||
|
}
|
|||
|
|
|||
|
if (l.size() > 0)
|
|||
|
{
|
|||
|
add_one_option(l, option);
|
|||
|
}
|
|||
|
|
|||
|
//add the help details
|
|||
|
auto& options = m_help[group];
|
|||
|
|
|||
|
options.options.emplace_back(HelpOptionDetails{s, l, stringDesc,
|
|||
|
value->has_arg(),
|
|||
|
value->has_default(), value->get_default_value(),
|
|||
|
value->has_implicit(), value->get_implicit_value(),
|
|||
|
std::move(arg_help)});
|
|||
|
}
|
|||
|
|
|||
|
void
|
|||
|
Options::add_one_option
|
|||
|
(
|
|||
|
const std::string& option,
|
|||
|
std::shared_ptr<OptionDetails> details
|
|||
|
)
|
|||
|
{
|
|||
|
auto in = m_options.emplace(option, details);
|
|||
|
|
|||
|
if (!in.second)
|
|||
|
{
|
|||
|
throw option_exists_error(option);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
String
|
|||
|
Options::help_one_group(const std::string& g) const
|
|||
|
{
|
|||
|
typedef std::vector<std::pair<String, String>> OptionHelp;
|
|||
|
|
|||
|
auto group = m_help.find(g);
|
|||
|
if (group == m_help.end())
|
|||
|
{
|
|||
|
return "";
|
|||
|
}
|
|||
|
|
|||
|
OptionHelp format;
|
|||
|
|
|||
|
size_t longest = 0;
|
|||
|
|
|||
|
String result;
|
|||
|
|
|||
|
if (!g.empty())
|
|||
|
{
|
|||
|
result += toLocalString(" " + g + " options:\n\n");
|
|||
|
}
|
|||
|
|
|||
|
for (const auto& o : group->second.options)
|
|||
|
{
|
|||
|
auto s = format_option(o);
|
|||
|
longest = std::max(longest, stringLength(s));
|
|||
|
format.push_back(std::make_pair(s, String()));
|
|||
|
}
|
|||
|
|
|||
|
longest = std::min(longest, static_cast<size_t>(OPTION_LONGEST));
|
|||
|
|
|||
|
//widest allowed description
|
|||
|
auto allowed = size_t{76} - longest - OPTION_DESC_GAP;
|
|||
|
|
|||
|
auto fiter = format.begin();
|
|||
|
for (const auto& o : group->second.options)
|
|||
|
{
|
|||
|
auto d = format_description(o, longest + OPTION_DESC_GAP, allowed);
|
|||
|
|
|||
|
result += fiter->first;
|
|||
|
if (stringLength(fiter->first) > longest)
|
|||
|
{
|
|||
|
result += "\n";
|
|||
|
result += toLocalString(std::string(longest + OPTION_DESC_GAP, ' '));
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
result += toLocalString(std::string(longest + OPTION_DESC_GAP -
|
|||
|
stringLength(fiter->first),
|
|||
|
' '));
|
|||
|
}
|
|||
|
result += d;
|
|||
|
result += "\n";
|
|||
|
|
|||
|
++fiter;
|
|||
|
}
|
|||
|
|
|||
|
return result;
|
|||
|
}
|
|||
|
|
|||
|
std::string
|
|||
|
Options::help(const std::vector<std::string>& groups) const
|
|||
|
{
|
|||
|
String result = "Usage:\n " + toLocalString(m_program) + " [OPTION...]"
|
|||
|
+ m_help_string + "\n\n";
|
|||
|
|
|||
|
for (std::size_t i = 0; i < groups.size(); ++i)
|
|||
|
{
|
|||
|
result += help_one_group(groups[i]);
|
|||
|
if (i < groups.size() - 1)
|
|||
|
{
|
|||
|
result += "\n";
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
return toUTF8String(result);
|
|||
|
}
|
|||
|
|
|||
|
const std::vector<std::string>
|
|||
|
Options::groups() const
|
|||
|
{
|
|||
|
std::vector<std::string> g;
|
|||
|
|
|||
|
std::transform(
|
|||
|
m_help.begin(),
|
|||
|
m_help.end(),
|
|||
|
std::back_inserter(g),
|
|||
|
[] (const std::map<std::string, HelpGroupDetails>::value_type& pair)
|
|||
|
{
|
|||
|
return pair.first;
|
|||
|
}
|
|||
|
);
|
|||
|
|
|||
|
return g;
|
|||
|
}
|
|||
|
|
|||
|
const HelpGroupDetails&
|
|||
|
Options::group_help(const std::string& group) const
|
|||
|
{
|
|||
|
return m_help.at(group);
|
|||
|
}
|
|||
|
|
|||
|
}
|
|||
|
|
|||
|
#if defined(__GNU__)
|
|||
|
#pragma GCC diagnostic pop
|
|||
|
#endif
|
|||
|
|
|||
|
#endif //CXX_OPTS_HPP
|