135 lines
3.9 KiB
Text
135 lines
3.9 KiB
Text
# z4rr3t
|
||
# by Garret Jennings
|
||
# https://github.com/inimicus/z4rr3t
|
||
# Based on Pure
|
||
# by Sindre Sorhus
|
||
# https://github.com/sindresorhus/pure
|
||
# MIT License
|
||
|
||
# For my own and others sanity
|
||
# git:
|
||
# %b => current branch
|
||
# %a => current action (rebase/merge)
|
||
# prompt:
|
||
# %F => color dict
|
||
# %f => reset color
|
||
# %~ => current path
|
||
# %* => time
|
||
# %n => username
|
||
# %m => shortname host
|
||
# %(?..) => prompt conditional - %(condition.true.false)
|
||
|
||
|
||
# turns seconds into human readable time
|
||
# 165392 => 1d 21h 56m 32s
|
||
prompt_z4rr3t_human_time() {
|
||
echo -n " "
|
||
local tmp=$1
|
||
local days=$(( tmp / 60 / 60 / 24 ))
|
||
local hours=$(( tmp / 60 / 60 % 24 ))
|
||
local minutes=$(( tmp / 60 % 60 ))
|
||
local seconds=$(( tmp % 60 ))
|
||
(( $days > 0 )) && echo -n "${days}d "
|
||
(( $hours > 0 )) && echo -n "${hours}h "
|
||
(( $minutes > 0 )) && echo -n "${minutes}m "
|
||
echo "${seconds}s"
|
||
}
|
||
|
||
# fastest possible way to check if repo is dirty
|
||
prompt_z4rr3t_git_dirty() {
|
||
# check if we're in a git repo
|
||
[[ "$(command git rev-parse --is-inside-work-tree 2>/dev/null)" == "true" ]] || return
|
||
# check if it's dirty
|
||
[[ "$Z4RR3T_GIT_UNTRACKED_DIRTY" == 0 ]] && local umode="-uno" || local umode="-unormal"
|
||
command test -n "$(git status --porcelain --ignore-submodules ${umode})"
|
||
|
||
(($? == 0)) && echo '*'
|
||
}
|
||
|
||
# displays the exec time of the last command if set threshold was exceeded
|
||
prompt_z4rr3t_cmd_exec_time() {
|
||
local stop=$EPOCHSECONDS
|
||
local start=${cmd_timestamp:-$stop}
|
||
integer elapsed=$stop-$start
|
||
(($elapsed > ${Z4RR3T_CMD_MAX_EXEC_TIME:=10})) && prompt_z4rr3t_human_time $elapsed
|
||
}
|
||
|
||
prompt_z4rr3t_preexec() {
|
||
cmd_timestamp=$EPOCHSECONDS
|
||
|
||
# shows the current dir and executed command in the title when a process is active
|
||
print -Pn "\e]0;"
|
||
echo -nE "$PWD:t: $2"
|
||
print -Pn "\a"
|
||
}
|
||
|
||
# string length ignoring ansi escapes
|
||
prompt_z4rr3t_string_length() {
|
||
echo $(( ${#${(S%%)1//(\%([KF1]|)\{*\}|\%[Bbkf])}} ))
|
||
}
|
||
|
||
prompt_z4rr3t_precmd() {
|
||
# shows the full path in the title
|
||
print -Pn '\e]0;%~\a'
|
||
|
||
# git info
|
||
vcs_info
|
||
|
||
local prompt_z4rr3t_preprompt="$prompt_z4rr3t_username%F{blue}%~%F{242}$vcs_info_msg_0_`prompt_z4rr3t_git_dirty`%f%F{yellow}`prompt_z4rr3t_cmd_exec_time`%f"
|
||
print -P $prompt_z4rr3t_preprompt
|
||
|
||
# check async if there is anything to pull
|
||
(( ${Z4RR3T_GIT_PULL:-1} )) && {
|
||
# check if we're in a git repo
|
||
[[ "$(command git rev-parse --is-inside-work-tree 2>/dev/null)" == "true" ]] &&
|
||
# make sure working tree is not $HOME
|
||
[[ "$(command git rev-parse --show-toplevel)" != "$HOME" ]] &&
|
||
# check check if there is anything to pull
|
||
command git fetch &>/dev/null &&
|
||
# check if there is an upstream configured for this branch
|
||
command git rev-parse --abbrev-ref @'{u}' &>/dev/null && {
|
||
local arrows=''
|
||
(( $(command git rev-list --right-only --count HEAD...@'{u}' 2>/dev/null) > 0 )) && arrows='⇣'
|
||
(( $(command git rev-list --left-only --count HEAD...@'{u}' 2>/dev/null) > 0 )) && arrows+='⇡'
|
||
print -Pn "\e7\e[A\e[1G\e[`prompt_z4rr3t_string_length $prompt_z4rr3t_preprompt`C%F{cyan}${arrows}%f\e8"
|
||
}
|
||
} &!
|
||
|
||
# reset value since `preexec` isn't always triggered
|
||
unset cmd_timestamp
|
||
}
|
||
|
||
|
||
prompt_z4rr3t_setup() {
|
||
# prevent percentage showing up
|
||
# if output doesn't end with a newline
|
||
export PROMPT_EOL_MARK=''
|
||
|
||
# disable auth prompting on git 2.3+
|
||
export GIT_TERMINAL_PROMPT=0
|
||
|
||
prompt_opts=(cr subst percent)
|
||
|
||
zmodload zsh/datetime
|
||
autoload -Uz add-zsh-hook
|
||
autoload -Uz vcs_info
|
||
|
||
add-zsh-hook precmd prompt_z4rr3t_precmd
|
||
add-zsh-hook preexec prompt_z4rr3t_preexec
|
||
|
||
zstyle ':vcs_info:*' enable git
|
||
zstyle ':vcs_info:git*' formats ' %b'
|
||
zstyle ':vcs_info:git*' actionformats ' %b|%a'
|
||
|
||
# show username@host if logged in through SSH
|
||
[[ "$SSH_CONNECTION" != '' ]] && prompt_z4rr3t_username='%F{yellow}%n%F{white}@%F{242}%m '
|
||
|
||
# show username@host if root, with username in white
|
||
[[ $UID -eq 0 ]] && prompt_z4rr3t_username='%F{white}%n%F{red}@%F{242}%m '
|
||
|
||
# prompt turns red if the previous command didn't exit with 0
|
||
PROMPT="%(?.%F{magenta}.%F{red})${Z4RR3T_PROMPT_SYMBOL:-❯}%f "
|
||
}
|
||
|
||
prompt_z4rr3t_setup "$@"
|
||
|