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
2022-04-03 18:08:49 -04:00
print_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 )
2022-04-03 18:08:49 -04:00
if [ -e '~/.bash_profile' ]; then
profile='~/.bash_profile'
else
profile='~/.profile'
fi
profile_explain="~/.bash_profile if it exists, otherwise ~/.profile"
2021-05-13 15:57:14 -04:00
rc='~/.bashrc'
2012-08-31 02:23:41 -04:00
;;
zsh )
2021-05-13 15:57:14 -04:00
profile='~/.zprofile'
rc='~/.zshrc'
2012-08-31 02:23:41 -04:00
;;
ksh )
profile='~/.profile'
2021-05-13 15:57:14 -04:00
rc='~/.profile'
2012-08-31 02:23:41 -04:00
;;
* )
2021-05-13 15:57:14 -04:00
profile='your shell'\''s login startup file'
rc='your shell'\''s interactive startup file'
2012-08-31 02:23:41 -04:00
;;
esac
2021-05-11 15:35:15 -04:00
{
2022-04-03 18:08:49 -04:00
case "$shell" in
fish )
echo "# Add pyenv executable to PATH by running"
echo "# the following interactively:"
echo
echo 'set -Ux PYENV_ROOT $HOME/.pyenv'
echo 'set -U fish_user_paths $PYENV_ROOT/bin $fish_user_paths'
echo
echo "# Load pyenv automatically by appending"
echo "# the following to ~/.config/fish/config.fish:"
echo
echo 'pyenv init - | source'
echo
;;
* )
echo '# Load pyenv automatically by appending'
echo -n "# the following to "
if [ "$profile" == "$rc" ]; then
echo "$profile :"
else
echo
echo "${profile_explain:-$profile} (for login shells)"
echo "and $rc (for interactive shells) :"
fi
echo
echo 'export PYENV_ROOT="$HOME/.pyenv"'
echo 'command -v pyenv >/dev/null || export PATH="$PYENV_ROOT/bin:$PATH"'
echo 'eval "$(pyenv init -)"'
;;
esac
echo
echo '# Restart your shell for the changes to take effect.'
2021-05-13 15:57:14 -04:00
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() {
case "$shell" in
fish )
2022-04-03 18:08:49 -04:00
echo 'while set index (contains -i -- '\'"${PYENV_ROOT}/shims"\'' $PATH)'
echo 'set -eg PATH[$index]; end; set -e index'
echo 'set -gx PATH '\'"${PYENV_ROOT}/shims"\'' $PATH'
2021-05-03 21:30:52 -04:00
;;
* )
2022-04-03 18:08:49 -04:00
echo 'PATH="$(bash -ec '\''IFS=:; paths=($PATH); for i in ${!paths[@]}; do if [[ ${paths[i]} == "'\'"${PYENV_ROOT}/shims"\''" ]]; then unset '\'\\\'\''paths[i]'\'\\\'\''; fi; done; echo "${paths[*]}"'\'')"'
echo 'export PATH="'"${PYENV_ROOT}"'/shims:${PATH}"'
2021-05-03 21:30:52 -04:00
;;
esac
}
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