Merge pull request #227 from KuhlTime/master

Add GNU Screen bullet

README.md

@@ -241,6 +241,14 @@ This environment var is used by aws-cli and other tools to use the right access
 
 The git prompt can be disabled for a specific repository by setting a git config flag: `get config oh-my-zsh.hide-status 1`. This is useful to avoid performance issues for particularly huge repositories.
 
+### Screen
+
+|Variable|Default|Meaning
+|--------|-------|-------|
+|`BULLETTRAIN_SCREEN_BG`|`white`|Background color
+|`BULLETTRAIN_SCREEN_FG`|`black`|Foreground color
+|`BULLETTRAIN_SCREEN_PREFIX`|`⬗`|Prefix of the segment
+
 ### Mercurial/HG
 
 |Variable|Default|Meaning

bullet-train.zsh-theme

@@ -23,6 +23,7 @@ if [ ! -n "${BULLETTRAIN_PROMPT_ORDER+1}" ]; then
     custom
     context
     dir
+    screen
     perl
     ruby
     virtualenv
@@ -275,6 +276,17 @@ else
   ZSH_THEME_GIT_PROMPT_DIVERGED=$BULLETTRAIN_GIT_PROMPT_DIVERGED
 fi
 
+# SCREEN
+if [ ! -n "${BULLETTRAIN_SCREEN_BG+1}" ]; then
+  BULLETTRAIN_SCREEN_BG=white
+fi
+if [ ! -n "${BULLETTRAIN_SCREEN_FG+1}" ]; then
+  BULLETTRAIN_SCREEN_FG=black
+fi
+if [ ! -n "${BULLETTRAIN_SCREEN_PREFIX+1}" ]; then
+  BULLETTRAIN_SCREEN_PREFIX="⬗"
+fi
+
 # COMMAND EXECUTION TIME
 if [ ! -n "${BULLETTRAIN_EXEC_TIME_ELAPSED+1}" ]; then
   BULLETTRAIN_EXEC_TIME_ELAPSED=5
@@ -540,6 +552,14 @@ prompt_aws() {
   fi
 }
 
+# SCREEN Session
+prompt_screen() {
+  local session_name="$STY"
+  if [[ "$session_name" != "" ]]; then
+    prompt_segment $BULLETTRAIN_SCREEN_BG $BULLETTRAIN_SCREEN_FG $BULLETTRAIN_SCREEN_PREFIX" $session_name"
+  fi
+}
+
 prompt_time() {
   if [[ $BULLETTRAIN_TIME_12HR == true ]]; then
     prompt_segment $BULLETTRAIN_TIME_BG $BULLETTRAIN_TIME_FG %D{%r}