mirror of
https://github.com/Tarrasch/zsh-autoenv.git
synced 2024-11-05 01:10:59 +02:00
147 lines
3.5 KiB
Bash
147 lines
3.5 KiB
Bash
# Initially based on
|
|
# https://github.com/joshuaclayton/dotfiles/blob/master/zsh_profile.d/autoenv.zsh
|
|
|
|
# TODO: move this to DOTENV_*?!
|
|
export ENV_AUTHORIZATION_FILE=$HOME/.env_auth
|
|
|
|
# Name of file to look for when entering directories.
|
|
: ${DOTENV_FILE_ENTER:=.env}
|
|
|
|
# Name of file to look for when leaving directories.
|
|
# Requires DOTENV_HANDLE_LEAVE=1.
|
|
: ${DOTENV_FILE_LEAVE:=.env.leave}
|
|
|
|
# Look for .env in parent dirs?
|
|
: ${DOTENV_LOOK_UPWARDS:=1}
|
|
|
|
# Handle leave events when changing away from a subtree, where an "enter"
|
|
# event was handled?
|
|
: ${DOTENV_HANDLE_LEAVE:=1}
|
|
|
|
|
|
# Internal: stack of entered (and handled) directories.
|
|
_dotenv_stack_entered=()
|
|
|
|
|
|
_dotenv_hash_pair() {
|
|
local env_file=$1
|
|
echo "$env_file:$env_shasum:1"
|
|
}
|
|
|
|
_dotenv_authorized_env_file() {
|
|
local env_file=$1
|
|
local pair=$(_dotenv_hash_pair $env_file)
|
|
test -f $ENV_AUTHORIZATION_FILE \
|
|
&& \grep -qF $pair $ENV_AUTHORIZATION_FILE
|
|
}
|
|
|
|
_dotenv_authorize() {
|
|
local env_file=$1
|
|
_dotenv_deauthorize $env_file
|
|
_dotenv_hash_pair $env_file >> $ENV_AUTHORIZATION_FILE
|
|
}
|
|
|
|
_dotenv_deauthorize() {
|
|
local env_file=$1
|
|
if [[ -f $ENV_AUTHORIZATION_FILE ]]; then
|
|
echo $(\grep -vF $env_file $ENV_AUTHORIZATION_FILE) > $ENV_AUTHORIZATION_FILE
|
|
fi
|
|
}
|
|
|
|
# This function can be mocked in tests
|
|
_dotenv_read_answer() {
|
|
local answer
|
|
read $=_AUTOENV_TEST_READ_ARGS -q answer
|
|
echo $answer
|
|
}
|
|
|
|
# Args: 1: absolute path to env file (resolved symlinks).
|
|
_dotenv_check_authorized_env_file() {
|
|
if ! [[ -f $1 ]]; then
|
|
return 1
|
|
fi
|
|
if ! _dotenv_authorized_env_file $1; then
|
|
echo "Attempting to load unauthorized env file: $1"
|
|
echo ""
|
|
echo "**********************************************"
|
|
echo ""
|
|
cat $1
|
|
echo ""
|
|
echo "**********************************************"
|
|
echo ""
|
|
echo -n "Would you like to authorize it? [y/N] "
|
|
|
|
local answer=$(_dotenv_read_answer)
|
|
echo
|
|
if [[ $answer != 'y' ]]; then
|
|
return 1
|
|
fi
|
|
|
|
_dotenv_authorize $1
|
|
fi
|
|
return 0
|
|
}
|
|
|
|
_dotenv_source() {
|
|
local env_file=$1
|
|
_dotenv_event=$2
|
|
_dotenv_cwd=$PWD
|
|
|
|
builtin cd -q ${env_file:h}
|
|
source $env_file
|
|
builtin cd -q $_dotenv_cwd
|
|
|
|
unset _dotenv_event _dotenv_cwd
|
|
}
|
|
|
|
_dotenv_chpwd_handler() {
|
|
local env_file="$PWD/$DOTENV_FILE_ENTER"
|
|
|
|
# Handle leave event for previously sourced env files.
|
|
if [[ $DOTENV_HANDLE_LEAVE == 1 ]] && (( $#_dotenv_stack_entered )); then
|
|
for prev_dir in ${_dotenv_stack_entered}; do
|
|
if ! [[ ${PWD}/ == ${prev_dir}/* ]]; then
|
|
local env_file_leave=$prev_dir/$DOTENV_FILE_LEAVE
|
|
if _dotenv_check_authorized_env_file $env_file_leave; then
|
|
_dotenv_source $env_file_leave leave
|
|
fi
|
|
# Remove this entry from the stack.
|
|
_dotenv_stack_entered=(${_dotenv_stack_entered#$prev_dir})
|
|
fi
|
|
done
|
|
fi
|
|
|
|
if ! [[ -f $env_file ]] && [[ $DOTENV_LOOK_UPWARDS == 1 ]]; then
|
|
# Look for files in parent dirs, using an extended Zsh glob.
|
|
setopt localoptions extendedglob
|
|
local m
|
|
m=((../)#${DOTENV_FILE_ENTER}(N))
|
|
if (( $#m )); then
|
|
env_file=${${m[1]}:A}
|
|
else
|
|
return
|
|
fi
|
|
fi
|
|
|
|
if ! _dotenv_check_authorized_env_file $env_file; then
|
|
return
|
|
fi
|
|
|
|
# Load the env file only once: check if $env_file's parent
|
|
# is in $_dotenv_stack_entered.
|
|
local env_file_dir=${env_file:A:h}
|
|
if (( ${+_dotenv_stack_entered[(r)${env_file_dir}]} )); then
|
|
return
|
|
fi
|
|
|
|
_dotenv_stack_entered+=(${env_file_dir})
|
|
|
|
_dotenv_source $env_file enter
|
|
}
|
|
|
|
autoload -U add-zsh-hook
|
|
add-zsh-hook chpwd _dotenv_chpwd_handler
|
|
|
|
# Look in current directory already.
|
|
_dotenv_chpwd_handler
|