workbench.zsh-theme

# check defined theme custom vars
[[ -z $OMZSH_WORKBENCH_PROMPT_INLINE ]] && OMZSH_WORKBENCH_PROMPT_INLINE="false"
[[ -z $OMZSH_WORKBENCH_PROMPT_STATUS ]] && OMZSH_WORKBENCH_PROMPT_STATUS="true"
[[ -z $OMZSH_WORKBENCH_SHORT_PWD ]] && OMZSH_WORKBENCH_SHORT_PWD="false"

# disable default virtualenv display
VIRTUAL_ENV_DISABLE_PROMPT="false"

# zsh git info theme
ZSH_THEME_GIT_PROMPT_PREFIX=" on %{$FG[197]%}⎇ %{$FG[123]%}"
ZSH_THEME_GIT_PROMPT_SUFFIX="%{$reset_color%}"
ZSH_THEME_GIT_PROMPT_DIRTY="%{$FG[214]%}±"
ZSH_THEME_GIT_PROMPT_CLEAN="%{$FG[154]%}✔"

# zsh git status theme
ZSH_THEME_GIT_PROMPT_ADDED="✚ "
ZSH_THEME_GIT_PROMPT_DELETED="✖ "
ZSH_THEME_GIT_PROMPT_MODIFIED="✸ "
ZSH_THEME_GIT_PROMPT_UNTRACKED="✻ "
ZSH_THEME_GIT_PROMPT_RENAMED="➜ "
ZSH_THEME_GIT_PROMPT_UNMERGED="⤔ "

# working directory
function get_pwd() {
    [[ $OMZSH_WORKBENCH_SHORT_PWD = "true" ]] && workbench_pwd="${PWD##*/}" || workbench_pwd="${PWD/$HOME/~}"
    echo -ne "⚒ %{$FG[031]%}$workbench_pwd%{$reset_color%}"
}

# virtualenv
function get_virtualenv() {
  if [[ -n $VIRTUAL_ENV ]]; then
    echo -ne "%{$FG[177]%}[$(basename $VIRTUAL_ENV)]%{$reset_color%} "
  fi
}

# add a line break
function end_line() {
    if [[ $OMZSH_WORKBENCH_PROMPT_INLINE = "false" ]]; then
        echo ""
    fi
}

# check prompt status
function prompt_status() {
    if [[ $RETURN -ne 0 ]]; then
        echo -ne "%{$FG[160]%}✘ "
    fi
    if [[ $(jobs -l | wc -l) -gt 0 ]]; then
        echo -ne "%{$FG[105]%}⚙ "
    fi
}

# display git informations
function git_prompt() {
    echo -ne "$(git_prompt_info) %{$FG[192]%}$(git_prompt_status)%{$reset_color%}"
}

# input line
function prompt_input() {
    echo -ne "%{$FG[154]%}▷%{$reset_color%} "
}

# prompt builder
function display_prompt() {
    RETURN=$?
    end_line
    get_virtualenv
    get_pwd
    git_prompt
    end_line
    [[ $OMZSH_WORKBENCH_PROMPT_STATUS = "true" ]] && prompt_status
    prompt_input
}

PROMPT='$(display_prompt)'