nas/app/models/form/admin_settings.rb

# frozen_string_literal: true

class Form::AdminSettings
  include ActiveModel::Model

  include AuthorizedFetchHelper

  KEYS = %i(
    site_contact_username
    site_contact_email
    site_title
    site_short_description
    site_extended_description
    site_terms
    registrations_mode
    closed_registrations_message
    registration_button_message
    registrations_limit
    registrations_limit_per_day
    registrations_start_hour
    registrations_end_hour
    registrations_secondary_start_hour
    registrations_secondary_end_hour
    timeline_preview
    bootstrap_timeline_accounts
    theme
    activity_api_enabled
    peers_api_enabled
    preview_sensitive_media
    custom_css
    profile_directory
    thumbnail
    mascot
    trends
    trends_as_landing_page
    trendable_by_default
    show_domain_blocks
    show_domain_blocks_rationale
    noindex
    require_invite_text
    media_cache_retention_period
    content_cache_retention_period
    backups_retention_period
    delete_content_cache_without_reaction
    status_page_url
    captcha_enabled
    stranger_mention_from_local_ng
    hide_local_users_for_anonymous
    post_hash_tags_max
    post_mentions_max
    post_stranger_mentions_max
    auto_warning_text
    authorized_fetch
    receive_other_servers_emoji_reaction
    streaming_other_servers_emoji_reaction
    streaming_local_emoji_reaction
    enable_emoji_reaction
    check_lts_version_only
    enable_public_visibility
    enable_public_unlisted_visibility
    unlocked_friend
    enable_local_timeline
    emoji_reaction_disallow_domains
    block_unfollow_account_mention
    hold_remote_new_accounts
  ).freeze

  INTEGER_KEYS = %i(
    media_cache_retention_period
    content_cache_retention_period
    backups_retention_period
    post_hash_tags_max
    post_mentions_max
    post_stranger_mentions_max
    registrations_limit
    registrations_limit_per_day
    registrations_start_hour
    registrations_end_hour
    registrations_secondary_start_hour
    registrations_secondary_end_hour
  ).freeze

  BOOLEAN_KEYS = %i(
    timeline_preview
    activity_api_enabled
    peers_api_enabled
    preview_sensitive_media
    profile_directory
    trends
    trends_as_landing_page
    trendable_by_default
    noindex
    require_invite_text
    captcha_enabled
    hide_local_users_for_anonymous
    authorized_fetch
    receive_other_servers_emoji_reaction
    streaming_other_servers_emoji_reaction
    streaming_local_emoji_reaction
    enable_emoji_reaction
    check_lts_version_only
    enable_public_visibility
    enable_public_unlisted_visibility
    unlocked_friend
    stranger_mention_from_local_ng
    enable_local_timeline
    delete_content_cache_without_reaction
    block_unfollow_account_mention
    hold_remote_new_accounts
  ).freeze

  UPLOAD_KEYS = %i(
    thumbnail
    mascot
  ).freeze

  OVERRIDEN_SETTINGS = {
    authorized_fetch: :authorized_fetch_mode?,
  }.freeze

  STRING_ARRAY_KEYS = %i(
    emoji_reaction_disallow_domains
  ).freeze

  attr_accessor(*KEYS)

  validates :registrations_mode, inclusion: { in: %w(open approved none) }, if: -> { defined?(@registrations_mode) }
  validates :site_contact_email, :site_contact_username, presence: true, if: -> { defined?(@site_contact_username) || defined?(@site_contact_email) }
  validates :site_contact_username, existing_username: true, if: -> { defined?(@site_contact_username) }
  validates :bootstrap_timeline_accounts, existing_username: { multiple: true }, if: -> { defined?(@bootstrap_timeline_accounts) }
  validates :show_domain_blocks, inclusion: { in: %w(disabled users all) }, if: -> { defined?(@show_domain_blocks) }
  validates :show_domain_blocks_rationale, inclusion: { in: %w(disabled users all) }, if: -> { defined?(@show_domain_blocks_rationale) }
  validates :media_cache_retention_period, :content_cache_retention_period, :backups_retention_period, numericality: { only_integer: true }, allow_blank: true, if: -> { defined?(@media_cache_retention_period) || defined?(@content_cache_retention_period) || defined?(@backups_retention_period) }
  validates :site_short_description, length: { maximum: 200 }, if: -> { defined?(@site_short_description) }
  validates :status_page_url, url: true, allow_blank: true
  validate :validate_site_uploads

  KEYS.each do |key|
    define_method(key) do
      return instance_variable_get(:"@#{key}") if instance_variable_defined?(:"@#{key}")

      stored_value = if UPLOAD_KEYS.include?(key)
                       SiteUpload.where(var: key).first_or_initialize(var: key)
                     elsif STRING_ARRAY_KEYS.include?(key)
                       Setting.public_send(key)&.join("\n") || ''
                     elsif OVERRIDEN_SETTINGS.include?(key)
                       public_send(OVERRIDEN_SETTINGS[key])
                     else
                       Setting.public_send(key)
                     end

      instance_variable_set(:"@#{key}", stored_value)
    end
  end

  UPLOAD_KEYS.each do |key|
    define_method(:"#{key}=") do |file|
      value = public_send(key)
      value.file = file
    rescue Mastodon::DimensionsValidationError => e
      errors.add(key.to_sym, e.message)
    end
  end

  def save
    # NOTE: Annoyingly, files are processed and can error out before
    # validations are called, and `valid?` clears errors…
    # So for now, return early if errors aren't empty.
    return false unless errors.empty? && valid?

    KEYS.each do |key|
      next unless instance_variable_defined?(:"@#{key}")

      if UPLOAD_KEYS.include?(key)
        public_send(key).save
      else
        setting = Setting.where(var: key).first_or_initialize(var: key)
        setting.update(value: typecast_value(key, instance_variable_get(:"@#{key}")))
      end
    end
  end

  private

  def typecast_value(key, value)
    if BOOLEAN_KEYS.include?(key)
      value == '1'
    elsif INTEGER_KEYS.include?(key)
      value.blank? ? value : Integer(value)
    elsif STRING_ARRAY_KEYS.include?(key)
      value&.split(/\r\n|\r|\n/)&.filter(&:present?)&.uniq || []
    else
      value
    end
  end

  def validate_site_uploads
    UPLOAD_KEYS.each do |key|
      next unless instance_variable_defined?(:"@#{key}")

      upload = instance_variable_get(:"@#{key}")
      next if upload.valid?

      upload.errors.each do |error|
        errors.import(error, attribute: key)
      end
    end
  end
end