1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
local function parse(arg, config)
local short_params = config and config.short_params or {};
local value_params = config and config.value_params or {};
local array_params = config and config.array_params or {};
local kv_params = config and config.kv_params or {};
local strict = config and config.strict;
local stop_on_positional = not config or config.stop_on_positional ~= false;
local parsed_opts = {};
if #arg == 0 then
return parsed_opts;
end
while true do
local raw_param = arg[1];
if not raw_param then
break;
end
local prefix = raw_param:match("^%-%-?");
if not prefix and stop_on_positional then
break;
elseif prefix == "--" and raw_param == "--" then
table.remove(arg, 1);
break;
end
if prefix then
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
return nil, "param-not-found", raw_param;
end
local uparam = param:match("^[^=]*"):gsub("%-", "_");
local param_k, param_v;
if value_params[uparam] or array_params[uparam] then
param_k, param_v = uparam, table.remove(arg, 1);
if not param_v then
return nil, "missing-value", raw_param;
end
else
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("%-", "_");
if strict and not kv_params[param_k] then
return nil, "param-not-found", raw_param;
end
end
if array_params[uparam] then
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
if stop_on_positional then
for i = 1, #arg do
parsed_opts[i] = arg[i];
end
end
return parsed_opts;
end
return {
parse = parse;
}
|
