blob: 378512fa49652a0dfcc6196a2118a6f9e256957a [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
Ryan Kanno0611f862011-05-31 21:38:41 -100013DIRECTORY_COLOR='\[${green}\]'
Ryan1a83a4d2011-05-31 13:19:20 -100014
Ryan Kanno0611f862011-05-31 21:38:41 -100015VE_COLOR='\[${cyan}\]'
16RVM_COLOR='\[${cyan}\]'
Ryan1a83a4d2011-05-31 13:19:20 -100017
Ryan Kanno9bd5b542011-05-31 21:23:33 -100018SCM_COLOR=$ORANGE
19REF_COLOR='\[${purple}\]'
20
Ryan1a83a4d2011-05-31 13:19:20 -100021# SCM prompts
22SCM_THEME_PROMPT_DIRTY=' ${bold_red}✗${normal}'
23SCM_THEME_PROMPT_CLEAN=' ${bold_green}✓${normal}'
24
25# Max length of PWD to display
26MAX_PWD_LENGTH=20
27
28# Max length of Git Hex to display
29MAX_GIT_HEX_LENGTH=5
30
Ryan Kanno0611f862011-05-31 21:38:41 -100031# Removed prefix/suffix
32SCM_THEME_PROMPT_PREFIX=" "
33SCM_THEME_PROMPT_SUFFIX=""
Ryan1a83a4d2011-05-31 13:19:20 -100034
Ryan09dfe752011-05-31 19:17:10 -100035function ip {
36 echo $(ifconfig en1 | grep "inet " | awk '{ print $2 }')
37}
38
Ryanaa2f1cb2011-05-31 18:41:07 -100039# Override function scm
40function scm {
41 if [[ -d .git ]]; then SCM=$GIT
42 elif [[ -n "$(git symbolic-ref HEAD 2> /dev/null)" ]]; then SCM=$GIT
43 elif [[ -n "$(hg summary 2> /dev/null)" ]]; then SCM=$HG
44 elif [[ -d .svn ]]; then SCM=$SVN
45 else SCM='NONE'
46 fi
47}
48
Ryan1a83a4d2011-05-31 13:19:20 -100049# Displays the current virtualenv information
50function curr_virtualenv_info() {
51 [ ! -z "$VIRTUAL_ENV" ] && echo "`basename $VIRTUAL_ENV`"
52}
53
54# Displays the current rvm information w/gemset
55function curr_rvm_info() {
56 local ruby_version=$(echo $MY_RUBY_HOME | awk -F'-' '{print $2}')
57 local ruby_gemset=$(echo $GEM_HOME | awk -F'@' '{print $2}')
58
59 if [ "$ruby_version" != "" ]; then
60 [ "$ruby_gemset" != "" ] && ruby_gemset="@$ruby_gemset"
61 echo "$ruby_version$ruby_gemset"
62 fi
63}
64
65# Displays using ...
66function virtual_info() {
67 local virtual_env_info=$(curr_virtualenv_info)
68 local rvm_info=$(curr_rvm_info)
Ryan1a83a4d2011-05-31 13:19:20 -100069
70 # If no virtual info, just return
71 [ "$virtual_env_info" == "" -a "$rvm_info" == "" ] && return
72
Ryan Kanno0611f862011-05-31 21:38:41 -100073 local prompt=" using"
74
Ryan1a83a4d2011-05-31 13:19:20 -100075 # If virtual_env info present, append to prompt
76 [ "$virtual_env_info" != "" ] && prompt="$prompt virtualenv: ${VE_COLOR}$virtual_env_info${DEFAULT_COLOR}"
77
78 if [ "$rvm_info" != "" ]
79 then
80 [ "$virtual_env_info" != "" ] && prompt="$prompt,"
81 prompt="$prompt rvm: ${RVM_COLOR}$rvm_info${DEFAULT_COLOR}"
82 fi
Ryan Kanno0611f862011-05-31 21:38:41 -100083 echo "$prompt"
Ryan1a83a4d2011-05-31 13:19:20 -100084}
85
Ryan1a83a4d2011-05-31 13:19:20 -100086# SCM information
87function scm_info() {
88 SCM_CHAR=$(scm_char)
89 [ "$SCM_CHAR" == "$SCM_NONE_CHAR" ] && return
Ryan Kanno0611f862011-05-31 21:38:41 -100090 local prompt=" on"
Ryan1a83a4d2011-05-31 13:19:20 -100091 [ "$SCM_CHAR" == "$SCM_GIT_CHAR" ] && echo "$prompt$(parse_git_info)" && return
Ryan06d71ab2011-05-31 16:10:30 -100092 [ "$SCM_CHAR" == "$SCM_SVN_CHAR" ] && echo "$prompt$(parse_svn_info)" && return
93 [ "$SCM_CHAR" == "$SCM_HG_CHAR" ] && echo "$prompt$(parse_hg_info)" && return
Ryan1a83a4d2011-05-31 13:19:20 -100094}
95
96# Parse git info
97function parse_git_info() {
98 if [[ -n $(git status -s 2> /dev/null |grep -v ^# |grep -v "working directory clean") ]]; then
99 state=${GIT_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
100 else
101 state=${GIT_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
102 fi
103 prefix=${GIT_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
104 suffix=${GIT_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
105 ref=$(git symbolic-ref HEAD 2> /dev/null) || return
106 rawhex=$(git rev-parse HEAD 2>/dev/null) || return
107
Ryan Kanno9bd5b542011-05-31 21:23:33 -1000108 echo "$prefix${REF_COLOR}${ref#refs/heads/}${DEFAULT_COLOR}:${rawhex:0:$MAX_GIT_HEX_LENGTH}$state$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -1000109}
110
Ryan06d71ab2011-05-31 16:10:30 -1000111# Parse hg info
112function parse_hg_info() {
Ryanaa2f1cb2011-05-31 18:41:07 -1000113 if [[ -n $(hg status 2> /dev/null) ]]; then
Ryan06d71ab2011-05-31 16:10:30 -1000114 state=${HG_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
115 else
116 state=${HG_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
117 fi
118 prefix=${HG_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
119 suffix=${HG_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
Ryanaa2f1cb2011-05-31 18:41:07 -1000120 branch=$(hg summary 2> /dev/null | grep branch | awk '{print $2}')
121 changeset=$(hg summary 2> /dev/null | grep parent | awk '{print $2}')
Ryan06d71ab2011-05-31 16:10:30 -1000122
Ryan Kanno9bd5b542011-05-31 21:23:33 -1000123 echo "$prefix${REF_COLOR}${branch}${DEFAULT_COLOR}:${changeset#*:}$state$suffix"
Ryan06d71ab2011-05-31 16:10:30 -1000124}
125
Ryan2a6d1da2011-05-31 19:03:28 -1000126# Parse svn info
127function parse_svn_info() {
128 if [[ -n $(svn status --ignore-externals -q 2> /dev/null) ]]; then
129 state=${SVN_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
130 else
131 state=${SVN_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
132 fi
133 prefix=${SVN_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
134 suffix=${SVN_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
135 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
136 revision=$(svn info 2> /dev/null | sed -ne 's#^Revision: ##p' )
137 [[ -z $ref ]] && return
Ryan Kanno9bd5b542011-05-31 21:23:33 -1000138 echo -e "$prefix${REF_COLOR}$ref${DEFAULT_COLOR}:$revision$state$suffix"
Ryan2a6d1da2011-05-31 19:03:28 -1000139}
Ryan06d71ab2011-05-31 16:10:30 -1000140
Ryan1a83a4d2011-05-31 13:19:20 -1000141# Displays last X characters of pwd
142function limited_pwd() {
143
144 # Replace $HOME with ~ if possible
145 RELATIVE_PWD=${PWD/#$HOME/\~}
146
147 local offset=$((${#RELATIVE_PWD}-$MAX_PWD_LENGTH))
148
149 if [ $offset -gt "0" ]
150 then
151 local truncated_symbol="..."
152 TRUNCATED_PWD=${RELATIVE_PWD:$offset:$MAX_PWD_LENGTH}
153 echo "${truncated_symbol}/${TRUNCATED_PWD#*/}"
154 else
155 echo "${RELATIVE_PWD}"
156 fi
157}
158
159# Displays the current prompt
160function prompt() {
161
162 local UC=$USER_COLOR
163 [ $UID -eq "0" ] && UC=$SUPERUSER_COLOR
164
Ryan Kanno0611f862011-05-31 21:38:41 -1000165 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 -1000166 PS2='> '
167 PS4='+ '
168}
169
170PROMPT_COMMAND=prompt