stow/oh-my-zsh/.oh-my-zsh/plugins/git-auto-fetch/git-auto-fetch.plugin.zsh

   1 # Default auto-fetch interval: 60 seconds
   2 : ${GIT_AUTO_FETCH_INTERVAL:=60}
   3
   4 # Necessary for the git-fetch-all function
   5 zmodload zsh/datetime
   6 zmodload -F zsh/stat b:zstat  # only zstat command, not stat command
   7
   8 function git-fetch-all {
   9   (
  10     # Get git root directory
  11     if ! gitdir="$(command git rev-parse --git-dir 2>/dev/null)"; then
  12       return 0
  13     fi
  14
  15     # Do nothing if auto-fetch is disabled or don't have permissions
  16     if [[ ! -w "$gitdir" || -f "$gitdir/NO_AUTO_FETCH" ]] ||
  17        [[ -f "$gitdir/FETCH_LOG" && ! -w "$gitdir/FETCH_LOG" ]]; then
  18       return 0
  19     fi
  20
  21     # Get time (seconds) when auto-fetch was last run
  22     lastrun="$(zstat +mtime "$gitdir/FETCH_LOG" 2>/dev/null || echo 0)"
  23     # Do nothing if not enough time has passed since last auto-fetch
  24     if (( EPOCHSECONDS - lastrun < $GIT_AUTO_FETCH_INTERVAL )); then
  25       return 0
  26     fi
  27
  28     # Fetch all remotes (avoid ssh passphrase prompt)
  29     date -R &>! "$gitdir/FETCH_LOG"
  30     GIT_SSH_COMMAND="command ssh -o BatchMode=yes" \
  31     GIT_TERMINAL_PROMPT=0 \
  32       command git fetch --all 2>/dev/null &>> "$gitdir/FETCH_LOG"
  33   ) &|
  34 }
  35
  36 function git-auto-fetch {
  37   # Do nothing if not in a git repository
  38   command git rev-parse --is-inside-work-tree &>/dev/null || return 0
  39
  40   # Remove or create guard file depending on its existence
  41   local guard="$(command git rev-parse --git-dir)/NO_AUTO_FETCH"
  42   if [[ -f "$guard" ]]; then
  43     command rm "$guard" && echo "${fg_bold[green]}enabled${reset_color}"
  44   else
  45     command touch "$guard" && echo "${fg_bold[red]}disabled${reset_color}"
  46   fi
  47 }
  48
  49 # zle-line-init widget (don't redefine if already defined)
  50 (( ! ${+functions[_git-auto-fetch_zle-line-init]} )) || return 0
  51
  52 case "$widgets[zle-line-init]" in
  53   # Simply define the function if zle-line-init doesn't yet exist
  54   builtin|"") function _git-auto-fetch_zle-line-init() {
  55       git-fetch-all
  56     } ;;
  57   # Override the current zle-line-init widget, calling the old one
  58   user:*) zle -N _git-auto-fetch_orig_zle-line-init "${widgets[zle-line-init]#user:}"
  59     function _git-auto-fetch_zle-line-init() {
  60       git-fetch-all
  61       zle _git-auto-fetch_orig_zle-line-init -- "$@"
  62     } ;;
  63 esac
  64
  65 zle -N zle-line-init _git-auto-fetch_zle-line-init