mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2026-01-30 02:44:42 +01:00
acb9b5a460
The return value of the 'wd' command is always '0' by default so it cannot be processed by another program. The return value is given as degguage mode associated with an exit which does not allow it to be used with '$?'.
584 lines
14 KiB
Bash
Executable file
584 lines
14 KiB
Bash
Executable file
#!/bin/zsh
|
|
|
|
# WARP DIRECTORY
|
|
# ==============
|
|
# Jump to custom directories in terminal
|
|
# because `cd` takes too long...
|
|
#
|
|
# @github.com/mfaerevaag/wd
|
|
|
|
# version
|
|
readonly WD_VERSION=0.7.0
|
|
|
|
# colors
|
|
readonly WD_BLUE="\033[96m"
|
|
readonly WD_GREEN="\033[92m"
|
|
readonly WD_YELLOW="\033[93m"
|
|
readonly WD_RED="\033[91m"
|
|
readonly WD_NOC="\033[m"
|
|
|
|
## functions
|
|
|
|
# helpers
|
|
wd_yesorno()
|
|
{
|
|
# variables
|
|
local question="${1}"
|
|
local prompt="${question} "
|
|
local yes_RETVAL="0"
|
|
local no_RETVAL="3"
|
|
local RETVAL=""
|
|
local answer=""
|
|
|
|
# read-eval loop
|
|
while true ; do
|
|
printf $prompt
|
|
read -r answer
|
|
|
|
case ${answer:=${default}} in
|
|
"Y"|"y"|"YES"|"yes"|"Yes" )
|
|
RETVAL=${yes_RETVAL} && \
|
|
break
|
|
;;
|
|
"N"|"n"|"NO"|"no"|"No" )
|
|
RETVAL=${no_RETVAL} && \
|
|
break
|
|
;;
|
|
* )
|
|
echo "Please provide a valid answer (y or n)"
|
|
;;
|
|
esac
|
|
done
|
|
|
|
return ${RETVAL}
|
|
}
|
|
|
|
wd_print_msg()
|
|
{
|
|
if [[ -z $wd_quiet_mode ]]
|
|
then
|
|
local color="${1:-$WD_BLUE}" # Default to blue if no color is provided
|
|
local msg="$2"
|
|
|
|
if [[ -z "$msg" ]]; then
|
|
print "${WD_RED}*${WD_NOC} Could not print message. Sorry!"
|
|
else
|
|
print " ${color}*${WD_NOC} ${msg}"
|
|
fi
|
|
fi
|
|
}
|
|
|
|
wd_print_usage()
|
|
{
|
|
command cat <<- EOF
|
|
Usage: wd [command] [point]
|
|
|
|
Commands:
|
|
<point> Warps to the directory specified by the warp point
|
|
<point> <path> Warps to the directory specified by the warp point with path appended
|
|
add <point> Adds the current working directory to your warp points
|
|
add Adds the current working directory to your warp points with current directory's name
|
|
addcd <path> Adds a path to your warp points with the directory's name
|
|
addcd <path> <point> Adds a path to your warp points with a custom name
|
|
rm <point> Removes the given warp point
|
|
rm Removes the given warp point with current directory's name
|
|
show <point> Print path to given warp point
|
|
show Print warp points to current directory
|
|
list Print all stored warp points
|
|
ls <point> Show files from given warp point (ls)
|
|
path <point> Show the path to given warp point (pwd)
|
|
clean Remove points warping to nonexistent directories (will prompt unless --force is used)
|
|
|
|
-v | --version Print version
|
|
-c | --config Specify config file (default ~/.warprc)
|
|
-q | --quiet Suppress all output
|
|
-f | --force Allows overwriting without warning (for add & clean)
|
|
|
|
help Show this extremely helpful text
|
|
EOF
|
|
}
|
|
|
|
wd_exit_fail()
|
|
{
|
|
local msg=$1
|
|
|
|
wd_print_msg "$WD_RED" "$msg"
|
|
WD_EXIT_CODE=1
|
|
}
|
|
|
|
wd_exit_warn()
|
|
{
|
|
local msg=$1
|
|
|
|
wd_print_msg "$WD_YELLOW" "$msg"
|
|
WD_EXIT_CODE=1
|
|
}
|
|
|
|
wd_getdir()
|
|
{
|
|
local name_arg=$1
|
|
|
|
point=$(wd_show "$name_arg")
|
|
dir=${point:28+$#name_arg+7}
|
|
|
|
if [[ -z $name_arg ]]; then
|
|
wd_exit_fail "You must enter a warp point"
|
|
break
|
|
elif [[ -z $dir ]]; then
|
|
wd_exit_fail "Unknown warp point '${name_arg}'"
|
|
break
|
|
fi
|
|
}
|
|
|
|
# core
|
|
|
|
wd_warp()
|
|
{
|
|
local point=$1
|
|
local sub=$2
|
|
|
|
if [[ $point =~ "^\.+$" ]]
|
|
then
|
|
if [[ $#1 < 2 ]]
|
|
then
|
|
wd_exit_warn "Warping to current directory?"
|
|
else
|
|
(( n = $#1 - 1 ))
|
|
cd -$n > /dev/null
|
|
fi
|
|
elif [[ ${points[$point]} != "" ]]
|
|
then
|
|
if [[ $sub != "" ]]
|
|
then
|
|
cd ${points[$point]/#\~/$HOME}/$sub
|
|
else
|
|
cd ${points[$point]/#\~/$HOME}
|
|
fi
|
|
else
|
|
wd_exit_fail "Unknown warp point '${point}'"
|
|
fi
|
|
}
|
|
|
|
wd_add()
|
|
{
|
|
local point=$1
|
|
local force=$2
|
|
cmdnames=(add rm show list ls path clean help)
|
|
|
|
if [[ $point == "" ]]
|
|
then
|
|
point=$(basename "$PWD")
|
|
fi
|
|
|
|
if [[ $point =~ "^[\.]+$" ]]
|
|
then
|
|
wd_exit_fail "Warp point cannot be just dots"
|
|
elif [[ $point =~ "[[:space:]]+" ]]
|
|
then
|
|
wd_exit_fail "Warp point should not contain whitespace"
|
|
elif [[ $point =~ : ]] || [[ $point =~ / ]]
|
|
then
|
|
wd_exit_fail "Warp point contains illegal character (:/)"
|
|
elif (($cmdnames[(Ie)$point]))
|
|
then
|
|
wd_exit_fail "Warp point name cannot be a wd command (see wd -h for a full list)"
|
|
elif [[ ${points[$point]} == "" ]] || [ ! -z "$force" ]
|
|
then
|
|
wd_remove "$point" > /dev/null
|
|
printf "%q:%s\n" "${point}" "${PWD/#$HOME/~}" >> "$WD_CONFIG"
|
|
if (whence sort >/dev/null); then
|
|
local config_tmp=$(mktemp "${TMPDIR:-/tmp}/wd.XXXXXXXXXX")
|
|
# use 'cat' below to ensure we respect $WD_CONFIG as a symlink
|
|
command sort -o "${config_tmp}" "$WD_CONFIG" && command cat "${config_tmp}" >| "$WD_CONFIG" && command rm "${config_tmp}"
|
|
fi
|
|
|
|
wd_export_static_named_directories
|
|
|
|
wd_print_msg "$WD_GREEN" "Warp point added"
|
|
|
|
# override exit code in case wd_remove did not remove any points
|
|
# TODO: we should handle this kind of logic better
|
|
WD_EXIT_CODE=0
|
|
else
|
|
wd_exit_warn "Warp point '${point}' already exists. Use 'add --force' to overwrite."
|
|
fi
|
|
}
|
|
|
|
wd_addcd() {
|
|
local folder="$1"
|
|
local point=$2
|
|
local force=$3
|
|
local currentdir=$PWD
|
|
|
|
if [[ -z "$folder" ]]; then
|
|
wd_exit_fail "You must specify a path"
|
|
return
|
|
fi
|
|
|
|
if [[ ! -d "$folder" ]]; then
|
|
wd_exit_fail "The directory does not exist"
|
|
return
|
|
fi
|
|
|
|
cd "$folder" || return
|
|
wd_add "$point" "$force"
|
|
cd "$currentdir" || return
|
|
}
|
|
|
|
|
|
wd_remove()
|
|
{
|
|
local point_list=$1
|
|
|
|
if [[ "$point_list" == "" ]]
|
|
then
|
|
point_list=$(basename "$PWD")
|
|
fi
|
|
|
|
for point_name in $point_list ; do
|
|
if [[ ${points[$point_name]} != "" ]]
|
|
then
|
|
local config_tmp=$(mktemp "${TMPDIR:-/tmp}/wd.XXXXXXXXXX")
|
|
# Copy and delete in two steps in order to preserve symlinks
|
|
if sed -n "/^${point_name}:.*$/!p" "$WD_CONFIG" >| "$config_tmp" && command cp "$config_tmp" "$WD_CONFIG" && command rm "$config_tmp"
|
|
then
|
|
wd_print_msg "$WD_GREEN" "Warp point removed"
|
|
else
|
|
wd_exit_fail "Something bad happened! Sorry."
|
|
fi
|
|
else
|
|
wd_exit_fail "Warp point was not found"
|
|
fi
|
|
done
|
|
}
|
|
|
|
wd_browse() {
|
|
if ! command -v fzf >/dev/null; then
|
|
echo "This functionality requires fzf. Please install fzf first."
|
|
return 1
|
|
fi
|
|
local entries=("${(@f)$(sed "s:${HOME}:~:g" "$WD_CONFIG" | awk -F ':' '{print $1 " -> " $2}')}")
|
|
local script_path="${${(%):-%x}:h}"
|
|
local wd_remove_output=$(mktemp "${TMPDIR:-/tmp}/wd.XXXXXXXXXX")
|
|
entries=("All warp points:" "Press enter to select. Press delete to remove" "${entries[@]}")
|
|
local fzf_bind="delete:execute(echo {} | awk -F ' -> ' '{print \$1}' | xargs -I {} "$script_path/wd.sh" rm {} > "$wd_remove_output")+abort"
|
|
local selected_entry=$(printf '%s\n' "${entries[@]}" | fzf --height 100% --reverse --header-lines=2 --bind="$fzf_bind")
|
|
if [[ -e $wd_remove_output ]]; then
|
|
cat "$wd_remove_output"
|
|
rm "$wd_remove_output"
|
|
fi
|
|
if [[ -n $selected_entry ]]; then
|
|
local selected_point="${selected_entry%% ->*}"
|
|
selected_point=$(echo "$selected_point" | xargs)
|
|
wd $selected_point
|
|
fi
|
|
}
|
|
|
|
wd_browse_widget() {
|
|
if [[ -e $WD_CONFIG ]]; then
|
|
wd_browse
|
|
saved_buffer=$BUFFER
|
|
saved_cursor=$CURSOR
|
|
BUFFER=
|
|
zle redisplay
|
|
zle accept-line
|
|
fi
|
|
}
|
|
|
|
wd_restore_buffer() {
|
|
if [[ -n $saved_buffer ]]; then
|
|
BUFFER=$saved_buffer
|
|
CURSOR=$saved_cursor
|
|
fi
|
|
saved_buffer=
|
|
saved_cursor=1
|
|
}
|
|
|
|
wd_list_all()
|
|
{
|
|
wd_print_msg "$WD_BLUE" "All warp points:"
|
|
|
|
entries=$(sed "s:${HOME}:~:g" "$WD_CONFIG")
|
|
|
|
max_warp_point_length=0
|
|
while IFS= read -r line
|
|
do
|
|
arr=(${(s,:,)line})
|
|
key=${arr[1]}
|
|
|
|
length=${#key}
|
|
if [[ length -gt max_warp_point_length ]]
|
|
then
|
|
max_warp_point_length=$length
|
|
fi
|
|
done <<< "$entries"
|
|
|
|
while IFS= read -r line
|
|
do
|
|
if [[ $line != "" ]]
|
|
then
|
|
arr=(${(s,:,)line})
|
|
key=${arr[1]}
|
|
val=${line#"${arr[1]}:"}
|
|
|
|
if [[ -z $wd_quiet_mode ]]
|
|
then
|
|
printf "%${max_warp_point_length}s -> %s\n" "$key" "$val"
|
|
fi
|
|
fi
|
|
done <<< "$entries"
|
|
}
|
|
|
|
wd_ls()
|
|
{
|
|
wd_getdir "$1"
|
|
ls "${dir/#\~/$HOME}"
|
|
}
|
|
|
|
wd_path()
|
|
{
|
|
wd_getdir "$1"
|
|
echo "$(echo "$dir" | sed "s:~:${HOME}:g")"
|
|
}
|
|
|
|
wd_show()
|
|
{
|
|
local name_arg=$1
|
|
# if there's an argument we look up the value
|
|
if [[ -n $name_arg ]]
|
|
then
|
|
if [[ -z $points[$name_arg] ]]
|
|
then
|
|
wd_print_msg "$WD_BLUE" "No warp point named $name_arg"
|
|
else
|
|
wd_print_msg "$WD_GREEN" "Warp point: ${WD_GREEN}$name_arg${WD_NOC} -> $points[$name_arg]"
|
|
fi
|
|
else
|
|
# hax to create a local empty array
|
|
local wd_matches
|
|
wd_matches=()
|
|
# do a reverse lookup to check whether PWD is in $points
|
|
PWD="${PWD/$HOME/~}"
|
|
if [[ ${points[(r)$PWD]} == "$PWD" ]]
|
|
then
|
|
for name in ${(k)points}
|
|
do
|
|
if [[ $points[$name] == "$PWD" ]]
|
|
then
|
|
wd_matches[$(($#wd_matches+1))]=$name
|
|
fi
|
|
done
|
|
|
|
wd_print_msg "$WD_BLUE" "$#wd_matches warp point(s) to current directory: ${WD_GREEN}$wd_matches${WD_NOC}"
|
|
else
|
|
wd_print_msg "$WD_YELLOW" "No warp point to $(echo "$PWD" | sed "s:$HOME:~:")"
|
|
fi
|
|
fi
|
|
}
|
|
|
|
wd_clean() {
|
|
local force=$1
|
|
local count=0
|
|
local wd_tmp=""
|
|
|
|
while read -r line
|
|
do
|
|
if [[ $line != "" ]]
|
|
then
|
|
arr=(${(s,:,)line})
|
|
key=${arr[1]}
|
|
val=${arr[2]}
|
|
|
|
if [ -d "${val/#\~/$HOME}" ]
|
|
then
|
|
wd_tmp=$wd_tmp"\n"`echo "$line"`
|
|
else
|
|
wd_print_msg "$WD_YELLOW" "Nonexistent directory: ${key} -> ${val}"
|
|
count=$((count+1))
|
|
fi
|
|
fi
|
|
done < "$WD_CONFIG"
|
|
|
|
if [[ $count -eq 0 ]]
|
|
then
|
|
wd_print_msg "$WD_BLUE" "No warp points to clean, carry on!"
|
|
else
|
|
if [ ! -z "$force" ] || wd_yesorno "Removing ${count} warp points. Continue? (y/n)"
|
|
then
|
|
echo "$wd_tmp" >! "$WD_CONFIG"
|
|
wd_print_msg "$WD_GREEN" "Cleanup complete. ${count} warp point(s) removed"
|
|
else
|
|
wd_print_msg "$WD_BLUE" "Cleanup aborted"
|
|
fi
|
|
fi
|
|
}
|
|
|
|
wd_export_static_named_directories() {
|
|
if [[ ! -z $WD_EXPORT ]]
|
|
then
|
|
command grep '^[0-9a-zA-Z_-]\+:' "$WD_CONFIG" | sed -e "s,~,$HOME," -e 's/:/=/' | while read -r warpdir ; do
|
|
hash -d "$warpdir"
|
|
done
|
|
fi
|
|
}
|
|
|
|
WD_CONFIG=${WD_CONFIG:-$HOME/.warprc}
|
|
local WD_QUIET=0
|
|
local WD_EXIT_CODE=0
|
|
local WD_DEBUG=0
|
|
|
|
# Parse 'meta' options first to avoid the need to have them before
|
|
# other commands. The `-D` flag consumes recognized options so that
|
|
# the actual command parsing won't be affected.
|
|
|
|
zparseopts -D -E \
|
|
c:=wd_alt_config -config:=wd_alt_config \
|
|
q=wd_quiet_mode -quiet=wd_quiet_mode \
|
|
v=wd_print_version -version=wd_print_version \
|
|
f=wd_force_mode -force=wd_force_mode
|
|
|
|
if [[ ! -z $wd_print_version ]]
|
|
then
|
|
echo "wd version $WD_VERSION"
|
|
fi
|
|
|
|
if [[ ! -z $wd_alt_config ]]
|
|
then
|
|
WD_CONFIG=$wd_alt_config[2]
|
|
fi
|
|
|
|
# check if config file exists
|
|
if [ ! -e "$WD_CONFIG" ]
|
|
then
|
|
# if not, create config file
|
|
touch "$WD_CONFIG"
|
|
else
|
|
wd_export_static_named_directories
|
|
fi
|
|
|
|
# disable extendedglob for the complete wd execution time
|
|
setopt | grep -q extendedglob
|
|
wd_extglob_is_set=$?
|
|
if (( wd_extglob_is_set == 0 )); then
|
|
setopt noextendedglob
|
|
fi
|
|
|
|
# load warp points
|
|
typeset -A points
|
|
while read -r line
|
|
do
|
|
arr=(${(s,:,)line})
|
|
key=${arr[1]}
|
|
# join the rest, in case the path contains colons
|
|
val=${(j,:,)arr[2,-1]}
|
|
|
|
points[$key]=$val
|
|
done < "$WD_CONFIG"
|
|
|
|
# get opts
|
|
args=$(getopt -o a:r:c:lhs -l add:,rm:,clean,list,ls:,path:,help,show -- $*)
|
|
|
|
# check if no arguments were given, and that version is not set
|
|
if [[ ($? -ne 0 || $#* -eq 0) && -z $wd_print_version ]]
|
|
then
|
|
wd_print_usage
|
|
|
|
# check if config file is writeable
|
|
elif [ ! -w "$WD_CONFIG" ]
|
|
then
|
|
# do nothing
|
|
# can't run `exit`, as this would exit the executing shell
|
|
wd_exit_fail "\'$WD_CONFIG\' is not writeable."
|
|
|
|
else
|
|
# parse rest of options
|
|
local wd_o
|
|
for wd_o
|
|
do
|
|
case "$wd_o"
|
|
in
|
|
"-a"|"--add"|"add")
|
|
wd_add "$2" "$wd_force_mode"
|
|
break
|
|
;;
|
|
"-b"|"browse")
|
|
wd_browse
|
|
break
|
|
;;
|
|
"-c"|"--addcd"|"addcd")
|
|
wd_addcd "$2" "$3" "$wd_force_mode"
|
|
break
|
|
;;
|
|
"-e"|"export")
|
|
wd_export_static_named_directories
|
|
break
|
|
;;
|
|
"-r"|"--remove"|"rm")
|
|
# Passes all the arguments as a single string separated by whitespace to wd_remove
|
|
wd_remove "${@:2}"
|
|
break
|
|
;;
|
|
"-l"|"list")
|
|
wd_list_all
|
|
break
|
|
;;
|
|
"-ls"|"ls")
|
|
wd_ls "$2"
|
|
break
|
|
;;
|
|
"-p"|"--path"|"path")
|
|
wd_path "$2"
|
|
break
|
|
;;
|
|
"-h"|"--help"|"help")
|
|
wd_print_usage
|
|
break
|
|
;;
|
|
"-s"|"--show"|"show")
|
|
wd_show "$2"
|
|
break
|
|
;;
|
|
"-c"|"--clean"|"clean")
|
|
wd_clean "$wd_force_mode"
|
|
break
|
|
;;
|
|
*)
|
|
wd_warp "$wd_o" "$2"
|
|
break
|
|
;;
|
|
--)
|
|
break
|
|
;;
|
|
esac
|
|
done
|
|
fi
|
|
|
|
## garbage collection
|
|
# if not, next time warp will pick up variables from this run
|
|
# remember, there's no sub shell
|
|
|
|
if (( wd_extglob_is_set == 0 )); then
|
|
setopt extendedglob
|
|
fi
|
|
|
|
unset wd_extglob_is_set
|
|
unset wd_warp
|
|
unset wd_add
|
|
unset wd_addcd
|
|
unset wd_remove
|
|
unset wd_show
|
|
unset wd_list_all
|
|
unset wd_print_msg
|
|
unset wd_yesorno
|
|
unset wd_print_usage
|
|
unset wd_alt_config
|
|
unset wd_quiet_mode
|
|
unset wd_print_version
|
|
unset wd_export_static_named_directories
|
|
unset wd_o
|
|
|
|
unset args
|
|
unset points
|
|
unset val &> /dev/null # fixes issue #1
|
|
|
|
return $WD_EXIT_CODE
|