blob: 16ecf07c0eab7a92c94ae040b5f94eac21e5eef7 [file] [log] [blame]
Ryan1a83a4d2011-05-31 13:19:20 -10001#!/bin/basORANGEh
2
3# Colors
4ORANGE='\e[0;33m'
5GREY='\e[1:37m'
6
7DEFAULT_COLOR='\[${white}\]'
8
9USER_COLOR='\[${purple}\]'
10SUPERUSER_COLOR='\[${red}\]'
11MACHINE_COLOR=$ORANGE
12DIRECTORY_COLOR='\[${bold_green}\]'
13
14VE_COLOR='\[${red}\]'
15RVM_COLOR='\[${purple}\]'
16
17# SCM prompts
18SCM_THEME_PROMPT_DIRTY=' ${bold_red}✗${normal}'
19SCM_THEME_PROMPT_CLEAN=' ${bold_green}✓${normal}'
20
21# Max length of PWD to display
22MAX_PWD_LENGTH=20
23
24# Max length of Git Hex to display
25MAX_GIT_HEX_LENGTH=5
26
Ryan06d71ab2011-05-31 16:10:30 -100027GIT_THEME_PROMPT_PREFIX=' |git:'
28GIT_THEME_PROMPT_SUFFIX='|'
29
30HG_THEME_PROMPT_PREFIX=' |hg:'
31HG_THEME_PROMPT_SUFFIX='|'
32
Ryan2a6d1da2011-05-31 19:03:28 -100033SVN_THEME_PROMPT_PREFIX=' |svn:'
34SVN_THEME_PROMPT_SUFFIX='|'
35
Ryan1a83a4d2011-05-31 13:19:20 -100036# Use http://geoff.greer.fm/lscolors/
37
Ryanaa2f1cb2011-05-31 18:41:07 -100038# Override function scm
39function scm {
40 if [[ -d .git ]]; then SCM=$GIT
41 elif [[ -n "$(git symbolic-ref HEAD 2> /dev/null)" ]]; then SCM=$GIT
42 elif [[ -n "$(hg summary 2> /dev/null)" ]]; then SCM=$HG
43 elif [[ -d .svn ]]; then SCM=$SVN
44 else SCM='NONE'
45 fi
46}
47
Ryan1a83a4d2011-05-31 13:19:20 -100048# Displays the current virtualenv information
49function curr_virtualenv_info() {
50 [ ! -z "$VIRTUAL_ENV" ] && echo "`basename $VIRTUAL_ENV`"
51}
52
53# Displays the current rvm information w/gemset
54function curr_rvm_info() {
55 local ruby_version=$(echo $MY_RUBY_HOME | awk -F'-' '{print $2}')
56 local ruby_gemset=$(echo $GEM_HOME | awk -F'@' '{print $2}')
57
58 if [ "$ruby_version" != "" ]; then
59 [ "$ruby_gemset" != "" ] && ruby_gemset="@$ruby_gemset"
60 echo "$ruby_version$ruby_gemset"
61 fi
62}
63
64# Displays using ...
65function virtual_info() {
66 local virtual_env_info=$(curr_virtualenv_info)
67 local rvm_info=$(curr_rvm_info)
68 local prompt="using"
69
70 # If no virtual info, just return
71 [ "$virtual_env_info" == "" -a "$rvm_info" == "" ] && return
72
73 # If virtual_env info present, append to prompt
74 [ "$virtual_env_info" != "" ] && prompt="$prompt virtualenv: ${VE_COLOR}$virtual_env_info${DEFAULT_COLOR}"
75
76 if [ "$rvm_info" != "" ]
77 then
78 [ "$virtual_env_info" != "" ] && prompt="$prompt,"
79 prompt="$prompt rvm: ${RVM_COLOR}$rvm_info${DEFAULT_COLOR}"
80 fi
81 echo $prompt
82}
83
Ryan1a83a4d2011-05-31 13:19:20 -100084# SCM information
85function scm_info() {
86 SCM_CHAR=$(scm_char)
87 [ "$SCM_CHAR" == "$SCM_NONE_CHAR" ] && return
88 local prompt="on"
89 [ "$SCM_CHAR" == "$SCM_GIT_CHAR" ] && echo "$prompt$(parse_git_info)" && return
Ryan06d71ab2011-05-31 16:10:30 -100090 [ "$SCM_CHAR" == "$SCM_SVN_CHAR" ] && echo "$prompt$(parse_svn_info)" && return
91 [ "$SCM_CHAR" == "$SCM_HG_CHAR" ] && echo "$prompt$(parse_hg_info)" && return
Ryan1a83a4d2011-05-31 13:19:20 -100092}
93
94# Parse git info
95function parse_git_info() {
96 if [[ -n $(git status -s 2> /dev/null |grep -v ^# |grep -v "working directory clean") ]]; then
97 state=${GIT_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
98 else
99 state=${GIT_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
100 fi
101 prefix=${GIT_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
102 suffix=${GIT_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
103 ref=$(git symbolic-ref HEAD 2> /dev/null) || return
104 rawhex=$(git rev-parse HEAD 2>/dev/null) || return
105
Ryanaa2f1cb2011-05-31 18:41:07 -1000106 echo "$prefix${ref#refs/heads/}:${rawhex:0:$MAX_GIT_HEX_LENGTH}$state$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -1000107}
108
Ryan06d71ab2011-05-31 16:10:30 -1000109# Parse hg info
110function parse_hg_info() {
Ryanaa2f1cb2011-05-31 18:41:07 -1000111 if [[ -n $(hg status 2> /dev/null) ]]; then
Ryan06d71ab2011-05-31 16:10:30 -1000112 state=${HG_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
113 else
114 state=${HG_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
115 fi
116 prefix=${HG_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
117 suffix=${HG_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
Ryanaa2f1cb2011-05-31 18:41:07 -1000118 branch=$(hg summary 2> /dev/null | grep branch | awk '{print $2}')
119 changeset=$(hg summary 2> /dev/null | grep parent | awk '{print $2}')
Ryan06d71ab2011-05-31 16:10:30 -1000120
Ryanaa2f1cb2011-05-31 18:41:07 -1000121 echo "$prefix${branch}:${changeset#*:}$state$suffix"
Ryan06d71ab2011-05-31 16:10:30 -1000122}
123
Ryan2a6d1da2011-05-31 19:03:28 -1000124# Parse svn info
125function parse_svn_info() {
126 if [[ -n $(svn status --ignore-externals -q 2> /dev/null) ]]; then
127 state=${SVN_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
128 else
129 state=${SVN_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
130 fi
131 prefix=${SVN_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
132 suffix=${SVN_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
133 ref=$(svn info 2> /dev/null | awk -F/ '/^URL:/ { for (i=0; i<=NF; i++) { if ($i == "branches" || $i == "tags" ) { print $(i+1); break }; if ($i == "trunk") { print $i; break } } }') || return
134 revision=$(svn info 2> /dev/null | sed -ne 's#^Revision: ##p' )
135 [[ -z $ref ]] && return
136 echo -e "$prefix$ref:$revision$state$suffix"
137}
Ryan06d71ab2011-05-31 16:10:30 -1000138
Ryan1a83a4d2011-05-31 13:19:20 -1000139# Displays last X characters of pwd
140function limited_pwd() {
141
142 # Replace $HOME with ~ if possible
143 RELATIVE_PWD=${PWD/#$HOME/\~}
144
145 local offset=$((${#RELATIVE_PWD}-$MAX_PWD_LENGTH))
146
147 if [ $offset -gt "0" ]
148 then
149 local truncated_symbol="..."
150 TRUNCATED_PWD=${RELATIVE_PWD:$offset:$MAX_PWD_LENGTH}
151 echo "${truncated_symbol}/${TRUNCATED_PWD#*/}"
152 else
153 echo "${RELATIVE_PWD}"
154 fi
155}
156
157# Displays the current prompt
158function prompt() {
159
160 local UC=$USER_COLOR
161 [ $UID -eq "0" ] && UC=$SUPERUSER_COLOR
162
163 PS1="$(scm_char) ${UC}\u ${DEFAULT_COLOR}at ${MACHINE_COLOR}\h${DEFAULT_COLOR} in ${DIRECTORY_COLOR}$(limited_pwd)${DEFAULT_COLOR} $(virtual_info) $(scm_info) \$ "
164 PS2='> '
165 PS4='+ '
166}
167
168PROMPT_COMMAND=prompt