mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2024-11-22 12:09:36 +08:00
112 lines
3.6 KiB
Bash
112 lines
3.6 KiB
Bash
# Migrate .zsh-update file to $ZSH_CACHE_DIR
|
|
if [[ -f ~/.zsh-update && ! -f "${ZSH_CACHE_DIR}/.zsh-update" ]]; then
|
|
mv ~/.zsh-update "${ZSH_CACHE_DIR}/.zsh-update"
|
|
fi
|
|
|
|
# Get user's update preferences
|
|
#
|
|
# Supported update modes:
|
|
# - prompt (default): the user is asked before updating when it's time to update
|
|
# - auto: the update is performed automatically when it's time
|
|
# - reminder: a reminder is shown to the user when it's time to update
|
|
# - disabled: automatic update is turned off
|
|
zstyle -s ':omz:update' mode update_mode || update_mode=prompt
|
|
|
|
# Support old-style settings
|
|
[[ "$DISABLE_UPDATE_PROMPT" != true ]] || update_mode=auto
|
|
[[ "$DISABLE_AUTO_UPDATE" != true ]] || update_mode=disabled
|
|
|
|
# Cancel update if:
|
|
# - the automatic update is disabled.
|
|
# - the current user doesn't have write permissions nor owns the $ZSH directory.
|
|
# - git is unavailable on the system.
|
|
if [[ "$update_mode" = disabled ]] \
|
|
|| [[ ! -w "$ZSH" || ! -O "$ZSH" ]] \
|
|
|| ! command -v git &>/dev/null; then
|
|
unset update_mode
|
|
return
|
|
fi
|
|
|
|
function current_epoch() {
|
|
zmodload zsh/datetime
|
|
echo $(( EPOCHSECONDS / 60 / 60 / 24 ))
|
|
}
|
|
|
|
function update_last_updated_file() {
|
|
echo "LAST_EPOCH=$(current_epoch)" >! "${ZSH_CACHE_DIR}/.zsh-update"
|
|
}
|
|
|
|
function update_ohmyzsh() {
|
|
if ZSH="$ZSH" zsh -f "$ZSH/tools/upgrade.sh" --interactive; then
|
|
update_last_updated_file
|
|
fi
|
|
}
|
|
|
|
() {
|
|
emulate -L zsh
|
|
|
|
local epoch_target mtime option LAST_EPOCH
|
|
|
|
# Remove lock directory if older than a day
|
|
zmodload zsh/datetime
|
|
zmodload -F zsh/stat b:zstat
|
|
if mtime=$(zstat +mtime "$ZSH/log/update.lock" 2>/dev/null); then
|
|
if (( (mtime + 3600 * 24) < EPOCHSECONDS )); then
|
|
command rm -rf "$ZSH/log/update.lock"
|
|
fi
|
|
fi
|
|
|
|
# Check for lock directory
|
|
if ! command mkdir "$ZSH/log/update.lock" 2>/dev/null; then
|
|
return
|
|
fi
|
|
|
|
# Remove lock directory on exit. `return $ret` is important for when trapping a SIGINT:
|
|
# The return status from the function is handled specially. If it is zero, the signal is
|
|
# assumed to have been handled, and execution continues normally. Otherwise, the shell
|
|
# will behave as interrupted except that the return status of the trap is retained.
|
|
# This means that for a CTRL+C, the trap needs to return the same exit status so that
|
|
# the shell actually exits what it's running.
|
|
trap "
|
|
ret=\$?
|
|
command rm -rf '$ZSH/log/update.lock'
|
|
return \$ret
|
|
" EXIT INT QUIT
|
|
|
|
# Create or update .zsh-update file if missing or malformed
|
|
if ! source "${ZSH_CACHE_DIR}/.zsh-update" 2>/dev/null || [[ -z "$LAST_EPOCH" ]]; then
|
|
update_last_updated_file
|
|
return
|
|
fi
|
|
|
|
# Number of days before trying to update again
|
|
zstyle -s ':omz:update' frequency epoch_target || epoch_target=${UPDATE_ZSH_DAYS:-13}
|
|
# Test if enough time has passed until the next update
|
|
if (( ( $(current_epoch) - $LAST_EPOCH ) < $epoch_target )); then
|
|
return
|
|
fi
|
|
|
|
# Ask for confirmation before updating unless in auto mode
|
|
if [[ "$update_mode" = auto ]]; then
|
|
update_ohmyzsh
|
|
elif [[ "$update_mode" = reminder ]]; then
|
|
echo "[oh-my-zsh] It's time to update! You can do that by running \`omz update\`"
|
|
else
|
|
# input sink to swallow all characters typed before the prompt
|
|
# and add a newline if there wasn't one after characters typed
|
|
while read -t -k 1 option; do true; done
|
|
[[ "$option" != ($'\n'|"") ]] && echo
|
|
|
|
echo -n "[oh-my-zsh] Would you like to update? [Y/n] "
|
|
read -r -k 1 option
|
|
[[ "$option" != $'\n' ]] && echo
|
|
case "$option" in
|
|
[yY$'\n']) update_ohmyzsh ;;
|
|
[nN]) update_last_updated_file ;;
|
|
esac
|
|
fi
|
|
}
|
|
|
|
unset update_mode
|
|
unset -f current_epoch update_last_updated_file update_ohmyzsh
|