greencastle.zsh-theme

# Please note that the below function is very greatly inspired by the precmd function from the jonathan oh-my-zsh theme.
function theme_precmd {
    local TERMWIDTH=$(( COLUMNS - ${ZLE_RPROMPT_INDENT:-1} ))

    PR_FILLBAR=""
    PR_PWDLEN=""

    local prompt_size=${#${:-()---()--}}
    local pwd_size=${#${(%):-%~}}

    # Truncating path if it is too long.
    if (( prompt_size + pwd_size > TERMWIDTH )); then
        (( PR_PWDLEN = TERMWIDTH - prompt_size ))
    elif [[ "${langinfo[CODESET]}" = UTF-8 ]]; then
        PR_FILLBAR="\${(l:$(( TERMWIDTH - (prompt_size + pwd_size) ))::${PR_HBAR}:)}"
    else
        PR_FILLBAR="${PR_SHIFT_IN}\${(1:$(( TERMWIDTH - (prompt_size + pwd_size) ))::${altchar[q]:--}:)}${PR_SHIFT_OUT}"
    fi

    
    # Making the git prompt
    PR_MID_BAR=""
    PR_MID_FILLBAR=""
    PR_BOTTOM_BAR=""
    
    if [ -d .git ]; then
        GIT_BRANCH=$(git symbolic-ref --short HEAD)
        GIT_PREFIX="git::"
        local mid_prompt="┣ ${GIT_PREFIX}${GIT_BRANCH} ┫"
        local MID_WIDTH=$(( COLUMNS - ${ZLE_RPROMPT_INDENT:-1} - ${#mid_prompt} ))


        if (( ${#GIT_BRANCH} > COLUMNS/2 )); then
            # Determining the suffix
            if [ -z "$(git status --porcelain)" ]; then
                GIT_SUFFIX="%{$FG[064]%} ---"
            else
                GIT_SUFFIX="%{$FG[088]%} 🎁 "
            fi
            GIT_SUFFIX+="${HEAVY_RIGHT}"

            PR_MID_FILLBAR="\${(l:$(( ${MID_WIDTH} ))::${PR_HBAR}:)}"
            PR_MID_BAR="\
 %{$reset_color%}${GIT_PREFIX}%{$FG[088]%}${GIT_BRANCH}%{$FG[054]%} \
${(e)PR_MID_FILLBAR}${PR_TLCORNER}"$'\n'

            PR_BOTTOM_FILL_GITSPACE="\${(l:${#GIT_PREFIX}:: :)}"
            PR_BOTTOM_BAR="${(e)PR_BOTTOM_FILL_GITSPACE}${PR_CURVERIGHT}${PR_LIGHT_LEFTBAR}${GIT_SUFFIX}"
        else
            PR_BOTTOM_BAR=" %{$reset_color%}"
        fi
    fi

}

function theme_preexec {
    setopt local_options extended_glob
    if [[ "$TERM" = "screen" ]]; then
        local CMD=${1[(wr)^(*=*|sudo|-*)]}
        echo -n "\ek$CMD\e\\"
    fi
}


autoload -U add-zsh-hook
add-zsh-hook precmd theme_precmd
add-zsh-hook preexec theme_preexec


### Setting the prompt ###

# Modyfying the git prompts
ZSH_THEME_GIT_PROMPT_PREFIX="git::%{$FG[088]%}"
ZSH_THEME_GIT_PROMPT_SUFFIX="\u276f%{$reset_color%}"
ZSH_THEME_GIT_PROMPT_DIRTY="%{$FG[088]%} 🎁 "
ZSH_THEME_GIT_PROMPT_CLEAN="%{$FG[064]%} ---"


# Setting characters
PR_HBAR="━"
PR_ULCORNER="┏"
PR_LLCORNER="┗"
PR_TLCORNER="┫"
PR_CURVERIGHT="╰"
PR_LIGHT_LEFTBAR="╴"


HEAVY_RIGHT="❯"
HEAVY_LEFT="❮"

# Setting the prompt
PROMPT='\
${FG[054]%}${PR_ULCORNER}${PR_HBAR}${FG[247]%}${HEAVY_LEFT}\
%${PR_PWDLEN}<...<%~%<<\
${FG[247]%}${HEAVY_RIGHT}\
${FG[054]%}${PR_HBAR}${PR_HBAR}${(e)PR_FILLBAR}${PR_HBAR}\
${FG[064]%} %(?.✔.%{$fg[red]%}✘%f)\

${FG[054]%}${PR_LLCORNER}${PR_MID_BAR}${PR_BOTTOM_BAR}\
$(git_prompt_info) %{$reset_color%}'
# Fix this to work for really long branch names at some point.


# See https://geoff.greer.fm/lscolors/
export LSCOLORS="CxexhxdxBxDbedabagacad"
export LS_COLORS="di=1;32:ln=34:so=37:pi=33:ex=1;31:bd=1;33;41:cd=34;43:su=30;41:sg=30;46:tw=30;42:ow=30;43"