mirror of
https://github.com/dcarrillo/prezto.git
synced 2024-09-07 02:07:32 +00:00
371 lines
9.4 KiB
Plaintext
371 lines
9.4 KiB
Plaintext
#
|
|
# Displays Git repository information.
|
|
#
|
|
# Authors:
|
|
# Sorin Ionescu <sorin.ionescu@gmail.com>
|
|
#
|
|
|
|
# Gets the path to the Git directory.
|
|
function _git-dir {
|
|
local git_root="$(git-root)"
|
|
local git_dir_or_file="${git_root}/.git"
|
|
local git_dir
|
|
|
|
if [[ ! -d "$git_root" ]]; then
|
|
return 1
|
|
fi
|
|
|
|
if [[ -f "$git_dir_or_file" ]]; then
|
|
git_dir="${${${$(<"$git_dir_or_file")}[(fr)gitdir:*]}#gitdir: }"
|
|
else
|
|
git_dir="$git_dir_or_file"
|
|
fi
|
|
|
|
if [[ -d "$git_dir" ]]; then
|
|
print "$git_dir"
|
|
return 0
|
|
fi
|
|
|
|
return 1
|
|
}
|
|
|
|
# Gets the Git special action (am, merge, rebase, etc.).
|
|
# Borrowed from vcs_info and edited.
|
|
function _git-action {
|
|
local action=''
|
|
local action_dir
|
|
local git_dir="$(_git-dir)"
|
|
|
|
for action_dir in \
|
|
"${git_dir}/rebase-apply" \
|
|
"${git_dir}/rebase" \
|
|
"${git_dir}/../.dotest"; do
|
|
if [[ -d "$action_dir" ]] ; then
|
|
if [[ -f "${action_dir}/rebasing" ]] ; then
|
|
action='rebase'
|
|
elif [[ -f "${action_dir}/applying" ]] ; then
|
|
action='am'
|
|
else
|
|
action='am/rebase'
|
|
fi
|
|
print "$action"
|
|
return 0
|
|
fi
|
|
done
|
|
|
|
for action_dir in \
|
|
"${git_dir}/rebase-merge/interactive" \
|
|
"${git_dir}/.dotest-merge/interactive"; do
|
|
if [[ -f "$action_dir" ]]; then
|
|
print 'rebase-i'
|
|
return 0
|
|
fi
|
|
done
|
|
|
|
for action_dir in \
|
|
"${git_dir}/rebase-merge" \
|
|
"${git_dir}/.dotest-merge"; do
|
|
if [[ -d "$action_dir" ]]; then
|
|
print 'rebase-m'
|
|
return 0
|
|
fi
|
|
done
|
|
|
|
if [[ -f "${git_dir}/MERGE_HEAD" ]]; then
|
|
print 'merge'
|
|
return 0
|
|
fi
|
|
|
|
if [[ -f "${git_dir}/CHERRY_PICK_HEAD" ]]; then
|
|
print 'cherry-pick'
|
|
return 0
|
|
fi
|
|
|
|
if [[ -f "${git_dir}/BISECT_LOG" ]]; then
|
|
print 'bisect'
|
|
return 0
|
|
fi
|
|
|
|
return 1
|
|
}
|
|
|
|
# Turns off git-info for the current repository.
|
|
function _git-info-abort {
|
|
if ! is-true "$_git_info_executing"; then
|
|
return 1
|
|
fi
|
|
|
|
cat >&2 <<EOF
|
|
|
|
|
|
Gathering status for certain repositories is time intensive.
|
|
By pressing CTRL + C, you have turned off prompt Git status
|
|
for this repository.
|
|
|
|
To revert, execute:
|
|
git-info on
|
|
|
|
EOF
|
|
|
|
unset _git_info_executing
|
|
git config --bool prompt.showinfo false
|
|
git-info
|
|
return 0
|
|
}
|
|
add-zsh-trap INT _git-info-abort
|
|
|
|
# Gets the Git status information.
|
|
function git-info {
|
|
# Extended globbing is needed to parse repository status.
|
|
setopt LOCAL_OPTIONS
|
|
setopt EXTENDED_GLOB
|
|
|
|
local action
|
|
local action_format
|
|
local action_formatted
|
|
local added=0
|
|
local added_format
|
|
local added_formatted
|
|
local ahead
|
|
local ahead_and_behind
|
|
local ahead_and_behind_cmd
|
|
local ahead_format
|
|
local ahead_formatted
|
|
local ahead_or_behind
|
|
local behind
|
|
local behind_format
|
|
local behind_formatted
|
|
local branch
|
|
local branch_format
|
|
local branch_formatted
|
|
local branch_info
|
|
local clean
|
|
local clean_formatted
|
|
local commit
|
|
local commit_format
|
|
local commit_formatted
|
|
local deleted=0
|
|
local deleted_format
|
|
local deleted_formatted
|
|
local dirty=0
|
|
local dirty_format
|
|
local dirty_formatted
|
|
local ignore_submodule
|
|
local ignore_submodule_when
|
|
local line_number=0
|
|
local modified=0
|
|
local modified_format
|
|
local modified_formatted
|
|
local prompt_format
|
|
local remote
|
|
local remote_cmd
|
|
local remote_format
|
|
local remote_formatted
|
|
local renamed=0
|
|
local renamed_format
|
|
local renamed_formatted
|
|
local rprompt_format
|
|
local stashed=0
|
|
local stashed_format
|
|
local stashed_formatted
|
|
local status_cmd
|
|
local unmerged=0
|
|
local unmerged_format
|
|
local unmerged_formatted
|
|
local untracked=0
|
|
local untracked_format
|
|
local untracked_formatted
|
|
local -A git_info_vars
|
|
local git_info_var
|
|
|
|
# Clean up previous git-info.
|
|
unset git_prompt_info
|
|
unset git_rprompt_info
|
|
|
|
# Return if not inside a Git repository work tree.
|
|
if ! is-true "$(git rev-parse --is-inside-work-tree 2> /dev/null)"; then
|
|
return 1
|
|
fi
|
|
|
|
if (( $# > 0 )); then
|
|
if [[ "$1" == [Oo][Nn] ]]; then
|
|
git config --bool prompt.showinfo true
|
|
elif [[ "$1" == [Oo][Ff][Ff] ]]; then
|
|
git config --bool prompt.showinfo false
|
|
else
|
|
print "usage: $0 [ on | off ]" >&2
|
|
fi
|
|
return 0
|
|
fi
|
|
|
|
# Return if git-info is disabled.
|
|
if ! is-true "${$(git config --bool prompt.showinfo):-true}"; then
|
|
return 1
|
|
fi
|
|
|
|
# Used to abort and turn git-info off on SIGINT.
|
|
_git_info_executing=true
|
|
|
|
# Use porcelain status for easy parsing.
|
|
status_cmd='git status --porcelain'
|
|
|
|
# Gets the remote name.
|
|
remote_cmd='git rev-parse --symbolic-full-name --verify HEAD@{upstream}'
|
|
|
|
# Gets the commit difference counts between local and remote.
|
|
ahead_and_behind_cmd='git rev-list --count --left-right HEAD...@{upstream}'
|
|
|
|
# Ignore submodule status.
|
|
zstyle -b \
|
|
':omz:plugin:git:prompt:ignore' submodule 'ignore_submodule'
|
|
zstyle -s \
|
|
':omz:plugin:git:prompt:ignore:submodule' when 'ignore_submodule_when'
|
|
if is-true "$ignore_submodule"; then
|
|
status_cmd+=" --ignore-submodules=${ignore_submodule_when:-all}"
|
|
fi
|
|
|
|
# Format commit.
|
|
commit="$(git rev-parse HEAD 2> /dev/null)"
|
|
if [[ -n "$commit" ]]; then
|
|
zstyle -s ':omz:plugin:git:prompt' commit 'commit_format'
|
|
zformat -f commit_formatted "$commit_format" "c:$commit"
|
|
fi
|
|
|
|
# Format stashed.
|
|
if [[ -f "$(_git-dir)/refs/stash" ]]; then
|
|
stashed="$(git stash list 2> /dev/null | wc -l)"
|
|
zstyle -s ':omz:plugin:git:prompt' stashed 'stashed_format'
|
|
zformat -f stashed_formatted "$stashed_format" "S:$stashed"
|
|
fi
|
|
|
|
# Format action.
|
|
action="$(_git-action)"
|
|
if [[ -n "$action" ]]; then
|
|
zstyle -s ':omz:plugin:git:prompt' action 'action_format'
|
|
zformat -f action_formatted "$action_format" "s:$action"
|
|
fi
|
|
|
|
# Get current status.
|
|
while IFS=$'\n' read line; do
|
|
# Count added, deleted, modified, renamed, unmerged, untracked, dirty.
|
|
# T (type change) is undocumented, see http://git.io/FnpMGw.
|
|
# For a table of scenarii, see http://i.imgur.com/2YLu1.png.
|
|
[[ "$line" == ([ACDMT][\ MT]|[ACMT]D)\ * ]] && (( added++ ))
|
|
[[ "$line" == [\ ACMRT]D\ * ]] && (( deleted++ ))
|
|
[[ "$line" == ?[MT]\ * ]] && (( modified++ ))
|
|
[[ "$line" == R?\ * ]] && (( renamed++ ))
|
|
[[ "$line" == (AA|DD|U?|?U)\ * ]] && (( unmerged++ ))
|
|
[[ "$line" == \?\?\ * ]] && (( untracked++ ))
|
|
(( dirty++ ))
|
|
done < <(${(z)status_cmd} 2> /dev/null)
|
|
|
|
# Format branch.
|
|
branch="${$(git symbolic-ref -q HEAD)##refs/heads/}"
|
|
if [[ -n "$branch" ]]; then
|
|
zstyle -s ':omz:plugin:git:prompt' branch 'branch_format'
|
|
zformat -f branch_formatted "$branch_format" "b:$branch"
|
|
|
|
# Format remote.
|
|
remote="${$(${(z)remote_cmd} 2> /dev/null)##refs/remotes/}"
|
|
if [[ -n "$remote" ]]; then
|
|
zstyle -s ':omz:plugin:git:prompt' remote 'remote_format'
|
|
zformat -f remote_formatted "$remote_format" "R:$remote"
|
|
|
|
# Get ahead and behind counts.
|
|
ahead_and_behind="$(${(z)ahead_and_behind_cmd} 2> /dev/null)"
|
|
|
|
# Format ahead.
|
|
ahead="$ahead_and_behind[(w)1]"
|
|
if (( $ahead > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' ahead 'ahead_format'
|
|
zformat -f ahead_formatted "$ahead_format" "A:$ahead"
|
|
fi
|
|
|
|
# Format behind.
|
|
behind="$ahead_and_behind[(w)2]"
|
|
if (( $behind > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' behind 'behind_format'
|
|
zformat -f behind_formatted "$behind_format" "B:$behind"
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
# Format added.
|
|
if (( $added > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' added 'added_format'
|
|
zformat -f added_formatted "$added_format" "a:$added_format"
|
|
fi
|
|
|
|
# Format deleted.
|
|
if (( $deleted > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' deleted 'deleted_format'
|
|
zformat -f deleted_formatted "$deleted_format" "d:$deleted_format"
|
|
fi
|
|
|
|
# Format modified.
|
|
if (( $modified > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' modified 'modified_format'
|
|
zformat -f modified_formatted "$modified_format" "m:$modified"
|
|
fi
|
|
|
|
# Format renamed.
|
|
if (( $renamed > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' renamed 'renamed_format'
|
|
zformat -f renamed_formatted "$renamed_format" "r:$renamed"
|
|
fi
|
|
|
|
# Format unmerged.
|
|
if (( $unmerged > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' unmerged 'unmerged_format'
|
|
zformat -f unmerged_formatted "$unmerged_format" "U:$unmerged"
|
|
fi
|
|
|
|
# Format untracked.
|
|
if (( $untracked > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' untracked 'untracked_format'
|
|
zformat -f untracked_formatted "$untracked_format" "u:$untracked"
|
|
fi
|
|
|
|
# Format dirty.
|
|
if (( $dirty > 0 )); then
|
|
zstyle -s ':omz:plugin:git:prompt' dirty 'dirty_format'
|
|
zformat -f dirty_formatted "$dirty_format" "D:$dirty"
|
|
else
|
|
zstyle -s ':omz:plugin:git:prompt' clean 'clean_formatted'
|
|
fi
|
|
|
|
# Format prompts.
|
|
zstyle -s ':omz:plugin:git:prompt' prompt 'prompt_format'
|
|
zstyle -s ':omz:plugin:git:prompt' rprompt 'rprompt_format'
|
|
|
|
git_info_vars=(
|
|
git_prompt_info "$prompt_format"
|
|
git_rprompt_info "$rprompt_format"
|
|
)
|
|
|
|
for git_info_var in ${(k)git_info_vars}; do
|
|
zformat -f "$git_info_var" "$git_info_vars[$git_info_var]" \
|
|
"s:$action_formatted" \
|
|
"a:$added_formatted" \
|
|
"A:$ahead_formatted" \
|
|
"B:$behind_formatted" \
|
|
"b:$branch_formatted" \
|
|
"C:$clean_formatted" \
|
|
"c:$commit_formatted" \
|
|
"d:$deleted_formatted" \
|
|
"D:$dirty_formatted" \
|
|
"m:$modified_formatted" \
|
|
"R:$remote_formatted" \
|
|
"r:$renamed_formatted" \
|
|
"S:$stashed_formatted" \
|
|
"U:$unmerged_formatted" \
|
|
"u:$untracked_formatted"
|
|
done
|
|
|
|
unset _git_info_executing
|
|
return 0
|
|
}
|
|
|
|
git-info "$@"
|
|
|