From 2fa6df3d01ded606478e264759a57913f818051c Mon Sep 17 00:00:00 2001
From: Eryn Wells <eryn@erynwells.me>
Date: Mon, 8 Oct 2012 11:10:01 -0700
Subject: [PATCH] Add some tests for SSH_CONNECTION

---
 zshrc | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/zshrc b/zshrc
index b954157..d1e1a57 100644
--- a/zshrc
+++ b/zshrc
@@ -24,7 +24,7 @@ prompt_newline()
     # Don't print newlines the first time the prompt is displayed.
     if [[ $is_first_prompt == 1 ]]; then   
         is_first_prompt=0
-        return
+        [[ -z $SSH_CONNECTION ]] && return
     fi
 
     echo
@@ -83,7 +83,9 @@ precmd_flags_rprompt()
 
 precmd_assemble_prompt()
 {
-   local p="$PROMPT_NAME on $PROMPT_HOST at $PROMPT_CWD"
+   local p="$PROMPT_NAME "
+   [[ -z "$SSH_CONNECTION" ]] && p+="on $PROMPT_HOST "
+   p+="at $PROMPT_CWD"
    if [[ -n "$PROMPT_REPO" ]]; then
       p+=" on $PROMPT_REPO"
    fi