mirrors/argparse
1
0
Fork 0
mirror of https://github.com/TangentFoxy/argparse.git synced 2025-07-27 18:42:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make add_complete a Parser method

Browse Source
This commit is contained in:
Paul Ouellette
2019-06-10 16:39:15 -04:00
parent 03e7daf31f
commit 5b733ba50a
1 changed files with 27 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
src/argparse.lua
View File

@@ -227,27 +227,6 @@ local add_help = {"add_help", function(self, value)
end
end}
local add_complete = {"add_complete", function(self, value)
typecheck("add_complete", {"nil", "string", "table"}, value)
local complete = self:option()
:description "Output a shell completion script for the specified shell."
:args(1)
:choices {"bash", "zsh", "fish"}
:action(function(_, _, shell)
print(self["get_" .. shell .. "_complete"](self))
os.exit(0)
end)
if value then
complete = complete(value)
end
if not complete._name then
complete "--completion"
end
end}
local Parser = class({
_arguments = {},
_options = {},
@@ -273,8 +252,7 @@ local Parser = class({
typechecked("help_usage_margin", "number"),
typechecked("help_description_margin", "number"),
typechecked("help_max_width", "number"),
add_help,
add_complete
add_help
})
local Command = class({
@@ -1105,6 +1083,32 @@ function Parser:add_help_command(value)
return self
end
function Parser:add_complete(value)
if value then
assert(type(value) == "string" or type(value) == "table",
("bad argument #1 to 'add_complete' (string or table expected, got %s)"):format(type(value)))
end
local complete = self:option()
:description "Output a shell completion script for the specified shell."
:args(1)
:choices {"bash", "zsh", "fish"}
:action(function(_, _, shell)
print(self["get_" .. shell .. "_complete"](self))
os.exit(0)
end)
if value then
complete = complete(value)
end
if not complete._name then
complete "--completion"
end
return self
end
local function get_short_description(element)
local short = element._description:match("^(.-)%.%s")
return short or element._description:match("^(.-)%.?$")