chiark / gitweb /
dot/shell-rc: Relax the hard dependency on `rlwrap'.
[profile] / dot / shell-rc
1 ### -*-sh-*-
2 ###
3 ### Common per-shell configuration.
4
5 ###--------------------------------------------------------------------------
6 ### Utilities.
7
8 __mdw_programp () { type >/dev/null 2>&1 "$1"; }
9
10 __mdw_source_if_exists () {
11   local i
12   for i in "$@"; do
13     if [ -r "$i" ]; then . "$i"; fi
14   done
15 }
16
17 ###--------------------------------------------------------------------------
18 ### Hooks.
19
20 __mdw_addhook () {
21   local hk=$1 fn=$2 t
22
23   eval t=\${$hk+t}
24   case $t in t) ;; *) echo >&2 "unknown hook \`$hk'"; return 2; esac
25
26   eval t=\$$hk
27   case " $t " in
28     *" $fn "*) ;;
29     *) eval "$hk=\${$hk:+\$$hk }\$fn" ;;
30   esac
31 }
32
33 __mdw_delhook () {
34   local hk=$1 fn=$2 t l r
35
36   eval t=\${$hk+t}
37   case $t in t) ;; *) echo >&2 "unknown hook \`$hk'"; return 2; esac
38
39   eval t=\" \$$hk \"
40   case $t in
41     *" $fn "*)
42       l=${t%% $fn*} r=${t##*$fn }
43       l=${l# } r=${r% }
44       eval "$hk=\$l\${l:+ }\$r"
45       ;;
46   esac
47 }
48
49 __mdw_setrc () { return $1; }
50
51 __mdw_runhook () {
52   local hk=$1 saverc=$? t i; shift
53
54   eval t=\${$hk+t}
55   case $t in t) ;; *) echo >&2 "unknown hook \`$hk'"; return 2; esac
56
57   eval t=\$$hk
58   for i in $t; do __mdw_setrc $saverc; "$i" "$@"; done
59 }
60
61 ###--------------------------------------------------------------------------
62 ### Prompt machinery.
63
64 __mdw_host=$(hostname)
65 __mdw_hqual=
66 __mdw_hqual=$__mdw_hqual${SCHROOT_CHROOT_NAME+/$SCHROOT_CHROOT_NAME}
67 __mdw_hqual=$__mdw_hqual${CROSS_BUILDENV+/$CROSS_BUILDENV}
68 __mdw_set_prompt_hacks () { host=$__mdw_host; dir=""; }
69
70 __mdw_system=$(uname -s)
71 : ${USER-${LOGNAME-$(id -un)}}
72 __mdw_user=$USER
73
74 case $(id -u) in
75   0)
76     __mdw_rootp=t
77     ;;
78   *)
79     case $__mdw_system in
80       CYGWIN_*)
81         case " $(id -G) " in
82           *" 544 "*) __mdw_rootp=t __mdw_user="$__mdw_user%admin" ;;
83           *) __mdw_rootp=nil ;;
84         esac
85         ;;
86       *)
87         __mdw_rootp=nil
88         ;;
89     esac
90 esac
91
92 __mdw_set_prompt_pieces () {
93
94   ## Fancy highlighting in some terminals.
95   local bold unbold nl more host dir
96   local gitcolour extracolour rccolour uncolour
97   bold="" unbold="" nl="" more=""
98   gitcolour="" extracolour="" rccolour="" uncolour=""
99   __mdw_set_prompt_hacks
100
101   ## Choose the right delimiters.  Highlight root prompts specially;
102   ## highlight when I'm running as some other user.  Highlight when this
103   ## isn't the outermost shell on the terminal.
104   local left right u tty
105   case $__mdw_rootp in
106     t)
107       left=$(echo « | iconv -f UTF-8 -t //translit)
108       right=$(echo » | iconv -f UTF-8 -t //translit)
109       ;;
110     nil)
111       case $USER in
112         mdw | mwooding | nemo) u="" left="[" right="]" ;;
113         *) u="$__mdw_user@" left="{" right="}" ;;
114       esac
115       tty=$(tty)
116       case "$__mdw_tty" in
117         "$tty") left="<" right=">" ;;
118         *) __mdw_tty=$tty; export __mdw_tty ;;
119       esac
120       ;;
121   esac
122
123   ## If this session is insecure then highlight that.
124   local sec_l sec_r h
125   h=$(hostname)
126   case ${SSH_CLIENT-nil},${SCHROOT_CHROOT_NAME-nil},$__mdw_sechost in
127     nil,nil,"$h") sec_l="" sec_r="" ;;
128     nil,nil,*) sec_l="(" sec_r=")" ;;
129     *) sec_l="" sec_r=""
130   esac
131
132   ## If this is an schroot environment or some other interesting augmented
133   ## environment then point this out.
134
135   ## Put together the main pieces.
136   __mdw_prompt_left="$nl$bold$left$sec_l$u$host$__mdw_hqual$sec_r$dir"
137   __mdw_prompt_git_left="$unbold$gitcolour"
138   __mdw_prompt_git_right="$uncolour$bold"
139   __mdw_prompt_rc_left="$unbold$rccolour"
140   __mdw_prompt_rc_right="$uncolour$bold"
141   __mdw_prompt_extra_left="$unbold$extracolour"
142   __mdw_prompt_extra_right="$uncolour$bold"
143   __mdw_prompt_right="$right$unbold"
144   __mdw_prompt_more=" $more$bold>$unbold "
145 }
146
147 __mdw_set_prompt () {
148   case "${TERM-dumb}:${INSIDE_EMACS+$INSIDE_EMACS}" in
149     dumb:)
150       case $(id -u) in 0) PS1='# ' ;; *) PS1='$ ' ;; esac
151       PS2='> '
152       ;;
153     *)
154       __mdw_last_rc=$?
155       local git extra rc
156       if type __git_ps1 >/dev/null 2>&1; then
157         git="$__mdw_prompt_git_left$(__git_ps1)$__mdw_prompt_git_right"
158       else
159         git=""
160       fi
161       case $__mdw_last_rc in
162         0) rc="" ;;
163         *) rc="$__mdw_prompt_rc_left rc=$__mdw_last_rc$__mdw_prompt_rc_right" ;;
164       esac
165       case $__mdw_prompt_extra in
166         ?*) extra="$__mdw_prompt_extra_left [$__mdw_prompt_extra]$__mdw_prompt_extra_right" ;;
167         *) extra="" ;;
168       esac
169       PS1="$__mdw_prompt_left$git$extra$rc$__mdw_prompt_right"
170       PS2="$PS1$__mdw_prompt_more"
171       unset __mdw_last_rc
172       ;;
173   esac
174 }
175
176 __mdw_xterm_settitle () {
177   printf >/dev/tty \
178     "\e]0;%s@%s:%s – %s\e\\" \
179     "$__mdw_user" "$__mdw_host$__mdw_hqual" "$PWD" \
180     "$1"
181 }
182 __mdw_xterm_precmd () { __mdw_xterm_settitle "$__mdw_shell"; }
183 __mdw_xterm_preexec () { __mdw_xterm_settitle "$1"; }
184
185 __mdw_screen_settitle () {
186   printf >/dev/tty \
187     "\ek%s\e\\" \
188     "$1"
189 }
190 __mdw_screen_precmd () { __mdw_screen_settitle "$__mdw_shell"; }
191 __mdw_screen_preexec () { __mdw_screen_settitle "$1"; }
192
193 if [ -t 0 ]; then
194   case ${STY+t},${__mdw_precmd_hook+t},${__mdw_preexec_hook+t},${TERM} in
195     ,t,t,xterm*)
196       __mdw_addhook __mdw_precmd_hook __mdw_xterm_precmd
197       __mdw_addhook __mdw_preexec_hook __mdw_xterm_preexec
198       ;;
199     t,t,t,*)
200       __mdw_addhook __mdw_precmd_hook __mdw_screen_precmd
201       __mdw_addhook __mdw_preexec_hook __mdw_screen_preexec
202       ;;
203   esac
204   case ${__mdw_precmd_hook+t} in
205     t) __mdw_addhook __mdw_precmd_hook __mdw_set_prompt ;;
206   esac
207 fi
208
209 ###--------------------------------------------------------------------------
210 ### Some handy aliases.
211
212 alias cx='chmod +x'
213 alias which="command -v"
214 alias rc="rc -l"
215 rootly () {
216   case $# in 0) set -- "${SHELL-/bin/sh}" ;; esac
217   $__MDW_ROOTLY "$@"
218 }
219 alias r=rootly
220 alias re="rootly $EDITOR"
221 alias pstree="pstree -hl"
222 alias cdtmp='cd ${TMPDIR-/tmp}'
223 alias pushtmp='pushd ${TMPDIR-/tmp}'
224 alias e="$EDITOR"
225 alias svn="svnwrap svn"
226 alias @="ssh"
227 alias make="nice make"
228 alias cross-run="nice cross-run"
229 alias gdb="gdb -q"
230
231 ## Shut up Lisp interpreters.
232 alias clisp="clisp -q -q"
233 if command -v ionice >/dev/null 2>&1; then
234   alias maybe-rlwrap="rlwrap"
235 else
236   alias maybe-rlwrap=""
237 fi
238 alias cmucl="maybe-rlwrap cmucl -quiet"
239 alias ecl="maybe-rlwrap ecl"
240 alias sbcl="maybe-rlwrap sbcl --noinform"
241 alias ccl="maybe-rlwrap ccl"
242 alias ccl32="maybe-rlwrap ccl32"
243 alias ccl64="maybe-rlwrap ccl64"
244 alias abcl="maybe-rlwrap abcl --noinform"
245
246 ###--------------------------------------------------------------------------
247 ### Colour output.
248
249 mdw_colour_p () {
250   if ! [ -t 1 ]; then return 1; fi
251   case $TERM in dumb) return 1 ;; esac
252   return 0
253 }
254
255 ## Arrange for `ls' output to be in colour.
256 unalias ls 2>/dev/null || :
257 ls () {
258   if mdw_colour_p; then command ls $LS_OPTIONS ${LS_COLORS+--color=always} "$@"
259   else command ls "$@"; fi
260 }
261
262 ## Arrange for `grep' output to be in colour.
263 greplike () {
264   local grep=$1; shift
265   if mdw_colour_p; then
266     command $grep ${GREP_COLORS+--color=always} "$@" | mdw-pager
267   else
268     command $grep "$@"
269   fi
270 }
271 alias grep="greplike grep"
272 alias egrep="greplike egrep"
273 alias fgrep="greplike fgrep"
274 alias zgrep="greplike zgrep"
275
276 ## Arrange for `diff' output to be in colour.
277 difflike () {
278   local diff=$1; shift
279   if mdw_colour_p; then
280     command $diff \
281             ${DIFF_COLORS+--color=always} \
282             ${DIFF_COLORS+--palette="$DIFF_COLORS"} \
283             "$@" | mdw-pager
284   else
285     command $diff "$@" | cat
286   fi
287 }
288 alias diff="difflike diff"
289
290 ###--------------------------------------------------------------------------
291 ### Other hacks.
292
293 ## Turn off pagers inside Emacs shell buffers.
294 case "$INSIDE_EMACS" in
295   2[2-9].*,comint | [3-9][0-9].*,comint) export PAGER=cat ;;
296 esac
297
298 ###--------------------------------------------------------------------------
299 ### More complicated shell functions.
300
301 ## xt [@HOST] XTERM-ARGS
302 ##
303 ## Open a terminal, maybe on a remote host.
304 xt () {
305   case "$1" in
306     @*)
307       local remote=${1#@} title
308       shift
309       if [ $# -gt 0 ]; then
310         title="xterm [$remote] $1"
311       else
312         title="xterm [$remote]"
313       fi
314       (xterm -title "$title" -e ssh $remote "$@" &)
315       ;;
316     *)
317       (xterm "$@" &)
318       ;;
319   esac
320 }
321
322 ## core [y|n]
323 ##
324 ## Tweak core dumps on and off, or show the current status.
325 core () {
326   case "x$1" in
327     xon|xy|xyes) ulimit -Sc $(ulimit -Hc) ;;
328     xoff|xn|xno) ulimit -Sc 0 ;;
329     x)
330       local l=$(ulimit -Sc)
331       case $l in
332         0) echo "Core dumps disabled" ;;
333         unlimited) echo "Core dumps enabled" ;;
334         *) echo "Core dump limit is $l blocks" ;;
335       esac
336       ;;
337     *)
338       echo >&2 "usage: core [y|n]"
339       return 1
340       ;;
341   esac
342 }
343
344 ## world [NAME]
345 ##
346 ## Set current security world to NAME.  With no NAME, print the currently
347 ## selected world.
348 world () {
349   local nfast=${NFAST_HOME-/opt/nfast}
350   local kmdata
351   case "$#" in
352     0)
353       echo "${NFAST_KMDATA#$nfast/kmdata-}"
354       ;;
355     *)
356       if [ -d "$1" ]; then
357         kmdata=$1
358       elif [ -d "$nfast/kmdata-$1" ]; then
359         kmdata=$nfast/kmdata-$1
360       else
361         echo >&2 "world: can't find world $1"
362         return 1
363       fi
364       shift
365       case "$#" in
366         0) export NFAST_KMDATA=$kmdata ;;
367         *) "$@" ;;
368       esac
369       ;;
370   esac
371 }
372
373 ## path-add [VAR] DIR
374 ##
375 ## Add DIR to the beginning of PATH-like variable VAR (defaults to PATH) if
376 ## it's not there already.
377 path_add () {
378   local pathvar export dir val
379   case $# in
380     1) pathvar=PATH dir=$1 export="export PATH" ;;
381     2) pathvar=$1 dir=$2 export=: ;;
382     *) echo >&2 "Usage: $0 [VAR] DIR"; return 1 ;;
383   esac
384   eval val=\$$pathvar
385   case ":$val:" in
386     *:"$dir":*) ;;
387     *) val=$dir:$val ;;
388   esac
389   eval $pathvar=\$val
390   eval $export
391 }
392
393 ## path-remove [VAR] DIR
394 ##
395 ## Remove DIR from PATH-like variable VAR (defaults to PATH); it's not an
396 ## error if DIR isn't in VAR.
397 path_remove () {
398   local pathvar export dir val
399   case $# in
400     1) pathvar=PATH dir=$1 export="export PATH" ;;
401     2) pathvar=$1 dir=$2 export=: ;;
402     *) echo >&2 "Usage: $0 [VAR] DIR"; return 1 ;;
403   esac
404   eval val=\$$pathvar
405   case ":$val:" in
406     :"$dir":) val= ;;
407     :"$dir":*) val=${val#$dir:} ;;
408     *:"$dir":) val=${val%:$dir} ;;
409     *:"$dir":*) val=${val%%:$dir:*}:${val#*:$dir:} ;;
410   esac
411   eval $pathvar=\$val
412   eval $export
413 }
414
415 ## pathhack [-f] +HACK|-HACK...
416 ##
417 ## Each HACK refers to a subdirectory of `~/bin/hacks'.  A hack name preceded
418 ## by `+' adds the directory to the PATH; a `-' removes.  Adding a hack
419 ## that's already on the PATH doesn't do anything unless `-f' is set, in
420 ## which case it gets moved to the beginning.  With no arguments, print the
421 ## currently installed hacks.
422 pathhack () {
423   local p e force arg hack dir
424   p=$PATH
425   if [ $# -eq 0 ]; then
426     while :; do
427       e=${p%%:*}
428       case "$e" in "$HOME/bin/hacks/"*) echo ${e#$HOME/bin/hacks/} ;; esac
429       case "$p" in *:*) p=${p#*:} ;; *) break ;; esac
430     done
431     return
432   fi
433   force=nil
434   while [ $# -gt 0 ]; do
435     arg=$1
436     case "$arg" in
437       -f | --force) force=t; shift; continue ;;
438       --) shift; break ;;
439       [-+]*) ;;
440       *) break; ;;
441     esac
442     hack=${arg#[+-]}
443     dir=$HOME/bin/hacks/$hack
444     if ! [ -d "$dir" ]; then
445       echo "$0: path hack $hack not found"
446       return 1
447     fi
448     case "$arg,$force,:$PATH:" in
449       -*,*,*:"$dir":*) path_remove p "$dir" ;;
450       +*,t,*:"$dir":*) path_remove p "$dir"; path_add p "$dir" ;;
451       +*,nil,*:"$dir":*) ;;
452       +*,*) path_add p "$dir" ;;
453     esac
454     shift
455   done
456   if [ $# -eq 0 ]; then PATH=$p; export PATH
457   else PATH=$p "$@"; fi
458 }
459
460 ###--------------------------------------------------------------------------
461 ### Finishing touches.
462
463 ## Make sure `$HOME/bin' is on the path.
464 path_add "$HOME/bin"
465
466 ## Set the temporary directory again.  (A setuid or setgid program may have
467 ## unhelpfully forgotten this for us.)
468 case ${TMPDIR+t} in
469   t) ;;
470   *) if __mdw_programp tmpdir; then eval $(tmpdir -b); fi ;;
471 esac
472
473 ## For `root' use -- some simple molly-guards.
474 case $(id -u) in
475   0)
476     alias rm="rm -i" cp="cp -i" mv="mv -i"
477     set -o noclobber
478     ;;
479 esac
480
481 ## Install wakeup machinery for long-running programs.
482 case ${DISPLAY+t},${WINDOWID+t},${__mdw_precmd_hook+t},${__mdw_preexec_hook+t} in
483   t,t,t,t)
484     if [ -t 0 ] && __mdw_programp notify-send; then . $HOME/bin/wakey.sh; fi
485     ;;
486 esac
487
488 ## Run any local hooks.
489 __mdw_source_if_exists "$HOME/.shell-local"
490
491 ###----- That's all, folks --------------------------------------------------