discourse/app/jobs/scheduled/bookmark_reminder_notifications.rb

# frozen_string_literal: true

module Jobs

  # Runs periodically to send out bookmark reminders, capped at 300 at a time.
  # Any leftovers will be caught in the next run, because the reminder_at column
  # is set to NULL once a reminder has been sent.
  class BookmarkReminderNotifications < ::Jobs::Scheduled
    JOB_RUN_NUMBER_KEY ||= 'jobs_bookmark_reminder_notifications_job_run_num'.freeze
    AT_DESKTOP_CONSISTENCY_RUN_NUMBER ||= 6

    every 5.minutes

    def self.max_reminder_notifications_per_run
      @@max_reminder_notifications_per_run ||= 3
      @@max_reminder_notifications_per_run
    end

    def self.max_reminder_notifications_per_run=(max)
      @@max_reminder_notifications_per_run = max
    end

    def execute(args = nil)
      bookmarks = Bookmark.pending_reminders
        .where.not(reminder_type: Bookmark.reminder_types[:at_desktop])
        .includes(:user).order('reminder_at ASC')

      bookmarks.limit(BookmarkReminderNotifications.max_reminder_notifications_per_run).each do |bookmark|
        BookmarkReminderNotificationHandler.send_notification(bookmark)
      end

      # we only want to ensure the desktop consistency every X runs of this job
      # (every 30 mins in this case) so we don't bother redis too much, and the
      # at desktop consistency problem should not really happen unless people
      # are setting the "at desktop" reminder, going out for milk, and never coming
      # back
      current_job_run_number = Discourse.redis.get(JOB_RUN_NUMBER_KEY).to_i
      if current_job_run_number == AT_DESKTOP_CONSISTENCY_RUN_NUMBER
        ensure_at_desktop_consistency
      end

      increment_job_run_number(current_job_run_number)
    end

    def increment_job_run_number(current_job_run_number)
      if current_job_run_number.zero? || current_job_run_number == AT_DESKTOP_CONSISTENCY_RUN_NUMBER
        new_job_run_number = 1
      else
        new_job_run_number = current_job_run_number + 1
      end
      Discourse.redis.set(JOB_RUN_NUMBER_KEY, new_job_run_number)
    end

    def ensure_at_desktop_consistency
      pending_at_desktop_bookmark_reminders = \
        Bookmark.includes(:user)
          .references(:user)
          .pending_at_desktop_reminders
          .where('users.last_seen_at >= :one_day_ago', one_day_ago: 1.day.ago.utc)

      return if pending_at_desktop_bookmark_reminders.count.zero?

      unique_users = pending_at_desktop_bookmark_reminders.map(&:user).uniq.map { |u| [u.id, u] }.flatten
      unique_users = Hash[*unique_users]
      pending_reminders_for_redis_check = unique_users.keys.map do |user_id|
        "#{BookmarkReminderNotificationHandler::PENDING_AT_DESKTOP_KEY_PREFIX}#{user_id}"
      end

      Discourse.redis.mget(pending_reminders_for_redis_check).each.with_index do |value, idx|
        next if value.present?
        user_id = pending_reminders_for_redis_check[idx][/\d+/].to_i
        user = unique_users[user_id]

        user_pending_bookmark_reminders = pending_at_desktop_bookmark_reminders.select do |bookmark|
          bookmark.user == user
        end

        user_expired_bookmark_reminders = user_pending_bookmark_reminders.select do |bookmark|
          bookmark.reminder_set_at <= expiry_limit_datetime
        end

        next if user_pending_bookmark_reminders.length == user_expired_bookmark_reminders.length

        # only tell the cache-gods that this user has pending "at desktop" reminders
        # if they haven't let them all expire before coming back to their desktop
        #
        # the next time they visit the desktop the reminders will be cleared out once
        # the notifications are sent
        BookmarkReminderNotificationHandler.cache_pending_at_desktop_reminder(user)
      end

      clear_expired_at_desktop_reminders
    end

    def clear_expired_at_desktop_reminders
      Bookmark
        .pending_at_desktop_reminders
        .where('reminder_set_at <= :expiry_limit_datetime', expiry_limit_datetime: expiry_limit_datetime)
        .update_all(
          reminder_set_at: nil, reminder_type: nil
        )
    end

    def expiry_limit_datetime
      BookmarkReminderNotificationHandler::PENDING_AT_DESKTOP_EXPIRY_DAYS.days.ago.utc
    end
  end
end