Improved usage message generation

Put options taking variable number of arguments after positional arguments

spec/usage_spec.lua

@@ -41,6 +41,21 @@ describe("tests related to usage message generation", function()
       )
    end)
 
+   it("creates correct usage message for options with variable argument count", function()
+      local parser = Parser "foo"
+         :add_help(false)
+      parser:argument "files"
+         :args "+"
+      parser:flag "-q" "--quiet"
+      parser:option "--globals"
+         :args "*"
+
+      assert.equal(
+         [=[Usage: foo [-q] <files> [<files>] ... [--globals [<globals>] ...]]=],
+         parser:get_usage()
+      )
+   end)
+
    it("creates correct usage message for arguments with default value", function()
       local parser = Parser "foo"
          :add_help(false)

src/argparse.lua

@@ -373,6 +373,10 @@ function Option:_get_target()
    return self._name:sub(2)
 end
 
+function Option:_is_vararg()
+   return self._maxargs ~= self._minargs
+end
+
 function Parser:_get_fullname()
    local parent = self._parent
    local buf = {self._name}
@@ -459,25 +463,53 @@ function Parser:get_usage()
       end
    end
 
-   -- set of mentioned elements
+   -- This can definitely be refactored into something cleaner
-   local used = {}
+   local mutex_options = {}
+   local vararg_mutexes = {}
 
+   -- First, put mutexes which do not contain vararg options and remember those which do
    for _, mutex in ipairs(self._mutexes) do
       local buf = {}
+      local is_vararg = false
 
       for _, option in ipairs(mutex) do
+         if option:_is_vararg() then
+            is_vararg = true
+         end
+
          table.insert(buf, option:_get_usage())
-         used[option] = true
+         mutex_options[option] = true
       end
 
-      add("(" .. table.concat(buf, " | ") .. ")")
+      local repr = "(" .. table.concat(buf, " | ") .. ")"
+
+      if is_vararg then
+         table.insert(vararg_mutexes, repr)
+      else
+         add(repr)
+      end
    end
 
-   for _, elements in ipairs{self._options, self._arguments} do
+   -- Second, put regular options
-      for _, element in ipairs(elements) do
+   for _, option in ipairs(self._options) do
-         if not used[element] then
+      if not mutex_options[option] and not option:_is_vararg() then
-            add(element:_get_usage())
+         add(option:_get_usage())
-         end
+      end
+   end
+
+   -- Put positional arguments
+   for _, argument in ipairs(self._arguments) do
+      add(argument:_get_usage())
+   end
+
+   -- Put mutexes containing vararg options
+   for _, mutex_repr in ipairs(vararg_mutexes) do
+      add(mutex_repr)
+   end
+
+   for _, option in ipairs(self._options) do
+      if not mutex_options[option] and option:_is_vararg() then
+         add(option:_get_usage())
       end
    end