#!/usr/bin/env zsh

zmodload zsh/datetime

function _current_epoch() {
  echo $(( $EPOCHSECONDS / 60 / 60 / 24 ))
}

function _update_dotfiles_update() {
  echo "LAST_EPOCH=$(_current_epoch)" >! ~/.dotfiles-update
}

function _upgrade_dotfiles() {
  env _DOTFILES=$_DOTFILES sh $_DOTFILES/upgrade.sh
  # update the zsh file
  _update_dotfiles_update
}

# Configure this in shell/env
epoch_target=$UPDATE_DOTFILES_DAYS
if [[ -z "$epoch_target" ]]; then
  # Default to every 2 weeks
  epoch_target=13
fi

# Cancel upgrade if the current user doesn't have write permissions for the
# dotfiles directory.
[[ -w "$_DOTFILES" ]] || echo "You can't write to $(dotfiles)"; return 1

# Cancel upgrade if git is unavailable on the system
whence git >/dev/null || echo "git is not available"; return 2

if mkdir "$_DOTFILES/log/update.lock" 2>/dev/null; then
  if [ -f ~/.dotfiles-update ]; then
    . ~/.dotfiles-update

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

    epoch_diff=$(($(_current_epoch) - $LAST_EPOCH))
    if [ $epoch_diff -gt $epoch_target ]; then
      if [ "$DISABLE_UPDATE_PROMPT" = "true" ]; then
        _upgrade_dotfiles
      else
        echo "[Dotfiles] Would you like to check for updates? [Y/n]: \c"
        read line
        if [[ "$line" == Y* ]] || [[ "$line" == y* ]] || [ -z "$line" ]; then
          _upgrade_dotfiles
        else
          _update_dotfiles_update
        fi
      fi
    fi
  else
    # create the zsh file
    _update_dotfiles_update
  fi

  rmdir $_DOTFILES/log/update.lock
fi