summaryrefslogtreecommitdiffstats
path: root/bash/j.sh
blob: e0306b3dbd9fa83f5f7c1144c0325dd7b799f99c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
# maintains a jump-list of the directories you actually use
# old directories eventually fall off the list
# inspired by Joel Schaerer's http://wiki.github.com/joelthelion/autojump
# and something similar i had - but i could never get the dir list right.
#
# INSTALL:
#   source into .bashrc under your '[-z "$PS1" ] || return' line
#   cd around for a while
#
# USE:
#   j [--h[elp]] [--l] [--r] [--s] [regex1 ... regexn]
#                       with no args, returns full list (same as j --l)
#     regex1 ... regexn jump to the the weighted directory matching all masks
#     --l               show the list instead of jumping
#     --r               order by recently used instead of most used.
#     --s               shortest matching path
#
# TIPS:
#   Some handy aliases:
#     alias jl='j --l'
#     alias jr='j --r'
#     alias js='j --s'
#
# CREDITS:
#   Joel Schaerer aka joelthelion for autojump
#   Daniel Drucker aka dmd for finding bugs and making me late for lunch
j() {
 # change jfile if you already have a .j file for something else
 local jfile=$HOME/.j
 [ "$1" = "--add" ] && {
  shift
  # we're in $HOME all the time, let something else get all the good letters
  [ "$*" = "$HOME" ] && return
  awk -v q="$*" -v t="$(date +%s)" -F"|" '
   BEGIN { l[q] = 1; d[q] = t }
   $2 >= 1 { 
    if( $1 == q ) {
     l[$1] = $2 + 1
     d[$1] = t
    } else {
     l[$1] = $2
     d[$1] = $3
    }
    count += $2
   }
   END {
    if( count > 1000 ) {
     for( i in l ) print i "|" 0.9*l[i] "|" d[i] # aging
    } else for( i in l ) print i "|" l[i] "|" d[i]
   }
  ' $jfile 2>/dev/null > $jfile.tmp
  mv -f $jfile.tmp $jfile
  return
 }
 # for tab completion
 [ "$1" = "--complete" ] && {
  awk -v q="$2" -F"|" '
   BEGIN { split(substr(q,3),a," ") }
   { 
    if( system("test -d \"" $1 "\"") ) next
    for( i in a ) $1 !~ a[i] && $1 = ""; if( $1 ) print $1
   }
  ' $jfile 2>/dev/null
  return
 }
 if [ $1 ]; then
  local x; local out
  for x do case $x in
   --h*) echo "j [--h[elp]] [--l] [--r] [--s] [regex1 ... regexn]"; return;;
   --l)local list=1;;
   --r)local recent=r;;
   --s)local short=1;;
     *)local out="$out $x";;
  esac; shift; done
  set -- $out
 else
  local list=1
 fi
 # remove directories that no longer exist
 awk -F"|" '
  { if( system("test -d \"" $1 "\"") ) next; print $0 }
 ' $jfile 2>/dev/null > $jfile.tmp
 mv -f $jfile.tmp $jfile
 if [ $list ]; then
  [ "$short" ] && return
  awk -v q="$*" -v t="$(date +%s)" -v r="$recent" -F"|" '
   BEGIN { f = 2; split(q,a," "); if( r ) f = 3 }
   {
    for( i in a ) $1 !~ a[i] && $1 = ""
    if( $1 ) if( f == 3 ) {
     print t - $f "\t" $1
    } else print $f "\t" $1
   }
  ' $jfile 2>/dev/null | sort -n$recent
 # if we hit enter on a completion just go there
 elif [ -d "/${out#*/}" ]; then
  cd "/${out#*/}"
 # prefer case sensitive
 else
  out=$(awk -v q="$*" -v s="$short" -v r="$recent" -F"|" '
   BEGIN { split(q,a," "); if( r ) { f = 3 } else f = 2 }
   { 
    for( i in a ) $1 !~ a[i] && $1 = ""
    if( $1 ) {
     if( s ) {
      if( length($1) <= length(x) ) {
       x = $1
      } else if( ! x ) x = $1
     } else if( $f >= dx ) { x = $1; dx = $f }
    }
   }
   END {
    if( ! x ) {
     close(FILENAME)
     while( getline < FILENAME ) {
      for( i in a ) tolower($1) !~ tolower(a[i]) && $1 = ""
      if( $1 ) {
       if( s ) {
        if( length($1) <= length(x) ) {
         x = $1
        } else if( ! x ) x = $1
       } else if( $f >= dx ) { x = $1; dx = $f }
      }
     }
    }
    if( x ) print x
   }
  ' $jfile)
  [ "$out" ] && cd "$out"
 fi
}
# tab completion for j
complete -C 'j --complete "$COMP_LINE"' j
# populate directory list. avoid clobbering other PROMPT_COMMANDs.
PROMPT_COMMAND='j --add "$(pwd -P)";'"$PROMPT_COMMAND"