blob: a86821c0fd0524e1ce6d37add34322624bae45b2 [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 ======================================================================
Ryan1a83a4d2011-05-31 13:19:20 -100034ORANGE='\e[0;33m'
35GREY='\e[1:37m'
36
37DEFAULT_COLOR='\[${white}\]'
38
39USER_COLOR='\[${purple}\]'
40SUPERUSER_COLOR='\[${red}\]'
41MACHINE_COLOR=$ORANGE
Ryan0df98ca2011-06-01 11:22:22 -100042IP_COLOR=$ORANGE
Ryan Kanno0611f862011-05-31 21:38:41 -100043DIRECTORY_COLOR='\[${green}\]'
Ryan1a83a4d2011-05-31 13:19:20 -100044
Ryan Kanno0611f862011-05-31 21:38:41 -100045VE_COLOR='\[${cyan}\]'
46RVM_COLOR='\[${cyan}\]'
Ryan1a83a4d2011-05-31 13:19:20 -100047
Ryan Kanno9bd5b542011-05-31 21:23:33 -100048REF_COLOR='\[${purple}\]'
49
Ryan1a83a4d2011-05-31 13:19:20 -100050# SCM prompts
51SCM_THEME_PROMPT_DIRTY=' ${bold_red}✗${normal}'
52SCM_THEME_PROMPT_CLEAN=' ${bold_green}✓${normal}'
Ryan09bd0ef2011-06-08 13:34:27 -100053SCM_THEME_PROMPT_PREFIX=' on '
Ryan0713fdc2011-06-08 11:32:50 -100054SCM_THEME_PROMPT_SUFFIX=''
55
Ryan Kannob96345e2011-06-09 12:18:11 -100056# rvm prompts
Ryan0713fdc2011-06-08 11:32:50 -100057RVM_THEME_PROMPT_PREFIX=''
58RVM_THEME_PROMPT_SUFFIX=''
59
Ryan Kannob96345e2011-06-09 12:18:11 -100060# virtualenv prompts
Ryan0713fdc2011-06-08 11:32:50 -100061VIRTUALENV_THEME_PROMPT_PREFIX=''
62VIRTUALENV_THEME_PROMPT_SUFFIX=''
Ryan1a83a4d2011-05-31 13:19:20 -100063
Ryan Kannob96345e2011-06-09 12:18:11 -100064# virtual prompts
65VIRTUAL_PROMPT_ENABLED=1
66
67VIRTUAL_THEME_PROMPT_PREFIX=' using '
68VIRTUAL_THEME_PROMPT_SUFFIX=''
69
Ryan1a83a4d2011-05-31 13:19:20 -100070# Max length of PWD to display
71MAX_PWD_LENGTH=20
72
73# Max length of Git Hex to display
74MAX_GIT_HEX_LENGTH=5
75
Ryan Kanno49f897a2011-06-09 13:20:41 -100076# IP address
77IP_SEPARATOR=', '
78
Ryan0df98ca2011-06-01 11:22:22 -100079# FUNCS =======================================================================
Ryan Kannodcb0dde2011-06-07 08:25:45 -100080
Ryan09dfe752011-05-31 19:17:10 -100081function ip {
Ryan Kanno49f897a2011-06-09 13:20:41 -100082 myip=$(curl -s checkip.dyndns.org | grep -Eo '[0-9\.]+')
83 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 -100084}
85
Ryan Kannob96345e2011-06-09 12:18:11 -100086# Displays virtual info prompt (virtualenv/rvm)
87function virtual_prompt_info() {
Ryan0713fdc2011-06-08 11:32:50 -100088 local virtual_env_info=$(virtualenv_prompt)
89 local rvm_info=$(rvm_version_prompt)
Ryan Kannob96345e2011-06-09 12:18:11 -100090 local virtual_prompt=""
91
92 local prefix=${VIRTUAL_THEME_PROMPT_PREFIX}
93 local suffix=${VIRTUAL_THEME_PROMPT_SUFFIX}
Ryan1a83a4d2011-05-31 13:19:20 -100094
95 # If no virtual info, just return
Ryan Kannob96345e2011-06-09 12:18:11 -100096 [[ -z "$virtual_env_info" && -z "$rvm_info" ]] && return
Ryan Kanno0611f862011-05-31 21:38:41 -100097
Ryan1a83a4d2011-05-31 13:19:20 -100098 # If virtual_env info present, append to prompt
Ryan Kannob96345e2011-06-09 12:18:11 -100099 [[ -n "$virtual_env_info" ]] && virtual_prompt="virtualenv: ${VE_COLOR}$virtual_env_info${DEFAULT_COLOR}"
Ryan1a83a4d2011-05-31 13:19:20 -1000100
Ryan Kannob96345e2011-06-09 12:18:11 -1000101 if [[ -n "$rvm_info" ]]
Ryan1a83a4d2011-05-31 13:19:20 -1000102 then
Ryan Kannob96345e2011-06-09 12:18:11 -1000103 [[ -n "$virtual_env_info" ]] && virtual_prompt="$virtual_prompt, "
104 virtual_prompt="${virtual_prompt}rvm: ${RVM_COLOR}$rvm_info${DEFAULT_COLOR}"
Ryan1a83a4d2011-05-31 13:19:20 -1000105 fi
Ryan Kannob96345e2011-06-09 12:18:11 -1000106 echo -e "$prefix$virtual_prompt$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -1000107}
108
Ryan1a83a4d2011-05-31 13:19:20 -1000109# Parse git info
Ryan28e2aed2011-06-08 11:57:16 -1000110function git_prompt_info() {
Ryan1a83a4d2011-05-31 13:19:20 -1000111 if [[ -n $(git status -s 2> /dev/null |grep -v ^# |grep -v "working directory clean") ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000112 state=${GIT_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan1a83a4d2011-05-31 13:19:20 -1000113 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000114 state=${GIT_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan1a83a4d2011-05-31 13:19:20 -1000115 fi
116 prefix=${GIT_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
117 suffix=${GIT_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
118 ref=$(git symbolic-ref HEAD 2> /dev/null) || return
Ryan Kannob96345e2011-06-09 12:18:11 -1000119 commit_id=$(git rev-parse HEAD 2>/dev/null) || return
Ryan1a83a4d2011-05-31 13:19:20 -1000120
Ryan Kannob96345e2011-06-09 12:18:11 -1000121 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 -1000122}
123
Ryan06d71ab2011-05-31 16:10:30 -1000124# Parse hg info
Ryan28e2aed2011-06-08 11:57:16 -1000125function hg_prompt_info() {
Ryanaa2f1cb2011-05-31 18:41:07 -1000126 if [[ -n $(hg status 2> /dev/null) ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000127 state=${HG_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan06d71ab2011-05-31 16:10:30 -1000128 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000129 state=${HG_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan06d71ab2011-05-31 16:10:30 -1000130 fi
131 prefix=${HG_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
132 suffix=${HG_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
Ryanaa2f1cb2011-05-31 18:41:07 -1000133 branch=$(hg summary 2> /dev/null | grep branch | awk '{print $2}')
134 changeset=$(hg summary 2> /dev/null | grep parent | awk '{print $2}')
Ryan06d71ab2011-05-31 16:10:30 -1000135
Ryan Kannob96345e2011-06-09 12:18:11 -1000136 echo -e "$prefix${REF_COLOR}${branch}${DEFAULT_COLOR}:${changeset#*:}$state$suffix"
Ryan06d71ab2011-05-31 16:10:30 -1000137}
138
Ryan2a6d1da2011-05-31 19:03:28 -1000139# Parse svn info
Ryan28e2aed2011-06-08 11:57:16 -1000140function svn_prompt_info() {
Ryan2a6d1da2011-05-31 19:03:28 -1000141 if [[ -n $(svn status --ignore-externals -q 2> /dev/null) ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000142 state=${SVN_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan2a6d1da2011-05-31 19:03:28 -1000143 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000144 state=${SVN_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan2a6d1da2011-05-31 19:03:28 -1000145 fi
146 prefix=${SVN_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
147 suffix=${SVN_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
148 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 -1000149 [[ -z $ref ]] && return
Ryan Kannob96345e2011-06-09 12:18:11 -1000150
151 revision=$(svn info 2> /dev/null | sed -ne 's#^Revision: ##p' )
152
Ryan Kanno9bd5b542011-05-31 21:23:33 -1000153 echo -e "$prefix${REF_COLOR}$ref${DEFAULT_COLOR}:$revision$state$suffix"
Ryan2a6d1da2011-05-31 19:03:28 -1000154}
Ryan06d71ab2011-05-31 16:10:30 -1000155
Ryan1a83a4d2011-05-31 13:19:20 -1000156# Displays last X characters of pwd
157function limited_pwd() {
158
159 # Replace $HOME with ~ if possible
160 RELATIVE_PWD=${PWD/#$HOME/\~}
161
162 local offset=$((${#RELATIVE_PWD}-$MAX_PWD_LENGTH))
163
164 if [ $offset -gt "0" ]
165 then
166 local truncated_symbol="..."
167 TRUNCATED_PWD=${RELATIVE_PWD:$offset:$MAX_PWD_LENGTH}
Ryan Kannob96345e2011-06-09 12:18:11 -1000168 echo -e "${truncated_symbol}/${TRUNCATED_PWD#*/}"
Ryan1a83a4d2011-05-31 13:19:20 -1000169 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000170 echo -e "${RELATIVE_PWD}"
Ryan1a83a4d2011-05-31 13:19:20 -1000171 fi
172}
173
174# Displays the current prompt
175function prompt() {
Ryan Kannob96345e2011-06-09 12:18:11 -1000176 local UC=$USER_COLOR
177 [ $UID -eq "0" ] && UC=$SUPERUSER_COLOR
Ryan1a83a4d2011-05-31 13:19:20 -1000178
Ryan Kannob96345e2011-06-09 12:18:11 -1000179 if [[ $VIRTUAL_PROMPT_ENABLED == 1 ]]; then
180 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_prompt_info)$(scm_prompt_info) \$ "
181 else
182 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}$(scm_prompt_info) \$ "
183 fi
184 PS2='> '
185 PS4='+ '
Ryan1a83a4d2011-05-31 13:19:20 -1000186}
187
188PROMPT_COMMAND=prompt