mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-04 05:37:39 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

util.argparse: Optionally continue processing past positional parameters

Browse source
This commit is contained in:
Matthew Wild 2025-02-17 16:38:48 +00:00
parent b8e4d5e840
commit edafb195e1
1 changed files with 42 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

74
util/argparse.lua
View file

@ -2,6 +2,7 @@ local function parse(arg, config)
local short_params = config and config.short_params or {}; local short_params = config and config.short_params or {};
local value_params = config and config.value_params or {}; local value_params = config and config.value_params or {};
local array_params = config and config.array_params or {}; local array_params = config and config.array_params or {};
local stop_on_positional = not config or config.stop_on_positional ~= false;
local parsed_opts = {}; local parsed_opts = {};
@ -15,51 +16,60 @@ local function parse(arg, config)
end end
local prefix = raw_param:match("^%-%-?"); local prefix = raw_param:match("^%-%-?");
if not prefix then if not prefix and stop_on_positional then
break; break;
elseif prefix == "--" and raw_param == "--" then elseif prefix == "--" and raw_param == "--" then
table.remove(arg, 1); table.remove(arg, 1);
break; break;
end end
local param = table.remove(arg, 1):sub(#prefix+1);
if #param == 1 and short_params then
param = short_params[param];
end
if not param then if prefix then
return nil, "param-not-found", raw_param; local param = table.remove(arg, 1):sub(#prefix+1);
end if #param == 1 and short_params then
param = short_params[param];
local param_k, param_v;
if value_params[param] or array_params[param] then
param_k, param_v = param, table.remove(arg, 1);
if not param_v then
return nil, "missing-value", raw_param;
end end
else
param_k, param_v = param:match("^([^=]+)=(.+)$"); if not param then
if not param_k then return nil, "param-not-found", raw_param;
if param:match("^no%-") then end
param_k, param_v = param:sub(4), false;
else local param_k, param_v;
param_k, param_v = param, true; if value_params[param] or array_params[param] then
param_k, param_v = param, table.remove(arg, 1);
if not param_v then
return nil, "missing-value", raw_param;
end end
end
param_k = param_k:gsub("%-", "_");
end
if array_params[param] then
if parsed_opts[param_k] then
table.insert(parsed_opts[param_k], param_v);
else else
parsed_opts[param_k] = { param_v }; param_k, param_v = param:match("^([^=]+)=(.+)$");
if not param_k then
if param:match("^no%-") then
param_k, param_v = param:sub(4), false;
else
param_k, param_v = param, true;
end
end
param_k = param_k:gsub("%-", "_");
end end
else if array_params[param] then
parsed_opts[param_k] = param_v; if parsed_opts[param_k] then
table.insert(parsed_opts[param_k], param_v);
else
parsed_opts[param_k] = { param_v };
end
else
parsed_opts[param_k] = param_v;
end
elseif not stop_on_positional then
table.insert(parsed_opts, table.remove(arg, 1));
end end
end end
for i = 1, #arg do
parsed_opts[i] = arg[i]; if stop_on_positional then
for i = 1, #arg do
parsed_opts[i] = arg[i];
end
end end
return parsed_opts; return parsed_opts;
end end