1
0
Fork 0
dotfiles/files/.config/zsh/custom/themes/custom.zsh-theme

139 lines
3.7 KiB
Bash

# enable command substitution in prompt
setopt prompt_subst
PROMPT='$(prompt_cmd)'
RPROMPT=''
[[ -v ZSH_THEME_HOSTNAME_COLOR ]] || ZSH_THEME_HOSTNAME_COLOR="$(printf "%03d" "$(hostname | md5sum | head -c2)")"
# we export this for motd
export ZSH_THEME_HOSTNAME_COLOR
ZSH_THEME_GIT_PROMPT_PREFIX="%{$FG[008]%} on %{$fg[cyan]%}"
ZSH_THEME_GIT_PROMPT_SUFFIX=""
ZSH_THEME_GIT_PROMPT_ADDED="%{$fg_bold[green]%}+"
ZSH_THEME_GIT_PROMPT_MODIFIED="%{$fg_bold[blue]%}!"
ZSH_THEME_GIT_PROMPT_DELETED="%{$fg_bold[red]%}-"
ZSH_THEME_GIT_PROMPT_RENAMED="%{$fg_bold[magenta]%}>"
ZSH_THEME_GIT_PROMPT_UNMERGED="%{$fg_bold[yellow]%}#"
ZSH_THEME_GIT_PROMPT_UNTRACKED="%{$fg_bold[cyan]%}?"
ZSH_THEME_GIT_PROMPT_AHEAD="%{$fg_bold[white]%}↑"
ZSH_THEME_GIT_PROMPT_BEHIND="%{$fg_bold[white]%}↓"
# Evaluate if root user
__is_root() {
if uname -s | grep -E '^(CYGWIN)|(MINGW)|(MSYS)' 2>&1 >/dev/null; then
id -G | grep 544 2>&1 >/dev/null
else
[ $UID -eq 0 ]
fi
}
# Get the tmp directory
function __get_runtime_dir() {
if [[ -n "$XDG_RUNTIME_DIR" ]]; then
echo -n "$XDG_RUNTIME_DIR"
else
echo -n "/tmp/"
fi
}
# Based on http://www.anishathalye.com/2015/02/07/an-asynchronous-shell-prompt/
ASYNC_PROC=0
function precmd() {
function async() {
# save to temp file
printf "%s" "$(rprompt_cmd)" > "$(__get_runtime_dir)/zsh_prompt_$$"
# signal parent
kill -s USR1 $$
}
# do not clear RPROMPT, let it persist
# kill child if necessary
if [[ "${ASYNC_PROC}" != 0 ]]; then
kill -s HUP $ASYNC_PROC >/dev/null 2>&1 || :
fi
# start background computation
async &!
ASYNC_PROC=$!
}
function TRAPUSR1() {
# read from tmp file
RPROMPT="$(cat $(__get_runtime_dir)/zsh_prompt_$$)"
rm "/tmp/zsh_prompt_$$" &>/dev/null
# reset proc number
ASYNC_PROC=0
# redisplay
zle && zle reset-prompt
}
prompt_cmd() {
# exit code
local exit_code=$?
local exit_code_hex="$(printf '(%02x)' $exit_code)"
if [[ exit_code -eq 0 ]]; then
exit_code_hex="%{$FG[008]%}$exit_code_hex"
else
exit_code_hex="%{$fg[red]%}$exit_code_hex"
fi
# kubernetes context
if which kubectl &>/dev/null; then
local kubectl_context="%{$FG[008]%}[$(kubectl config current-context)] "
fi
# name@hostname
local user_color="$ZSH_THEME_HOSTNAME_COLOR"
if __is_root; then
local user_color="red"
fi
local name_hostname="%{$FG[$user_color]%}$USER%{$FG[$ZSH_THEME_HOSTNAME_COLOR]%}@%m"
# working directory
local work_dir_base="${PWD/$HOME/~}"
local work_dir_post=$(echo "$work_dir_base" | grep -Eo '(^~|/[^/]+){1,4}$')
local work_dir_pre=$(echo "${work_dir_base/$work_dir_post/}" | grep -Eo '^~|/[^/]{2}' | tr -d '\n' )
local work_dir="$work_dir_pre$work_dir_post"
[[ "$work_dir" == "" ]] && work_dir="/"
work_dir="%{$fg[yellow]%}$work_dir"
# prompt
if __is_root; then
# #
local prompt="%{$fg[red]%}#"
else
# $
local prompt="%{$FG[008]%}$"
fi
printf "%s %s %s %s\n%s %s" "$exit_code_hex" "$name_hostname" "$work_dir" "$kubectl_context" "$prompt" "%{$reset_color%}"
}
rprompt_cmd() {
# git
if git rev-parse --git-dir >/dev/null 2>&1; then
local prompt_info="$(git_prompt_info)"
if [[ -z "$prompt_info" ]]; then
local git_rev="$(git_prompt_short_sha)"
else
local git_rev="$prompt_info"
fi
local prompt_status="$(git_prompt_status)"
if [[ -n "$prompt_status" ]]; then
local git_status="$(printf "%s[%s%s]" "%{$FG[008]%}" "$prompt_status" "%{$reset_color%}%{$FG[008]%}")"
fi
fi
printf "%s %s%s" "$git_rev" "$git_status" "%{$reset_color%}"
}
# vim: syn=zsh