From 4fc095d48d84d65c4f55ac58e6616b50b96e9b5e Mon Sep 17 00:00:00 2001
From: Caio Gondim <me@caiogondim.com>
Date: Tue, 5 Jan 2016 22:59:10 +0100
Subject: [PATCH] Add BG and FG variables for TIME_EXEC segment customisation

---
 README.md              | 2 ++
 bullet-train.zsh-theme | 9 ++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 136ad17..4cbeee5 100644
--- a/README.md
+++ b/README.md
@@ -229,6 +229,8 @@ or don't want to see. All options must be overridden in your **.zshrc** file.
 |--------|-------|-------|
 |`BULLETTRAIN_EXEC_TIME_SHOW`|`false`|Show/hide that segment
 |`BULLETTRAIN_EXEC_TIME_ELAPSED`|5|Elapsed time of command execution
+|`BULLETTRAIN_EXEC_TIME_BG`|`yellow`|Background color
+|`BULLETTRAIN_EXEC_TIME_FG`|`black`|Foreground color
 
 ## Contributors
 
diff --git a/bullet-train.zsh-theme b/bullet-train.zsh-theme
index c05aa35..48a3f22 100644
--- a/bullet-train.zsh-theme
+++ b/bullet-train.zsh-theme
@@ -251,6 +251,13 @@ fi
 if [ ! -n "${BULLETTRAIN_EXEC_TIME_ELAPSED+1}" ]; then
   BULLETTRAIN_EXEC_TIME_ELAPSED=5
 fi
+if [ ! -n "${BULLETTRAIN_EXEC_TIME_BG+1}" ]; then
+  BULLETTRAIN_EXEC_TIME_BG=yellow
+fi
+if [ ! -n "${BULLETTRAIN_EXEC_TIME_FG+1}" ]; then
+  BULLETTRAIN_EXEC_TIME_FG=black
+fi
+
 
 # ------------------------------------------------------------------------------
 # SEGMENT DRAWING
@@ -318,7 +325,7 @@ prompt_cmd_exec_time() {
   local stop=`date +%s`
   local start=${cmd_timestamp:-$stop}
   let local elapsed=$stop-$start
-  [ $elapsed -gt $BULLETTRAIN_EXEC_TIME_ELAPSED ] && prompt_segment yellow white "${elapsed}s"
+  [ $elapsed -gt $BULLETTRAIN_EXEC_TIME_ELAPSED ] && prompt_segment $BULLETTRAIN_EXEC_TIME_BG $BULLETTRAIN_EXEC_TIME_FG "${elapsed}s"
 }
 
 # Custom