check_for_upgrade.sh 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. #!/usr/bin/env zsh
  2. zmodload zsh/datetime
  3. function _current_epoch() {
  4. echo $(( $EPOCHSECONDS / 60 / 60 / 24 ))
  5. }
  6. function _update_zsh_update() {
  7. echo "LAST_EPOCH=$(_current_epoch)" >! ~/.zsh-update
  8. }
  9. function _upgrade_zsh() {
  10. env ZSH=$ZSH sh $ZSH/tools/upgrade.sh
  11. # update the zsh file
  12. _update_zsh_update
  13. }
  14. epoch_target=$UPDATE_ZSH_DAYS
  15. if [[ -z "$epoch_target" ]]; then
  16. # Default to old behavior
  17. epoch_target=13
  18. fi
  19. # Cancel upgrade if the current user doesn't have write permissions for the
  20. # oh-my-zsh directory.
  21. [[ -w "$ZSH" ]] || return 0
  22. # Cancel upgrade if git is unavailable on the system
  23. whence git >/dev/null || return 0
  24. if mkdir "$ZSH/log/update.lock" 2>/dev/null; then
  25. if [ -f ~/.zsh-update ]; then
  26. . ~/.zsh-update
  27. if [[ -z "$LAST_EPOCH" ]]; then
  28. _update_zsh_update && return 0;
  29. fi
  30. epoch_diff=$(($(_current_epoch) - $LAST_EPOCH))
  31. if [ $epoch_diff -gt $epoch_target ]; then
  32. if [ "$DISABLE_UPDATE_PROMPT" = "true" ]; then
  33. _upgrade_zsh
  34. else
  35. echo "[Oh My Zsh] Would you like to update? [Y/n]: \c"
  36. read line
  37. if [[ "$line" == Y* ]] || [[ "$line" == y* ]] || [ -z "$line" ]; then
  38. _upgrade_zsh
  39. else
  40. _update_zsh_update
  41. fi
  42. fi
  43. fi
  44. else
  45. # create the zsh file
  46. _update_zsh_update
  47. fi
  48. rmdir $ZSH/log/update.lock
  49. fi