check_for_upgrade.sh 1.23 KB
Newer Older
1
2
3
#!/usr/bin/env zsh

zmodload zsh/datetime
4

5
function _current_epoch() {
6
  echo $EPOCHSECONDS
7
8
9
}

function _update_zsh_update() {
10
  echo "LAST_EPOCH=$(_current_epoch)" >! ~/.zsh-update
11
}
12

13
14
15
16
17
18
function _upgrade_zsh() {
  /usr/bin/env ZSH=$ZSH /bin/sh $ZSH/tools/upgrade.sh
  # update the zsh file
  _update_zsh_update
}

Eric Danielson's avatar
Eric Danielson committed
19
20
21
22
23
24
epoch_target=$UPDATE_ZSH_DAYS
if [[ -z "$epoch_target" ]]; then
  # Default to old behavior
  epoch_target=13
fi

25
26
epoch_target_seconds=$((epoch_target * 86400))

n.st's avatar
n.st committed
27
28
29
30
31
32
[ -f ~/.profile ] && source ~/.profile

# Cancel upgrade if the current user doesn't have write permissions for the
# oh-my-zsh directory.
[[ -w "$ZSH" ]] || return 0

33
34
if [ -f ~/.zsh-update ]
then
Gerhard Lazu's avatar
Gerhard Lazu committed
35
  . ~/.zsh-update
36
37
38
39
40

  if [[ -z "$LAST_EPOCH" ]]; then
    _update_zsh_update && return 0;
  fi

Christopher Chow's avatar
Christopher Chow committed
41
  epoch_diff=$(($(_current_epoch) - $LAST_EPOCH))
42
  if [ $epoch_diff -gt $epoch_target_seconds ]
43
  then
44
    if [ "$DISABLE_UPDATE_PROMPT" = "true" ]
45
    then
46
47
48
49
50
      _upgrade_zsh
    else
      echo "[Oh My Zsh] Would you like to check for updates?"
      echo "Type Y to update oh-my-zsh: \c"
      read line
51
      if [ "$line" = Y ] || [ "$line" = y ]; then
52
        _upgrade_zsh
53
54
      else
        _update_zsh_update
55
      fi
56
57
    fi
  fi
nebirhos's avatar
nebirhos committed
58
59
60
else
  # create the zsh file
  _update_zsh_update
61
fi
62