mirror of
https://github.com/pyenv/pyenv.git
synced 2024-11-07 20:31:01 -05:00
1381c2ca79
It doesn't need to be a bash array and we don't need a separate index of shims registered. Simply keep everything in a space-separated string and use that as an index as well. This assumes that executable names *never* have spaces in them.
149 lines
3.7 KiB
Bash
Executable file
149 lines
3.7 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
# Summary: Rehash rbenv shims (run this after installing executables)
|
|
|
|
set -e
|
|
[ -n "$RBENV_DEBUG" ] && set -x
|
|
|
|
SHIM_PATH="${RBENV_ROOT}/shims"
|
|
PROTOTYPE_SHIM_PATH="${SHIM_PATH}/.rbenv-shim"
|
|
|
|
# Create the shims directory if it doesn't already exist.
|
|
mkdir -p "$SHIM_PATH"
|
|
|
|
# Ensure only one instance of rbenv-rehash is running at a time by
|
|
# setting the shell's `noclobber` option and attempting to write to
|
|
# the prototype shim file. If the file already exists, print a warning
|
|
# to stderr and exit with a non-zero status.
|
|
set -o noclobber
|
|
{ echo > "$PROTOTYPE_SHIM_PATH"
|
|
} 2>/dev/null ||
|
|
{ if [ -w "$SHIM_PATH" ]; then
|
|
echo "rbenv: cannot rehash: $PROTOTYPE_SHIM_PATH exists"
|
|
else
|
|
echo "rbenv: cannot rehash: $SHIM_PATH isn't writable"
|
|
fi
|
|
exit 1
|
|
} >&2
|
|
set +o noclobber
|
|
|
|
# If we were able to obtain a lock, register a trap to clean up the
|
|
# prototype shim when the process exits.
|
|
trap remove_prototype_shim EXIT
|
|
|
|
remove_prototype_shim() {
|
|
rm -f "$PROTOTYPE_SHIM_PATH"
|
|
}
|
|
|
|
# The prototype shim file is a script that re-execs itself, passing
|
|
# its filename and any arguments to `rbenv exec`. This file is
|
|
# hard-linked for every executable and then removed. The linking
|
|
# technique is fast, uses less disk space than unique files, and also
|
|
# serves as a locking mechanism.
|
|
create_prototype_shim() {
|
|
cat > "$PROTOTYPE_SHIM_PATH" <<SH
|
|
#!/usr/bin/env bash
|
|
set -e
|
|
[ -n "\$RBENV_DEBUG" ] && set -x
|
|
|
|
program="\${0##*/}"
|
|
if [ "\$program" = "ruby" ]; then
|
|
for arg; do
|
|
case "\$arg" in
|
|
-e* | -- ) break ;;
|
|
*/* )
|
|
if [ -f "\$arg" ]; then
|
|
export RBENV_DIR="\${arg%/*}"
|
|
break
|
|
fi
|
|
;;
|
|
esac
|
|
done
|
|
fi
|
|
|
|
export RBENV_ROOT="$RBENV_ROOT"
|
|
exec "$(command -v rbenv)" exec "\$program" "\$@"
|
|
SH
|
|
chmod +x "$PROTOTYPE_SHIM_PATH"
|
|
}
|
|
|
|
# If the contents of the prototype shim file differ from the contents
|
|
# of the first shim in the shims directory, assume rbenv has been
|
|
# upgraded and the existing shims need to be removed.
|
|
remove_outdated_shims() {
|
|
local shim
|
|
for shim in "$SHIM_PATH"/*; do
|
|
if ! diff "$PROTOTYPE_SHIM_PATH" "$shim" >/dev/null 2>&1; then
|
|
rm -f "$SHIM_PATH"/*
|
|
fi
|
|
break
|
|
done
|
|
}
|
|
|
|
# List basenames of executables for every Ruby version
|
|
list_executable_names() {
|
|
local file
|
|
for file in "$RBENV_ROOT"/versions/*/bin/*; do
|
|
echo "${file##*/}"
|
|
done
|
|
}
|
|
|
|
# The basename of each argument passed to `make_shims` will be
|
|
# registered for installation as a shim. In this way, plugins may call
|
|
# `make_shims` with a glob to register many shims at once.
|
|
make_shims() {
|
|
local file shim
|
|
for file; do
|
|
shim="${file##*/}"
|
|
register_shim "$shim"
|
|
done
|
|
}
|
|
|
|
registered_shims=" "
|
|
|
|
# Registers the name of a shim to be generated.
|
|
register_shim() {
|
|
registered_shims="${registered_shims}${1} "
|
|
}
|
|
|
|
# Install all the shims registered via `make_shims` or `register_shim` directly.
|
|
install_registered_shims() {
|
|
local shim file
|
|
for shim in $registered_shims; do
|
|
file="${SHIM_PATH}/${shim}"
|
|
[ -e "$file" ] || ln -f "$PROTOTYPE_SHIM_PATH" "$file"
|
|
done
|
|
}
|
|
|
|
# Once the registered shims have been installed, we make a second pass
|
|
# over the contents of the shims directory. Any file that is present
|
|
# in the directory but has not been registered as a shim should be
|
|
# removed.
|
|
remove_stale_shims() {
|
|
local shim
|
|
for shim in "$SHIM_PATH"/*; do
|
|
if [[ "$registered_shims" != *" ${shim##*/} "* ]]; then
|
|
rm -f "$shim"
|
|
fi
|
|
done
|
|
}
|
|
|
|
shopt -s nullglob
|
|
|
|
# Create the prototype shim, then register shims for all known
|
|
# executables.
|
|
create_prototype_shim
|
|
remove_outdated_shims
|
|
make_shims $(list_executable_names | sort -u)
|
|
|
|
|
|
# Allow plugins to register shims.
|
|
OLDIFS="$IFS"
|
|
IFS=$'\n' scripts=(`rbenv-hooks rehash`)
|
|
IFS="$OLDIFS"
|
|
|
|
for script in "${scripts[@]}"; do
|
|
source "$script"
|
|
done
|
|
|
|
install_registered_shims
|
|
remove_stale_shims
|