Customization/ohmyzsh
1
0
Fork 0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2024-12-22 12:21:58 +00:00
Code Releases Activity

fix(dotenv): draw confirmation prompt in next empty line

Browse source 
Without this fix the confirmation prompt appears wherever the cursor is,
which means that it might appear in the command line when using a widget
that changes the directory without redrawing the prompt (an example of
this are the dircycle and dirhistory plugins).
This commit is contained in:
Marc Cornellà 2021-06-28 15:54:24 +02:00
parent 0e7d7b87f3
commit c44b99e901
No known key found for this signature in database
GPG key ID: 0314585E776A9C1B
1 changed files with 38 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

66
plugins/dotenv/dotenv.plugin.zsh
View file

@ -11,41 +11,51 @@
## Functions ## Functions
source_env() { source_env() {
if [[ -f $ZSH_DOTENV_FILE ]]; then if [[ ! -f "$ZSH_DOTENV_FILE" ]]; then
if [[ "$ZSH_DOTENV_PROMPT" != false ]]; then return
local confirmation dirpath="${PWD:A}" fi
# make sure there is an (dis-)allowed file if [[ "$ZSH_DOTENV_PROMPT" != false ]]; then
touch "$ZSH_DOTENV_ALLOWED_LIST" local confirmation dirpath="${PWD:A}"
touch "$ZSH_DOTENV_DISALLOWED_LIST"
# early return if disallowed # make sure there is an (dis-)allowed file
if grep -q "$dirpath" "$ZSH_DOTENV_DISALLOWED_LIST" &>/dev/null; then touch "$ZSH_DOTENV_ALLOWED_LIST"
return; touch "$ZSH_DOTENV_DISALLOWED_LIST"
fi
# check if current directory's .env file is allowed or ask for confirmation # early return if disallowed
if ! grep -q "$dirpath" "$ZSH_DOTENV_ALLOWED_LIST" &>/dev/null; then if command grep -q "$dirpath" "$ZSH_DOTENV_DISALLOWED_LIST" &>/dev/null; then
# print same-line prompt and output newline character if necessary return
echo -n "dotenv: found '$ZSH_DOTENV_FILE' file. Source it? ([Y]es/[n]o/[a]lways/n[e]ver) "
read -k 1 confirmation; [[ "$confirmation" != $'\n' ]] && echo
# check input
case "$confirmation" in
[nN]) return ;;
[aA]) echo "$dirpath" >> "$ZSH_DOTENV_ALLOWED_LIST" ;;
[eE]) echo "$dirpath" >> "$ZSH_DOTENV_DISALLOWED_LIST"; return ;;
*) ;; # interpret anything else as a yes
esac
fi
fi fi
# test .env syntax # check if current directory's .env file is allowed or ask for confirmation
zsh -fn $ZSH_DOTENV_FILE || echo "dotenv: error when sourcing '$ZSH_DOTENV_FILE' file" >&2 if ! command grep -q "$dirpath" "$ZSH_DOTENV_ALLOWED_LIST" &>/dev/null; then
# get cursor column and print new line before prompt if not at line beginning
local column
echo -ne "\e[6n" > /dev/tty
read -t 1 -s -d R column < /dev/tty
column="${column##*\[*;}"
[[ $column -eq 1 ]] || echo
setopt localoptions allexport # print same-line prompt and output newline character if necessary
source $ZSH_DOTENV_FILE echo -n "dotenv: found '$ZSH_DOTENV_FILE' file. Source it? ([Y]es/[n]o/[a]lways/n[e]ver) "
read -k 1 confirmation
[[ "$confirmation" = $'\n' ]] || echo
# check input
case "$confirmation" in
[nN]) return ;;
[aA]) echo "$dirpath" >> "$ZSH_DOTENV_ALLOWED_LIST" ;;
[eE]) echo "$dirpath" >> "$ZSH_DOTENV_DISALLOWED_LIST"; return ;;
*) ;; # interpret anything else as a yes
esac
fi
fi fi
# test .env syntax
zsh -fn $ZSH_DOTENV_FILE || echo "dotenv: error when sourcing '$ZSH_DOTENV_FILE' file" >&2
setopt localoptions allexport
source $ZSH_DOTENV_FILE
} }
autoload -U add-zsh-hook autoload -U add-zsh-hook