From bd88ade2631121770bd562d17b0c5f4ad1ddf7fd Mon Sep 17 00:00:00 2001
From: Caio Gondim <me@caiogondim.com>
Date: Sun, 6 Jul 2014 22:24:14 +0200
Subject: [PATCH] Uses agnoster as base

---
 bullet-train.zsh-theme | 250 +++++++++++++++++++++++++----------------
 1 file changed, 154 insertions(+), 96 deletions(-)

diff --git a/bullet-train.zsh-theme b/bullet-train.zsh-theme
index d08fb23..2f2be84 100644
--- a/bullet-train.zsh-theme
+++ b/bullet-train.zsh-theme
@@ -1,105 +1,163 @@
-# Basic colors: RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, BLACK, WHITE
+# README
+#
+# In order for this theme to render correctly, you will need a
+# [Powerline-patched font](https://github.com/Lokaltog/powerline-fonts).
+#
+# In addition, I recommend the
+# [Solarized theme](https://github.com/altercation/solarized/) and, if you're
+# using it on Mac OS X, [iTerm 2](http://www.iterm2.com/) over Terminal.app -
+# it has significantly better color fidelity.
 
-BULLETTRAIN_DATE=%D{%H:%M:%S}
+### Segment drawing
+# A few utility functions to make it easy and re-usable to draw segmented prompts
 
-if [ "$BULLETTRAIN_HIDE_USER_NAME" = "" ] && [ "$BULLETTRAIN_HIDE_HOST_NAME" = "" ]; then
-  BULLETTRAIN_USER_NAME="%n@%M"
-elif [ "$BULLETTRAIN_HIDE_USER_NAME" != "" ] && [ "$BULLETTRAIN_HIDE_HOST_NAME" = "" ]; then
-  BULLETTRAIN_USER_NAME="@%M"
-elif [ "$BULLETTRAIN_HIDE_USER_NAME" = "" ] && [ "$BULLETTRAIN_HIDE_HOST_NAME" != "" ]; then
-  BULLETTRAIN_USER_NAME="%n"
-else
-  BULLETTRAIN_USER_NAME=""
-fi
+CURRENT_BG='NONE'
+SEGMENT_SEPARATOR=''
 
-BULLETTRAIN_CURRENT_PATH="%d"
-
-if [ "$BULLETTRAIN_FULL_CURRENT_PATH" = "" ]; then
-  BULLETTRAIN_CURRENT_PATH="%1~"
-fi
-
-if [ "$BULLETTRAIN_GIT_CLEAN" = "" ]; then
-  BULLETTRAIN_GIT_CLEAN="✔"
-fi
-
-if [ "$BULLETTRAIN_GIT_DIRTY" = "" ]; then
-  BULLETTRAIN_GIT_DIRTY="✘"
-fi
-
-if [ "$BULLETTRAIN_GIT_ADDED" = "" ]; then
-  BULLETTRAIN_GIT_ADDED="%F{green}✚%F{black}"
-fi
-
-if [ "$BULLETTRAIN_GIT_MODIFIED" = "" ]; then
-  BULLETTRAIN_GIT_MODIFIED="%F{blue}✹%F{black}"
-fi
-
-if [ "$BULLETTRAIN_GIT_DELETED" = "" ]; then
-  BULLETTRAIN_GIT_DELETED="%F{red}✖%F{black}"
-fi
-
-if [ "$BULLETTRAIN_GIT_UNTRACKED" = "" ]; then
-  BULLETTRAIN_GIT_UNTRACKED="%F{yellow}✭%F{black}"
-fi
-
-if [ "$BULLETTRAIN_GIT_RENAMED" = "" ]; then
-  BULLETTRAIN_GIT_RENAMED="➜"
-fi
-
-if [ "$BULLETTRAIN_GIT_UNMERGED" = "" ]; then
-  BULLETTRAIN_GIT_UNMERGED="═"
-fi
-
-ZSH_THEME_GIT_PROMPT_PREFIX=" \ue0a0 "
-ZSH_THEME_GIT_PROMPT_SUFFIX=""
-ZSH_THEME_GIT_PROMPT_DIRTY=" $BULLETTRAIN_GIT_DIRTY"
-ZSH_THEME_GIT_PROMPT_CLEAN=" $BULLETTRAIN_GIT_CLEAN"
-
-ZSH_THEME_GIT_PROMPT_ADDED=" $BULLETTRAIN_GIT_ADDED"
-ZSH_THEME_GIT_PROMPT_MODIFIED=" $BULLETTRAIN_GIT_MODIFIED"
-ZSH_THEME_GIT_PROMPT_DELETED=" $BULLETTRAIN_GIT_DELETED"
-ZSH_THEME_GIT_PROMPT_UNTRACKED=" $BULLETTRAIN_GIT_UNTRACKED"
-ZSH_THEME_GIT_PROMPT_RENAMED=" $BULLETTRAIN_GIT_RENAMED"
-ZSH_THEME_GIT_PROMPT_UNMERGED=" $BULLETTRAIN_GIT_UNMERGED"
-ZSH_THEME_GIT_PROMPT_AHEAD=" ⬆"
-ZSH_THEME_GIT_PROMPT_BEHIND=" ⬇"
-ZSH_THEME_GIT_PROMPT_DIVERGED=" ⬍"
-
-if [ "$BULLETTRAIN_SHOW_GIT_ON_RIGHT" = "" ]; then
-  if [ "$BULLETTRAIN_HIDE_GIT_PROMPT_STATUS" = "" ]; then
-    BULLETTRAIN_GIT_INFO_LEFT=" %F{blue}%K{white}"$'\ue0b0'"%F{white}%F{black}%K{white}"$'$(git_prompt_info)$(git_prompt_status)%F{white}'
+# Begin a segment
+# Takes two arguments, background and foreground. Both can be omitted,
+# rendering default background/foreground.
+prompt_segment() {
+  local bg fg
+  [[ -n $1 ]] && bg="%K{$1}" || bg="%k"
+  [[ -n $2 ]] && fg="%F{$2}" || fg="%f"
+  if [[ $CURRENT_BG != 'NONE' && $1 != $CURRENT_BG ]]; then
+    echo -n " %{$bg%F{$CURRENT_BG}%}$SEGMENT_SEPARATOR%{$fg%} "
   else
-    BULLETTRAIN_GIT_INFO_LEFT=" %F{blue}%K{white}"$'\ue0b0'"%F{white}%F{black}%K{white}"$'$(git_prompt_info)%F{white}'
+    echo -n "%{$bg%}%{$fg%} "
   fi
-  BULLETTRAIN_GIT_INFO_RIGHT=""
-else
-  BULLETTRAIN_GIT_INFO_LEFT=""
-  BULLETTRAIN_GIT_INFO_RIGHT="%F{white}"$'\ue0b2'"%F{black}%K{white}"$'$(git_prompt_info)'" %K{white}"
-fi
+  CURRENT_BG=$1
+  [[ -n $3 ]] && echo -n $3
+}
 
-if [ $(id -u) -eq 0 ]; then
-  BULLETTRAIN_SEC1_BG=%K{red}
-  BULLETTRAIN_SEC1_FG=%F{red}
-else
-  BULLETTRAIN_SEC1_BG=%K{green}
-  BULLETTRAIN_SEC1_FG=%F{green}
-fi
-BULLETTRAIN_SEC1_TXT=%F{black}
-if [ "$BULLETTRAIN_DETECT_SSH" != "" ]; then
-  if [ -n "$SSH_CLIENT" ]; then
-    BULLETTRAIN_SEC1_BG=%K{red}
-    BULLETTRAIN_SEC1_FG=%F{red}
-    BULLETTRAIN_SEC1_TXT=%F{white}
+# End the prompt, closing any open segments
+prompt_end() {
+  if [[ -n $CURRENT_BG ]]; then
+    echo -n " %{%k%F{$CURRENT_BG}%}$SEGMENT_SEPARATOR"
+  else
+    echo -n "%{%k%}"
   fi
-fi
+  echo -n "%{%f%}"
+  CURRENT_BG=''
+}
 
-PROMPT="$BULLETTRAIN_SEC1_BG$BULLETTRAIN_SEC1_TXT %k%f$BULLETTRAIN_SEC1_FG%K{blue}"$'\ue0b0'"%k%f%F{white}%K{blue} "$BULLETTRAIN_CURRENT_PATH"%F{blue}"$BULLETTRAIN_GIT_INFO_LEFT" %k"$'\ue0b0'"%f "
-PROMPT="%K{red} %F{red}%K{yellow}"$'\ue0b0'"%F{white}%K{yellow} "$BULLETTRAIN_DATE" %F{yellow}%K{green}"$'\ue0b0'$PROMPT
-PROMPT=$PROMPT"
-%{${fg_bold[default]}%}\$ %{$reset_color%}"
+### Prompt components
+# Each component will draw itself, and hide itself if no information needs to be shown
 
-# Adds blankline before?
-if [ "$BULLETTRAIN_NO_BLANK_LINE" = "" ]; then
-  PROMPT="
-"$PROMPT
-fi
+# Context: user@hostname (who am I and where am I)
+prompt_context() {
+  local user=`whoami`
+
+  if [[ "$user" != "$DEFAULT_USER" || -n "$SSH_CLIENT" ]]; then
+    prompt_segment black default "%(!.%{%F{yellow}%}.)$user@%m"
+  fi
+}
+
+# Git: branch/detached head, dirty status
+prompt_git() {
+  local ref dirty mode repo_path
+  repo_path=$(git rev-parse --git-dir 2>/dev/null)
+
+  if $(git rev-parse --is-inside-work-tree >/dev/null 2>&1); then
+
+    ZSH_THEME_GIT_PROMPT_PREFIX=" \ue0a0 "
+    ZSH_THEME_GIT_PROMPT_SUFFIX=""
+    ZSH_THEME_GIT_PROMPT_DIRTY=" ✘"
+    ZSH_THEME_GIT_PROMPT_CLEAN=" ✔"
+    ZSH_THEME_GIT_PROMPT_ADDED=" %F{green}✚%F{black}"
+    ZSH_THEME_GIT_PROMPT_MODIFIED=" %F{blue}✹%F{black}"
+    ZSH_THEME_GIT_PROMPT_DELETED=" %F{red}✖%F{black}"
+    ZSH_THEME_GIT_PROMPT_UNTRACKED=" %F{yellow}✭%F{black}"
+    ZSH_THEME_GIT_PROMPT_RENAMED=" ➜"
+    ZSH_THEME_GIT_PROMPT_UNMERGED=" ═"
+    ZSH_THEME_GIT_PROMPT_AHEAD=" ⬆"
+    ZSH_THEME_GIT_PROMPT_BEHIND=" ⬇"
+    ZSH_THEME_GIT_PROMPT_DIVERGED=" ⬍"
+
+    prompt_segment white black
+    echo -n $(git_prompt_info)
+  fi
+}
+
+prompt_hg() {
+  local rev status
+  if $(hg id >/dev/null 2>&1); then
+    if $(hg prompt >/dev/null 2>&1); then
+      if [[ $(hg prompt "{status|unknown}") = "?" ]]; then
+        # if files are not added
+        prompt_segment red white
+        st='±'
+      elif [[ -n $(hg prompt "{status|modified}") ]]; then
+        # if any modification
+        prompt_segment yellow black
+        st='±'
+      else
+        # if working copy is clean
+        prompt_segment green black
+      fi
+      echo -n $(hg prompt "☿ {rev}@{branch}") $st
+    else
+      st=""
+      rev=$(hg id -n 2>/dev/null | sed 's/[^-0-9]//g')
+      branch=$(hg id -b 2>/dev/null)
+      if `hg st | grep -Eq "^\?"`; then
+        prompt_segment red black
+        st='±'
+      elif `hg st | grep -Eq "^(M|A)"`; then
+        prompt_segment yellow black
+        st='±'
+      else
+        prompt_segment green black
+      fi
+      echo -n "☿ $rev@$branch" $st
+    fi
+  fi
+}
+
+# Dir: current working directory
+prompt_dir() {
+  prompt_segment blue white '%1~'
+}
+
+# Virtualenv: current working virtualenv
+prompt_virtualenv() {
+  local virtualenv_path="$VIRTUAL_ENV"
+  if [[ -n $virtualenv_path && -n $VIRTUAL_ENV_DISABLE_PROMPT ]]; then
+    prompt_segment blue black "(`basename $virtualenv_path`)"
+  fi
+}
+
+prompt_time() {
+  prompt_segment green white %D{%H:%M:%S}
+}
+
+# Status:
+# - was there an error
+# - am I root
+# - are there background jobs?
+prompt_status() {
+  local symbols
+  symbols=()
+  [[ $RETVAL -ne 0 ]] && symbols+="%{%F{red}%}✘"
+  [[ $UID -eq 0 ]] && symbols+="%{%F{yellow}%}⚡"
+  [[ $(jobs -l | wc -l) -gt 0 ]] && symbols+="%{%F{cyan}%}⚙"
+
+  [[ -n "$symbols" ]] && prompt_segment black default "$symbols"
+}
+
+## Main prompt
+build_prompt() {
+  RETVAL=$?
+  prompt_status
+  prompt_time
+  prompt_virtualenv
+  # prompt_context
+  prompt_dir
+  prompt_git
+  prompt_hg
+  prompt_end
+}
+
+PROMPT='
+%{%f%b%k%}$(build_prompt)
+%{${fg_bold[default]}%}\$ %{$reset_color%}'