summaryrefslogtreecommitdiff
path: root/oh-my-zsh/plugins/wd/wd.sh
diff options
context:
space:
mode:
Diffstat (limited to 'oh-my-zsh/plugins/wd/wd.sh')
-rw-r--r--oh-my-zsh/plugins/wd/wd.sh513
1 files changed, 513 insertions, 0 deletions
diff --git a/oh-my-zsh/plugins/wd/wd.sh b/oh-my-zsh/plugins/wd/wd.sh
new file mode 100644
index 0000000..e51cf90
--- /dev/null
+++ b/oh-my-zsh/plugins/wd/wd.sh
@@ -0,0 +1,513 @@
+#!/bin/zsh
+
+# WARP DIRECTORY
+# ==============
+# Jump to custom directories in terminal
+# because `cd` takes too long...
+#
+# @github.com/mfaerevaag/wd
+
+# version
+readonly WD_VERSION=0.5.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
+ local msg=$2
+
+ if [[ $color == "" || $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
+ 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
+ -d | --debug Exit after execution with exit codes (for testing)
+ -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_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_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
+}
+
+local 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 \
+ d=wd_debug_mode -debug=wd_debug_mode \
+ 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=$?
+[[ $wd_extglob_is_set ]] && setopt noextendedglob
+
+# 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
+ ;;
+ "-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
+
+[[ $wd_extglob_is_set ]] && setopt extendedglob
+
+unset wd_extglob_is_set
+unset wd_warp
+unset wd_add
+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
+
+if [[ -n $wd_debug_mode ]]
+then
+ exit $WD_EXIT_CODE
+else
+ unset wd_debug_mode
+fi
generated by cgit v1.2.3 (git 2.25.1) at 2025-02-23 08:42:15 +0000