2019-04-23 10:23:33 -04:00
|
|
|
#!/usr/bin/env bash
|
2013-01-18 03:41:41 -05:00
|
|
|
# Summary: Configure the shell environment for pyenv
|
2021-05-03 21:30:52 -04:00
|
|
|
# Usage: eval "$(pyenv init [-|--path] [--no-rehash] [<shell>])"
|
2013-01-18 03:41:41 -05:00
|
|
|
|
2012-08-31 02:23:41 -04:00
|
|
|
set -e
|
|
|
|
[ -n "$PYENV_DEBUG" ] && set -x
|
|
|
|
|
2015-11-20 23:13:52 -05:00
|
|
|
# Provide pyenv completions
|
2015-11-20 09:20:01 -05:00
|
|
|
if [ "$1" = "--complete" ]; then
|
|
|
|
echo -
|
2021-05-03 21:30:52 -04:00
|
|
|
echo --path
|
2015-11-20 09:20:01 -05:00
|
|
|
echo --no-rehash
|
|
|
|
echo bash
|
|
|
|
echo fish
|
|
|
|
echo ksh
|
|
|
|
echo zsh
|
|
|
|
exit
|
|
|
|
fi
|
|
|
|
|
2021-05-03 21:30:52 -04:00
|
|
|
mode="help"
|
2012-08-31 02:23:41 -04:00
|
|
|
no_rehash=""
|
|
|
|
for args in "$@"
|
|
|
|
do
|
|
|
|
if [ "$args" = "-" ]; then
|
2021-05-03 21:30:52 -04:00
|
|
|
mode="print"
|
2013-09-30 05:02:12 -04:00
|
|
|
shift
|
2012-08-31 02:23:41 -04:00
|
|
|
fi
|
|
|
|
|
2021-05-03 21:30:52 -04:00
|
|
|
if [ "$args" = "--path" ]; then
|
|
|
|
mode="path"
|
|
|
|
shift
|
|
|
|
fi
|
|
|
|
|
2012-08-31 02:23:41 -04:00
|
|
|
if [ "$args" = "--no-rehash" ]; then
|
|
|
|
no_rehash=1
|
|
|
|
shift
|
|
|
|
fi
|
|
|
|
done
|
|
|
|
|
|
|
|
shell="$1"
|
|
|
|
if [ -z "$shell" ]; then
|
2015-05-10 10:03:06 -04:00
|
|
|
shell="$(ps -p "$PPID" -o 'args=' 2>/dev/null || true)"
|
2014-01-02 14:58:29 -05:00
|
|
|
shell="${shell%% *}"
|
2015-12-24 07:23:01 -05:00
|
|
|
shell="${shell##-}"
|
2015-05-10 10:17:35 -04:00
|
|
|
shell="${shell:-$SHELL}"
|
|
|
|
shell="${shell##*/}"
|
2021-03-01 00:49:47 -05:00
|
|
|
shell="${shell%%-*}"
|
2012-08-31 02:23:41 -04:00
|
|
|
fi
|
|
|
|
|
2014-11-30 10:20:53 -05:00
|
|
|
root="${0%/*}/.."
|
2012-08-31 02:23:41 -04:00
|
|
|
|
2021-05-03 21:12:17 -04:00
|
|
|
function main() {
|
|
|
|
case "$mode" in
|
|
|
|
"help")
|
|
|
|
help_
|
|
|
|
exit 1
|
|
|
|
;;
|
|
|
|
"path")
|
|
|
|
print_path
|
|
|
|
exit 0
|
|
|
|
;;
|
|
|
|
"print")
|
|
|
|
init_dirs
|
2021-05-04 09:59:56 -04:00
|
|
|
warn_path
|
2021-05-03 21:12:17 -04:00
|
|
|
print_env
|
|
|
|
print_completion
|
|
|
|
print_shell_function
|
|
|
|
exit 0
|
|
|
|
;;
|
|
|
|
esac
|
|
|
|
# should never get here
|
|
|
|
exit 2
|
|
|
|
}
|
|
|
|
|
|
|
|
function help_() {
|
2012-08-31 02:23:41 -04:00
|
|
|
case "$shell" in
|
|
|
|
bash )
|
2021-05-03 21:30:52 -04:00
|
|
|
profile='~/.bashrc'
|
2012-08-31 02:23:41 -04:00
|
|
|
;;
|
|
|
|
zsh )
|
|
|
|
profile='~/.zshrc'
|
|
|
|
;;
|
|
|
|
ksh )
|
|
|
|
profile='~/.profile'
|
|
|
|
;;
|
2013-08-15 08:27:52 -04:00
|
|
|
fish )
|
|
|
|
profile='~/.config/fish/config.fish'
|
|
|
|
;;
|
2012-08-31 02:23:41 -04:00
|
|
|
* )
|
|
|
|
profile='your profile'
|
|
|
|
;;
|
|
|
|
esac
|
|
|
|
|
2015-11-20 23:13:52 -05:00
|
|
|
{ echo "# Load pyenv automatically by appending"
|
2011-08-04 01:45:40 -04:00
|
|
|
echo "# the following to ${profile}:"
|
2012-08-31 02:23:41 -04:00
|
|
|
echo
|
2013-09-30 05:02:12 -04:00
|
|
|
case "$shell" in
|
|
|
|
fish )
|
2020-10-01 21:28:29 -04:00
|
|
|
echo 'pyenv init - | source'
|
2013-09-30 05:02:12 -04:00
|
|
|
;;
|
|
|
|
* )
|
|
|
|
echo 'eval "$(pyenv init -)"'
|
|
|
|
;;
|
|
|
|
esac
|
2012-08-31 02:23:41 -04:00
|
|
|
echo
|
2021-05-03 21:30:52 -04:00
|
|
|
echo "# And the following to ~/.profile:"
|
|
|
|
echo
|
|
|
|
case "$shell" in
|
|
|
|
fish )
|
|
|
|
echo 'pyenv init --path | source'
|
|
|
|
;;
|
|
|
|
* )
|
|
|
|
echo 'eval "$(pyenv init --path)"'
|
|
|
|
;;
|
|
|
|
esac
|
|
|
|
echo
|
2021-05-07 09:21:17 -04:00
|
|
|
echo '# Make sure to restart your entire logon session'
|
|
|
|
echo '# for changes to ~/.profile to take effect.'
|
|
|
|
echo
|
2012-08-31 02:23:41 -04:00
|
|
|
} >&2
|
2021-05-03 21:12:17 -04:00
|
|
|
}
|
2012-08-31 02:23:41 -04:00
|
|
|
|
2021-05-03 21:12:17 -04:00
|
|
|
function init_dirs() {
|
|
|
|
mkdir -p "${PYENV_ROOT}/"{shims,versions}
|
|
|
|
}
|
2012-08-31 02:23:41 -04:00
|
|
|
|
2021-05-03 21:30:52 -04:00
|
|
|
function print_path() {
|
|
|
|
# Need to use the login shell rather than the current one
|
|
|
|
case "$shell" in
|
|
|
|
fish )
|
|
|
|
echo "set -gx PATH '${PYENV_ROOT}/shims' \$PATH"
|
|
|
|
;;
|
|
|
|
* )
|
|
|
|
echo 'export PATH="'${PYENV_ROOT}'/shims:${PATH}"'
|
|
|
|
;;
|
|
|
|
esac
|
|
|
|
}
|
|
|
|
|
2021-05-04 09:59:56 -04:00
|
|
|
function warn_path() {
|
|
|
|
if ! perl -ls0x3A -e 'while (<>) { chomp; ($_ eq $d) && exit 0; } exit 1' -- -d="${PYENV_ROOT}/shims" <<<"$PATH" ; then
|
|
|
|
echo 'echo '\''WARNING: `pyenv init -` no longer sets PATH.'\'
|
|
|
|
echo 'echo '\''Run `pyenv init` to see the necessary changes to make to your configuration.'\'
|
|
|
|
fi
|
|
|
|
}
|
|
|
|
|
2021-05-03 21:12:17 -04:00
|
|
|
function print_env() {
|
|
|
|
case "$shell" in
|
|
|
|
fish )
|
|
|
|
echo "set -gx PYENV_SHELL $shell"
|
|
|
|
;;
|
|
|
|
* )
|
|
|
|
echo "export PYENV_SHELL=$shell"
|
|
|
|
;;
|
|
|
|
esac
|
|
|
|
}
|
2012-08-31 02:23:41 -04:00
|
|
|
|
2021-05-03 21:12:17 -04:00
|
|
|
function print_completion() {
|
|
|
|
completion="${root}/completions/pyenv.${shell}"
|
|
|
|
if [ -r "$completion" ]; then
|
|
|
|
echo "source '$completion'"
|
|
|
|
fi
|
2012-08-31 02:23:41 -04:00
|
|
|
|
2021-05-03 21:12:17 -04:00
|
|
|
if [ -z "$no_rehash" ]; then
|
|
|
|
echo 'command pyenv rehash 2>/dev/null'
|
|
|
|
fi
|
|
|
|
}
|
|
|
|
|
|
|
|
function print_shell_function() {
|
|
|
|
commands=(`pyenv-commands --sh`)
|
|
|
|
case "$shell" in
|
|
|
|
fish )
|
|
|
|
cat <<EOS
|
2013-09-30 05:02:12 -04:00
|
|
|
function pyenv
|
|
|
|
set command \$argv[1]
|
|
|
|
set -e argv[1]
|
|
|
|
|
|
|
|
switch "\$command"
|
|
|
|
case ${commands[*]}
|
2017-06-05 09:02:24 -04:00
|
|
|
source (pyenv "sh-\$command" \$argv|psub)
|
2013-09-30 05:02:12 -04:00
|
|
|
case '*'
|
|
|
|
command pyenv "\$command" \$argv
|
|
|
|
end
|
|
|
|
end
|
|
|
|
EOS
|
2021-05-03 21:12:17 -04:00
|
|
|
;;
|
|
|
|
ksh )
|
|
|
|
cat <<EOS
|
2013-09-30 05:02:12 -04:00
|
|
|
function pyenv {
|
|
|
|
typeset command
|
2013-08-15 08:27:52 -04:00
|
|
|
EOS
|
2021-05-03 21:12:17 -04:00
|
|
|
;;
|
|
|
|
* )
|
|
|
|
cat <<EOS
|
2012-08-31 02:23:41 -04:00
|
|
|
pyenv() {
|
2013-09-30 05:02:12 -04:00
|
|
|
local command
|
|
|
|
EOS
|
2021-05-03 21:12:17 -04:00
|
|
|
;;
|
|
|
|
esac
|
2013-09-30 05:02:12 -04:00
|
|
|
|
2021-05-03 21:12:17 -04:00
|
|
|
if [ "$shell" != "fish" ]; then
|
|
|
|
IFS="|"
|
|
|
|
cat <<EOS
|
2017-01-02 01:06:39 -05:00
|
|
|
command="\${1:-}"
|
2012-08-31 02:23:41 -04:00
|
|
|
if [ "\$#" -gt 0 ]; then
|
|
|
|
shift
|
|
|
|
fi
|
|
|
|
|
|
|
|
case "\$command" in
|
|
|
|
${commands[*]})
|
2021-05-03 21:12:17 -04:00
|
|
|
eval "\$(pyenv "sh-\$command" "\$@")"
|
|
|
|
;;
|
2012-08-31 02:23:41 -04:00
|
|
|
*)
|
2021-05-03 21:12:17 -04:00
|
|
|
command pyenv "\$command" "\$@"
|
|
|
|
;;
|
2012-08-31 02:23:41 -04:00
|
|
|
esac
|
|
|
|
}
|
|
|
|
EOS
|
2021-05-03 21:12:17 -04:00
|
|
|
fi
|
|
|
|
}
|
|
|
|
|
|
|
|
main
|