X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=.profile.d%2Fprompt.bashrc;h=f874d595e3e00ea13d96f9abeb026c37c6105d4a;hb=95120cbd748ebe21f1cf944911632f9afdad1958;hp=58c2222421123284dc0fccd9857b04b84fcf2445;hpb=6d1a3c415482b587dbadc1323ff37c1319ecfd2e;p=profile.git

diff --git a/.profile.d/prompt.bashrc b/.profile.d/prompt.bashrc
index 58c2222..f874d59 100644
--- a/.profile.d/prompt.bashrc
+++ b/.profile.d/prompt.bashrc
@@ -5,13 +5,28 @@
 
 SSH_FORWARDED=$(get_remote_ip -n)
 if [ ! -z "$SSH_FORWARDED" ]; then
-  if [ ! "$SSH_FORWARDED" = "10.99.10.121" -a ! "$SSH_FORWARDED" = "ip01pc" ]; then
+  if [ ! "$SSH_FORWARDED" = "62.189.242.202" -a ! "$SSH_FORWARDED" = "62.189.241.202" -a ! "$SSH_FORWARDED" = "193.128.72.68" -a ! "$SSH_FORWARDED" = "localhost" ]; then
     SSH_FORWARDED=" (from $SSH_FORWARDED)"
   else
     SSH_FORWARDED=
   fi
 fi
+
+# Unscientific check for grid jobs.
+case ${SSH_CLIENT##* } in
+  ""|22|2222);;
+  *) SSH_FORWARDED="$SSH_FORWARDED [grid]"
+esac
+
 info="$HOSTNAME$SSH_FORWARDED"
-blank=$(echo $info | sed 's/./ /g')
-export PROMPT_COMMAND="echo -ne '\033]0;$info\007\r$blank\r'"
-unset info blank SSH_FORWARDED
+
+# Check for vim.
+vimchild=" [vim]"
+if ps -o comm= -p $PPID 2>/dev/null | grep '^vim*$' &>/dev/null; then
+  # Clear [vim] when exiting.
+  trap "echo -ne '\\033]0;$info\\007'" exit
+  info="$info$vimchild"
+fi
+blank=$(echo "$info" | sed 's/./ /g')
+export PROMPT_COMMAND="echo -ne '\033]0;$info\007\r$blank \r'"
+unset blank info vimchild SSH_FORWARDED