5ae18574fc
extraFlagsArray should not be exposed outside of `makeWrapper`, it should only be possible to set it inside a script supplied via the `--run` argument.
134 lines
4.3 KiB
Bash
134 lines
4.3 KiB
Bash
# construct an executable file that wraps the actual executable
|
||
# makeWrapper EXECUTABLE ARGS
|
||
|
||
# ARGS:
|
||
# --argv0 NAME : set name of executed process to NAME
|
||
# (otherwise it’s called …-wrapped)
|
||
# --set VAR VAL : add VAR with value VAL to the executable’s environment
|
||
# --unset VAR : remove VAR from the environment
|
||
# --run COMMAND : run command before the executable
|
||
# The command can push extra flags to a magic list variable
|
||
# extraFlagsArray, which are then added to the invocation
|
||
# of the executable
|
||
# --add-flags FLAGS : add FLAGS to invocation of executable
|
||
|
||
# --prefix ENV SEP VAL : suffix/prefix ENV with VAL, separated by SEP
|
||
# --suffix
|
||
# --suffix-each ENV SEP VALS : like --suffix, but VALS is a list
|
||
# --prefix-contents ENV SEP FILES : like --suffix-each, but contents of FILES
|
||
# are read first and used as VALS
|
||
# --suffix-contents
|
||
makeWrapper() {
|
||
local original=$1
|
||
local wrapper=$2
|
||
local params varName value command separator n fileNames
|
||
local argv0 flagsBefore flags extraFlagsArray
|
||
|
||
mkdir -p "$(dirname $wrapper)"
|
||
|
||
echo "#! $SHELL -e" > $wrapper
|
||
|
||
params=("$@")
|
||
for ((n = 2; n < ${#params[*]}; n += 1)); do
|
||
p=${params[$n]}
|
||
|
||
if test "$p" = "--set"; then
|
||
varName=${params[$((n + 1))]}
|
||
value=${params[$((n + 2))]}
|
||
n=$((n + 2))
|
||
echo "export $varName=\"$value\"" >> $wrapper
|
||
fi
|
||
|
||
if test "$p" = "--unset"; then
|
||
varName=${params[$((n + 1))]}
|
||
n=$((n + 1))
|
||
echo "unset $varName" >> "$wrapper"
|
||
fi
|
||
|
||
if test "$p" = "--run"; then
|
||
command=${params[$((n + 1))]}
|
||
n=$((n + 1))
|
||
echo "$command" >> $wrapper
|
||
fi
|
||
|
||
if test "$p" = "--suffix" -o "$p" = "--prefix"; then
|
||
varName=${params[$((n + 1))]}
|
||
separator=${params[$((n + 2))]}
|
||
value=${params[$((n + 3))]}
|
||
n=$((n + 3))
|
||
if test -n "$value"; then
|
||
if test "$p" = "--suffix"; then
|
||
echo "export $varName=\$$varName\${$varName:+$separator}$value" >> $wrapper
|
||
else
|
||
echo "export $varName=$value\${$varName:+$separator}\$$varName" >> $wrapper
|
||
fi
|
||
fi
|
||
fi
|
||
|
||
if test "$p" = "--suffix-each"; then
|
||
varName=${params[$((n + 1))]}
|
||
separator=${params[$((n + 2))]}
|
||
values=${params[$((n + 3))]}
|
||
n=$((n + 3))
|
||
for value in $values; do
|
||
echo "export $varName=\$$varName\${$varName:+$separator}$value" >> $wrapper
|
||
done
|
||
fi
|
||
|
||
if test "$p" = "--suffix-contents" -o "$p" = "--prefix-contents"; then
|
||
varName=${params[$((n + 1))]}
|
||
separator=${params[$((n + 2))]}
|
||
fileNames=${params[$((n + 3))]}
|
||
n=$((n + 3))
|
||
for fileName in $fileNames; do
|
||
if test "$p" = "--suffix-contents"; then
|
||
echo "export $varName=\$$varName\${$varName:+$separator}$(cat $fileName)" >> $wrapper
|
||
else
|
||
echo "export $varName=$(cat $fileName)\${$varName:+$separator}\$$varName" >> $wrapper
|
||
fi
|
||
done
|
||
fi
|
||
|
||
if test "$p" = "--add-flags"; then
|
||
flags=${params[$((n + 1))]}
|
||
n=$((n + 1))
|
||
flagsBefore="$flagsBefore $flags"
|
||
fi
|
||
|
||
if test "$p" = "--argv0"; then
|
||
argv0=${params[$((n + 1))]}
|
||
n=$((n + 1))
|
||
fi
|
||
done
|
||
|
||
# Note: extraFlagsArray is an array containing additional flags
|
||
# that may be set by --run actions.
|
||
echo exec ${argv0:+-a $argv0} "$original" \
|
||
$flagsBefore '"${extraFlagsArray[@]}"' '"$@"' >> $wrapper
|
||
|
||
chmod +x $wrapper
|
||
}
|
||
|
||
addSuffix() {
|
||
suffix=$1
|
||
shift
|
||
for name in "$@"; do
|
||
echo "$name$suffix"
|
||
done
|
||
}
|
||
|
||
filterExisting() {
|
||
for fn in "$@"; do
|
||
if test -e "$fn"; then
|
||
echo "$fn"
|
||
fi
|
||
done
|
||
}
|
||
|
||
# Syntax: wrapProgram <PROGRAM> <MAKE-WRAPPER FLAGS...>
|
||
wrapProgram() {
|
||
local prog="$1"
|
||
local hidden="$(dirname "$prog")/.$(basename "$prog")"-wrapped
|
||
mv $prog $hidden
|
||
makeWrapper $hidden $prog --argv0 '"$0"' "$@"
|
||
}
|