|
|
|
|
@ -1,4 +1,4 @@ |
|
|
|
|
# Copyright (c) 2009 rupa deadwyler under the WTFPL license |
|
|
|
|
# Copyright (c) 2009 rupa deadwyler. Licensed under the WTFPL license, Version 2 |
|
|
|
|
|
|
|
|
|
# maintains a jump-list of the directories you actually use |
|
|
|
|
# |
|
|
|
|
@ -21,6 +21,7 @@ |
|
|
|
|
# * z -r foo # cd to highest ranked dir matching foo |
|
|
|
|
# * z -t foo # cd to most recently accessed dir matching foo |
|
|
|
|
# * z -l foo # list matches instead of cd |
|
|
|
|
# * z -e foo # echo the best match, don't cd |
|
|
|
|
# * z -c foo # restrict matches to subdirs of $PWD |
|
|
|
|
|
|
|
|
|
[ -d "${_Z_DATA:-$HOME/.z}" ] && { |
|
|
|
|
@ -31,9 +32,21 @@ _z() { |
|
|
|
|
|
|
|
|
|
local datafile="${_Z_DATA:-$HOME/.z}" |
|
|
|
|
|
|
|
|
|
# if symlink, dereference |
|
|
|
|
[ -h "$datafile" ] && datafile=$(readlink "$datafile") |
|
|
|
|
|
|
|
|
|
# bail if we don't own ~/.z and $_Z_OWNER not set |
|
|
|
|
[ -z "$_Z_OWNER" -a -f "$datafile" -a ! -O "$datafile" ] && return |
|
|
|
|
|
|
|
|
|
_z_dirs () { |
|
|
|
|
local line |
|
|
|
|
while read line; do |
|
|
|
|
# only count directories |
|
|
|
|
[ -d "${line%%\|*}" ] && echo "$line" |
|
|
|
|
done < "$datafile" |
|
|
|
|
return 0 |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
# add entries |
|
|
|
|
if [ "$1" = "--add" ]; then |
|
|
|
|
shift |
|
|
|
|
@ -49,10 +62,7 @@ _z() { |
|
|
|
|
|
|
|
|
|
# maintain the data file |
|
|
|
|
local tempfile="$datafile.$RANDOM" |
|
|
|
|
while read line; do |
|
|
|
|
# only count directories |
|
|
|
|
[ -d "${line%%\|*}" ] && echo $line |
|
|
|
|
done < "$datafile" | awk -v path="$*" -v now="$(date +%s)" -F"|" ' |
|
|
|
|
_z_dirs | awk -v path="$*" -v now="$(date +%s)" -F"|" ' |
|
|
|
|
BEGIN { |
|
|
|
|
rank[path] = 1 |
|
|
|
|
time[path] = now |
|
|
|
|
@ -75,7 +85,7 @@ _z() { |
|
|
|
|
} else for( x in rank ) print x "|" rank[x] "|" time[x] |
|
|
|
|
} |
|
|
|
|
' 2>/dev/null >| "$tempfile" |
|
|
|
|
# do our best to avoid clobbering the datafile in a race condition |
|
|
|
|
# do our best to avoid clobbering the datafile in a race condition. |
|
|
|
|
if [ $? -ne 0 -a -f "$datafile" ]; then |
|
|
|
|
env rm -f "$tempfile" |
|
|
|
|
else |
|
|
|
|
@ -85,17 +95,15 @@ _z() { |
|
|
|
|
|
|
|
|
|
# tab completion |
|
|
|
|
elif [ "$1" = "--complete" -a -s "$datafile" ]; then |
|
|
|
|
while read line; do |
|
|
|
|
[ -d "${line%%\|*}" ] && echo $line |
|
|
|
|
done < "$datafile" | awk -v q="$2" -F"|" ' |
|
|
|
|
_z_dirs | awk -v q="$2" -F"|" ' |
|
|
|
|
BEGIN { |
|
|
|
|
if( q == tolower(q) ) imatch = 1 |
|
|
|
|
q = substr(q, 3) |
|
|
|
|
gsub(" ", ".*", q) |
|
|
|
|
if( q == tolower(q) ) imatch = 1 |
|
|
|
|
gsub(/ /, ".*", q) |
|
|
|
|
} |
|
|
|
|
{ |
|
|
|
|
if( imatch ) { |
|
|
|
|
if( tolower($1) ~ tolower(q) ) print $1 |
|
|
|
|
if( tolower($1) ~ q ) print $1 |
|
|
|
|
} else if( $1 ~ q ) print $1 |
|
|
|
|
} |
|
|
|
|
' 2>/dev/null |
|
|
|
|
@ -106,11 +114,12 @@ _z() { |
|
|
|
|
--) while [ "$1" ]; do shift; local fnd="$fnd${fnd:+ }$1";done;; |
|
|
|
|
-*) local opt=${1:1}; while [ "$opt" ]; do case ${opt:0:1} in |
|
|
|
|
c) local fnd="^$PWD $fnd";; |
|
|
|
|
h) echo "${_Z_CMD:-z} [-chlrtx] args" >&2; return;; |
|
|
|
|
x) sed -i -e "\:^${PWD}|.*:d" "$datafile";; |
|
|
|
|
e) local echo=1;; |
|
|
|
|
h) echo "${_Z_CMD:-z} [-cehlrtx] args" >&2; return;; |
|
|
|
|
l) local list=1;; |
|
|
|
|
r) local typ="rank";; |
|
|
|
|
t) local typ="recent";; |
|
|
|
|
x) sed -i -e "\:^${PWD}|.*:d" "$datafile";; |
|
|
|
|
esac; opt=${opt:1}; done;; |
|
|
|
|
*) local fnd="$fnd${fnd:+ }$1";; |
|
|
|
|
esac; local last=$1; [ "$#" -gt 0 ] && shift; done |
|
|
|
|
@ -119,16 +128,14 @@ _z() { |
|
|
|
|
# if we hit enter on a completion just go there |
|
|
|
|
case "$last" in |
|
|
|
|
# completions will always start with / |
|
|
|
|
/*) [ -z "$list" -a -d "$last" ] && cd "$last" && return;; |
|
|
|
|
/*) [ -z "$list" -a -d "$last" ] && builtin cd "$last" && return;; |
|
|
|
|
esac |
|
|
|
|
|
|
|
|
|
# no file yet |
|
|
|
|
[ -f "$datafile" ] || return |
|
|
|
|
|
|
|
|
|
local cd |
|
|
|
|
cd="$(while read line; do |
|
|
|
|
[ -d "${line%%\|*}" ] && echo $line |
|
|
|
|
done < "$datafile" | awk -v t="$(date +%s)" -v list="$list" -v typ="$typ" -v q="$fnd" -F"|" ' |
|
|
|
|
cd="$( < <( _z_dirs ) awk -v t="$(date +%s)" -v list="$list" -v typ="$typ" -v q="$fnd" -F"|" ' |
|
|
|
|
function frecent(rank, time) { |
|
|
|
|
# relate frequency and time |
|
|
|
|
dx = t - time |
|
|
|
|
@ -137,19 +144,21 @@ _z() { |
|
|
|
|
if( dx < 604800 ) return rank / 2 |
|
|
|
|
return rank / 4 |
|
|
|
|
} |
|
|
|
|
function output(files, out, common) { |
|
|
|
|
function output(matches, best_match, common) { |
|
|
|
|
# list or return the desired directory |
|
|
|
|
if( list ) { |
|
|
|
|
cmd = "sort -n >&2" |
|
|
|
|
for( x in files ) { |
|
|
|
|
if( files[x] ) printf "%-10s %s\n", files[x], x | cmd |
|
|
|
|
for( x in matches ) { |
|
|
|
|
if( matches[x] ) { |
|
|
|
|
printf "%-10s %s\n", matches[x], x | cmd |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
if( common ) { |
|
|
|
|
printf "%-10s %s\n", "common:", common > "/dev/stderr" |
|
|
|
|
} |
|
|
|
|
} else { |
|
|
|
|
if( common ) out = common |
|
|
|
|
print out |
|
|
|
|
if( common ) best_match = common |
|
|
|
|
print best_match |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
function common(matches) { |
|
|
|
|
@ -160,11 +169,9 @@ _z() { |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
if( short == "/" ) return |
|
|
|
|
# use a copy to escape special characters, as we want to return |
|
|
|
|
# the original. yeah, this escaping is awful. |
|
|
|
|
clean_short = short |
|
|
|
|
gsub(/\[\(\)\[\]\|\]/, "\\\\&", clean_short) |
|
|
|
|
for( x in matches ) if( matches[x] && x !~ clean_short ) return |
|
|
|
|
for( x in matches ) if( matches[x] && index(x, short) != 1 ) { |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
return short |
|
|
|
|
} |
|
|
|
|
BEGIN { |
|
|
|
|
@ -197,8 +204,10 @@ _z() { |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
')" |
|
|
|
|
[ $? -gt 0 ] && return |
|
|
|
|
[ "$cd" ] && cd "$cd" |
|
|
|
|
|
|
|
|
|
[ $? -eq 0 ] && [ "$cd" ] && { |
|
|
|
|
if [ "$echo" ]; then echo "$cd"; else builtin cd "$cd"; fi |
|
|
|
|
} |
|
|
|
|
fi |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
@ -212,11 +221,11 @@ if type compctl >/dev/null 2>&1; then |
|
|
|
|
# populate directory list, avoid clobbering any other precmds. |
|
|
|
|
if [ "$_Z_NO_RESOLVE_SYMLINKS" ]; then |
|
|
|
|
_z_precmd() { |
|
|
|
|
_z --add "${PWD:a}" |
|
|
|
|
(_z --add "${PWD:a}" &) |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
_z_precmd() { |
|
|
|
|
_z --add "${PWD:A}" |
|
|
|
|
(_z --add "${PWD:A}" &) |
|
|
|
|
} |
|
|
|
|
fi |
|
|
|
|
[[ -n "${precmd_functions[(r)_z_precmd]}" ]] || { |
|
|
|
|
@ -237,7 +246,7 @@ elif type complete >/dev/null 2>&1; then |
|
|
|
|
[ "$_Z_NO_PROMPT_COMMAND" ] || { |
|
|
|
|
# populate directory list. avoid clobbering other PROMPT_COMMANDs. |
|
|
|
|
grep "_z --add" <<< "$PROMPT_COMMAND" >/dev/null || { |
|
|
|
|
PROMPT_COMMAND="$PROMPT_COMMAND"$'\n''_z --add "$(command pwd '$_Z_RESOLVE_SYMLINKS' 2>/dev/null)" 2>/dev/null;' |
|
|
|
|
PROMPT_COMMAND="$PROMPT_COMMAND"$'\n''(_z --add "$(command pwd '$_Z_RESOLVE_SYMLINKS' 2>/dev/null)" 2>/dev/null &);' |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
fi |
|
|
|
|
|