Merge remote-tracking branch 'myk002/myk_getopt' into develop
lethosor
commit
d4aef78b26
@ -0,0 +1,169 @@
|
||||
-- Copyright (c) 2009 Aleksey Cheusov <vle@gmx.net>
|
||||
--
|
||||
-- Permission is hereby granted, free of charge, to any person obtaining
|
||||
-- a copy of this software and associated documentation files (the
|
||||
-- "Software"), to deal in the Software without restriction, including
|
||||
-- without limitation the rights to use, copy, modify, merge, publish,
|
||||
-- distribute, sublicense, and/or sell copies of the Software, and to
|
||||
-- permit persons to whom the Software is furnished to do so, subject to
|
||||
-- the following conditions:
|
||||
--
|
||||
-- The above copyright notice and this permission notice shall be
|
||||
-- included in all copies or substantial portions of the Software.
|
||||
--
|
||||
-- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||||
-- EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||||
-- MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||||
-- NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
||||
-- LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
||||
-- OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
||||
-- WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
||||
-- based on https://github.com/LuaDist/alt-getopt/blob/master/alt_getopt.lua
|
||||
-- MIT licence
|
||||
-- modified to support aggregation of non-options and to call qerror instead of
|
||||
-- os.exit() on error. can be used directly or via the utils.processArgs2()
|
||||
-- wrapper.
|
||||
|
||||
-- sh_opts should be in standard getopt format: a string of letters that
|
||||
-- represent options, each followed by a colon if that option takes an argument.
|
||||
-- e.g.: 'ak:hv' has three flags (options with no arguments): 'a', 'h', and 'v'
|
||||
-- and one option that takes an argument: 'k'.
|
||||
--
|
||||
-- Options passed to the module to parse can be in any of the following formats:
|
||||
-- -kVALUE, -k VALUE, --key=VALUE, --key VALUE
|
||||
-- -abcd is equivalent to -a -b -c -d if none of them accept arguments.
|
||||
-- -abckVALUE and -abck VALUE are also acceptable (where k is the only option
|
||||
-- in the string that takes a value).
|
||||
|
||||
local _ENV = mkmodule('3rdparty.alt_getopt')
|
||||
|
||||
local function get_opt_map(opts)
|
||||
local i = 1
|
||||
local len = #opts
|
||||
local options = {}
|
||||
|
||||
for short_opt, accept_arg in opts:gmatch('(%w)(:?)') do
|
||||
options[short_opt] = #accept_arg
|
||||
end
|
||||
|
||||
return options
|
||||
end
|
||||
|
||||
local function err_unknown_opt(opt)
|
||||
qerror(string.format('Unknown option "-%s%s"', #opt > 1 and '-' or '', opt))
|
||||
end
|
||||
|
||||
-- resolve aliases into their canonical forms
|
||||
local function canonicalize(options, opt)
|
||||
if not options[opt] then
|
||||
err_unknown_opt(opt)
|
||||
end
|
||||
|
||||
while type(options[opt]) == 'string' do
|
||||
opt = options[opt]
|
||||
|
||||
if not options[opt] then
|
||||
err_unknown_opt(opt)
|
||||
end
|
||||
end
|
||||
|
||||
if type(options[opt]) ~= 'number' then
|
||||
qerror(string.format(
|
||||
'Option "%s" resolves to non-number for has_arg flag', opt))
|
||||
end
|
||||
|
||||
return opt
|
||||
end
|
||||
|
||||
local function has_arg(options, opt)
|
||||
return options[canonicalize(options, opt)] == 1
|
||||
end
|
||||
|
||||
-- returns vectors for opts, optargs, and nonoptions
|
||||
function get_ordered_opts(args, sh_opts, long_opts)
|
||||
local optind, count, opts, optargs, nonoptions = 1, 1, {}, {}, {}
|
||||
|
||||
local options = get_opt_map(sh_opts)
|
||||
for k,v in pairs(long_opts) do
|
||||
options[k] = v
|
||||
end
|
||||
|
||||
while optind <= #args do
|
||||
local a = args[optind]
|
||||
if a == '--' then
|
||||
optind = optind + 1
|
||||
elseif a:sub(1, 2) == '--' then
|
||||
local pos = a:find('=', 1, true)
|
||||
if pos then
|
||||
local opt = a:sub(3, pos-1)
|
||||
if not has_arg(options, opt) then
|
||||
qerror(string.format('Bad usage of option "%s"', a))
|
||||
end
|
||||
opts[count] = opt
|
||||
optargs[count] = a:sub(pos+1)
|
||||
else
|
||||
local opt = a:sub(3)
|
||||
opts[count] = opt
|
||||
if has_arg(options, opt) then
|
||||
if i == #args then
|
||||
qerror(string.format(
|
||||
'Missing value for option "%s"', a))
|
||||
end
|
||||
optargs[count] = args[optind+1]
|
||||
optind = optind + 1
|
||||
end
|
||||
end
|
||||
count = count + 1
|
||||
elseif a:sub(1, 1) == '-' then
|
||||
local j
|
||||
for j=2,#a do
|
||||
local opt = canonicalize(options, a:sub(j, j))
|
||||
if not has_arg(options, opt) then
|
||||
opts[count] = opt
|
||||
count = count + 1
|
||||
elseif j == #a then
|
||||
if optind == #args then
|
||||
qerror(string.format(
|
||||
'Missing value for option "-%s"', opt))
|
||||
end
|
||||
opts[count] = opt
|
||||
optargs[count] = args[optind+1]
|
||||
optind = optind + 1
|
||||
count = count + 1
|
||||
else
|
||||
opts[count] = opt
|
||||
optargs[count] = a:sub(j+1)
|
||||
count = count + 1
|
||||
break
|
||||
end
|
||||
end
|
||||
else
|
||||
table.insert(nonoptions, args[optind])
|
||||
end
|
||||
optind = optind + 1
|
||||
end
|
||||
for i=optind,#args do
|
||||
table.insert(nonoptions, args[i])
|
||||
end
|
||||
return opts, optargs, nonoptions
|
||||
end
|
||||
|
||||
-- returns a map of options to their optargs (or 1 if the option doesn't take an
|
||||
-- argument), and a vector for nonoptions
|
||||
function get_opts(args, sh_opts, long_opts)
|
||||
local ret = {}
|
||||
|
||||
local opts,optargs,nonoptions = get_ordered_opts(args, sh_opts, long_opts)
|
||||
for i,v in ipairs(opts) do
|
||||
if optarg[i] then
|
||||
ret[v] = optarg[i]
|
||||
else
|
||||
ret[v] = 1
|
||||
end
|
||||
end
|
||||
|
||||
return ret, nonoptions
|
||||
end
|
||||
|
||||
return _ENV
|
||||
Loading…
Reference in New Issue