rc/functions

   1 #!/bin/bash
   2 ###############################################################################
   3 #
   4 # File:         $RCSfile: functions,v $
   5 # Revision:     $Revision: 1.20 $
   6 # Description:  Common bash functions
   7 # Author:       Andrew@DeFaria.com
   8 # Created:      Thu Jun  6 08:31:57 PDT 1996
   9 # Modified:     $Date: 2013/03/26 20:38:23 $
  10 # Language:     bash
  11 #
  12 # (c) Copyright 2000-2005, Andrew@DeFaria.com, all rights reserved.
  13 #
  14 ###############################################################################
  15 ESC=$(echo "\033")
  16 CR=$(echo "\015")
  17
  18 view_name=
  19
  20 # Function to set the title bar. Works on the terminal emulators listed.
  21 function title_bar {
  22   if [ $# -gt 1 ]; then
  23     ROOT=shift
  24     ROOT="${NORMAL}\($ROOT\)"
  25   fi
  26
  27   prefix="$@"
  28   # Change $HOME -> ~
  29   if [ "${PWD#$HOME*}" != "$PWD" ]; then
  30     current_dir="~${PWD#$HOME*}"
  31   elif [ "$PWD" = "$HOME" ]; then
  32     current_dir=~
  33   else
  34     current_dir="$PWD"
  35   fi
  36
  37   # Remove view name
  38   current_dir="${current_dir#/view/$view_name*}"
  39   current_dir="${current_dir#/sview/$view_name*}"
  40
  41   # Add CVS/Root if there is one
  42   if [ -f "CVS/Root" ]; then
  43     current_dir="$current_dir ($(cat CVS/Root | tr -d $CR))"
  44   fi
  45
  46   # Add GIT info if available
  47   if in_git_repo; then
  48     current_dir="$current_dir [git: $(git branch | sed -n -e 's/^\* \(.*\)/\1/p')]"
  49   fi
  50
  51   if [ "$TERM" = "hpterm" -o \
  52        "$TERM" = "hp"     -o \
  53        "$TERM" = "2392A" ]; then
  54     string=$(echo "${SYSNAME##*:}:$@")
  55     echo -n "${ESC}&f0k${#string}D$string"
  56   elif [ "$TERM" = "dtterm" -o \
  57          "$TERM" = "vt221" ]; then
  58     string=$(echo "${SYSNAME##*:}:$@")
  59     echo -n "${ESC}]2;$string\007"
  60   elif [ "$TERM" = "cygwin" -o \
  61          "$TERM" = "vt100"  -o \
  62          "$TERM" = "xterm"  -o \
  63          "$TERM" = "xtermc" -o \
  64          "$TERM" = "xterm-256color" ]; then
  65     PS1="\[\e]0;$prefix$current_dir\007\]$ROOT\[$B_YELLOW\]$SYSNAME:\[$B_WHITE\]"
  66   fi
  67 } # title_bar
  68
  69 # Function to set the icon name. Works on the terminal emulators listed.
  70 function icon_name {
  71   if [ "$TERM" = "hpterm" -o \
  72        "$TERM" = "hp"     -o \
  73        "$TERM" = "2392A" ]; then
  74     string=$(echo "$1")
  75     echo -n "${ESC}&f-1k${#string}D$string"
  76   elif [ "$TERM" = "dtterm" -o \
  77          "$TERM" = "vt100"  -a "$DTTERM" = "True" ]; then
  78     # Note setting icon_name on vt100 overwrites the title bar so skip it
  79     echo -n "${ESC}]1;$@\007"
  80   fi
  81 } # icon_name
  82
  83 # Sets both the title bar and the icon name.
  84 function title {
  85   title_bar "$@"
  86   icon_name "${SYSNAME##*:}"
  87 } # title
  88
  89 # Sets title bar to machine name and path. Will include a view name if in a
  90 # view and a string to indicate that you are root.
  91 function set_title {
  92   if [ $($id -u) -eq 0 ]; then
  93     root="Wizard "
  94   else
  95     root=
  96   fi
  97
  98   view_name=$(scm pwv -short 2> /dev/null);
  99
 100   if [ $? -ne 0 -o -z "$view_name" ]; then
 101     view_name='*NONE*'
 102   fi
 103
 104   if [[ $view_name = *NONE* ]]; then
 105     view_name=""
 106     title_bar "$root"
 107   else
 108     title_bar "${root}View: $view_name: "
 109   fi
 110
 111   icon_name "${SYSNAME##*:}"
 112 } # set_title
 113
 114 # Sets prompt on terminals listed.
 115 function set_prompt {
 116   if [ $($id -u) -eq 0 ]; then
 117     if [ "$TERM" = "hpterm"         -o \
 118          "$TERM" = "hp"             -o \
 119          "$TERM" = "2392A"          -o \
 120          "$TERM" = "dtterm"         -o \
 121          "$TERM" = "vt100"          -o \
 122          "$TERM" = "xterm"          -o \
 123          "$TERM" = "xtermc"         -o \
 124          "$TERM" = "xterm-256color" -o \
 125          "$TERM" = "vt220" ]; then
 126       ROOT="\[${ROOT_COLOR}\]Wizard\[$NORMAL\] "
 127     else
 128       ROOT="Wizard "
 129     fi
 130   else
 131     ROOT=""
 132   fi
 133
 134   if [ "$TERM" = "vt100"          -o \
 135        "$TERM" = "xterm"          -o \
 136        "$TERM" = "xtermc"         -o \
 137        "$TERM" = "xterm-256color" -o \
 138        "$TERM" = "vt220" ]; then
 139     PS1="$ROOT\[$B_YELLOW\]$SYSNAME:\[$B_WHITE\]"
 140   else
 141     PS1="$ROOT$SYSNAME:"
 142   fi
 143
 144   set_title
 145 } # set_prompt
 146
 147 # Function to override the normal cd command, setting title and prompt.
 148 function mycd {
 149   if [ -z "$1" ]; then
 150     \cd ~
 151   else
 152     \cd "$1"
 153   fi
 154   set_title
 155   set_prompt
 156 } # mycd
 157 export mycd
 158
 159 # Functions to override the normal push/popd commands, setting title and prompt.
 160 function mypushd {
 161   if [ -z "$1" ]; then
 162     \pushd > /dev/null
 163   else
 164     \pushd "$1" > /dev/null
 165   fi
 166   set_title
 167   set_prompt
 168 } # mypushd
 169
 170 function mypopd {
 171   if [ -z "$1" ]; then
 172     cd - > /dev/null
 173   else
 174     \popd "$1" > /dev/null
 175   fi
 176   set_title
 177   set_prompt
 178 } # mypopd
 179
 180 # Function to override rlogin. Note that it fixes up the title and prompt
 181 # upon return.
 182 function rlogin {
 183   /usr/bin/rlogin "$@"
 184   set_title
 185   set_prompt
 186 } # rlogin
 187
 188 # Function to override rsh. Note that it fixes up the title and prompt
 189 # upon return.
 190 function rsh {
 191   /usr/bin/rsh "$@"
 192   set_title
 193   set_prompt
 194 } # rsh
 195
 196 # Function to override ssh. Note that it fixes up the title and prompt
 197 # upon return.
 198 function ssh {
 199   /usr/bin/ssh "$@"
 200   set_title
 201   set_prompt
 202 } # ssh
 203
 204 function processRunning {
 205   if [ $ARCHITECTURE = "FreeBSD" ]; then
 206     psopts="-aux"
 207   else
 208     psopts="-ef"
 209   fi
 210
 211   if [ $1 != '' ]; then
 212     return $(ps $psopts | grep "$1" | grep -v "grep $1" | grep -v "grep -d skip" | wc -l)
 213   fi
 214 } # processRunning
 215
 216 function sj {
 217   if [ $ARCHITECTURE = "FreeBSD" ]; then
 218     psopts="-aux"
 219   else
 220     psopts="-ef"
 221   fi
 222
 223   if [ $# = 0 ]; then
 224     ps $psopts | $PAGER
 225   else
 226     for str; do
 227       ps $psopts | grep "$str" | grep -v "grep $str" | grep -v "grep -d skip"
 228     done
 229   fi
 230 } # sj
 231
 232 function start_imap {
 233   # Starts an ssh tunnel for IMAP
 234   ssh -C -L 143:defaria.com:143 andrew@defaria.com
 235 } # start_imap
 236
 237 function cmdline {
 238   # Displays the command line from the /proc filesystem (if present)
 239
 240   me=$0;
 241
 242   if [ $# -ne 1 ]; then
 243     error "Usage: cmdline <pid>"
 244     return 1
 245   fi
 246
 247   pid=$1;
 248
 249   if [ ! -d "/proc" ]; then
 250     error "This OS has no /proc filesystem"
 251     return 1
 252   fi
 253
 254   if [ ! -d "/proc/$pid" ]; then
 255     error "PID $pid does not exist"
 256     return 1
 257   fi
 258
 259   if [ ! -f "/proc/$pid/cmdline" ]; then
 260     error "PID $pid has no cmdline!"
 261     return 1
 262   fi
 263
 264   cat /proc/$pid/cmdline | tr -c [:print:] " "
 265   display
 266 } # cmdline
 267
 268 function user {
 269   processRunning ypbind;  ypbind=$?
 270   processRunning winbind; winbind=$?
 271
 272   if [ $# -gt 0 ]; then
 273     if [ $ypbind -gt 0 ]; then
 274       ypcat passwd | grep -i $@
 275     elif [ $winbind -gt 0 ]; then
 276       for user in $(wbinfo -u | grep -i $@); do
 277         wbinfo --user-info $user
 278       done
 279     fi
 280   else
 281     if [ $ypbind -gt 0 ]; then
 282       ypcat passwd | $PAGER
 283     elif [ $wbind -gt 0 ]; then
 284       for user in $(wbinfo -u); do
 285         wbinfo --user-info $user
 286       done | $PAGER
 287     fi
 288   fi
 289 } # user
 290
 291 function group {
 292   processRunning ypbind;  ypbind=$?
 293   processRunning winbind; winbind=$?
 294
 295   if [ $# -gt 0 ]; then
 296     if [ $ypbind -gt 0 ]; then
 297       ypcat group | grep -i $@
 298     elif [ $winbind -gt 0 ]; then
 299       for group in $(wbinfo -g | grep -i $@); do
 300         wbinfo --group-info $group
 301       done
 302     fi
 303   else
 304     if [ $ypbind -gt 0 ]; then
 305       ypcat group | $PAGER
 306     elif [ $winbind -gt 0 ]; then
 307       for group in $(wbinfo -g); do
 308         wbinfo --group-info $group
 309       done | $PAGER
 310     fi
 311   fi
 312 } # group