blob: ad87e787a3bfa380b9a4441087234990a31718f7 [file] [log] [blame]
Ryan0df98ca2011-06-01 11:22:22 -10001#!/bin/bash
Ryanf3787fd2011-06-01 11:33:41 -10002#
Ryan0df98ca2011-06-01 11:22:22 -10003# This theme was obviously inspired a lot by
Ryanf3787fd2011-06-01 11:33:41 -10004#
Ryan0df98ca2011-06-01 11:22:22 -10005# - Demula theme
6#
7# which in itself was inspired by :
8#
9# - Ronacher's dotfiles (mitsuhikos) - http://github.com/mitsuhiko/dotfiles/tree/master/bash/
10# - Glenbot - http://theglenbot.com/custom-bash-shell-for-development/
11# - My extravagant zsh - http://stevelosh.com/blog/2010/02/my-extravagant-zsh-prompt/
12# - Monokai colors - http://monokai.nl/blog/2006/07/15/textmate-color-theme/
13# - Bash_it modern theme
14#
Ryanf3787fd2011-06-01 11:33:41 -100015# Hawaii50 theme supports :
16#
17# - configurable directory length
18# - hg, svn, git detection (I work in all of them)
19# - virtualenv, rvm + gemsets
Ryan0df98ca2011-06-01 11:22:22 -100020#
Ryan Kannodcb0dde2011-06-07 08:25:45 -100021# Screenshot: http://i.imgur.com/4IAMJ.png
Ryan0df98ca2011-06-01 11:22:22 -100022#
23# by Ryan Kanno <ryankanno@localkinegrinds.com>
24#
25# And yes, we code out in Hawaii. :D
26#
27# Note: I also am really new to this bash scripting game, so if you see things
28# that are flat out wrong, or if you think of something neat, just send a pull
Ryanf3787fd2011-06-01 11:33:41 -100029# request. This probably only works on a Mac - as some functions are OS
30# specific like getting ip, etc.
31#
Ryan0df98ca2011-06-01 11:22:22 -100032
33# COLORS ======================================================================
Ryan Kanno8e110772011-06-17 10:31:49 -100034ORANGE='\[\e[0;33m\]'
Ryan1a83a4d2011-05-31 13:19:20 -100035
36DEFAULT_COLOR='\[${white}\]'
37
38USER_COLOR='\[${purple}\]'
39SUPERUSER_COLOR='\[${red}\]'
40MACHINE_COLOR=$ORANGE
Ryan0df98ca2011-06-01 11:22:22 -100041IP_COLOR=$ORANGE
Ryan Kanno0611f862011-05-31 21:38:41 -100042DIRECTORY_COLOR='\[${green}\]'
Ryan1a83a4d2011-05-31 13:19:20 -100043
Ryan Kanno0611f862011-05-31 21:38:41 -100044VE_COLOR='\[${cyan}\]'
45RVM_COLOR='\[${cyan}\]'
Ryan1a83a4d2011-05-31 13:19:20 -100046
Ryan Kanno9bd5b542011-05-31 21:23:33 -100047REF_COLOR='\[${purple}\]'
48
Ryan1a83a4d2011-05-31 13:19:20 -100049# SCM prompts
Ryan Kanno8e110772011-06-17 10:31:49 -100050SCM_THEME_PROMPT_DIRTY=' \[${bold_red}\]✗\[${normal}\]'
51SCM_THEME_PROMPT_CLEAN=' \[${bold_green}\]✓\[${normal}\]'
Ryan09bd0ef2011-06-08 13:34:27 -100052SCM_THEME_PROMPT_PREFIX=' on '
Ryan0713fdc2011-06-08 11:32:50 -100053SCM_THEME_PROMPT_SUFFIX=''
54
Ryan Kannob96345e2011-06-09 12:18:11 -100055# rvm prompts
Ryan0713fdc2011-06-08 11:32:50 -100056RVM_THEME_PROMPT_PREFIX=''
57RVM_THEME_PROMPT_SUFFIX=''
58
Ryan Kannob96345e2011-06-09 12:18:11 -100059# virtualenv prompts
Ryan0713fdc2011-06-08 11:32:50 -100060VIRTUALENV_THEME_PROMPT_PREFIX=''
61VIRTUALENV_THEME_PROMPT_SUFFIX=''
Ryan1a83a4d2011-05-31 13:19:20 -100062
Ryan Kannob96345e2011-06-09 12:18:11 -100063# virtual prompts
64VIRTUAL_PROMPT_ENABLED=1
65
66VIRTUAL_THEME_PROMPT_PREFIX=' using '
67VIRTUAL_THEME_PROMPT_SUFFIX=''
68
Ryan1a83a4d2011-05-31 13:19:20 -100069# Max length of PWD to display
70MAX_PWD_LENGTH=20
71
72# Max length of Git Hex to display
73MAX_GIT_HEX_LENGTH=5
74
Ryan Kanno49f897a2011-06-09 13:20:41 -100075# IP address
76IP_SEPARATOR=', '
77
Ryan0df98ca2011-06-01 11:22:22 -100078# FUNCS =======================================================================
Ryan Kannodcb0dde2011-06-07 08:25:45 -100079
Ryan09dfe752011-05-31 19:17:10 -100080function ip {
Ryan Kanno49f897a2011-06-09 13:20:41 -100081 myip=$(curl -s checkip.dyndns.org | grep -Eo '[0-9\.]+')
82 echo -e "$(ips | sed -e :a -e '$!N;s/\n/${IP_SEPARATOR}/;ta' | sed -e 's/127\.0\.0\.1\${IP_SEPARATOR}//g'), ${myip}"
Ryan09dfe752011-05-31 19:17:10 -100083}
84
Ryan Kannob96345e2011-06-09 12:18:11 -100085# Displays virtual info prompt (virtualenv/rvm)
86function virtual_prompt_info() {
Ryan0713fdc2011-06-08 11:32:50 -100087 local virtual_env_info=$(virtualenv_prompt)
88 local rvm_info=$(rvm_version_prompt)
Ryan Kannob96345e2011-06-09 12:18:11 -100089 local virtual_prompt=""
90
91 local prefix=${VIRTUAL_THEME_PROMPT_PREFIX}
92 local suffix=${VIRTUAL_THEME_PROMPT_SUFFIX}
Ryan1a83a4d2011-05-31 13:19:20 -100093
94 # If no virtual info, just return
Ryan Kannob96345e2011-06-09 12:18:11 -100095 [[ -z "$virtual_env_info" && -z "$rvm_info" ]] && return
Ryan Kanno0611f862011-05-31 21:38:41 -100096
Ryan1a83a4d2011-05-31 13:19:20 -100097 # If virtual_env info present, append to prompt
Ryan Kannob96345e2011-06-09 12:18:11 -100098 [[ -n "$virtual_env_info" ]] && virtual_prompt="virtualenv: ${VE_COLOR}$virtual_env_info${DEFAULT_COLOR}"
Ryan1a83a4d2011-05-31 13:19:20 -100099
Ryan Kannob96345e2011-06-09 12:18:11 -1000100 if [[ -n "$rvm_info" ]]
Ryan1a83a4d2011-05-31 13:19:20 -1000101 then
Ryan Kannob96345e2011-06-09 12:18:11 -1000102 [[ -n "$virtual_env_info" ]] && virtual_prompt="$virtual_prompt, "
103 virtual_prompt="${virtual_prompt}rvm: ${RVM_COLOR}$rvm_info${DEFAULT_COLOR}"
Ryan1a83a4d2011-05-31 13:19:20 -1000104 fi
Ryan Kannob96345e2011-06-09 12:18:11 -1000105 echo -e "$prefix$virtual_prompt$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -1000106}
107
Ryan1a83a4d2011-05-31 13:19:20 -1000108# Parse git info
Ryan28e2aed2011-06-08 11:57:16 -1000109function git_prompt_info() {
Ryan1a83a4d2011-05-31 13:19:20 -1000110 if [[ -n $(git status -s 2> /dev/null |grep -v ^# |grep -v "working directory clean") ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000111 state=${GIT_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan1a83a4d2011-05-31 13:19:20 -1000112 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000113 state=${GIT_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan1a83a4d2011-05-31 13:19:20 -1000114 fi
115 prefix=${GIT_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
116 suffix=${GIT_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
117 ref=$(git symbolic-ref HEAD 2> /dev/null) || return
Ryan Kannob96345e2011-06-09 12:18:11 -1000118 commit_id=$(git rev-parse HEAD 2>/dev/null) || return
Ryan1a83a4d2011-05-31 13:19:20 -1000119
Ryan Kannob96345e2011-06-09 12:18:11 -1000120 echo -e "$prefix${REF_COLOR}${ref#refs/heads/}${DEFAULT_COLOR}:${commit_id:0:$MAX_GIT_HEX_LENGTH}$state$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -1000121}
122
Ryan06d71ab2011-05-31 16:10:30 -1000123# Parse hg info
Ryan28e2aed2011-06-08 11:57:16 -1000124function hg_prompt_info() {
Ryanaa2f1cb2011-05-31 18:41:07 -1000125 if [[ -n $(hg status 2> /dev/null) ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000126 state=${HG_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan06d71ab2011-05-31 16:10:30 -1000127 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000128 state=${HG_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan06d71ab2011-05-31 16:10:30 -1000129 fi
130 prefix=${HG_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
131 suffix=${HG_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
Ryanaa2f1cb2011-05-31 18:41:07 -1000132 branch=$(hg summary 2> /dev/null | grep branch | awk '{print $2}')
133 changeset=$(hg summary 2> /dev/null | grep parent | awk '{print $2}')
Ryan06d71ab2011-05-31 16:10:30 -1000134
Ryan Kannob96345e2011-06-09 12:18:11 -1000135 echo -e "$prefix${REF_COLOR}${branch}${DEFAULT_COLOR}:${changeset#*:}$state$suffix"
Ryan06d71ab2011-05-31 16:10:30 -1000136}
137
Ryan2a6d1da2011-05-31 19:03:28 -1000138# Parse svn info
Ryan28e2aed2011-06-08 11:57:16 -1000139function svn_prompt_info() {
Ryan2a6d1da2011-05-31 19:03:28 -1000140 if [[ -n $(svn status --ignore-externals -q 2> /dev/null) ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000141 state=${SVN_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan2a6d1da2011-05-31 19:03:28 -1000142 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000143 state=${SVN_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan2a6d1da2011-05-31 19:03:28 -1000144 fi
145 prefix=${SVN_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
146 suffix=${SVN_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
147 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
Ryan2a6d1da2011-05-31 19:03:28 -1000148 [[ -z $ref ]] && return
Ryan Kannob96345e2011-06-09 12:18:11 -1000149
150 revision=$(svn info 2> /dev/null | sed -ne 's#^Revision: ##p' )
151
Ryan Kanno9bd5b542011-05-31 21:23:33 -1000152 echo -e "$prefix${REF_COLOR}$ref${DEFAULT_COLOR}:$revision$state$suffix"
Ryan2a6d1da2011-05-31 19:03:28 -1000153}
Ryan06d71ab2011-05-31 16:10:30 -1000154
Ryan1a83a4d2011-05-31 13:19:20 -1000155# Displays last X characters of pwd
156function limited_pwd() {
157
158 # Replace $HOME with ~ if possible
159 RELATIVE_PWD=${PWD/#$HOME/\~}
160
161 local offset=$((${#RELATIVE_PWD}-$MAX_PWD_LENGTH))
162
163 if [ $offset -gt "0" ]
164 then
165 local truncated_symbol="..."
166 TRUNCATED_PWD=${RELATIVE_PWD:$offset:$MAX_PWD_LENGTH}
Ryan Kannob96345e2011-06-09 12:18:11 -1000167 echo -e "${truncated_symbol}/${TRUNCATED_PWD#*/}"
Ryan1a83a4d2011-05-31 13:19:20 -1000168 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000169 echo -e "${RELATIVE_PWD}"
Ryan1a83a4d2011-05-31 13:19:20 -1000170 fi
171}
172
173# Displays the current prompt
174function prompt() {
Ryan Kannob96345e2011-06-09 12:18:11 -1000175 local UC=$USER_COLOR
176 [ $UID -eq "0" ] && UC=$SUPERUSER_COLOR
Ryan1a83a4d2011-05-31 13:19:20 -1000177
Ryan Kannob96345e2011-06-09 12:18:11 -1000178 if [[ $VIRTUAL_PROMPT_ENABLED == 1 ]]; then
Ryan Kanno8e110772011-06-17 10:31:49 -1000179 PS1="$(scm_char) ${UC}\u ${DEFAULT_COLOR}at ${MACHINE_COLOR}\h ${DEFAULT_COLOR}(${IP_COLOR}$(ip)${DEFAULT_COLOR}) in ${DIRECTORY_COLOR}$(limited_pwd)${DEFAULT_COLOR}$(virtual_prompt_info)$(scm_prompt_info) \$ "
Ryan Kannob96345e2011-06-09 12:18:11 -1000180 else
Ryan Kanno8e110772011-06-17 10:31:49 -1000181 PS1="$(scm_char) ${UC}\u ${DEFAULT_COLOR}at ${MACHINE_COLOR}\h ${DEFAULT_COLOR}(${IP_COLOR}$(ip)${DEFAULT_COLOR}) in ${DIRECTORY_COLOR}$(limited_pwd)${DEFAULT_COLOR}$(scm_prompt_info) \$ "
Ryan Kannob96345e2011-06-09 12:18:11 -1000182 fi
183 PS2='> '
184 PS4='+ '
Ryan1a83a4d2011-05-31 13:19:20 -1000185}
186
187PROMPT_COMMAND=prompt