| 1 | CURRENT_DIR=$(pwd) |
| 2 | cd ~/.hygo |
| 3 | HYGOSHELL_UPDATE=$(git pull) |
| 4 | if [[ "$HYGOSHELL_UPDATE" == *"Already"* ]]; then |
| 5 | echo -e "\n\e[38;5;214mHygoShell is up to date.\e[0m\n" |
| 6 | else |
| 7 | echo -e "\n\e[38;5;214mHygoShell updating.\e[0m\n" |
| 8 | fi |
| 9 | cd $CURRENT_DIR |
| 10 | |
| 11 | |
| 12 | get_git_branch() { |
| 13 | git branch 2>/dev/null | grep '*' | sed 's/* //' |
| 14 | } |
| 15 | |
| 16 | git_branch_str() { |
| 17 | git_branch=$(get_git_branch) |
| 18 | if [ -n "$git_branch" ]; then |
| 19 | echo "($git_branch)" |
| 20 | fi |
| 21 | } |
| 22 | |
| 23 | update_prompt() { |
| 24 | local git_branch=$(get_git_branch) |
| 25 | if [ -n "$git_branch" ]; then |
| 26 | PROMPT="%F{214}%n@%m %F{white}%~ %F{51}($git_branch)$NEWLINE%F{214}%#%k%f " |
| 27 | else |
| 28 | PROMPT="%F{214}%n@%m %F{white}%~$NEWLINE%F{214}%#%k%f " |
| 29 | fi |
| 30 | } |
| 31 | |
| 32 | |
| 33 | if [ -n "$ZSH_VERSION" ]; then |
| 34 | NEWLINE=$'\n' |
| 35 | autoload -Uz add-zsh-hook |
| 36 | add-zsh-hook precmd update_prompt |
| 37 | update_prompt |
| 38 | fi |
| 39 | |
| 40 | |
| 41 | if [ -n "$BASH_VERSION" ]; then |
| 42 | PS1='\[\e[38;5;214m\]\u@\h \w$(git_branch=$(get_git_branch); if [ -n "$git_branch" ]; then echo " \[\e[37m\]($git_branch\[\e[0m\])"; fi)\n\$\[\e[0m\] ' |
| 43 | fi |