Add autoenv command to manage autoenv's
usage: autoenv [-h] {init,edit,deinit}
options:
-h, --help show this help message and exit
commands:
init add .enter and .exit scripts in current directory
edit edit .enter and .exit scripts in current directory
deinit remove .enter and .exit scripts in current directory
This commit is contained in:
parent
40c2395f43
commit
9b434a1f85
24
autoenv/_autoenv
Normal file
24
autoenv/_autoenv
Normal file
@ -0,0 +1,24 @@
|
||||
#compdef autoenv
|
||||
|
||||
# Completion for the autoenv command.
|
||||
_autoenv() {
|
||||
local ret=1 context curcontext="$curcontext" state line
|
||||
typeset -A opt_args
|
||||
|
||||
_arguments -C -w -s \
|
||||
'(-h --help)'{-h,--help}'[show this help message and exit]' \
|
||||
'1: :->command'
|
||||
|
||||
case $state in
|
||||
(command)
|
||||
declare -a commands
|
||||
local commands=(
|
||||
init:'add .enter and .exit scripts in current directory'
|
||||
edit:'edit .enter and .exit scripts in current directory'
|
||||
deinit:'remove .enter and .exit scripts in current directory'
|
||||
)
|
||||
_describe -t commands command commands && ret=0 ;;
|
||||
esac
|
||||
|
||||
return ret
|
||||
}
|
||||
@ -1,108 +1,4 @@
|
||||
# Automatically update the environment when the current working directory
|
||||
# changes, this is a reimplementation of the ideas found in the repository
|
||||
# https://github.com/Tarrasch/zsh-autoenv stripped down to bare essentials.
|
||||
#
|
||||
# The secret sauce can be found at the bottom of this file, where the `chpwd`
|
||||
# hook function `_autoenv_chpwd` is added.
|
||||
|
||||
# Global entered directories array.
|
||||
_autoenv_entered=()
|
||||
|
||||
# Check if the given file is authorized, if not prompt the user to authorize,
|
||||
# ignore, or view the file. Authorized files and their modified times are
|
||||
# stored in the ~/.cache/autoenv/authorized file to make authorization
|
||||
# persistent.
|
||||
_autoenv_authorized() {
|
||||
local file=$1
|
||||
# If autoenv cache directory does not exist, create it.
|
||||
local cache_dir=~/.cache/autoenv
|
||||
! [ -d $cache_dir ] && mkdir -p $cache_dir
|
||||
local authorized_file=$cache_dir/authorized
|
||||
# Define a map to hold authorized key value pairs.
|
||||
typeset -A authorized=()
|
||||
# If the authorized file exists, load it into the map.
|
||||
[ -f $authorized_file ] && typeset -A authorized=(`cat $authorized_file`)
|
||||
# If the given file has been authorized, i.e. the modified time matches that
|
||||
# held in the authorized file, return.
|
||||
local modified_time=`zstat +mtime $file`
|
||||
[ "$authorized[$file]" = "$modified_time" ] && return
|
||||
# Prompt to authorize file.
|
||||
while true; do
|
||||
read "answer?Authorize $file [Y/n/v]? "
|
||||
case "$answer" in
|
||||
y|Y|yes|'') # Authorize the file.
|
||||
authorized[$file]=$modified_time; break ;;
|
||||
n|N|no) return 1 ;; # Do not authorize the file.
|
||||
v|V|view) cat $file ;; # View the file.
|
||||
esac
|
||||
done
|
||||
# Store authorized associative array in authorized file.
|
||||
echo ${(kv)authorized} > $authorized_file
|
||||
}
|
||||
|
||||
# Source an enter script and add its directory to the global entered
|
||||
# directories array.
|
||||
_autoenv_enter() {
|
||||
local entered=$1
|
||||
# If entered exists in the entered directories array, return.
|
||||
(( ${+_autoenv_entered[${_autoenv_entered[(i)$entered]}]} )) && return
|
||||
# If the enter script is not authorized, return.
|
||||
_autoenv_authorized $entered/.enter || return
|
||||
# Source the enter script.
|
||||
source $entered/.enter
|
||||
# Add the entered directory to the global entered array.
|
||||
_autoenv_entered+=$entered
|
||||
}
|
||||
|
||||
# Source an exit script and remove its directory from the global entered
|
||||
# directories array.
|
||||
_autoenv_exit() {
|
||||
local entered=$1
|
||||
# If the exit script is not authorized, return.
|
||||
_autoenv_authorized $entered/.exit || return
|
||||
# Source the exit script.
|
||||
source $entered/.exit
|
||||
# Remove the entered directory from the global entered array.
|
||||
_autoenv_entered[${_autoenv_entered[(i)$entered]}]=()
|
||||
}
|
||||
|
||||
# Find all directories containing a .enter file by searching up the directory
|
||||
# tree starting in the current directory.
|
||||
_autoenv_find_enter_directories() {
|
||||
local current=$PWD
|
||||
# If an enter script is found in the current directory, return it.
|
||||
[ -f $current/.enter ] && echo $current
|
||||
# Loop until an enter script or the root directory is found.
|
||||
while true; do
|
||||
# Go up one directory and make the path absolute.
|
||||
local next=$current/..; local next=${next:A}
|
||||
# If an enter script is found in the current directory, return it.
|
||||
[ -f $next/.enter ] && echo $next
|
||||
# If the current directory equals the next directory we are done, otherwise
|
||||
# update the current directory.
|
||||
[[ $current == $next ]] && return || local current=$next
|
||||
done
|
||||
}
|
||||
|
||||
# A chpwd hook function which automatically sources enter and exit scripts to
|
||||
# setup local environments for directory and its subdirectories.
|
||||
_autoenv_chpwd() {
|
||||
local entered
|
||||
# Loop over the reversed entered directory array.
|
||||
for entered in ${(aO)_autoenv_entered}; do
|
||||
# If the the current directory was previously entered then exit.
|
||||
! [[ $PWD/ == $entered/* ]] && _autoenv_exit $entered
|
||||
done
|
||||
# Find all enter script directories, store them in an array.
|
||||
local enter_dirs=(`_autoenv_find_enter_directories`)
|
||||
# Loop over reversed enter script directories array, so enter scripts found
|
||||
# last are sourced first, then source all enter scripts.
|
||||
for enter in ${(aO)enter_dirs}; do _autoenv_enter $enter; done
|
||||
}
|
||||
|
||||
# Register the autoenv chpwd hook.
|
||||
autoload -U add-zsh-hook
|
||||
add-zsh-hook chpwd _autoenv_chpwd
|
||||
|
||||
# Ensure autoenv is activated in the current directory on first load.
|
||||
_autoenv_chpwd
|
||||
# Source the autoenv implementation script.
|
||||
source ${0:a:h}/autoenv.zsh
|
||||
# Add the autoenv directory to fpath for completion.
|
||||
fpath+=${0:a:h}
|
||||
|
||||
179
autoenv/autoenv.zsh
Normal file
179
autoenv/autoenv.zsh
Normal file
@ -0,0 +1,179 @@
|
||||
# Automatically update the environment when the current working directory
|
||||
# changes, this is a reimplementation of the ideas found in the repository
|
||||
# https://github.com/Tarrasch/zsh-autoenv stripped down to bare essentials.
|
||||
#
|
||||
# The secret sauce can be found at the bottom of this file, where the chpwd
|
||||
# hook function _autoenv_chpwd is added.
|
||||
|
||||
# The autoenv command provides a convenient way to create, edit, and remove
|
||||
# enter and exit scripts in the current directory.
|
||||
autoenv() {
|
||||
local cmd=$1
|
||||
case "$cmd" in
|
||||
-h|--help) # Display help.
|
||||
echo "\
|
||||
usage: autoenv [-h] {init,edit,deinit}
|
||||
|
||||
options:
|
||||
-h, --help show this help message and exit
|
||||
|
||||
commands:
|
||||
init add .enter and .exit scripts in current directory
|
||||
edit edit .enter and .exit scripts in current directory
|
||||
deinit remove .enter and .exit scripts in current directory"
|
||||
;;
|
||||
|
||||
init) # Create enter and exit scripts in current directory.
|
||||
if [ -f $PWD/.enter ] || [ -f $PWD/.exit ]; then
|
||||
echo '.enter or .exit already exists'; return 1
|
||||
fi
|
||||
# Create then edit enter and exit scripts.
|
||||
touch .enter .exit && autoenv edit
|
||||
# If enter script exists, authorize it.
|
||||
[ -f $PWD/.enter ] && _autoenv_authorized $PWD/.enter yes
|
||||
# If exit script exists, authorize it.
|
||||
[ -f $PWD/.exit ] && _autoenv_authorized $PWD/.exit yes
|
||||
# Enter the new autoenv.
|
||||
_autoenv_enter $PWD ;;
|
||||
|
||||
edit) # Edit enter and exit scripts in current directory.
|
||||
if ! [ -f $PWD/.enter ] || ! [ -f $PWD/.exit ]; then
|
||||
echo '.enter or .exit not found'; return 1
|
||||
fi
|
||||
# If vim exists, edit enter and exit scripts.
|
||||
if which vim &> /dev/null; then
|
||||
vim -p .enter .exit
|
||||
else
|
||||
echo 'vim not found'; return 1
|
||||
fi ;;
|
||||
|
||||
deinit) # Remove enter and exit scripts in current directory.
|
||||
if [ -f $PWD/.enter ] || [ -f $PWD/.exit ]; then
|
||||
# Prompt user to confirm removal of enter and exit scripts.
|
||||
while true; do
|
||||
read "answer?Are you sure [y/N]? "
|
||||
case "$answer" in
|
||||
y|Y|yes)
|
||||
# Remove enter and exit scripts if they exist.
|
||||
[ -f $PWD/.enter ] && rm $PWD/.enter
|
||||
[ -f $PWD/.exit ] && rm $PWD/.exit
|
||||
break ;;
|
||||
*) break ;;
|
||||
esac
|
||||
done
|
||||
else
|
||||
echo '.enter and .exit not found'; return 1
|
||||
fi ;;
|
||||
|
||||
*) # Invalid arguments, show help then error.
|
||||
echo "invalid arguments: $@"
|
||||
autoenv --help
|
||||
return 1 ;;
|
||||
esac
|
||||
}
|
||||
|
||||
# Global entered directories array.
|
||||
_autoenv_entered=()
|
||||
|
||||
# Check if the given file is authorized, if not prompt the user to authorize,
|
||||
# ignore, or view the file. Authorized files and their modified times are
|
||||
# stored in the ~/.cache/autoenv/authorized file to make authorization
|
||||
# persistent.
|
||||
_autoenv_authorized() {
|
||||
local file=$1 yes=$2
|
||||
# If autoenv cache directory does not exist, create it.
|
||||
! [ -d ~/.cache/autoenv ] && mkdir -p ~/.cache/autoenv
|
||||
# If the authorized file does not exist, create it.
|
||||
! [ -f ~/.cache/autoenv/authorized ] && touch ~/.cache/autoenv/authorized
|
||||
# Load the authorized file into a map of authorized key value pairs.
|
||||
typeset -A authorized=(`cat ~/.cache/autoenv/authorized`)
|
||||
# If the file has been removed, return.
|
||||
! [ -f $file ] && return 1
|
||||
# If the given file has been authorized, i.e. the modified time matches that
|
||||
# held in the authorized file, return.
|
||||
local modified_time=`zstat +mtime $file`
|
||||
[ "$authorized[$file]" = "$modified_time" ] && return
|
||||
# If yes, don't prompt for user confirmation.
|
||||
if [ "$yes" != "yes" ]; then
|
||||
# Prompt to authorize file.
|
||||
while true; do
|
||||
read "answer?Authorize $file [Y/n/v]? "
|
||||
case "$answer" in
|
||||
y|Y|yes|'') break ;; # Authorize the file.
|
||||
n|N|no) return 1 ;; # Do not authorize the file.
|
||||
v|V|view) cat $file ;; # View the file.
|
||||
esac
|
||||
done
|
||||
fi
|
||||
# Add file to the authorized map.
|
||||
authorized[$file]=$modified_time
|
||||
# Store authorized map in authorized file.
|
||||
echo ${(kv)authorized} > ~/.cache/autoenv/authorized
|
||||
}
|
||||
|
||||
# Source an enter script and add its directory to the global entered
|
||||
# directories array.
|
||||
_autoenv_enter() {
|
||||
local entered=$1
|
||||
# If entered exists in the entered directories array, return.
|
||||
(( ${+_autoenv_entered[${_autoenv_entered[(i)$entered]}]} )) && return
|
||||
# If the enter script is not authorized, return.
|
||||
_autoenv_authorized $entered/.enter || return
|
||||
# Source the enter script.
|
||||
source $entered/.enter
|
||||
# Add the entered directory to the global entered array.
|
||||
_autoenv_entered+=$entered
|
||||
}
|
||||
|
||||
# Source an exit script and remove its directory from the global entered
|
||||
# directories array.
|
||||
_autoenv_exit() {
|
||||
local entered=$1
|
||||
# If the exit script is not authorized, return.
|
||||
_autoenv_authorized $entered/.exit || return
|
||||
# Source the exit script.
|
||||
source $entered/.exit
|
||||
# Remove the entered directory from the global entered array.
|
||||
_autoenv_entered[${_autoenv_entered[(i)$entered]}]=()
|
||||
}
|
||||
|
||||
# Find all directories containing a .enter file by searching up the directory
|
||||
# tree starting in the current directory.
|
||||
_autoenv_find_enter_directories() {
|
||||
local current=$PWD
|
||||
# If an enter script is found in the current directory, return it.
|
||||
[ -f $current/.enter ] && echo $current
|
||||
# Loop until an enter script or the root directory is found.
|
||||
while true; do
|
||||
# Go up one directory and make the path absolute.
|
||||
local next=$current/..; local next=${next:A}
|
||||
# If an enter script is found in the current directory, return it.
|
||||
[ -f $next/.enter ] && echo $next
|
||||
# If the current directory equals the next directory we are done, otherwise
|
||||
# update the current directory.
|
||||
[[ $current == $next ]] && return || local current=$next
|
||||
done
|
||||
}
|
||||
|
||||
# A chpwd hook function which automatically sources enter and exit scripts to
|
||||
# setup local environments for directory and its subdirectories.
|
||||
_autoenv_chpwd() {
|
||||
local entered
|
||||
# Loop over the reversed entered directory array.
|
||||
for entered in ${(aO)_autoenv_entered}; do
|
||||
# If the the current directory was previously entered then exit.
|
||||
! [[ $PWD/ == $entered/* ]] && _autoenv_exit $entered
|
||||
done
|
||||
# Find all enter script directories, store them in an array.
|
||||
local enter_dirs=(`_autoenv_find_enter_directories`)
|
||||
# Loop over reversed enter script directories array, so enter scripts found
|
||||
# last are sourced first, then source all enter scripts.
|
||||
for enter in ${(aO)enter_dirs}; do _autoenv_enter $enter; done
|
||||
}
|
||||
|
||||
# Register the autoenv chpwd hook.
|
||||
autoload -U add-zsh-hook
|
||||
add-zsh-hook chpwd _autoenv_chpwd
|
||||
|
||||
# Ensure autoenv is activated in the current directory on first load.
|
||||
_autoenv_chpwd
|
||||
Loading…
x
Reference in New Issue
Block a user