98 lines
2.4 KiB
Text
98 lines
2.4 KiB
Text
|
#! /usr/bin/env bash
|
||
|
|
#
|
||
|
|
# This script wraps pip to run `asdf reshim` after installs and uninstalls.
|
||
|
|
# Any other cases are passed-through to pip.
|
||
|
|
#
|
||
|
|
# Based on the npm shim: https://github.com/asdf-vm/asdf-nodejs/blob/b2d06a768d9a14186db72018df10604bdb384436/shims/npm
|
||
|
|
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
this_dir=$(dirname "${BASH_SOURCE[0]}")
|
||
|
|
this_dir=$(cd "$this_dir" && pwd -P) # Normalizes the directory; see https://stackoverflow.com/a/7666/2308068
|
||
|
|
plugin_name=$(basename "$(dirname "$this_dir")")
|
||
|
|
|
||
|
|
asdf_data_dir=$(cd "${ASDF_DATA_DIR:-$HOME/.asdf}" && pwd -P)
|
||
|
|
asdf_shims_dir="$asdf_data_dir/shims"
|
||
|
|
|
||
|
|
plugin_dir="$asdf_data_dir/plugins/$plugin_name"
|
||
|
|
|
||
|
|
should_reshim() {
|
||
|
|
if [ "${ASDF_PYTHON_SKIP_RESHIM:-}" ]; then
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
for arg; do
|
||
|
|
case "$arg" in
|
||
|
|
install|uninstall)
|
||
|
|
return 0
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
done
|
||
|
|
|
||
|
|
return 1
|
||
|
|
}
|
||
|
|
|
||
|
|
resolve_pip() {
|
||
|
|
# Try searching in current path (asdf core from 0.7 onwards adds all binaries candidates directories to PATH)
|
||
|
|
# if that doesn't works (when calling the shim directly for example) it tries manually searching the binary in
|
||
|
|
# the installed version provided by "asdf where"
|
||
|
|
local pip_location="${ASDF_NODEJS_CANON_PIP_PATH:-$(search_pip_on_current_path || manually_search_pip_bin)}"
|
||
|
|
|
||
|
|
if ! [ "$pip_location" ]; then
|
||
|
|
echo "asdf-python couldn't find a suitable pip executable"
|
||
|
|
echo "This is probably a problem with the plugin, please report this issue"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "$pip_location"
|
||
|
|
}
|
||
|
|
|
||
|
|
remove_current_dir_from_path() {
|
||
|
|
local filtered_path= dir= normalized_dir=
|
||
|
|
|
||
|
|
while read -rd : dir; do
|
||
|
|
if [ -d "$dir" ]; then
|
||
|
|
normalized_dir=$(cd "$dir" && pwd -P)
|
||
|
|
|
||
|
|
if [ "$normalized_dir" = "$this_dir" ] || [ "$normalized_dir" = "$asdf_shims_dir" ]; then
|
||
|
|
continue
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
filtered_path+="$dir:"
|
||
|
|
done <<< "$PATH:"
|
||
|
|
|
||
|
|
echo "${filtered_path%:}"
|
||
|
|
}
|
||
|
|
|
||
|
|
search_pip_on_current_path() {
|
||
|
|
# Tries to prevent recursion by removing the current script and asdf-shim from PATH
|
||
|
|
local filtered_path=$(remove_current_dir_from_path)
|
||
|
|
PATH="$filtered_path" command -v pip
|
||
|
|
}
|
||
|
|
|
||
|
|
manually_search_pip_bin() {
|
||
|
|
local probably_pip="$(asdf where python)/bin/pip"
|
||
|
|
|
||
|
|
if [ -x "$probably_pip" ]; then
|
||
|
|
echo "$probably_pip"
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
return 1
|
||
|
|
}
|
||
|
|
|
||
|
|
wrap_pip() {
|
||
|
|
local pip=$(resolve_pip)
|
||
|
|
|
||
|
|
if should_reshim "$@"; then
|
||
|
|
"$pip" "$@"
|
||
|
|
echo "Reshimming asdf $plugin_name..."
|
||
|
|
asdf reshim "$plugin_name"
|
||
|
|
else
|
||
|
|
exec "$pip" "$@"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
wrap_pip "$@"
|