z upgraded to the latest version

plugins/z/README

@@ -6,7 +6,7 @@ NAME
        z - jump around
 
 SYNOPSIS
-       z [-chlrt] [regex1 regex2 ... regexn]
+       z [-chlrtx] [regex1 regex2 ... regexn]
 
 AVAILABILITY
        bash, zsh
@@ -18,7 +18,7 @@ DESCRIPTION
        directory that matches ALL of the regexes given on the command line.
 
 OPTIONS
-       -c     restrict matches to subdirectories of the current directory.
+       -c     restrict matches to subdirectories of the current directory
 
        -h     show a brief help message
 
@@ -28,6 +28,8 @@ OPTIONS
 
        -t     match by recent access only
 
+       -x     remove the current directory from the datafile
+
 EXAMPLES
        z foo         cd to most frecent dir matching foo
 
@@ -56,7 +58,7 @@ NOTES
               Set $_Z_NO_PROMPT_COMMAND to handle PROMPT_COMMAND/precmd  your-
               self.
               Set $_Z_EXCLUDE_DIRS to an array of directories to exclude.
-              (These  settings  should  go  in .bashrc/.zshrc before the lines
+              (These  settings  should  go  in  .bashrc/.zshrc before the line
               added above.)
               Install   the   provided   man   page   z.1    somewhere    like
               /usr/local/man/man1.
@@ -68,8 +70,8 @@ NOTES
        multiplied by 0.99. Entries with a rank lower than 1 are forgotten.
 
    Frecency:
-       Frecency is a portmantaeu of 'recent' and 'frequency'. It is a weighted
+       Frecency is a portmanteau of 'recent' and 'frequency'. It is a weighted
-       rank  that  depends on how often and how recently something occured. As
+       rank  that depends on how often and how recently something occurred. As
        far as I know, Mozilla came up with the term.
 
        To z, a directory that has low ranking but has been  accessed  recently
@@ -107,9 +109,9 @@ ENVIRONMENT
        resolving  of  symlinks.  If  it  is  not  set,  symbolic links will be
        resolved when added to the datafile.
 
-       In bash, z prepends a command to the PROMPT_COMMAND  environment  vari-
+       In bash, z appends a command to the PROMPT_COMMAND environment variable
-       able  to  maintain its database. In zsh, z appends a function _z_precmd
+       to maintain its database. In zsh, z appends a function _z_precmd to the
-       to the precmd_functions array.
+       precmd_functions array.
 
        The environment variable $_Z_NO_PROMPT_COMMAND can be set if  you  want
        to handle PROMPT_COMMAND or precmd yourself.

plugins/z/z.1

@@ -4,7 +4,7 @@ NAME
 z \- jump around
 .SH
 SYNOPSIS
-z [\-chlrt] [regex1 regex2 ... regexn]
+z [\-chlrtx] [regex1 regex2 ... regexn]
 .SH
 AVAILABILITY
 bash, zsh
@@ -18,7 +18,7 @@ directory that matches ALL of the regexes given on the command line.
 OPTIONS
 .TP
 \fB\-c\fR
-restrict matches to subdirectories of the current directory.
+restrict matches to subdirectories of the current directory
 .TP
 \fB\-h\fR
 show a brief help message
@@ -31,6 +31,9 @@ match by rank only
 .TP
 \fB\-t\fR
 match by recent access only
+.TP
+\fB\-x\fR
+remove the current directory from the datafile
 .SH EXAMPLES
 .TP 14
 \fBz foo\fR
@@ -79,7 +82,7 @@ Set \fB$_Z_NO_PROMPT_COMMAND\fR to handle \fBPROMPT_COMMAND/precmd\fR yourself.
 Set \fB$_Z_EXCLUDE_DIRS\fR to an array of directories to exclude.
 .RE
 .RS
-(These settings should go in .bashrc/.zshrc before the lines added above.)
+(These settings should go in .bashrc/.zshrc before the line added above.)
 .RE
 .RS
 Install the provided man page \fBz.1\fR somewhere like \fB/usr/local/man/man1\fR.
@@ -92,8 +95,8 @@ the sum of ranks is greater than 6000, all ranks are multiplied by 0.99. Entries
 with a rank lower than 1 are forgotten.
 .SS
 Frecency:
-Frecency is a portmantaeu of 'recent' and 'frequency'. It is a weighted rank
+Frecency is a portmanteau of 'recent' and 'frequency'. It is a weighted rank
-that depends on how often and how recently something occured. As far as I
+that depends on how often and how recently something occurred. As far as I
 know, Mozilla came up with the term.
 .P
 To \fBz\fR, a directory that has low ranking but has been accessed recently
@@ -131,7 +134,7 @@ The environment variable \fB$_Z_NO_RESOLVE_SYMLINKS\fR can be set to prevent
 resolving of symlinks. If it is not set, symbolic links will be resolved when
 added to the datafile.
 .P
-In bash, \fBz\fR prepends a command to the \fBPROMPT_COMMAND\fR environment
+In bash, \fBz\fR appends a command to the \fBPROMPT_COMMAND\fR environment
 variable to maintain its database. In zsh, \fBz\fR appends a function
 \fB_z_precmd\fR to the \fBprecmd_functions\fR array.
 .P

plugins/z/z.sh

@@ -22,11 +22,6 @@
 #     * z -l foo  # list matches instead of cd
 #     * z -c foo  # restrict matches to subdirs of $PWD
 
-case $- in
- *i*) ;;
-   *) echo 'ERROR: z.sh is meant to be sourced, not directly executed.'
-esac
-
 [ -d "${_Z_DATA:-$HOME/.z}" ] && {
     echo "ERROR: z.sh's datafile (${_Z_DATA:-$HOME/.z}) is a directory."
 }
@@ -35,7 +30,7 @@ _z() {
 
     local datafile="${_Z_DATA:-$HOME/.z}"
 
- # bail out if we don't own ~/.z (we're another user but our ENV is still set)
+    # bail if we don't own ~/.z (we're another user but our ENV is still set)
     [ -f "$datafile" -a ! -O "$datafile" ] && return
 
     # add entries
@@ -51,10 +46,10 @@ _z() {
             [ "$*" = "$exclude" ] && return
         done
 
-  # maintain the file
+        # maintain the data file
-  local tempfile
+        local tempfile="$datafile.$RANDOM"
-  tempfile="$(mktemp "$datafile.XXXXXX")" || return
         while read line; do
+            # only count directories
             [ -d "${line%%\|*}" ] && echo $line
         done < "$datafile" | awk -v path="$*" -v now="$(date +%s)" -F"|" '
             BEGIN {
@@ -62,6 +57,7 @@ _z() {
                 time[path] = now
             }
             $2 >= 1 {
+                # drop ranks below 1
                 if( $1 == path ) {
                     rank[$1] = $2 + 1
                     time[$1] = now
@@ -73,14 +69,16 @@ _z() {
             }
             END {
                 if( count > 6000 ) {
-     for( i in rank ) print i "|" 0.99*rank[i] "|" time[i] # aging
+                    # aging
-    } else for( i in rank ) print i "|" rank[i] "|" time[i]
+                    for( x in rank ) print x "|" 0.99*rank[x] "|" time[x]
+                } 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
         if [ $? -ne 0 -a -f "$datafile" ]; then
             env rm -f "$tempfile"
         else
-   env mv -f "$tempfile" "$datafile"
+            env mv -f "$tempfile" "$datafile" || env rm -f "$tempfile"
         fi
 
     # tab completion
@@ -89,14 +87,14 @@ _z() {
             [ -d "${line%%\|*}" ] && echo $line
         done < "$datafile" | awk -v q="$2" -F"|" '
             BEGIN {
-    if( q == tolower(q) ) nocase = 1
+                if( q == tolower(q) ) imatch = 1
-    split(substr(q,3),fnd," ")
+                split(substr(q, 3), fnd, " ")
             }
             {
-    if( nocase ) {
+                if( imatch ) {
-     for( i in fnd ) tolower($1) !~ tolower(fnd[i]) && $1 = ""
+                    for( x in fnd ) tolower($1) !~ tolower(fnd[x]) && $1 = ""
                 } else {
-     for( i in fnd ) $1 !~ fnd[i] && $1 = ""
+                    for( x in fnd ) $1 !~ fnd[x] && $1 = ""
                 }
                 if( $1 ) print $1
             }
@@ -105,15 +103,16 @@ _z() {
     else
         # list/go
         while [ "$1" ]; do case "$1" in
-   --) while [ "$1" ]; do shift; local fnd="$fnd $1";done;;
+            --) 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} [-chlrt] args" >&2; return;;
+                    h) echo "${_Z_CMD:-z} [-chlrtx] args" >&2; return;;
+                    x) sed -i -e "\:^${PWD}|.*:d" "$datafile";;
                     l) local list=1;;
                     r) local typ="rank";;
                     t) local typ="recent";;
                 esac; opt=${opt:1}; done;;
-    *) local fnd="$fnd $1";;
+             *) local fnd="$fnd${fnd:+ }$1";;
         esac; local last=$1; shift; done
         [ "$fnd" -a "$fnd" != "^$PWD " ] || local list=1
 
@@ -131,59 +130,70 @@ _z() {
             [ -d "${line%%\|*}" ] && echo $line
         done < "$datafile" | awk -v t="$(date +%s)" -v list="$list" -v typ="$typ" -v q="$fnd" -F"|" '
             function frecent(rank, time) {
-    dx = t-time
+                # relate frequency and time
-    if( dx < 3600 ) return rank*4
+                dx = t - time
-    if( dx < 86400 ) return rank*2
+                if( dx < 3600 ) return rank * 4
-    if( dx < 604800 ) return rank/2
+                if( dx < 86400 ) return rank * 2
-    return rank/4
+                if( dx < 604800 ) return rank / 2
+                return rank / 4
             }
-   function output(files, toopen, override) {
+            function output(files, out, common) {
+                # list or return the desired directory
                 if( list ) {
                     cmd = "sort -n >&2"
-     for( i in files ) if( files[i] ) printf "%-10s %s\n", files[i], i | cmd
+                    for( x in files ) {
-     if( override ) printf "%-10s %s\n", "common:", override > "/dev/stderr"
+                        if( files[x] ) printf "%-10s %s\n", files[x], x | cmd
+                    }
+                    if( common ) {
+                        printf "%-10s %s\n", "common:", common > "/dev/stderr"
+                    }
                 } else {
-     if( override ) toopen = override
+                    if( common ) out = common
-     print toopen
+                    print out
                 }
             }
             function common(matches) {
-    # shortest match
+                # find the common root of a list of matches, if it exists
-    for( i in matches ) {
+                for( x in matches ) {
-     if( matches[i] && (!short || length(i) < length(short)) ) short = i
+                    if( matches[x] && (!short || length(x) < length(short)) ) {
+                        short = x
+                    }
                 }
                 if( short == "/" ) return
-    # shortest match must be common to each match. escape special characters in
+                # use a copy to escape special characters, as we want to return
-    # a copy when testing, so we can return the original.
+                # the original. yeah, this escaping is awful.
                 clean_short = short
                 gsub(/[\(\)\[\]\|]/, "\\\\&", clean_short)
-    for( i in matches ) if( matches[i] && i !~ clean_short ) return
+                for( x in matches ) if( matches[x] && x !~ clean_short ) return
                 return short
             }
-   BEGIN { split(q, a, " "); oldf = noldf = -9999999999 }
+            BEGIN { split(q, words, " "); hi_rank = ihi_rank = -9999999999 }
             {
                 if( typ == "rank" ) {
-     f = $2
+                    rank = $2
                 } else if( typ == "recent" ) {
-     f = $3-t
+                    rank = $3 - t
-    } else f = frecent($2, $3)
+                } else rank = frecent($2, $3)
-    wcase[$1] = nocase[$1] = f
+                matches[$1] = imatches[$1] = rank
-    for( i in a ) {
+                for( x in words ) {
-     if( $1 !~ a[i] ) delete wcase[$1]
+                    if( $1 !~ words[x] ) delete matches[$1]
-     if( tolower($1) !~ tolower(a[i]) ) delete nocase[$1]
+                    if( tolower($1) !~ tolower(words[x]) ) delete imatches[$1]
                 }
-    if( wcase[$1] && wcase[$1] > oldf ) {
+                if( matches[$1] && matches[$1] > hi_rank ) {
-     cx = $1
+                    best_match = $1
-     oldf = wcase[$1]
+                    hi_rank = matches[$1]
-    } else if( nocase[$1] && nocase[$1] > noldf ) {
+                } else if( imatches[$1] && imatches[$1] > ihi_rank ) {
-     ncx = $1
+                    ibest_match = $1
-     noldf = nocase[$1]
+                    ihi_rank = imatches[$1]
                 }
             }
             END {
-    if( cx ) {
+                # prefer case sensitive
-     output(wcase, cx, common(wcase))
+                if( best_match ) {
-    } else if( ncx ) output(nocase, ncx, common(nocase))
+                    output(matches, best_match, common(matches))
+                } else if( ibest_match ) {
+                    output(imatches, ibest_match, common(imatches))
+                }
             }
         ')"
         [ $? -gt 0 ] && return
@@ -195,9 +205,10 @@ alias ${_Z_CMD:-z}='_z 2>&1'
 
 [ "$_Z_NO_RESOLVE_SYMLINKS" ] || _Z_RESOLVE_SYMLINKS="-P"
 
-if compctl &> /dev/null; then
+if compctl >/dev/null 2>&1; then
+    # zsh
     [ "$_Z_NO_PROMPT_COMMAND" ] || {
-  # zsh populate directory list, avoid clobbering any other precmds
+        # populate directory list, avoid clobbering any other precmds.
         if [ "$_Z_NO_RESOLVE_SYMLINKS" ]; then
             _z_precmd() {
                 _z --add "${PWD:a}"
@@ -207,22 +218,25 @@ if compctl &> /dev/null; then
                 _z --add "${PWD:A}"
             }
         fi
-  precmd_functions+=(_z_precmd)
+        [[ -n "${precmd_functions[(r)_z_precmd]}" ]] || {
+            precmd_functions[$(($#precmd_functions+1))]=_z_precmd
+        }
     }
- # zsh tab completion
     _z_zsh_tab_completion() {
+        # tab completion
         local compl
         read -l compl
         reply=(${(f)"$(_z --complete "$compl")"})
     }
     compctl -U -K _z_zsh_tab_completion _z
-elif complete &> /dev/null; then
+elif complete >/dev/null 2>&1; then
- # bash tab completion
+    # bash
+    # tab completion
     complete -o filenames -C '_z --complete "$COMP_LINE"' ${_Z_CMD:-z}
     [ "$_Z_NO_PROMPT_COMMAND" ] || {
-  # bash populate directory list. avoid clobbering other PROMPT_COMMANDs.
+        # populate directory list. avoid clobbering other PROMPT_COMMANDs.
-  echo $PROMPT_COMMAND | grep -q "_z --add" || {
+        grep "_z --add" <<< "$PROMPT_COMMAND" >/dev/null || {
-   PROMPT_COMMAND='_z --add "$(pwd '$_Z_RESOLVE_SYMLINKS' 2>/dev/null)" 2>/dev/null;'"$PROMPT_COMMAND"
+            PROMPT_COMMAND="$PROMPT_COMMAND"$'\n''_z --add "$(command pwd '$_Z_RESOLVE_SYMLINKS' 2>/dev/null)" 2>/dev/null;'
         }
     }
 fi