1
0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2024-12-12 08:25:53 +08:00
ohmyzsh/plugins/git-auto-fetch/git-auto-fetch.plugin.zsh

66 lines
2.2 KiB
Bash

# Default auto-fetch interval: 60 seconds
: ${GIT_AUTO_FETCH_INTERVAL:=60}
# Necessary for the git-fetch-all function
zmodload zsh/datetime
zmodload -F zsh/stat b:zstat # only zstat command, not stat command
function git-fetch-all {
(
# Get git root directory
if ! gitdir="$(command git rev-parse --git-dir 2>/dev/null)"; then
return 0
fi
# Do nothing if auto-fetch is disabled or don't have permissions
if [[ ! -w "$gitdir" || -f "$gitdir/NO_AUTO_FETCH" ]] ||
[[ -f "$gitdir/FETCH_LOG" && ! -w "$gitdir/FETCH_LOG" ]]; then
return 0
fi
# Get time (seconds) when auto-fetch was last run
lastrun="$(zstat +mtime "$gitdir/FETCH_LOG" 2>/dev/null || echo 0)"
# Do nothing if not enough time has passed since last auto-fetch
if (( EPOCHSECONDS - lastrun < $GIT_AUTO_FETCH_INTERVAL )); then
return 0
fi
# Fetch all remotes (avoid ssh passphrase prompt)
date -R &>! "$gitdir/FETCH_LOG"
GIT_SSH_COMMAND="command ssh -o BatchMode=yes" \
GIT_TERMINAL_PROMPT=0 \
command git fetch --all --recurse-submodules=yes 2>/dev/null &>> "$gitdir/FETCH_LOG"
) &|
}
function git-auto-fetch {
# Do nothing if not in a git repository
command git rev-parse --is-inside-work-tree &>/dev/null || return 0
# Remove or create guard file depending on its existence
local guard="$(command git rev-parse --git-dir)/NO_AUTO_FETCH"
if [[ -f "$guard" ]]; then
command rm "$guard" && echo "${fg_bold[green]}enabled${reset_color}"
else
command touch "$guard" && echo "${fg_bold[red]}disabled${reset_color}"
fi
}
# zle-line-init widget (don't redefine if already defined)
(( ! ${+functions[_git-auto-fetch_zle-line-init]} )) || return 0
case "$widgets[zle-line-init]" in
# Simply define the function if zle-line-init doesn't yet exist
builtin|"") function _git-auto-fetch_zle-line-init() {
git-fetch-all
} ;;
# Override the current zle-line-init widget, calling the old one
user:*) zle -N _git-auto-fetch_orig_zle-line-init "${widgets[zle-line-init]#user:}"
function _git-auto-fetch_zle-line-init() {
git-fetch-all
zle _git-auto-fetch_orig_zle-line-init -- "$@"
} ;;
esac
zle -N zle-line-init _git-auto-fetch_zle-line-init