# 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=${${(%):-%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. 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() { emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent 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 else while true; do _gitstatus_process_response $name $timeout $req_id || return [[ $VCS_STATUS_RESULT == *-async ]] || break done fi [[ $VCS_STATUS_RESULT != tout || -n $callback ]] } function gitstatus_process_results() { emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent 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 $name $timeout '' || return [[ $VCS_STATUS_RESULT == *-async ]] || break done return 0 } function _gitstatus_clear() { 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() { 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 fi return 0 else gitstatus_stop $name return 1 fi } while [[ $buf != *$'\x1e' ]]; do if ! sysread -i $resp_fd 'buf[$#buf+1]'; then gitstatus_stop $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 fi (( --_GITSTATUS_NUM_INFLIGHT_$name )) [[ $VCS_STATUS_RESULT == *-async ]] && emulate zsh -c "${resp[1]#* }" done return 0 } # 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() { emulate -L zsh || return setopt no_aliases no_bg_nice extended_glob typeset_silent monitor || return print -rnu2 || return 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 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 local log_level=$GITSTATUS_LOG_LEVEL local file_prefix=${TMPDIR:-/tmp}/gitstatus.$EUID.$sysparams[pid].$EPOCHSECONDS (( 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 local -i stderr_fd { if [[ $xtrace != /dev/null && -o no_xtrace ]]; then exec {stderr_fd}>&2 exec 2>>$xtrace || return setopt xtrace fi if (( ! _GITSTATUS_STATE_$name )); then print -rn >$file_prefix.lock || return zsystem flock -f lock_fd $file_prefix.lock || return typeset -gi _GITSTATUS_LOCK_FD_$name=lock_fd { () { if [[ $sysparams[procsubstpid] == <1-> ]]; then typeset -gi GITSTATUS_DAEMON_PID_$name="sysparams[procsubstpid]" fi sysopen -r -o cloexec -u resp_fd -- $1 || return typeset -gi _GITSTATUS_RESP_FD_$name=resp_fd } <( exec 2>&3 3>&- || return local pgid=$sysparams[pid] [[ $pgid == <1-> ]] || return { { trap '' PIPE local os daemon=$GITSTATUS_DAEMON if [[ -z $daemon ]]; then os="$(uname -s)" || return [[ -n $os ]] || return case $os in Linux) [[ "$(uname -o)" == Android ]] && os=android;; (#i)cygwin_nt-*) os=cygwin_nt-10.0;; (#i)(mingw|msys)) os=msys_nt-10.0;; esac local arch arch="$(uname -m)" || return [[ -n $arch ]] || return daemon=$_gitstatus_plugin_dir/bin/gitstatusd-${os:l}-${arch:l} fi [[ -x $daemon ]] || return if [[ $GITSTATUS_NUM_THREADS == <1-> ]]; then args+=(-t $GITSTATUS_NUM_THREADS) else local cpus if [[ $+commands[sysctl] == 0 || "${os:-$(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 -rn -- ${(l:20:)pgid} || return exec <$file_prefix.fifo || return zf_rm -- $file_prefix.fifo || return $daemon "${(@)args}" if [[ $? != (0|10) && $? -le 128 && -x ${daemon}-static ]]; then ${daemon}-static "${(@)args}" || return fi } always { kill -- -$pgid } } &! { if zsystem flock -- $file_prefix.lock && [[ -e $file_prefix.lock ]]; then zf_rm -f -- $file_prefix.lock $file_prefix.fifo kill -- -$pgid fi } &! ) || return } 3>>$daemon_log /dev/null || return 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 typeset -gi _GITSTATUS_REQ_FD_$name=req_fd function _gitstatus_process_response_$name() { emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent local name=${${(%):-%N}#_gitstatus_process_response_} if (( ARGC == 1 )); then _gitstatus_process_response $name 0 '' else gitstatus_stop $name fi } if ! zle -F $resp_fd _gitstatus_process_response_$name; then unfunction _gitstatus_process_response_$name return 1 fi function _gitstatus_cleanup_$name() { emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent local name=${${(%):-%N}#_gitstatus_cleanup_} (( _GITSTATUS_CLIENT_PID_$name == sysparams[pid] )) || return gitstatus_stop $name } if ! add-zsh-hook zshexit _gitstatus_cleanup_$name; then unfunction _gitstatus_cleanup_$name return 1 fi print -nru $req_fd -- $'hello\x1f\x1e' || return local expected=$'hello\x1f0' actual while (( $#actual < $#expected )); do [[ -t $resp_fd ]] sysread -s $(($#expected - $#actual)) -t $timeout -i $resp_fd 'actual[$#actual+1]' || return done [[ $actual == $expected ]] || return zf_rm -- $file_prefix.lock || return zsystem flock -u $lock_fd || return typeset -gi _GITSTATUS_STATE_$name=2 fi } always { local err=$? if (( stderr_fd )); then unsetopt xtrace exec 2>&$stderr_fd fi (( err )) || return gitstatus_stop $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 -Pnru2 -- '%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 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 } } # Stops gitstatusd if it's running. # # Usage: gitstatus_stop NAME. function gitstatus_stop() { emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent 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 local process=_gitstatus_process_response_$name [[ -n $daemon_pid ]] && kill -- -$daemon_pid 2>/dev/null 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 [[ -n $file_prefix ]] && zf_rm -f -- "$file_prefix.lock" "$file_prefix.fifo" [[ -n $lock_fd ]] && zsystem flock -u $lock_fd [[ -n $req_fd ]] && exec {req_fd}>&- [[ -n $resp_fd ]] && 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 } # 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() { emulate -L zsh -o no_aliases -o extended_glob -o typeset_silent 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'