Flag kick-out should work

2 files changed, 39 insertions, 8 deletions

diff --git a/args.hxx b/args.hxx
index 9a5681a..509a5c3 100644
--- a/args.hxx
+++ b/args.hxx
@@ -380,9 +380,10 @@ namespace args
     {
         protected:
             const std::string name;
+            bool kickout;
 
         public:
-            NamedBase(const std::string &name, const std::string &help) : Base(help), name(name) {}
+            NamedBase(const std::string &name, const std::string &help) : Base(help), name(name), kickout(false) {}
             virtual ~NamedBase() {}
 
             virtual std::tuple<std::string, std::string> GetDescription(const std::string &shortPrefix, const std::string &longPrefi, const std::string &shortSeparator, const std::string &longSeparator) const override
@@ -396,6 +397,16 @@ namespace args
             {
                 return name;
             }
+
+            void KickOut(bool kickout) noexcept
+            {
+                this->kickout = kickout;
+            }
+
+            bool KickOut() const noexcept
+            {
+                return kickout;
+            }
     };
 
     /** Base class for all flag options
@@ -1093,9 +1104,10 @@ namespace args
              *
              * \param begin an iterator to the beginning of the argument list
              * \param end an iterator to the past-the-end element of the argument list
+             * \return the iterator after the last parsed value.  Only useful for kick-out
              */
             template <typename It>
-            void ParseArgs(It begin, It end)
+            It ParseArgs(It begin, It end)
             {
                 // Reset all Matched statuses to false, for validation.  Don't reset values.
                 ResetMatched();
@@ -1160,6 +1172,11 @@ namespace args
                                 problem << "Passed an argument into a non-argument flag: " << chunk;
                                 throw ParseError(problem.str());
                             }
+
+                            if (base->KickOut())
+                            {
+                                return ++it;
+                            }
                         } else
                         {
                             std::ostringstream problem;
@@ -1174,7 +1191,7 @@ namespace args
                         {
                             const char arg = *argit;
 
-                            if (Base *base = Match(arg))
+                            if (FlagBase *base = Match(arg))
                             {
                                 if (ValueFlagBase *argbase = dynamic_cast<ValueFlagBase *>(base))
                                 {
@@ -1213,6 +1230,11 @@ namespace args
                                     // Because this argchunk is done regardless
                                     break;
                                 }
+
+                                if (base->KickOut())
+                                {
+                                    return ++it;
+                                }
                             } else
                             {
                                 std::ostringstream problem;
@@ -1226,6 +1248,11 @@ namespace args
                         if (pos)
                         {
                             pos->ParseValue(chunk);
+
+                            if (pos->KickOut())
+                            {
+                                return ++it;
+                            }
                         } else
                         {
                             std::ostringstream problem;
@@ -1240,30 +1267,34 @@ namespace args
                     problem << "Group validation failed somewhere!";
                     throw ValidationError(problem.str());
                 }
+                return end;
             }
 
             /** Parse all arguments.
              *
              * \param args an iterable of the arguments
+             * \return the iterator after the last parsed value.  Only useful for kick-out
              */
             template <typename T>
-            void ParseArgs(const T &args)
+            auto ParseArgs(const T &args) -> decltype(std::begin(args))
             {
-                ParseArgs(std::begin(args), std::end(args));
+                return ParseArgs(std::begin(args), std::end(args));
             }
 
             /** Convenience function to parse the CLI from argc and argv
              *
              * Just assigns the program name and vectorizes arguments for passing into ParseArgs()
+             *
+             * \return whether or not all arguments were parsed.  This works for detecting kick-out, but is generally useless as it can't do anything with it.
              */
-            void ParseCLI(const int argc, const char * const * const argv)
+            bool ParseCLI(const int argc, const char * const * const argv)
             {
                 if (prog.empty())
                 {
                     prog.assign(argv[0]);
                 }
                 const std::vector<std::string> args(argv + 1, argv + argc);
-                ParseArgs(args);
+                return ParseArgs(args) == std::end(args);
             }
     };
 
diff --git a/test.cxx b/test.cxx
index 1b6bea7..0fc703d 100644
--- a/test.cxx
+++ b/test.cxx
@@ -466,7 +466,7 @@ TEST_CASE("Sub-parsers should work through kick-out", "[args]")
     args::Flag foo2(parser2, "Foo", "Foo", {'f', "foo"});
     args::Flag bar2(parser2, "Bar", "Bar", {'b', "bar"});
 
-    parser1.ParseArgs(next, std::end(args));
+    parser2.ParseArgs(next, std::end(args));
 
     REQUIRE(foo1);
     REQUIRE_FALSE(bar1);