Danny Robson
0e3fa05f05
ipp files weren't a great way of keeping things clean, and IDEs have a little trouble dealing with the split configuration. this simplifies debugging a great deal.
337 lines
9.3 KiB
C++
337 lines
9.3 KiB
C++
/*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
* Copyright 2013-2016 Danny Robson <danny@nerdcruft.net>
|
|
*/
|
|
|
|
|
|
#ifndef CRUFT_UTIL_CMDLINE_HPP
|
|
#define CRUFT_UTIL_CMDLINE_HPP
|
|
|
|
#include "introspection.hpp"
|
|
#include "iterator.hpp"
|
|
|
|
#include <exception>
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <tuple>
|
|
#include <vector>
|
|
#include <sstream>
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
namespace util::cmdopt {
|
|
///////////////////////////////////////////////////////////////////////////
|
|
class error : public std::exception { };
|
|
|
|
//-------------------------------------------------------------------------
|
|
class invalid_key : public error {
|
|
public:
|
|
explicit invalid_key (std::string _key);
|
|
const char* what (void) const noexcept override;
|
|
|
|
private:
|
|
const std::string m_key;
|
|
};
|
|
|
|
//-------------------------------------------------------------------------
|
|
class invalid_value : public error {
|
|
public:
|
|
explicit invalid_value (std::string _value);
|
|
const char* what (void) const noexcept override;
|
|
|
|
private:
|
|
const std::string m_value;
|
|
};
|
|
|
|
//-------------------------------------------------------------------------
|
|
class invalid_null : public error {
|
|
public:
|
|
const char* what (void) const noexcept override;
|
|
};
|
|
|
|
//-------------------------------------------------------------------------
|
|
class invalid_required : public error {
|
|
public:
|
|
const char* what (void) const noexcept override;
|
|
};
|
|
|
|
//-------------------------------------------------------------------------
|
|
class unhandled_argument : public error {
|
|
public:
|
|
explicit unhandled_argument (int index);
|
|
const char* what (void) const noexcept override;
|
|
|
|
int index (void) const noexcept;
|
|
|
|
private:
|
|
const int m_index;
|
|
};
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
namespace option {
|
|
class base {
|
|
public:
|
|
// we deal almost exclusively with vtables, so disable copying
|
|
// just in case we do something stupid.
|
|
base () = default;
|
|
base (const base&) = delete;
|
|
base& operator= (const base&) = delete;
|
|
|
|
virtual ~base ();
|
|
|
|
virtual void execute (void);
|
|
virtual void execute (const char *restrict);
|
|
virtual void start (void);
|
|
virtual void finish (void);
|
|
|
|
virtual const std::string& example (void) const = 0;
|
|
|
|
bool required (void) const;
|
|
bool required (bool);
|
|
|
|
bool seen (void) const;
|
|
bool seen (bool);
|
|
|
|
private:
|
|
bool m_required = false;
|
|
bool m_seen = false;
|
|
};
|
|
|
|
|
|
class null : public base {
|
|
public:
|
|
virtual void execute (void) override;
|
|
virtual void execute (const char *restrict) override;
|
|
|
|
virtual const std::string& example (void) const override;
|
|
};
|
|
|
|
|
|
class present : public base {
|
|
public:
|
|
explicit present (bool&);
|
|
explicit present (bool&&) = delete;
|
|
|
|
using base::execute;
|
|
virtual void execute (void) override;
|
|
|
|
virtual const std::string& example (void) const override;
|
|
|
|
virtual void finish (void) override;
|
|
|
|
private:
|
|
bool &m_data;
|
|
};
|
|
|
|
|
|
namespace detail {
|
|
template <typename T>
|
|
std::enable_if_t<!std::is_enum<T>::value, const std::string&>
|
|
value_example (void)
|
|
{
|
|
static const std::string EXAMPLE =
|
|
std::string {"<"} +
|
|
std::string {to_string<T> ()} +
|
|
std::string {">"};
|
|
|
|
return EXAMPLE;
|
|
}
|
|
|
|
template <typename T>
|
|
std::enable_if_t<std::is_enum<T>::value, const std::string&>
|
|
value_example (void)
|
|
{
|
|
static const std::string EXAMPLE = [] (void) {
|
|
std::ostringstream os;
|
|
std::copy (std::cbegin (enum_traits<T>::names),
|
|
std::cend (enum_traits<T>::names),
|
|
infix_iterator<const char*> (os, "|"));
|
|
return os.str ();
|
|
} ();
|
|
return EXAMPLE;
|
|
}
|
|
}
|
|
|
|
|
|
template <typename T>
|
|
class value : public base {
|
|
public:
|
|
explicit value (T &_data): m_data (_data) { }
|
|
explicit value (T&&) = delete;
|
|
|
|
using base::execute;
|
|
|
|
void execute (const char *restrict str) override
|
|
{
|
|
try {
|
|
std::istringstream is (str);
|
|
is.exceptions (
|
|
std::istringstream::failbit
|
|
| std::istringstream::badbit
|
|
);
|
|
|
|
is >> m_data;
|
|
} catch (...) {
|
|
throw invalid_value (__func__);
|
|
}
|
|
|
|
seen (true);
|
|
}
|
|
|
|
const std::string& example (void) const override
|
|
{
|
|
return detail::value_example<T> ();
|
|
}
|
|
|
|
const T& data (void) const&
|
|
{
|
|
return m_data;
|
|
}
|
|
|
|
|
|
T& data (void) &
|
|
{
|
|
return m_data;
|
|
}
|
|
|
|
|
|
T& data (T _data) &
|
|
{
|
|
return m_data = _data;
|
|
}
|
|
|
|
private:
|
|
T& m_data;
|
|
};
|
|
|
|
template <>
|
|
inline void
|
|
value<bool>::execute (const char *restrict str)
|
|
{
|
|
static const std::string TRUE_STRING[] = {
|
|
"true",
|
|
"yes",
|
|
"y",
|
|
"1"
|
|
};
|
|
|
|
if (std::any_of (std::begin (TRUE_STRING),
|
|
std::end (TRUE_STRING),
|
|
[str] (auto i) { return i == str; }))
|
|
{
|
|
m_data = true;
|
|
return;
|
|
}
|
|
|
|
static const std::string FALSE_STRING[] = {
|
|
"false",
|
|
"no",
|
|
"n",
|
|
"0"
|
|
};
|
|
|
|
if (std::any_of (std::begin (FALSE_STRING),
|
|
std::end (FALSE_STRING),
|
|
[str] (auto i) { return i == str; }))
|
|
{
|
|
m_data = false;
|
|
return;
|
|
}
|
|
|
|
base::execute (str);
|
|
seen (true);
|
|
}
|
|
|
|
|
|
|
|
template <typename T = unsigned>
|
|
class count : public value<T> {
|
|
public:
|
|
explicit count (T&);
|
|
explicit count (T&&) = delete;
|
|
|
|
using value<T>::execute;
|
|
void execute (void) override;
|
|
};
|
|
|
|
|
|
class bytes : public value<size_t> {
|
|
public:
|
|
explicit bytes (size_t &_value): value (_value) { }
|
|
|
|
using value<size_t>::execute;
|
|
void execute (const char *restrict) override;
|
|
};
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
class parser {
|
|
public:
|
|
template <typename T, typename ...Args>
|
|
T& add (char shortname,
|
|
std::string longname,
|
|
std::string description,
|
|
Args&&... args)
|
|
{
|
|
auto handler = std::make_unique<T> (std::forward<Args> (args)...);
|
|
T& ref = *handler;
|
|
|
|
m_short.emplace_back (shortname, ref);
|
|
m_long .emplace_back (std::move (longname), ref);
|
|
|
|
m_options.emplace_back (std::move (description), std::move (handler));
|
|
|
|
return ref;
|
|
}
|
|
|
|
template <typename T, typename ...Args>
|
|
T&
|
|
append (std::string description, Args&&...args)
|
|
{
|
|
auto handler = std::make_unique<T> (std::forward<Args> (args)...);
|
|
auto &ref = *handler;
|
|
m_positional.push_back (ref);
|
|
m_options.emplace_back (std::move (description), std::move (handler));
|
|
return ref;
|
|
}
|
|
|
|
int scan (int argc, const char *const *argv);
|
|
|
|
private:
|
|
int parse_long (int pos, int argc, const char *const *argv);
|
|
int parse_short (int pos, int argc, const char *const *argv);
|
|
|
|
void print_help [[noreturn]] (int argc, const char *const *argv) const;
|
|
|
|
using short_t = std::tuple<char,option::base&>;
|
|
using long_t = std::tuple<std::string,option::base&>;
|
|
|
|
std::vector<short_t> m_short;
|
|
std::vector<long_t> m_long;
|
|
std::vector<std::reference_wrapper<option::base>> m_positional;
|
|
|
|
std::vector<
|
|
std::tuple<
|
|
std::string, // description
|
|
std::unique_ptr<option::base>
|
|
>
|
|
> m_options;
|
|
};
|
|
}
|
|
|
|
#endif
|