122 lines
3.1 KiB
Nix
122 lines
3.1 KiB
Nix
with import <stockholm/lib>;
|
|
{ coreutils, quote, utillinux, writeDash }:
|
|
|
|
opt-spec: cmd-spec: let
|
|
|
|
cmd = cmd-spec opts;
|
|
|
|
cmd-script =
|
|
if typeOf cmd == "set"
|
|
then "exec ${cmd}"
|
|
else cmd;
|
|
|
|
opts = mapAttrs (name: value: value // rec {
|
|
long = value.long or (replaceStrings ["_"] ["-"] name);
|
|
ref = value.ref or "\"\$${varname}\"";
|
|
short = value.short or null;
|
|
switch = value.switch or false;
|
|
varname = value.varname or (replaceStrings ["-"] ["_"] name);
|
|
}) opt-spec;
|
|
|
|
# true if b requires a to define its default value
|
|
opts-before = a: b:
|
|
test ".*[$]${stringAsChars (c: "[${c}]") a.varname}([^0-9A-Za-z_].*)?" (b.default or "");
|
|
|
|
opts-list = let
|
|
sort-out = toposort opts-before (attrValues opts);
|
|
in
|
|
if sort-out ? result
|
|
then sort-out.result
|
|
else throw "toposort output: ${toJSON sort-out}";
|
|
|
|
wrapper-name =
|
|
if typeOf cmd == "set" && cmd ? name
|
|
then "${cmd.name}-getopt"
|
|
else "getopt-wrapper";
|
|
|
|
in writeDash wrapper-name ''
|
|
set -efu
|
|
|
|
wrapper_name=${shell.escape wrapper-name}
|
|
|
|
${concatStringsSep "\n" (mapAttrsToList (name: opt: /* sh */ ''
|
|
unset ${opt.varname}
|
|
'') opts)}
|
|
|
|
WITHGETOPT_ORIG_ARGS=$(${quote}/bin/quote "$@")
|
|
export WITHGETOPT_ORIG_ARGS
|
|
|
|
args=$(${utillinux}/bin/getopt \
|
|
-l ${shell.escape
|
|
(concatMapStringsSep ","
|
|
(opt: opt.long + optionalString (!opt.switch) ":")
|
|
(filter (opt: opt.long != null)
|
|
(attrValues opts)))} \
|
|
-n "$wrapper_name" \
|
|
-o ${shell.escape
|
|
(concatMapStringsSep ""
|
|
(opt: opt.short + optionalString (!opt.switch) ":")
|
|
(filter (opt: opt.short != null)
|
|
(attrValues opts)))} \
|
|
-s sh \
|
|
-- "$@")
|
|
if \test $? != 0; then exit 1; fi
|
|
eval set -- "$args"
|
|
|
|
while :; do
|
|
case $1 in
|
|
${concatStringsSep "\n" (mapAttrsToList (name: opt: /* sh */ ''
|
|
(${concatMapStringsSep "|" shell.escape (filter (x: x != "") [
|
|
(optionalString (opt.long != null) "--${opt.long}")
|
|
(optionalString (opt.short != null) "-${opt.short}")
|
|
])})
|
|
${if opt.switch then /* sh */ ''
|
|
${opt.varname}=true
|
|
shift
|
|
'' else /* sh */ ''
|
|
${opt.varname}=$2
|
|
shift 2
|
|
''}
|
|
;;
|
|
'') (filterAttrs
|
|
(_: opt: opt.long != null || opt.short != null)
|
|
opts))}
|
|
--)
|
|
shift
|
|
break
|
|
esac
|
|
done
|
|
|
|
${concatMapStringsSep "\n"
|
|
(opt: /* sh */ ''
|
|
if \test "''${${opt.varname}+1}" != 1; then
|
|
printf '%s: missing mandatory option '--%s'\n' \
|
|
"$wrapper_name" \
|
|
${shell.escape opt.long}
|
|
error=1
|
|
fi
|
|
'')
|
|
(filter
|
|
(x: ! hasAttr "default" x)
|
|
(attrValues opts))}
|
|
if test "''${error+1}" = 1; then
|
|
exit 1
|
|
fi
|
|
|
|
${concatMapStringsSep "\n"
|
|
(opt: /* sh */ ''
|
|
if \test "''${${opt.varname}+1}" != 1; then
|
|
${opt.varname}=${opt.default}
|
|
fi
|
|
'')
|
|
(filter
|
|
(hasAttr "default")
|
|
opts-list)}
|
|
|
|
${concatStringsSep "\n" (mapAttrsToList (name: opt: /* sh */ ''
|
|
export ${opt.varname}
|
|
'') opts)}
|
|
|
|
${cmd-script} "$@"
|
|
''
|