From 2d250c2099274b4a66b0de21e6b1d49fe0818dc2 Mon Sep 17 00:00:00 2001
From: Caio Gondim <me@caiogondim.com>
Date: Tue, 6 Jan 2015 07:20:26 +0100
Subject: [PATCH] Improves context segment code

---
 README.md              | 17 +++++++++++++++++
 bullet-train.zsh-theme |  4 ++--
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 7cd6962..ffca455 100644
--- a/README.md
+++ b/README.md
@@ -114,6 +114,23 @@ BULLETTRAIN_TIME_BG=''
 BULLETTRAIN_TIME_FG=''
 ```
 
+### Context
+
+Shows username and machine name when in some different value than default. To
+set the default user use the `BULLETTRAIN_CONTEXT_DEFAULT_USER` variable. To
+show the context segment in when ssh'ing, set the `BULLETTRAIN_IS_SSH_CLIENT` in
+the remote machine.
+
+Default values:
+
+```bash
+BULLETTRAIN_CONTEXT_SHOW=false
+BULLETTRAIN_CONTEXT_BG=black
+BULLETTRAIN_CONTEXT_FG=default
+# BULLETTRAIN_CONTEXT_DEFAULT_USER
+# BULLETTRAIN_IS_SSH_CLIENT
+```
+
 Python virtualenv background, foreground color, visibility and the symbol used
 for the segment:
 
diff --git a/bullet-train.zsh-theme b/bullet-train.zsh-theme
index c753110..94b00f9 100644
--- a/bullet-train.zsh-theme
+++ b/bullet-train.zsh-theme
@@ -248,8 +248,8 @@ prompt_context() {
 
   local user=$(whoami)
 
-  if [[ "$user" != "$DEFAULT_USER" || -n "$SSH_CLIENT" ]]; then
-    prompt_segment $BULLETTRAIN_CONTEXT_BG $BULLETTRAIN_CONTEXT_FG "%(!.%{%F{yellow}%}.)$user@%m"
+  if [[ "$user" != "$BULLETTRAIN_CONTEXT_DEFAULT_USER" || -n "$BULLETTRAIN_IS_SSH_CLIENT" ]]; then
+    prompt_segment $BULLETTRAIN_CONTEXT_BG $BULLETTRAIN_CONTEXT_FG "$user@%m"
   fi
 }