discourse/script/import_scripts/mbox/support/settings.rb

# frozen_string_literal: true

require "yaml"

module ImportScripts::Mbox
  class Settings
    def self.load(filename)
      yaml = YAML.load_file(filename)
      Settings.new(yaml)
    end

    attr_reader :data_dir
    attr_reader :split_regex
    attr_reader :batch_size
    attr_reader :trust_level
    attr_reader :prefer_html
    attr_reader :staged
    attr_reader :index_only
    attr_reader :group_messages_by_subject
    attr_reader :subject_prefix_regex
    attr_reader :automatically_remove_list_name_prefix
    attr_reader :show_trimmed_content
    attr_reader :tags
    attr_reader :fix_mailman_via_addresses
    attr_reader :elide_equals_in_addresses

    def initialize(yaml)
      @data_dir = yaml["data_dir"]
      @split_regex = Regexp.new(yaml["split_regex"]) unless yaml["split_regex"].empty?
      @batch_size = 1000 # no need to make this actually configurable at the moment
      @trust_level = yaml["default_trust_level"]
      @prefer_html = yaml["prefer_html"]
      @staged = yaml["staged"]
      @index_only = yaml["index_only"]
      @group_messages_by_subject = yaml["group_messages_by_subject"]

      if yaml["remove_subject_prefixes"].present?
        prefix_regexes = yaml["remove_subject_prefixes"].map { |p| Regexp.new(p) }
        @subject_prefix_regex = /^#{Regexp.union(prefix_regexes).source}/i
      end

      @automatically_remove_list_name_prefix = yaml["automatically_remove_list_name_prefix"]
      @show_trimmed_content = yaml["show_trimmed_content"]
      @fix_mailman_via_addresses = yaml["fix_mailman_via_addresses"]
      @elide_equals_in_addresses = yaml["elide_equals_in_addresses"]

      @tags = []
      if yaml["tags"].present?
        yaml["tags"].each do |tag_name, value|
          prefixes = Regexp.union(value).source
          @tags << {
            regex: /^(?:(?:\[(?:#{prefixes})\])|(?:\((?:#{prefixes})\)))\s*/i,
            name: tag_name,
          }
        end
      end
    end
  end
end