discourse/lib/email_controller_helper/digest_email_unsubscriber.rb

# frozen_string_literal: true

module EmailControllerHelper
  class DigestEmailUnsubscriber < BaseEmailUnsubscriber
    def prepare_unsubscribe_options(controller)
      super(controller)
      controller.instance_variable_set(:@digest_unsubscribe, !SiteSetting.disable_digest_emails)

      frequency_in_minutes = key_owner.user_option.digest_after_minutes
      email_digests = key_owner.user_option.email_digests
      frequencies = DigestEmailSiteSetting.values.dup
      never = frequencies.delete_at(0)
      allowed_frequencies = %w[never weekly every_month every_six_months]

      result =
        frequencies.reduce(
          frequencies: [],
          current: nil,
          selected: nil,
          take_next: false,
        ) do |memo, v|
          memo[:current] = v[:name] if v[:value] == frequency_in_minutes && email_digests
          next(memo) unless allowed_frequencies.include?(v[:name])

          memo.tap do |m|
            m[:selected] = v[:value] if m[:take_next] && email_digests
            m[:frequencies] << [I18n.t("unsubscribe.digest_frequency.#{v[:name]}"), v[:value]]
            m[:take_next] = !m[:take_next] && m[:current]
          end
        end

      digest_frequencies =
        result
          .slice(:frequencies, :current, :selected)
          .tap do |r|
            r[:frequencies] << [
              I18n.t("unsubscribe.digest_frequency.#{never[:name]}"),
              never[:value],
            ]
            r[:selected] ||= never[:value]
            r[:current] ||= never[:name]
          end

      controller.instance_variable_set(:@digest_frequencies, digest_frequencies)
    end

    def unsubscribe(params)
      updated = super(params)

      if params[:digest_after_minutes]
        digest_frequency = params[:digest_after_minutes].to_i

        unsubscribe_key.user.user_option.update_columns(
          digest_after_minutes: digest_frequency,
          email_digests: digest_frequency.positive?,
        )
        updated = true
      end

      updated
    end
  end
end