128 lines
3.9 KiB
Bash
128 lines
3.9 KiB
Bash
# Based off Pure <https://github.com/sindresorhus/pure>
|
|
|
|
# turns seconds into human readable time
|
|
# 165392 => 1d 21h 56m 32s
|
|
_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"
|
|
}
|
|
|
|
# string length ignoring ansi escapes
|
|
_string_length() {
|
|
# Subtract one since newline is counted as two characters
|
|
echo $(( ${#${(S%%)1//(\%([KF1]|)\{*\}|\%[Bbkf])}} - 1 ))
|
|
}
|
|
|
|
# fastest possible way to check if repo is dirty
|
|
prompt_git_dirty() {
|
|
# disable auth prompting on git 2.3+
|
|
GIT_TERMINAL_PROMPT=0
|
|
|
|
# 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
|
|
command test -n "$(git status --porcelain --ignore-submodules -unormal)" || return
|
|
|
|
echo -n "%F{red}[+]"
|
|
local r=$(command git rev-list --right-only --count HEAD...@'{u}' 2>/dev/null)
|
|
local l=$(command git rev-list --left-only --count HEAD...@'{u}' 2>/dev/null)
|
|
|
|
(( ${r:-0} > 0 )) && echo -n " %F{green}${r}⇣"
|
|
(( ${l:-0} > 0 )) && echo -n " %F{yellow}${l}⇡"
|
|
echo -n '%f'
|
|
}
|
|
|
|
# displays the exec time of the last command if set threshold was exceeded
|
|
prompt_cmd_exec_time() {
|
|
local stop=$EPOCHSECONDS
|
|
local start=${cmd_timestamp:-$stop}
|
|
integer elapsed=$stop-$start
|
|
(($elapsed > 2)) && _human_time $elapsed
|
|
}
|
|
|
|
## Hooks ###############################
|
|
prompt_hook_preexec() {
|
|
cmd_timestamp=$EPOCHSECONDS
|
|
}
|
|
|
|
prompt_hook_precmd() {
|
|
# shows the full path in the title
|
|
print -Pn '\e]0;%~\a'
|
|
# git info
|
|
vcs_info
|
|
}
|
|
|
|
# Updates cursor shape and prompt symbol based on vim mode
|
|
prompt_hook_update_vim_mode() {
|
|
case $KEYMAP in
|
|
vicmd) print -n -- "\E]50;CursorShape=0\C-G";
|
|
PROMPT_SYMBOL=$N_MODE;
|
|
;; # block cursor
|
|
main|viins) print -n -- "\E]50;CursorShape=1\C-G";
|
|
PROMPT_SYMBOL=$I_MODE;
|
|
;; # line cursor
|
|
esac
|
|
zle reset-prompt
|
|
zle -R
|
|
}
|
|
prompt_hook_restore_cursor() {
|
|
print -n -- "\E]50;CursorShape=0\C-G" # block cursor
|
|
}
|
|
|
|
## Initialization ######################
|
|
prompt_init() {
|
|
# prevent percentage showing up
|
|
# if output doesn't end with a newline
|
|
export PROMPT_EOL_MARK=''
|
|
|
|
prompt_opts=(cr subst percent)
|
|
|
|
setopt PROMPTSUBST
|
|
zmodload zsh/datetime
|
|
autoload -Uz add-zsh-hook
|
|
autoload -Uz vcs_info
|
|
|
|
add-zsh-hook precmd prompt_hook_precmd
|
|
add-zsh-hook preexec prompt_hook_preexec
|
|
|
|
# Show $N-MODE if in normal mode and Change cursor depending on vim mode
|
|
hooks-add-hook zle_keymap_select_hook prompt_hook_update_vim_mode
|
|
hooks-add-hook zle_line_init_hook prompt_hook_update_vim_mode
|
|
hooks-add-hook zle_line_finish_hook prompt_hook_restore_cursor
|
|
|
|
zstyle ':vcs_info:*' enable git
|
|
zstyle ':vcs_info:*' use-simple true
|
|
zstyle ':vcs_info:*' max-exports 2
|
|
zstyle ':vcs_info:git*' formats ':%b'
|
|
zstyle ':vcs_info:git*' actionformats ':%b (%a)'
|
|
|
|
# show username@host if logged in through SSH
|
|
[[ "$SSH_CONNECTION" != '' ]] && prompt_username='%n@%m '
|
|
|
|
# show username@host if root, with username in white
|
|
[[ $UID -eq 0 ]] && prompt_username='%F{white}%n%F{242}@%m '
|
|
|
|
## Vim cursors
|
|
N_MODE="%F{blue}## "
|
|
I_MODE=">> "
|
|
|
|
RPROMPT='%F{blue}%~' # PWD
|
|
RPROMPT+='%F{242}${vcs_info_msg_0_}$(prompt_git_dirty)' # Branch + dirty indicator
|
|
RPROMPT+='%F{yellow}$(prompt_cmd_exec_time)' # Exec time
|
|
RPROMPT+='%f' # end
|
|
|
|
PROMPT='$prompt_username' # username
|
|
PROMPT+='%(?.%F{yellow}.%F{red})${PROMPT_SYMBOL}' # Prompt (red if $? == 0)
|
|
PROMPT+='%f' #end
|
|
}
|
|
|
|
prompt_init "$@"
|