X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=.profile.d%2Fprompt.bashrc;h=e6a8236c27063f0774ba6e92e9bff0dcc2314c52;hb=4fedd438b177cc8adf227da193642b139a1ee5f4;hp=6c2e2bb685387936fe0da223742f69d74f3e608e;hpb=8ec5790d6c5eb9f833f1bb3c0e36c578c419a70d;p=profile.git

diff --git a/.profile.d/prompt.bashrc b/.profile.d/prompt.bashrc
index 6c2e2bb..e6a8236 100644
--- a/.profile.d/prompt.bashrc
+++ b/.profile.d/prompt.bashrc
@@ -4,22 +4,23 @@
 #
 
 SSH_FORWARDED=$(get_remote_ip -n)
-if [ ! -z "$SSH_FORWARDED" -a ! "$SSH_ORIGINAL_COMMAND" = "/usr/NX/bin/nxnode" ]; then
+if [ ! -z "$SSH_FORWARDED" ]; then
   if [ ! "$SSH_FORWARDED" = "62.189.242.202" -a ! "$SSH_FORWARDED" = "62.189.241.202" ]; then
     SSH_FORWARDED=" (from $SSH_FORWARDED)"
   else
     SSH_FORWARDED=
   fi
 fi
-else
-  SSH_FORWARDED=""
-fi
+
+info="$HOSTNAME$SSH_FORWARDED"
+
+# Check for vim.
+vimchild=" [vim]"
 if ps -o comm= -p $PPID 2>/dev/null | grep '^vim*$' &>/dev/null; then
-  VIMCHILD=" [vim]"
-else
-  VIMCHILD=
+  # Clear [vim] when exiting.
+  trap "echo -ne '\\033]0;$info\\007'" exit
+  info="$info$vimchild"
 fi
-info="$HOSTNAME$SSH_FORWARDED$VIMCHILD"
-blank=$(echo $info | sed 's/./ /g')
-export PROMPT_COMMAND="echo -ne '\033]0;$info\007\r$blank\r'"
-unset info blank SSH_FORWARDED VIMCHILD
+blank=$(echo "$info" | sed 's/./ /g')
+export PROMPT_COMMAND="echo -ne '\033]0;$info\007\r$blank \r'"
+unset blank info vimchild SSH_FORWARDED