+++ /dev/null
-#!/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
-
- 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 [[ ${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=${arr[2]}
-
- 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
-
-# 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
-
-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
projects / dotfiles.git / blobdiff