From b55126e7d3a1d4f74f9e272bb6001ba1deefddcd Mon Sep 17 00:00:00 2001
From: nojhan <nojhan@gmail.com>
Date: Tue, 26 Feb 2013 14:18:58 +0100
Subject: [PATCH] Display colored brackets in a term multiplexer

A pair of square brackets, colored in blue if your current shell is running in a
terminal multiplexer (screen or tmux).
---
 README.md    |  5 +++++
 liquidprompt | 25 +++++++++++++++++++++----
 2 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 69659ec..5ffaba1 100644
--- a/README.md
+++ b/README.md
@@ -44,6 +44,8 @@ colormap that became more and more noticeable with increasing load;
 and bring it back with `fg`), if there is any;
 * the number of attached running jobs (commands started with a `&`), if there is
 any;
+* a pair of square brackets, in blue if your current shell is running in a
+terminal multiplexer (screen or tmux);
 * the current user, in bold yellow if it is root, in light white if it is not
 the same as the login user;
 * a blue @ if the connection has X11 support;
@@ -224,6 +226,7 @@ Set to a null string "" if you do not want color.
     * `LP_COLOR_JOB_D` Detached (aka screen sessions)
     * `LP_COLOR_JOB_R` Running (xterm &)
     * `LP_COLOR_JOB_Z` Sleeping (Ctrl-Z)
+    * `LP_COLOR_IN_MULTIPLEXER` currently running in a terminal multiplexer
 * Last error code
     * `LP_COLOR_ERR`
 * Prompt mark
@@ -267,6 +270,8 @@ Special characters:
 * `LP_MARK_FOSSIL` (default: "⌘") prompt mark in fossil repositories
 * `LP_MARK_UNTRACKED` (default: "*") if git has untracked files
 * `LP_MARK_STASH` (default: "+") if git has stashed modifications
+* `LP_MARK_BRACKET_OPEN` (default: "[") marks around the main part of the prompt
+* `LP_MARK_BRACKET_CLOSE` (default: "]") marks around the main part of the prompt
 * `LP_TITLE_OPEN` (default: "\e]0;") escape character opening a window's title
 * `LP_TITLE_CLOSE` (default: "\a") escape character closing a window's title
 
diff --git a/liquidprompt b/liquidprompt
index b3e974c..bed3d51 100755
--- a/liquidprompt
+++ b/liquidprompt
@@ -226,6 +226,8 @@ _lp_source_config()
     LP_MARK_BZR=${LP_MARK_BZR:-"⚯"}
     LP_MARK_UNTRACKED=${LP_MARK_UNTRACKED:-"*"}
     LP_MARK_STASH=${LP_MARK_STASH:-"+"}
+    LP_MARK_BRACKET_OPEN=${LP_MARK_BRACKET_OPEN:-"["}
+    LP_MARK_BRACKET_CLOSE=${LP_MARK_BRACKET_CLOSE:-"]"}
 
     LP_COLOR_PATH=${LP_COLOR_PATH:-$BOLD_WHITE}
     LP_COLOR_PATH_ROOT=${LP_COLOR_PATH_ROOT:-$BOLD_YELLOW}
@@ -255,6 +257,7 @@ _lp_source_config()
     LP_COLOR_DISCHARGING_ABOVE=${LP_COLOR_DISCHARGING_ABOVE:-$YELLOW}
     LP_COLOR_DISCHARGING_UNDER=${LP_COLOR_DISCHARGING_UNDER:-$RED}
     LP_COLOR_TIME=${LP_COLOR_TIME:-$BLUE}
+    LP_COLOR_IN_MULTIPLEXER=${LP_COLOR_IN_MULTIPLEXER:-$BOLD_BLUE}
 
     LP_COLORMAP_0=${LP_COLORMAP_0:-""}
     LP_COLORMAP_1=${LP_COLORMAP_1:-$GREEN}
@@ -570,7 +573,7 @@ _lp_jobcount_color()
 
     local running=$(( $(jobs -r | wc -l) ))
     local stopped=$(( $(jobs -s | wc -l) ))
-    local screens=$(screen -ls 2> /dev/null | grep -c Detach )
+    local screens=$(screen -ls 2> /dev/null | grep -c Detach ) # FIXME add tmux support
     local m_detached="d"
     local m_stop="z"
     local m_run="&"
@@ -607,6 +610,16 @@ _lp_jobcount_color()
     echo -ne "$rep"
 }
 
+
+# Tells if we are running in a terminal multiplexer
+_lp_in_multiplexer()
+{
+    if [[ "$TERM" == "screen" ]] ; then
+        echo -ne "in_multiplexer"
+    fi
+}
+
+
 # Display the return value of the last command, if different from zero
 _lp_return_value()
 {
@@ -1214,6 +1227,10 @@ _lp_set_prompt()
     LP_TIME=$(_lp_sr "$(_lp_time)")
 
     # in main prompt: no space
+    if [[ $(_lp_in_multiplexer) ]] ; then
+        LP_MARK_BRACKET_OPEN="${LP_COLOR_IN_MULTIPLEXER}${LP_MARK_BRACKET_OPEN}${NO_COL}"
+        LP_MARK_BRACKET_CLOSE="${LP_COLOR_IN_MULTIPLEXER}${LP_MARK_BRACKET_CLOSE}${NO_COL}"
+    fi
     LP_PROXY="$(_lp_proxy)"
 
     # right of main prompt: space at left
@@ -1276,18 +1293,18 @@ _lp_set_prompt()
         # add title escape time, jobs, load and battery
         PS1="${LP_PS1_PREFIX}${LP_TIME}${LP_BATT}${LP_LOAD}${LP_JOBS}"
         # add user, host and permissions colon
-        PS1="${PS1}[${LP_USER}${LP_HOST}${LP_PERM}"
+        PS1="${PS1}${LP_MARK_BRACKET_OPEN}${LP_USER}${LP_HOST}${LP_PERM}"
 
         # if not root
         if [[ "$EUID" -ne "0" ]]
         then
             # path in foreground color
-            PS1="${PS1}${LP_PWD}]${LP_VENV}${LP_PROXY}"
+            PS1="${PS1}${LP_PWD}${LP_MARK_BRACKET_CLOSE}${LP_VENV}${LP_PROXY}"
             # add VCS infos
             PS1="${PS1}${LP_VCS}"
         else
             # path in yellow
-            PS1="${PS1}${LP_PWD}]${LP_VENV}${LP_PROXY}"
+            PS1="${PS1}${LP_PWD}${LP_MARK_BRACKET_CLOSE}${LP_VENV}${LP_PROXY}"
             # do not add VCS infos unless told otherwise (LP_ENABLE_VCS_ROOT)
             [[ "$LP_ENABLE_VCS_ROOT" = "1" ]] && PS1="${PS1}${LP_VCS}"
         fi