You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
869 lines
32 KiB
Bash
869 lines
32 KiB
Bash
# Zsh bindings for gitstatus.
|
|
#
|
|
# ------------------------------------------------------------------
|
|
#
|
|
# Example: Start gitstatusd, send it a request, wait for response and print it.
|
|
#
|
|
# source ~/gitstatus/gitstatus.plugin.zsh
|
|
# gitstatus_start MY
|
|
# gitstatus_query -d $PWD MY
|
|
# typeset -m 'VCS_STATUS_*'
|
|
#
|
|
# Output:
|
|
#
|
|
# VCS_STATUS_ACTION=''
|
|
# VCS_STATUS_COMMIT=c000eddcff0fb38df2d0137efe24d9d2d900f209
|
|
# VCS_STATUS_COMMITS_AHEAD=0
|
|
# VCS_STATUS_COMMITS_BEHIND=0
|
|
# VCS_STATUS_HAS_CONFLICTED=0
|
|
# VCS_STATUS_HAS_STAGED=0
|
|
# VCS_STATUS_HAS_UNSTAGED=1
|
|
# VCS_STATUS_HAS_UNTRACKED=1
|
|
# VCS_STATUS_INDEX_SIZE=33
|
|
# VCS_STATUS_LOCAL_BRANCH=master
|
|
# VCS_STATUS_NUM_ASSUME_UNCHANGED=0
|
|
# VCS_STATUS_NUM_CONFLICTED=0
|
|
# VCS_STATUS_NUM_STAGED=0
|
|
# VCS_STATUS_NUM_UNSTAGED=1
|
|
# VCS_STATUS_NUM_SKIP_WORKTREE=0
|
|
# VCS_STATUS_NUM_STAGED_NEW=0
|
|
# VCS_STATUS_NUM_STAGED_DELETED=0
|
|
# VCS_STATUS_NUM_UNSTAGED_DELETED=0
|
|
# VCS_STATUS_NUM_UNTRACKED=1
|
|
# VCS_STATUS_PUSH_COMMITS_AHEAD=0
|
|
# VCS_STATUS_PUSH_COMMITS_BEHIND=0
|
|
# VCS_STATUS_PUSH_REMOTE_NAME=''
|
|
# VCS_STATUS_PUSH_REMOTE_URL=''
|
|
# VCS_STATUS_REMOTE_BRANCH=master
|
|
# VCS_STATUS_REMOTE_NAME=origin
|
|
# VCS_STATUS_REMOTE_URL=git@github.com:romkatv/powerlevel10k.git
|
|
# VCS_STATUS_RESULT=ok-sync
|
|
# VCS_STATUS_STASHES=0
|
|
# VCS_STATUS_TAG=''
|
|
# VCS_STATUS_WORKDIR=/home/romka/powerlevel10k
|
|
|
|
[[ -o 'interactive' ]] || 'return'
|
|
|
|
# Temporarily change options.
|
|
'builtin' 'local' '-a' '_gitstatus_opts'
|
|
[[ ! -o 'aliases' ]] || _gitstatus_opts+=('aliases')
|
|
[[ ! -o 'sh_glob' ]] || _gitstatus_opts+=('sh_glob')
|
|
[[ ! -o 'no_brace_expand' ]] || _gitstatus_opts+=('no_brace_expand')
|
|
'builtin' 'setopt' 'no_aliases' 'no_sh_glob' 'brace_expand'
|
|
|
|
autoload -Uz add-zsh-hook || return
|
|
zmodload zsh/datetime zsh/system || return
|
|
zmodload -F zsh/files b:zf_rm || return
|
|
|
|
typeset -g _gitstatus_plugin_dir"${1:-}"="${${(%):-%x}:A:h}"
|
|
|
|
# Retrives status of a git repo from a directory under its working tree.
|
|
#
|
|
## Usage: gitstatus_query [OPTION]... NAME
|
|
#
|
|
# -d STR Directory to query. Defaults to the current directory. Has no effect if GIT_DIR
|
|
# is set.
|
|
# -c STR Callback function to call once the results are available. Called only after
|
|
# gitstatus_query returns 0 with VCS_STATUS_RESULT=tout.
|
|
# -t FLOAT Timeout in seconds. Negative value means infinity. Will block for at most this long.
|
|
# If no results are available by then: if -c isn't specified, will return 1; otherwise
|
|
# will set VCS_STATUS_RESULT=tout and return 0.
|
|
# -p Don't compute anything that requires reading Git index. If this option is used,
|
|
# the following parameters will be 0: VCS_STATUS_INDEX_SIZE,
|
|
# VCS_STATUS_{NUM,HAS}_{STAGED,UNSTAGED,UNTRACKED,CONFLICTED}.
|
|
#
|
|
# On success sets VCS_STATUS_RESULT to one of the following values:
|
|
#
|
|
# tout Timed out waiting for data; will call the user-specified callback later.
|
|
# norepo-sync The directory isn't a git repo.
|
|
# ok-sync The directory is a git repo.
|
|
#
|
|
# When the callback is called, VCS_STATUS_RESULT is set to one of the following values:
|
|
#
|
|
# norepo-async The directory isn't a git repo.
|
|
# ok-async The directory is a git repo.
|
|
#
|
|
# If VCS_STATUS_RESULT is ok-sync or ok-async, additional variables are set:
|
|
#
|
|
# VCS_STATUS_WORKDIR Git repo working directory. Not empty.
|
|
# VCS_STATUS_COMMIT Commit hash that HEAD is pointing to. Either 40 hex digits or
|
|
# empty if there is no HEAD (empty repo).
|
|
# VCS_STATUS_LOCAL_BRANCH Local branch name or empty if not on a branch.
|
|
# VCS_STATUS_REMOTE_NAME The remote name, e.g. "upstream" or "origin".
|
|
# VCS_STATUS_REMOTE_BRANCH Upstream branch name. Can be empty.
|
|
# VCS_STATUS_REMOTE_URL Remote URL. Can be empty.
|
|
# VCS_STATUS_ACTION Repository state, A.K.A. action. Can be empty.
|
|
# VCS_STATUS_INDEX_SIZE The number of files in the index.
|
|
# VCS_STATUS_NUM_STAGED The number of staged changes.
|
|
# VCS_STATUS_NUM_CONFLICTED The number of conflicted changes.
|
|
# VCS_STATUS_NUM_UNSTAGED The number of unstaged changes.
|
|
# VCS_STATUS_NUM_UNTRACKED The number of untracked files.
|
|
# VCS_STATUS_HAS_STAGED 1 if there are staged changes, 0 otherwise.
|
|
# VCS_STATUS_HAS_CONFLICTED 1 if there are conflicted changes, 0 otherwise.
|
|
# VCS_STATUS_HAS_UNSTAGED 1 if there are unstaged changes, 0 if there aren't, -1 if
|
|
# unknown.
|
|
# VCS_STATUS_NUM_STAGED_NEW The number of staged new files. Note that renamed files
|
|
# are reported as deleted plus new.
|
|
# VCS_STATUS_NUM_STAGED_DELETED The number of staged deleted files. Note that renamed files
|
|
# are reported as deleted plus new.
|
|
# VCS_STATUS_NUM_UNSTAGED_DELETED The number of unstaged deleted files. Note that renamed files
|
|
# are reported as deleted plus new.
|
|
# VCS_STATUS_HAS_UNTRACKED 1 if there are untracked files, 0 if there aren't, -1 if
|
|
# unknown.
|
|
# VCS_STATUS_COMMITS_AHEAD Number of commits the current branch is ahead of upstream.
|
|
# Non-negative integer.
|
|
# VCS_STATUS_COMMITS_BEHIND Number of commits the current branch is behind upstream.
|
|
# Non-negative integer.
|
|
# VCS_STATUS_STASHES Number of stashes. Non-negative integer.
|
|
# VCS_STATUS_TAG The last tag (in lexicographical order) that points to the same
|
|
# commit as HEAD.
|
|
# VCS_STATUS_PUSH_REMOTE_NAME The push remote name, e.g. "upstream" or "origin".
|
|
# VCS_STATUS_PUSH_REMOTE_URL Push remote URL. Can be empty.
|
|
# VCS_STATUS_PUSH_COMMITS_AHEAD Number of commits the current branch is ahead of push remote.
|
|
# Non-negative integer.
|
|
# VCS_STATUS_PUSH_COMMITS_BEHIND Number of commits the current branch is behind push remote.
|
|
# Non-negative integer.
|
|
# VCS_STATUS_NUM_SKIP_WORKTREE The number of files in the index with skip-worktree bit set.
|
|
# Non-negative integer.
|
|
# VCS_STATUS_NUM_ASSUME_UNCHANGED The number of files in the index with assume-unchanged bit set.
|
|
# Non-negative integer.
|
|
#
|
|
# The point of reporting -1 via VCS_STATUS_HAS_* is to allow the command to skip scanning files in
|
|
# large repos. See -m flag of gitstatus_start.
|
|
#
|
|
# gitstatus_query returns an error if gitstatus_start hasn't been called in the same shell or
|
|
# the call had failed.
|
|
#
|
|
# !!!!! WARNING: CONCURRENT CALLS WITH THE SAME NAME ARE NOT ALLOWED !!!!!
|
|
#
|
|
# It's illegal to call gitstatus_query if the last asynchronous call with the same NAME hasn't
|
|
# completed yet. If you need to issue concurrent requests, use different NAME arguments.
|
|
function gitstatus_query"${1:-}"() {
|
|
emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent
|
|
|
|
local fsuf=${${(%):-%N}#gitstatus_query}
|
|
|
|
unset VCS_STATUS_RESULT
|
|
|
|
local opt dir callback OPTARG
|
|
local -i no_diff OPTIND
|
|
local -F timeout=-1
|
|
while getopts ":d:c:t:p" opt; do
|
|
case $opt in
|
|
+p) no_diff=0;;
|
|
p) no_diff=1;;
|
|
d) dir=$OPTARG;;
|
|
c) callback=$OPTARG;;
|
|
t)
|
|
if [[ $OPTARG != (|+|-)<->(|.<->)(|[eE](|-|+)<->) ]]; then
|
|
print -ru2 -- "gitstatus_query: invalid -t argument: $OPTARG"
|
|
return 1
|
|
fi
|
|
timeout=OPTARG
|
|
;;
|
|
\?) print -ru2 -- "gitstatus_query: invalid option: $OPTARG" ; return 1;;
|
|
:) print -ru2 -- "gitstatus_query: missing required argument: $OPTARG"; return 1;;
|
|
*) print -ru2 -- "gitstatus_query: invalid option: $opt" ; return 1;;
|
|
esac
|
|
done
|
|
|
|
if (( OPTIND != ARGC )); then
|
|
print -ru2 -- "gitstatus_start: exactly one positional argument is required"
|
|
return 1
|
|
fi
|
|
|
|
local name=$*[OPTIND]
|
|
if [[ $name != [[:IDENT:]]## ]]; then
|
|
print -ru2 -- "gitstatus_start: invalid positional argument: $name"
|
|
return 1
|
|
fi
|
|
|
|
(( _GITSTATUS_STATE_$name == 2 )) || return
|
|
|
|
if [[ -z $GIT_DIR ]]; then
|
|
[[ $dir == /* ]] || dir=${(%):-%/}/$dir
|
|
else
|
|
[[ $GIT_DIR == /* ]] && dir=:$GIT_DIR || dir=:${(%):-%/}/$GIT_DIR
|
|
fi
|
|
|
|
local -i req_fd=${(P)${:-_GITSTATUS_REQ_FD_$name}}
|
|
local req_id=$EPOCHREALTIME
|
|
print -rnu $req_fd -- $req_id' '$callback$'\x1f'$dir$'\x1f'$no_diff$'\x1e' || return
|
|
|
|
(( ++_GITSTATUS_NUM_INFLIGHT_$name ))
|
|
|
|
if (( timeout == 0 )); then
|
|
typeset -g VCS_STATUS_RESULT=tout
|
|
_gitstatus_clear$fsuf
|
|
else
|
|
while true; do
|
|
_gitstatus_process_response$fsuf $name $timeout $req_id || return
|
|
[[ $VCS_STATUS_RESULT == *-async ]] || break
|
|
done
|
|
fi
|
|
|
|
[[ $VCS_STATUS_RESULT != tout || -n $callback ]]
|
|
}
|
|
|
|
# If the last call to gitstatus_query timed out (VCS_STATUS_RESULT=tout), wait for the callback
|
|
# to be called. Otherwise do nothing.
|
|
#
|
|
# Usage: gitstatus_process_results [OPTION]... NAME
|
|
#
|
|
# -t FLOAT Timeout in seconds. Negative value means infinity. Will block for at most this long.
|
|
#
|
|
# Returns an error only when invoked with incorrect arguments and when gitstatusd isn't running or
|
|
# broken.
|
|
#
|
|
# If a callback gets called, VCS_STATUS_* parameters are set as in gitstatus_query.
|
|
# VCS_STATUS_RESULT is either norepo-async or ok-async.
|
|
function gitstatus_process_results"${1:-}"() {
|
|
emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent
|
|
|
|
local fsuf=${${(%):-%N}#gitstatus_process_results}
|
|
|
|
local opt OPTARG
|
|
local -i OPTIND
|
|
local -F timeout=-1
|
|
while getopts ":t:" opt; do
|
|
case $opt in
|
|
t)
|
|
if [[ $OPTARG != (|+|-)<->(|.<->)(|[eE](|-|+)<->) ]]; then
|
|
print -ru2 -- "gitstatus_process_results: invalid -t argument: $OPTARG"
|
|
return 1
|
|
fi
|
|
timeout=OPTARG
|
|
;;
|
|
\?) print -ru2 -- "gitstatus_process_results: invalid option: $OPTARG" ; return 1;;
|
|
:) print -ru2 -- "gitstatus_process_results: missing required argument: $OPTARG"; return 1;;
|
|
*) print -ru2 -- "gitstatus_process_results: invalid option: $opt" ; return 1;;
|
|
esac
|
|
done
|
|
|
|
if (( OPTIND != ARGC )); then
|
|
print -ru2 -- "gitstatus_process_results: exactly one positional argument is required"
|
|
return 1
|
|
fi
|
|
|
|
local name=$*[OPTIND]
|
|
if [[ $name != [[:IDENT:]]## ]]; then
|
|
print -ru2 -- "gitstatus_process_results: invalid positional argument: $name"
|
|
return 1
|
|
fi
|
|
|
|
(( _GITSTATUS_STATE_$name == 2 )) || return
|
|
|
|
while (( _GITSTATUS_NUM_INFLIGHT_$name )); do
|
|
_gitstatus_process_response$fsuf $name $timeout '' || return
|
|
[[ $VCS_STATUS_RESULT == *-async ]] || break
|
|
done
|
|
|
|
return 0
|
|
}
|
|
|
|
function _gitstatus_clear"${1:-}"() {
|
|
unset VCS_STATUS_{WORKDIR,COMMIT,LOCAL_BRANCH,REMOTE_BRANCH,REMOTE_NAME,REMOTE_URL,ACTION,INDEX_SIZE,NUM_STAGED,NUM_UNSTAGED,NUM_CONFLICTED,NUM_UNTRACKED,HAS_STAGED,HAS_UNSTAGED,HAS_CONFLICTED,HAS_UNTRACKED,COMMITS_AHEAD,COMMITS_BEHIND,STASHES,TAG,NUM_UNSTAGED_DELETED,NUM_STAGED_NEW,NUM_STAGED_DELETED,PUSH_REMOTE_NAME,PUSH_REMOTE_URL,PUSH_COMMITS_AHEAD,PUSH_COMMITS_BEHIND,NUM_SKIP_WORKTREE,NUM_ASSUME_UNCHANGED}
|
|
}
|
|
|
|
function _gitstatus_process_response"${1:-}"() {
|
|
local name=$1 timeout req_id=$3 buf
|
|
local -i resp_fd=_GITSTATUS_RESP_FD_$name
|
|
local -i dirty_max_index_size=_GITSTATUS_DIRTY_MAX_INDEX_SIZE_$name
|
|
|
|
(( $2 >= 0 )) && timeout=-t$2 && [[ -t $resp_fd ]]
|
|
sysread $timeout -i $resp_fd 'buf[$#buf+1]' || {
|
|
if (( $? == 4 )); then
|
|
if [[ -n $req_id ]]; then
|
|
typeset -g VCS_STATUS_RESULT=tout
|
|
_gitstatus_clear$fsuf
|
|
fi
|
|
return 0
|
|
else
|
|
gitstatus_stop$fsuf $name
|
|
return 1
|
|
fi
|
|
}
|
|
while [[ $buf != *$'\x1e' ]]; do
|
|
if ! sysread -i $resp_fd 'buf[$#buf+1]'; then
|
|
gitstatus_stop$fsuf $name
|
|
return 1
|
|
fi
|
|
done
|
|
|
|
local s
|
|
for s in ${(ps:\x1e:)buf}; do
|
|
local -a resp=("${(@ps:\x1f:)s}")
|
|
if (( resp[2] )); then
|
|
if [[ $resp[1] == $req_id' '* ]]; then
|
|
typeset -g VCS_STATUS_RESULT=ok-sync
|
|
else
|
|
typeset -g VCS_STATUS_RESULT=ok-async
|
|
fi
|
|
for VCS_STATUS_WORKDIR \
|
|
VCS_STATUS_COMMIT \
|
|
VCS_STATUS_LOCAL_BRANCH \
|
|
VCS_STATUS_REMOTE_BRANCH \
|
|
VCS_STATUS_REMOTE_NAME \
|
|
VCS_STATUS_REMOTE_URL \
|
|
VCS_STATUS_ACTION \
|
|
VCS_STATUS_INDEX_SIZE \
|
|
VCS_STATUS_NUM_STAGED \
|
|
VCS_STATUS_NUM_UNSTAGED \
|
|
VCS_STATUS_NUM_CONFLICTED \
|
|
VCS_STATUS_NUM_UNTRACKED \
|
|
VCS_STATUS_COMMITS_AHEAD \
|
|
VCS_STATUS_COMMITS_BEHIND \
|
|
VCS_STATUS_STASHES \
|
|
VCS_STATUS_TAG \
|
|
VCS_STATUS_NUM_UNSTAGED_DELETED \
|
|
VCS_STATUS_NUM_STAGED_NEW \
|
|
VCS_STATUS_NUM_STAGED_DELETED \
|
|
VCS_STATUS_PUSH_REMOTE_NAME \
|
|
VCS_STATUS_PUSH_REMOTE_URL \
|
|
VCS_STATUS_PUSH_COMMITS_AHEAD \
|
|
VCS_STATUS_PUSH_COMMITS_BEHIND \
|
|
VCS_STATUS_NUM_SKIP_WORKTREE \
|
|
VCS_STATUS_NUM_ASSUME_UNCHANGED in "${(@)resp[3,27]}"; do
|
|
done
|
|
typeset -gi VCS_STATUS_{INDEX_SIZE,NUM_STAGED,NUM_UNSTAGED,NUM_CONFLICTED,NUM_UNTRACKED,COMMITS_AHEAD,COMMITS_BEHIND,STASHES,NUM_UNSTAGED_DELETED,NUM_STAGED_NEW,NUM_STAGED_DELETED,PUSH_COMMITS_AHEAD,PUSH_COMMITS_BEHIND,NUM_SKIP_WORKTREE,NUM_ASSUME_UNCHANGED}
|
|
typeset -gi VCS_STATUS_HAS_STAGED=$((VCS_STATUS_NUM_STAGED > 0))
|
|
if (( dirty_max_index_size >= 0 && VCS_STATUS_INDEX_SIZE > dirty_max_index_size )); then
|
|
typeset -gi \
|
|
VCS_STATUS_HAS_UNSTAGED=-1 \
|
|
VCS_STATUS_HAS_CONFLICTED=-1 \
|
|
VCS_STATUS_HAS_UNTRACKED=-1
|
|
else
|
|
typeset -gi \
|
|
VCS_STATUS_HAS_UNSTAGED=$((VCS_STATUS_NUM_UNSTAGED > 0)) \
|
|
VCS_STATUS_HAS_CONFLICTED=$((VCS_STATUS_NUM_CONFLICTED > 0)) \
|
|
VCS_STATUS_HAS_UNTRACKED=$((VCS_STATUS_NUM_UNTRACKED > 0))
|
|
fi
|
|
else
|
|
if [[ $resp[1] == $req_id' '* ]]; then
|
|
typeset -g VCS_STATUS_RESULT=norepo-sync
|
|
else
|
|
typeset -g VCS_STATUS_RESULT=norepo-async
|
|
fi
|
|
_gitstatus_clear$fsuf
|
|
fi
|
|
(( --_GITSTATUS_NUM_INFLIGHT_$name ))
|
|
[[ $VCS_STATUS_RESULT == *-async ]] && emulate zsh -c "${resp[1]#* }"
|
|
done
|
|
|
|
return 0
|
|
}
|
|
|
|
function _gitstatus_daemon"${1:-}"() {
|
|
local -i pipe_fd
|
|
exec 0<&- {pipe_fd}>&1 1>>$daemon_log 2>&1 || return
|
|
local pgid=$sysparams[pid]
|
|
[[ $pgid == <1-> ]] || return
|
|
builtin cd -q / || return
|
|
|
|
{
|
|
{
|
|
trap '' PIPE
|
|
|
|
local uname_sm
|
|
uname_sm="${(L)$(uname -sm)}" || return
|
|
[[ $uname_sm == [^' ']##' '[^' ']## ]] || return
|
|
local uname_s=${uname_sm% *}
|
|
local uname_m=${uname_sm#* }
|
|
|
|
if [[ $GITSTATUS_NUM_THREADS == <1-> ]]; then
|
|
args+=(-t $GITSTATUS_NUM_THREADS)
|
|
else
|
|
local cpus
|
|
if (( ! $+commands[sysctl] )) || [[ $uname_s == linux ]] ||
|
|
! cpus="$(sysctl -n hw.ncpu)"; then
|
|
if (( ! $+commands[getconf] )) || ! cpus="$(getconf _NPROCESSORS_ONLN)"; then
|
|
cpus=8
|
|
fi
|
|
fi
|
|
args+=(-t $((cpus > 16 ? 32 : cpus > 0 ? 2 * cpus : 16)))
|
|
fi
|
|
|
|
mkfifo -- $file_prefix.fifo || return
|
|
print -rnu $pipe_fd -- ${(l:20:)pgid} || return
|
|
exec <$file_prefix.fifo || return
|
|
zf_rm -- $file_prefix.fifo || return
|
|
|
|
local _gitstatus_zsh_daemon _gitstatus_zsh_version _gitstatus_zsh_downloaded
|
|
|
|
function _gitstatus_set_daemon$fsuf() {
|
|
_gitstatus_zsh_daemon="$1"
|
|
_gitstatus_zsh_version="$2"
|
|
_gitstatus_zsh_downloaded="$3"
|
|
}
|
|
|
|
local gitstatus_plugin_dir_var=_gitstatus_plugin_dir$fsuf
|
|
local gitstatus_plugin_dir=${(P)gitstatus_plugin_dir_var}
|
|
set -- -d $gitstatus_plugin_dir -s $uname_s -m $uname_m -p "printf . >&$pipe_fd" -- \
|
|
_gitstatus_set_daemon$fsuf
|
|
[[ ${GITSTATUS_AUTO_INSTALL:-1} == (|-|+)<1-> ]] || set -- -n "$@"
|
|
source $gitstatus_plugin_dir/install || return
|
|
[[ -n $_gitstatus_zsh_daemon ]] || return
|
|
[[ -n $_gitstatus_zsh_version ]] || return
|
|
[[ $_gitstatus_zsh_downloaded == [01] ]] || return
|
|
|
|
if [[ -x $_gitstatus_zsh_daemon ]]; then
|
|
$_gitstatus_zsh_daemon -G $_gitstatus_zsh_version "${(@)args}" >&$pipe_fd
|
|
local -i ret=$?
|
|
[[ $ret == (0|129|130|131|137|141|143) ]] && return ret
|
|
fi
|
|
|
|
(( ! _gitstatus_zsh_downloaded )) || return
|
|
[[ ${GITSTATUS_AUTO_INSTALL:-1} == (|-|+)<1-> ]] || return
|
|
set -- -f "$@"
|
|
_gitstatus_zsh_daemon=
|
|
_gitstatus_zsh_version=
|
|
_gitstatus_zsh_downloaded=
|
|
source $gitstatus_plugin_dir/install || return
|
|
[[ -n $_gitstatus_zsh_daemon ]] || return
|
|
[[ -n $_gitstatus_zsh_version ]] || return
|
|
[[ $_gitstatus_zsh_downloaded == 1 ]] || return
|
|
|
|
$_gitstatus_zsh_daemon -G $_gitstatus_zsh_version "${(@)args}" >&$pipe_fd
|
|
} always {
|
|
local -i ret=$?
|
|
zf_rm -f -- $file_prefix.lock $file_prefix.fifo
|
|
kill -- -$pgid
|
|
}
|
|
} &!
|
|
|
|
(( lock_fd == -1 )) && return
|
|
|
|
{
|
|
if zsystem flock -- $file_prefix.lock && [[ -e $file_prefix.lock ]]; then
|
|
zf_rm -f -- $file_prefix.lock $file_prefix.fifo
|
|
kill -- -$pgid
|
|
fi
|
|
} &!
|
|
}
|
|
|
|
# Starts gitstatusd in the background. Does nothing and succeeds if gitstatusd is already running.
|
|
#
|
|
# Usage: gitstatus_start [OPTION]... NAME
|
|
#
|
|
# -t FLOAT Fail the self-check on initialization if not getting a response from gitstatusd for
|
|
# this this many seconds. Defaults to 5.
|
|
#
|
|
# -s INT Report at most this many staged changes; negative value means infinity.
|
|
# Defaults to 1.
|
|
#
|
|
# -u INT Report at most this many unstaged changes; negative value means infinity.
|
|
# Defaults to 1.
|
|
#
|
|
# -c INT Report at most this many conflicted changes; negative value means infinity.
|
|
# Defaults to 1.
|
|
#
|
|
# -d INT Report at most this many untracked files; negative value means infinity.
|
|
# Defaults to 1.
|
|
#
|
|
# -m INT Report -1 unstaged, untracked and conflicted if there are more than this many
|
|
# files in the index. Negative value means infinity. Defaults to -1.
|
|
#
|
|
# -e Count files within untracked directories like `git status --untracked-files`.
|
|
#
|
|
# -U Unless this option is specified, report zero untracked files for repositories
|
|
# with status.showUntrackedFiles = false.
|
|
#
|
|
# -W Unless this option is specified, report zero untracked files for repositories
|
|
# with bash.showUntrackedFiles = false.
|
|
#
|
|
# -D Unless this option is specified, report zero staged, unstaged and conflicted
|
|
# changes for repositories with bash.showDirtyState = false.
|
|
function gitstatus_start"${1:-}"() {
|
|
emulate -L zsh -o no_aliases -o no_bg_nice -o extended_glob -o typeset_silent || return
|
|
print -rnu2 || return
|
|
|
|
local fsuf=${${(%):-%N}#gitstatus_start}
|
|
|
|
local opt OPTARG
|
|
local -i OPTIND
|
|
local -F timeout=5
|
|
local -i async=0
|
|
local -a args=()
|
|
local -i dirty_max_index_size=-1
|
|
|
|
while getopts ":t:s:u:c:d:m:eaUWD" opt; do
|
|
case $opt in
|
|
a) async=1;;
|
|
+a) async=0;;
|
|
t)
|
|
if [[ $OPTARG != (|+)<->(|.<->)(|[eE](|-|+)<->) ]] || (( ${timeout::=OPTARG} <= 0 )); then
|
|
print -ru2 -- "gitstatus_start: invalid -t argument: $OPTARG"
|
|
return 1
|
|
fi
|
|
;;
|
|
s|u|c|d|m)
|
|
if [[ $OPTARG != (|-|+)<-> ]]; then
|
|
print -ru2 -- "gitstatus_start: invalid -$opt argument: $OPTARG"
|
|
return 1
|
|
fi
|
|
args+=(-$opt $OPTARG)
|
|
[[ $opt == m ]] && dirty_max_index_size=OPTARG
|
|
;;
|
|
e|U|W|D) args+=$opt;;
|
|
+(e|U|W|D)) args=(${(@)args:#-$opt});;
|
|
\?) print -ru2 -- "gitstatus_start: invalid option: $OPTARG" ; return 1;;
|
|
:) print -ru2 -- "gitstatus_start: missing required argument: $OPTARG"; return 1;;
|
|
*) print -ru2 -- "gitstatus_start: invalid option: $opt" ; return 1;;
|
|
esac
|
|
done
|
|
|
|
if (( OPTIND != ARGC )); then
|
|
print -ru2 -- "gitstatus_start: exactly one positional argument is required"
|
|
return 1
|
|
fi
|
|
|
|
local name=$*[OPTIND]
|
|
if [[ $name != [[:IDENT:]]## ]]; then
|
|
print -ru2 -- "gitstatus_start: invalid positional argument: $name"
|
|
return 1
|
|
fi
|
|
|
|
local -i lock_fd resp_fd stderr_fd
|
|
local file_prefix xtrace=/dev/null daemon_log=/dev/null
|
|
if (( _GITSTATUS_STATE_$name )); then
|
|
(( async )) && return
|
|
(( _GITSTATUS_STATE_$name == 2 )) && return
|
|
lock_fd=_GITSTATUS_LOCK_FD_$name
|
|
resp_fd=_GITSTATUS_RESP_FD_$name
|
|
xtrace=${(P)${:-GITSTATUS_XTRACE_$name}}
|
|
daemon_log=${(P)${:-GITSTATUS_DAEMON_LOG_$name}}
|
|
file_prefix=${(P)${:-_GITSTATUS_FILE_PREFIX_$name}}
|
|
else
|
|
typeset -gi _GITSTATUS_START_COUNTER
|
|
local log_level=$GITSTATUS_LOG_LEVEL
|
|
local file_prefix=${${TMPDIR:-/tmp}:A}/gitstatus.$name.$EUID
|
|
file_prefix+=.$sysparams[pid].$EPOCHSECONDS.$((++_GITSTATUS_START_COUNTER))
|
|
(( GITSTATUS_ENABLE_LOGGING )) && : ${log_level:=INFO}
|
|
if [[ -n $log_level ]]; then
|
|
xtrace=$file_prefix.xtrace.log
|
|
daemon_log=$file_prefix.daemon.log
|
|
fi
|
|
args+=(-v ${log_level:-FATAL})
|
|
typeset -g GITSTATUS_XTRACE_$name=$xtrace
|
|
typeset -g GITSTATUS_DAEMON_LOG_$name=$daemon_log
|
|
typeset -g _GITSTATUS_FILE_PREFIX_$name=$file_prefix
|
|
typeset -gi _GITSTATUS_CLIENT_PID_$name="sysparams[pid]"
|
|
typeset -gi _GITSTATUS_DIRTY_MAX_INDEX_SIZE_$name=dirty_max_index_size
|
|
fi
|
|
|
|
() {
|
|
if [[ $xtrace != /dev/null && -o no_xtrace ]]; then
|
|
exec {stderr_fd}>&2 || return
|
|
exec 2>>$xtrace || return
|
|
setopt xtrace
|
|
fi
|
|
|
|
setopt monitor || return
|
|
|
|
if (( ! _GITSTATUS_STATE_$name )); then
|
|
if [[ -r /proc/version && "$(</proc/version)" == *Microsoft* ]]; then
|
|
lock_fd=-1
|
|
else
|
|
print -rn >$file_prefix.lock || return
|
|
zsystem flock -f lock_fd $file_prefix.lock || return
|
|
[[ $lock_fd == <1-> ]] || return
|
|
fi
|
|
|
|
typeset -gi _GITSTATUS_LOCK_FD_$name=lock_fd
|
|
|
|
if [[ -n $USERPROFILE && -d /cygdrive && -d /proc/self/fd ]]; then
|
|
# Work around bugs in Cygwin 32-bit.
|
|
#
|
|
# This hangs:
|
|
#
|
|
# emulate -L zsh
|
|
# () { exec {fd}< $1 } <(:)
|
|
# =true # hangs here
|
|
#
|
|
# This hangs:
|
|
#
|
|
# sysopen -r -u fd <(:)
|
|
local -i fd
|
|
exec {fd}< <(_gitstatus_daemon$fsuf) || return
|
|
{
|
|
[[ -r /proc/self/fd/$fd ]] || return
|
|
sysopen -r -o cloexec -u resp_fd /proc/self/fd/$fd || return
|
|
} always {
|
|
exec {fd} >&- || return
|
|
}
|
|
else
|
|
sysopen -r -o cloexec -u resp_fd <(_gitstatus_daemon$fsuf) || return
|
|
fi
|
|
|
|
typeset -gi GITSTATUS_DAEMON_PID_$name="${sysparams[procsubstpid]:--1}"
|
|
|
|
[[ $resp_fd == <1-> ]] || return
|
|
typeset -gi _GITSTATUS_RESP_FD_$name=resp_fd
|
|
typeset -gi _GITSTATUS_STATE_$name=1
|
|
fi
|
|
|
|
if (( ! async )); then
|
|
(( _GITSTATUS_CLIENT_PID_$name == sysparams[pid] )) || return
|
|
|
|
local pgid
|
|
while (( $#pgid < 20 )); do
|
|
[[ -t $resp_fd ]]
|
|
sysread -s $((20 - $#pgid)) -t $timeout -i $resp_fd 'pgid[$#pgid+1]' || return
|
|
done
|
|
[[ $pgid == ' '#<1-> ]] || return
|
|
typeset -gi GITSTATUS_DAEMON_PID_$name=pgid
|
|
|
|
sysopen -w -o cloexec -u req_fd -- $file_prefix.fifo || return
|
|
[[ $req_fd == <1-> ]] || return
|
|
typeset -gi _GITSTATUS_REQ_FD_$name=req_fd
|
|
|
|
function _gitstatus_process_response_$name-$fsuf() {
|
|
emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent
|
|
local pair=${${(%):-%N}#_gitstatus_process_response_}
|
|
local name=${pair%%-*}
|
|
local fsuf=${pair#*-}
|
|
if (( ARGC == 1 )); then
|
|
_gitstatus_process_response$fsuf $name 0 ''
|
|
else
|
|
gitstatus_stop$fsuf $name
|
|
fi
|
|
}
|
|
if ! zle -F $resp_fd _gitstatus_process_response_$name-$fsuf; then
|
|
unfunction _gitstatus_process_response_$name-$fsuf
|
|
return 1
|
|
fi
|
|
|
|
function _gitstatus_cleanup_$name-$fsuf() {
|
|
emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent
|
|
local pair=${${(%):-%N}#_gitstatus_cleanup_}
|
|
local name=${pair%%-*}
|
|
local fsuf=${pair#*-}
|
|
(( _GITSTATUS_CLIENT_PID_$name == sysparams[pid] )) || return
|
|
gitstatus_stop$fsuf $name
|
|
}
|
|
if ! add-zsh-hook zshexit _gitstatus_cleanup_$name-$fsuf; then
|
|
unfunction _gitstatus_cleanup_$name-$fsuf
|
|
return 1
|
|
fi
|
|
|
|
print -nru $req_fd -- $'hello\x1f\x1e' || return
|
|
local expected=$'hello\x1f0\x1e' actual
|
|
if (( $+functions[p10k] )) && [[ ! -t 1 && ! -t 0 ]]; then
|
|
local -F deadline='EPOCHREALTIME + 4'
|
|
else
|
|
local -F deadline='1'
|
|
fi
|
|
while true; do
|
|
[[ -t $resp_fd ]]
|
|
sysread -s 1 -t $timeout -i $resp_fd actual || return
|
|
[[ $actual == h ]] && break
|
|
[[ $actual == . ]] || return
|
|
(( EPOCHREALTIME < deadline )) && continue
|
|
if (( deadline > 0 )); then
|
|
deadline=0
|
|
if (( stderr_fd )); then
|
|
unsetopt xtrace
|
|
exec 2>&$stderr_fd {stderr_fd}>&-
|
|
stderr_fd=0
|
|
fi
|
|
if (( $+functions[p10k] )); then
|
|
p10k clear-instant-prompt || return
|
|
fi
|
|
if [[ $name == POWERLEVEL9K ]]; then
|
|
local label=powerlevel10k
|
|
else
|
|
local label=gitstatus
|
|
fi
|
|
if [[ -t 2 ]]; then
|
|
local spinner=($'\b%3F-%f' $'\b%3F\\%f' $'\b%3F|%f' $'\b%3F/%f')
|
|
print -Prnu2 -- "[%3F$label%f] fetching %2Fgitstatusd%f .. "
|
|
else
|
|
local spinner=('.')
|
|
print -rnu2 -- "[$label] fetching gitstatusd .."
|
|
fi
|
|
fi
|
|
print -Prnu2 -- $spinner[1]
|
|
spinner=($spinner[2,-1] $spinner[1])
|
|
done
|
|
|
|
if (( deadline == 0 )); then
|
|
if [[ -t 2 ]]; then
|
|
print -Pru2 -- $'\b[%2Fok%f]'
|
|
else
|
|
print -ru2 -- ' [ok]'
|
|
fi
|
|
if [[ $xtrace != /dev/null && -o no_xtrace ]]; then
|
|
exec {stderr_fd}>&2 || return
|
|
exec 2>>$xtrace || return
|
|
setopt xtrace
|
|
fi
|
|
fi
|
|
|
|
while (( $#actual < $#expected )); do
|
|
[[ -t $resp_fd ]]
|
|
sysread -s $(($#expected - $#actual)) -t $timeout -i $resp_fd 'actual[$#actual+1]' || return
|
|
done
|
|
[[ $actual == $expected ]] || return
|
|
|
|
if (( lock_fd != -1 )); then
|
|
zf_rm -- $file_prefix.lock || return
|
|
zsystem flock -u $lock_fd || return
|
|
fi
|
|
unset _GITSTATUS_LOCK_FD_$name
|
|
|
|
typeset -gi _GITSTATUS_STATE_$name=2
|
|
fi
|
|
}
|
|
|
|
local -i err=$?
|
|
(( stderr_fd )) && exec 2>&$stderr_fd {stderr_fd}>&-
|
|
(( err == 0 )) && return
|
|
|
|
gitstatus_stop$fsuf $name
|
|
|
|
setopt prompt_percent no_prompt_subst no_prompt_bang
|
|
print -Pru2 -- '[%F{red}ERROR%f]: gitstatus failed to initialize.'
|
|
print -ru2 -- ''
|
|
print -ru2 -- ' Your Git prompt may disappear or become slow.'
|
|
if [[ -s $xtrace ]]; then
|
|
print -ru2 -- ''
|
|
print -ru2 -- " The content of ${(q-)xtrace} (gitstatus_start xtrace):"
|
|
print -Pru2 -- '%F{yellow}'
|
|
>&2 awk '{print " " $0}' <$xtrace
|
|
print -Pru2 -- "%F{red} ^ this command failed ($err)%f"
|
|
fi
|
|
if [[ -s $daemon_log ]]; then
|
|
print -ru2 -- ''
|
|
print -ru2 -- " The content of ${(q-)daemon_log} (gitstatus daemon log):"
|
|
print -Pru2 -- '%F{yellow}'
|
|
>&2 awk '{print " " $0}' <$daemon_log
|
|
print -Pnru2 -- '%f'
|
|
fi
|
|
if [[ $GITSTATUS_LOG_LEVEL == DEBUG ]]; then
|
|
print -ru2 -- ''
|
|
print -ru2 -- ' Your system information:'
|
|
print -Pru2 -- '%F{yellow}'
|
|
print -ru2 -- " zsh: $ZSH_VERSION"
|
|
print -ru2 -- " uname -a: $(uname -a)"
|
|
print -Pru2 -- '%f'
|
|
print -ru2 -- ' If you need help, open an issue and attach this whole error message to it:'
|
|
print -ru2 -- ''
|
|
print -Pru2 -- ' %F{green}https://github.com/romkatv/gitstatus/issues/new%f'
|
|
else
|
|
print -ru2 -- ''
|
|
print -ru2 -- ' Run the following command to retry with extra diagnostics:'
|
|
print -Pru2 -- '%F{green}'
|
|
local env="GITSTATUS_LOG_LEVEL=DEBUG"
|
|
if [[ -n $GITSTATUS_NUM_THREADS ]]; then
|
|
env+=" GITSTATUS_NUM_THREADS=${(q)GITSTATUS_NUM_THREADS}"
|
|
fi
|
|
if [[ -n $GITSTATUS_DAEMON ]]; then
|
|
env+=" GITSTATUS_DAEMON=${(q)GITSTATUS_DAEMON}"
|
|
fi
|
|
if [[ -n $GITSTATUS_AUTO_INSTALL ]]; then
|
|
env+=" GITSTATUS_AUTO_INSTALL=${(q)GITSTATUS_AUTO_INSTALL}"
|
|
fi
|
|
if [[ -n $GITSTATUS_CACHE_DIR ]]; then
|
|
env+=" GITSTATUS_CACHE_DIR=${(q)GITSTATUS_CACHE_DIR}"
|
|
fi
|
|
print -nru2 -- " ${env} gitstatus_start ${(@q-)*}"
|
|
print -Pru2 -- '%f'
|
|
print -ru2 -- ''
|
|
local zshrc=${(D)ZDOTDIR:-~}/.zshrc
|
|
print -ru2 -- " If this command produces no output, add the following parameter to $zshrc:"
|
|
print -ru2 -- ''
|
|
print -Pru2 -- '%F{green} GITSTATUS_LOG_LEVEL=DEBUG%f'
|
|
print -ru2 -- ''
|
|
print -ru2 -- ' With this parameter gitstatus will print additional information on error.'
|
|
fi
|
|
|
|
return err
|
|
}
|
|
|
|
# Stops gitstatusd if it's running.
|
|
#
|
|
# Usage: gitstatus_stop NAME.
|
|
function gitstatus_stop"${1:-}"() {
|
|
emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent
|
|
|
|
local fsuf=${${(%):-%N}#gitstatus_stop}
|
|
|
|
if (( ARGC != 1 )); then
|
|
print -ru2 -- "gitstatus_stop: exactly one positional argument is required"
|
|
return 1
|
|
fi
|
|
|
|
local name=$1
|
|
if [[ $name != [[:IDENT:]]## ]]; then
|
|
print -ru2 -- "gitstatus_stop: invalid positional argument: $name"
|
|
return 1
|
|
fi
|
|
|
|
local state_var=_GITSTATUS_STATE_$name
|
|
local req_fd_var=_GITSTATUS_REQ_FD_$name
|
|
local resp_fd_var=_GITSTATUS_RESP_FD_$name
|
|
local lock_fd_var=_GITSTATUS_LOCK_FD_$name
|
|
local client_pid_var=_GITSTATUS_CLIENT_PID_$name
|
|
local daemon_pid_var=GITSTATUS_DAEMON_PID_$name
|
|
local inflight_var=_GITSTATUS_NUM_INFLIGHT_$name
|
|
local file_prefix_var=_GITSTATUS_FILE_PREFIX_$name
|
|
local dirty_max_index_size_var=_GITSTATUS_DIRTY_MAX_INDEX_SIZE_$name
|
|
|
|
local req_fd=${(P)req_fd_var}
|
|
local resp_fd=${(P)resp_fd_var}
|
|
local lock_fd=${(P)lock_fd_var}
|
|
local daemon_pid=${(P)daemon_pid_var}
|
|
local file_prefix=${(P)file_prefix_var}
|
|
|
|
local cleanup=_gitstatus_cleanup_$name-$fsuf
|
|
local process=_gitstatus_process_response_$name-$fsuf
|
|
|
|
if (( $+functions[$cleanup] )); then
|
|
add-zsh-hook -d zshexit $cleanup
|
|
unfunction -- $cleanup
|
|
fi
|
|
|
|
if (( $+functions[$process] )); then
|
|
[[ -n $resp_fd ]] && zle -F $resp_fd
|
|
unfunction -- $process
|
|
fi
|
|
|
|
[[ $daemon_pid == <1-> ]] && kill -- -$daemon_pid 2>/dev/null
|
|
[[ $file_prefix == /* ]] && zf_rm -f -- $file_prefix.lock $file_prefix.fifo
|
|
[[ $lock_fd == <1-> ]] && zsystem flock -u $lock_fd
|
|
[[ $req_fd == <1-> ]] && exec {req_fd}>&-
|
|
[[ $resp_fd == <1-> ]] && exec {resp_fd}>&-
|
|
|
|
unset $state_var $req_fd_var $lock_fd_var $resp_fd_var $client_pid_var $daemon_pid_var
|
|
unset $inflight_var $file_prefix_var $dirty_max_index_size_var
|
|
|
|
unset VCS_STATUS_RESULT
|
|
_gitstatus_clear$fsuf
|
|
}
|
|
|
|
# Usage: gitstatus_check NAME.
|
|
#
|
|
# Returns 0 if and only if `gitstatus_start NAME` has succeeded previously.
|
|
# If it returns non-zero, gitstatus_query NAME is guaranteed to return non-zero.
|
|
function gitstatus_check"${1:-}"() {
|
|
emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent
|
|
|
|
local fsuf=${${(%):-%N}#gitstatus_check}
|
|
|
|
if (( ARGC != 1 )); then
|
|
print -ru2 -- "gitstatus_check: exactly one positional argument is required"
|
|
return 1
|
|
fi
|
|
|
|
local name=$1
|
|
if [[ $name != [[:IDENT:]]## ]]; then
|
|
print -ru2 -- "gitstatus_check: invalid positional argument: $name"
|
|
return 1
|
|
fi
|
|
|
|
(( _GITSTATUS_STATE_$name == 2 ))
|
|
}
|
|
|
|
(( ${#_gitstatus_opts} )) && setopt ${_gitstatus_opts[@]}
|
|
'builtin' 'unset' '_gitstatus_opts'
|