CURRENT_DIR=$(pwd)
cd ~/.hygo
HYGOSHELL_UPDATE=$(git pull)
if [[ "$HYGOSHELL_UPDATE" == *"Already"* ]]; then
    echo -e "\n\e[38;5;214mHygoShell is up to date.\e[0m\n"
else
    echo -e "\n\e[38;5;214mHygoShell updating.\e[0m\n"
fi
cd $CURRENT_DIR

# Function to get the current Git branch
get_git_branch() {
    git branch 2>/dev/null | grep '*' | sed 's/* //'
}

git_branch_str() {
    git_branch=$(get_git_branch)
    if [ -n "$git_branch" ]; then
        echo "($git_branch)"
    fi
}

update_prompt() {
    local git_branch=$(get_git_branch)
    if [ -n "$git_branch" ]; then
        PROMPT="%F{214}%n@%m %F{white}%~ %F{51}($git_branch)$NEWLINE%F{214}%#%k%f "
    else
        PROMPT="%F{214}%n@%m %F{white}%~$NEWLINE%F{214}%#%k%f "
    fi
}

# Set up zsh prompt
if [ -n "$ZSH_VERSION" ]; then
    NEWLINE=$'\n'
    autoload -Uz add-zsh-hook
    add-zsh-hook precmd update_prompt
    update_prompt
fi

# Set up bash prompt
if [ -n "$BASH_VERSION" ]; then
    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\] '
fi