]> git.tdb.fi Git - libs/core.git/blobdiff - source/core/getopt.h
Add move semantics to Variant
[libs/core.git] / source / core / getopt.h
index 911934ac08adb7399bbd914b6778a8df8eb99f03..59c810c5e4a5ff4182cd42d9736f8af9d708eae9 100644 (file)
@@ -6,19 +6,20 @@
 #include <string>
 #include <vector>
 #include <msp/strings/lexicalcast.h>
+#include "mspcore_api.h"
+#include "noncopyable.h"
 
 namespace Msp {
 
-class usage_error: public std::runtime_error
+class MSPCORE_API usage_error: public std::runtime_error
 {
 private:
-       std::string help_;
+       std::string m_help;
 
 public:
-       usage_error(const std::string &w, const std::string &h = std::string()): std::runtime_error(w), help_(h) { }
-       ~usage_error() throw() { }
+       usage_error(const std::string &w, const std::string &h = std::string()): std::runtime_error(w), m_help(h) { }
 
-       const char *help() const throw() { return help_.c_str(); }
+       const char *help() const noexcept { return m_help.c_str(); }
 };
 
 
@@ -33,22 +34,32 @@ the string "-abc" could be interpreted as having the options 'a', 'b' and 'c'.
 If the option takes an argument and there are unused characters in the argv
 element, then those characters are interpreted as the argument.  Otherwise the
 next element is taken as the argument.  An optional argument must be given in
-the same element.
+the same element if it is given.
 
 Long options begin with a double dash and are identified by an arbitrary
 string.  An argument can be specified either in the same argv element,
 separated by an equals sign, or in the next element.  As with short options,
-an optional argument must be in the same element.
+an optional argument, if given, must be in the same element.
 
 A single option may have both alternative forms, but must always have at least
 a long form.  This is to encourage self-documenting options; it's much easier
 to remember words than letters.
 
-A built-in --help option is provided and will output a list of options and
-their associated help texts.  An application may override this by providing
-its own option with the same name.
+Positional arguments are also supported.  They are identified by an arbitrary
+string, but the identifier is only used in help text and error messages.  Any
+number of the final arguments may be optional.
+
+To support applications that take an arbitrary amount of arguments, a single
+positional argument list can be specified.  Fixed positional arguments are
+allowed together with a list, but they can't be optional.  An application that
+wants to do complex processing on the argument list can declare a list of
+string arguments.
+
+A built-in --help option is provided and will output a list of options,
+arguments and their associated help texts.  An application may override this by
+providing its own option with the same name.
 */
-class GetOpt
+class MSPCORE_API GetOpt: private NonCopyable
 {
 public:
        enum ArgType
@@ -58,12 +69,12 @@ public:
                REQUIRED_ARG
        };
 
-       class Option
+       class MSPCORE_API Option
        {
        protected:
-               Option() { }
+               Option() = default;
        public:
-               virtual ~Option() { }
+               virtual ~Option() = default;
 
                /// Sets help text for the option.
                virtual Option &set_help(const std::string &) = 0;
@@ -78,16 +89,27 @@ public:
                virtual unsigned get_seen_count() const = 0;
        };
 
+       class MSPCORE_API Argument
+       {
+       protected:
+               Argument() = default;
+       public:
+               virtual ~Argument() = default;
+
+               virtual Argument &set_help(const std::string &) = 0;
+       };
+
 private:
        class Store
        {
        protected:
-               Store() { }
+               Store() = default;
        public:
-               virtual ~Store() { }
+               virtual ~Store() = default;
 
                virtual Store *clone() const = 0;
 
+               virtual bool is_list() const = 0;
                virtual void store() = 0;
                virtual void store(const std::string &) = 0;
        };
@@ -95,32 +117,52 @@ private:
        class OptionImpl: public Option
        {
        protected:
-               char shrt;
+               char shrt = 0;
                std::string lng;
-               ArgType arg_type;
-               unsigned seen_count;
-               unsigned *ext_seen_count;
+               ArgType arg_type = NO_ARG;
+               unsigned seen_count = 0;
+               unsigned *ext_seen_count = nullptr;
                std::string help;
-               std::string metavar;
-               Store *store;
+               std::string metavar = "ARG";
+               Store *store = nullptr;
 
        public:
                OptionImpl(char, const std::string &, const Store &, ArgType);
-               virtual ~OptionImpl();
+               ~OptionImpl() override;
 
-               virtual OptionImpl &set_help(const std::string &);
-               virtual OptionImpl &set_help(const std::string &, const std::string &);
-               virtual OptionImpl &bind_seen_count(unsigned &);
+               OptionImpl &set_help(const std::string &) override;
+               OptionImpl &set_help(const std::string &, const std::string &) override;
+               OptionImpl &bind_seen_count(unsigned &) override;
                char get_short() const { return shrt; }
                const std::string &get_long() const { return lng; }
                ArgType get_arg_type() const { return arg_type; }
                const std::string &get_help() const { return help; }
                const std::string &get_metavar() const { return metavar; }
-               virtual unsigned get_seen_count() const { return seen_count; }
+               unsigned get_seen_count() const override { return seen_count; }
                void process();
                void process(const std::string &);
        };
 
+       class ArgumentImpl: public Argument
+       {
+       private:
+               std::string name;
+               ArgType type = REQUIRED_ARG;
+               std::string help;
+               Store *store = nullptr;
+
+       public:
+               ArgumentImpl(const std::string &, const Store &, ArgType);
+               ~ArgumentImpl() override;
+
+               ArgumentImpl &set_help(const std::string &) override;
+               const std::string &get_name() const { return name; }
+               ArgType get_type() const { return type; }
+               const std::string &get_help() const { return help; }
+               bool is_list_store() const { return store->is_list(); }
+               void process(const std::string &);
+       };
+
        template<typename T>
        class SimpleStore: public Store
        {
@@ -130,12 +172,14 @@ private:
        public:
                SimpleStore(T &d): data(d) { }
 
-               virtual SimpleStore *clone() const
+               SimpleStore *clone() const override
                { return new SimpleStore(data); }
 
-               virtual void store() { }
+               bool is_list() const override { return false; }
 
-               virtual void store(const std::string &a)
+               void store() override { }
+
+               void store(const std::string &a) override
                { data = lexical_cast<T>(a); }
        };
 
@@ -148,37 +192,42 @@ private:
        public:
                ListStore(T &d): data(d) { }
 
-               virtual ListStore *clone() const
+               ListStore *clone() const override
                { return new ListStore(data); }
 
-               virtual void store() { }
+               bool is_list() const override { return true; }
+
+               void store() override { }
 
-               virtual void store(const std::string &a)
+               void store(const std::string &a) override
                { data.push_back(lexical_cast<typename T::value_type>(a)); }
        };
 
-       typedef std::list<OptionImpl *> OptionList;
-
-       bool help;
-       OptionList opts;
-       std::vector<std::string> args;
+       bool help = false;
+       std::vector<OptionImpl *> opts;
+       std::vector<ArgumentImpl *> args;
+       std::vector<std::string> args_raw;
 
 public:
        GetOpt();
        ~GetOpt();
 
-       /// Returns any non-option arguments encountered during processing.
-       const std::vector<std::string> &get_args() const { return args; }
-
        /** Adds an option with both short and long forms.  Processing depends on
        the type of the destination variable and whether an argument is taken or
-       not.  With an argument, the value is lexical_cast to appropriate type and
-       stored in the destination.  Without an argument, a bool will be set to true
-       and an unsigned will be incremented; any other type will be ignored. */
+       not.  With an argument, the value is lexical_cast to the appropriate type
+       and stored in the destination.  Without an argument, a bool will be set to
+       true and an unsigned will be incremented; any other type will be ignored. */
        template<typename T>
        Option &add_option(char s, const std::string &l, T &d, ArgType a = NO_ARG)
        { return add_option(s, l, SimpleStore<T>(d), a); }
 
+       /** Adds an option with both short and long forms.  The option may be
+       specified multiple times, and the argument from each occurrence is stored in
+       the list.  The argument type must be REQUIRED_ARG. */
+       template<typename T>
+       Option &add_option(char s, const std::string &l, std::vector<T> &d, ArgType a = REQUIRED_ARG)
+       { return add_option(s, l, ListStore<std::vector<T> >(d), a); }
+
        /** Adds an option with both short and long forms.  The option may be
        specified multiple times, and the argument from each occurrence is stored in
        the list.  The argument type must be REQUIRED_ARG. */
@@ -191,8 +240,29 @@ public:
        Option &add_option(const std::string &l, T &d, ArgType a)
        { return add_option(0, l, d, a); }
 
+       /** Adds a positional argument.  The value will be lexical_cast to the
+       appropriate type and stored in the destination. */
+       template<typename T>
+       Argument &add_argument(const std::string &n, T &d, ArgType a = REQUIRED_ARG)
+       { return add_argument(n, SimpleStore<T>(d), a); }
+
+       /** Adds a positional argument list.  If the list is declared as required,
+       at least one element must be given; an optional list may be empty.  Only one
+       list may be added, and optional fixed arguments can't be used with it. */
+       template<typename T>
+       Argument &add_argument(const std::string &n, std::vector<T> &d, ArgType a = REQUIRED_ARG)
+       { return add_argument(n, ListStore<std::vector<T> >(d), a); }
+
+       /** Adds a positional argument list.  If the list is declared as required,
+       at least one element must be given; an optional list may be empty.  Only one
+       list may be added, and optional fixed arguments can't be used with it. */
+       template<typename T>
+       Argument &add_argument(const std::string &n, std::list<T> &d, ArgType a = REQUIRED_ARG)
+       { return add_argument(n, ListStore<std::list<T> >(d), a); }
+
 private:
        OptionImpl &add_option(char, const std::string &, const Store &, ArgType);
+       ArgumentImpl &add_argument(const std::string &, const Store &, ArgType);
 
        OptionImpl &get_option(char);
        OptionImpl &get_option(const std::string &);
@@ -210,11 +280,13 @@ private:
        unsigned process_short(const char *const *);
 
 public:
-       /** Generates a single line that describes known options. */
-       std::string generate_usage(const std::string &) const;
+       /** Generates a single line that describes known options and arguments.  If
+       compact is true, the options list is replaced with a placeholder.  This
+       provides cleaner output if full help text is printed. */
+       std::string generate_usage(const std::string &, bool compact = false) const;
 
-       /** Generates help for known options in tabular format, one option per
-       line.  The returned string will have a linefeed at the end. */
+       /** Generates help for known options and arguments in tabular format, one
+       item per line.  The returned string will have a linefeed at the end. */
        std::string generate_help() const;
 };
 
@@ -227,6 +299,9 @@ template<> inline void GetOpt::SimpleStore<unsigned>::store()
 template<> inline void GetOpt::SimpleStore<std::string>::store(const std::string &a)
 { data = a; }
 
+template<> inline void GetOpt::ListStore<std::vector<std::string> >::store(const std::string &a)
+{ data.push_back(a); }
+
 template<> inline void GetOpt::ListStore<std::list<std::string> >::store(const std::string &a)
 { data.push_back(a); }