blob: 1a9281df522758094074c3d7071123de1340768f [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
Ryan09dfe752011-05-31 19:17:10 -100012IP_COLOR=$MACHINE_COLOR
Ryan1a83a4d2011-05-31 13:19:20 -100013DIRECTORY_COLOR='\[${bold_green}\]'
14
15VE_COLOR='\[${red}\]'
16RVM_COLOR='\[${purple}\]'
17
18# SCM prompts
19SCM_THEME_PROMPT_DIRTY=' ${bold_red}✗${normal}'
20SCM_THEME_PROMPT_CLEAN=' ${bold_green}✓${normal}'
21
22# Max length of PWD to display
23MAX_PWD_LENGTH=20
24
25# Max length of Git Hex to display
26MAX_GIT_HEX_LENGTH=5
27
Ryan06d71ab2011-05-31 16:10:30 -100028GIT_THEME_PROMPT_PREFIX=' |git:'
29GIT_THEME_PROMPT_SUFFIX='|'
30
31HG_THEME_PROMPT_PREFIX=' |hg:'
32HG_THEME_PROMPT_SUFFIX='|'
33
Ryan2a6d1da2011-05-31 19:03:28 -100034SVN_THEME_PROMPT_PREFIX=' |svn:'
35SVN_THEME_PROMPT_SUFFIX='|'
36
Ryan1a83a4d2011-05-31 13:19:20 -100037# Use http://geoff.greer.fm/lscolors/
38
Ryan09dfe752011-05-31 19:17:10 -100039function ip {
40 echo $(ifconfig en1 | grep "inet " | awk '{ print $2 }')
41}
42
Ryanaa2f1cb2011-05-31 18:41:07 -100043# Override function scm
44function scm {
45 if [[ -d .git ]]; then SCM=$GIT
46 elif [[ -n "$(git symbolic-ref HEAD 2> /dev/null)" ]]; then SCM=$GIT
47 elif [[ -n "$(hg summary 2> /dev/null)" ]]; then SCM=$HG
48 elif [[ -d .svn ]]; then SCM=$SVN
49 else SCM='NONE'
50 fi
51}
52
Ryan1a83a4d2011-05-31 13:19:20 -100053# Displays the current virtualenv information
54function curr_virtualenv_info() {
55 [ ! -z "$VIRTUAL_ENV" ] && echo "`basename $VIRTUAL_ENV`"
56}
57
58# Displays the current rvm information w/gemset
59function curr_rvm_info() {
60 local ruby_version=$(echo $MY_RUBY_HOME | awk -F'-' '{print $2}')
61 local ruby_gemset=$(echo $GEM_HOME | awk -F'@' '{print $2}')
62
63 if [ "$ruby_version" != "" ]; then
64 [ "$ruby_gemset" != "" ] && ruby_gemset="@$ruby_gemset"
65 echo "$ruby_version$ruby_gemset"
66 fi
67}
68
69# Displays using ...
70function virtual_info() {
71 local virtual_env_info=$(curr_virtualenv_info)
72 local rvm_info=$(curr_rvm_info)
73 local prompt="using"
74
75 # If no virtual info, just return
76 [ "$virtual_env_info" == "" -a "$rvm_info" == "" ] && return
77
78 # If virtual_env info present, append to prompt
79 [ "$virtual_env_info" != "" ] && prompt="$prompt virtualenv: ${VE_COLOR}$virtual_env_info${DEFAULT_COLOR}"
80
81 if [ "$rvm_info" != "" ]
82 then
83 [ "$virtual_env_info" != "" ] && prompt="$prompt,"
84 prompt="$prompt rvm: ${RVM_COLOR}$rvm_info${DEFAULT_COLOR}"
85 fi
86 echo $prompt
87}
88
Ryan1a83a4d2011-05-31 13:19:20 -100089# SCM information
90function scm_info() {
91 SCM_CHAR=$(scm_char)
92 [ "$SCM_CHAR" == "$SCM_NONE_CHAR" ] && return
93 local prompt="on"
94 [ "$SCM_CHAR" == "$SCM_GIT_CHAR" ] && echo "$prompt$(parse_git_info)" && return
Ryan06d71ab2011-05-31 16:10:30 -100095 [ "$SCM_CHAR" == "$SCM_SVN_CHAR" ] && echo "$prompt$(parse_svn_info)" && return
96 [ "$SCM_CHAR" == "$SCM_HG_CHAR" ] && echo "$prompt$(parse_hg_info)" && return
Ryan1a83a4d2011-05-31 13:19:20 -100097}
98
99# Parse git info
100function parse_git_info() {
101 if [[ -n $(git status -s 2> /dev/null |grep -v ^# |grep -v "working directory clean") ]]; then
102 state=${GIT_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
103 else
104 state=${GIT_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
105 fi
106 prefix=${GIT_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
107 suffix=${GIT_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
108 ref=$(git symbolic-ref HEAD 2> /dev/null) || return
109 rawhex=$(git rev-parse HEAD 2>/dev/null) || return
110
Ryanaa2f1cb2011-05-31 18:41:07 -1000111 echo "$prefix${ref#refs/heads/}:${rawhex:0:$MAX_GIT_HEX_LENGTH}$state$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -1000112}
113
Ryan06d71ab2011-05-31 16:10:30 -1000114# Parse hg info
115function parse_hg_info() {
Ryanaa2f1cb2011-05-31 18:41:07 -1000116 if [[ -n $(hg status 2> /dev/null) ]]; then
Ryan06d71ab2011-05-31 16:10:30 -1000117 state=${HG_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
118 else
119 state=${HG_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
120 fi
121 prefix=${HG_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
122 suffix=${HG_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
Ryanaa2f1cb2011-05-31 18:41:07 -1000123 branch=$(hg summary 2> /dev/null | grep branch | awk '{print $2}')
124 changeset=$(hg summary 2> /dev/null | grep parent | awk '{print $2}')
Ryan06d71ab2011-05-31 16:10:30 -1000125
Ryanaa2f1cb2011-05-31 18:41:07 -1000126 echo "$prefix${branch}:${changeset#*:}$state$suffix"
Ryan06d71ab2011-05-31 16:10:30 -1000127}
128
Ryan2a6d1da2011-05-31 19:03:28 -1000129# Parse svn info
130function parse_svn_info() {
131 if [[ -n $(svn status --ignore-externals -q 2> /dev/null) ]]; then
132 state=${SVN_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
133 else
134 state=${SVN_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
135 fi
136 prefix=${SVN_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
137 suffix=${SVN_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
138 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
139 revision=$(svn info 2> /dev/null | sed -ne 's#^Revision: ##p' )
140 [[ -z $ref ]] && return
141 echo -e "$prefix$ref:$revision$state$suffix"
142}
Ryan06d71ab2011-05-31 16:10:30 -1000143
Ryan1a83a4d2011-05-31 13:19:20 -1000144# Displays last X characters of pwd
145function limited_pwd() {
146
147 # Replace $HOME with ~ if possible
148 RELATIVE_PWD=${PWD/#$HOME/\~}
149
150 local offset=$((${#RELATIVE_PWD}-$MAX_PWD_LENGTH))
151
152 if [ $offset -gt "0" ]
153 then
154 local truncated_symbol="..."
155 TRUNCATED_PWD=${RELATIVE_PWD:$offset:$MAX_PWD_LENGTH}
156 echo "${truncated_symbol}/${TRUNCATED_PWD#*/}"
157 else
158 echo "${RELATIVE_PWD}"
159 fi
160}
161
162# Displays the current prompt
163function prompt() {
164
165 local UC=$USER_COLOR
166 [ $UID -eq "0" ] && UC=$SUPERUSER_COLOR
167
Ryan09dfe752011-05-31 19:17:10 -1000168 PS1="$(scm_char) ${UC}\u ${DEFAULT_COLOR}at ${MACHINE_COLOR}\h ${DEFAULT_COLOR}(${IP_COLOR}$(ip)${DEFAULT_COLOR})${DEFAULT_COLOR} in ${DIRECTORY_COLOR}$(limited_pwd)${DEFAULT_COLOR} $(virtual_info) $(scm_info) \$ "
Ryan1a83a4d2011-05-31 13:19:20 -1000169 PS2='> '
170 PS4='+ '
171}
172
173PROMPT_COMMAND=prompt