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

# frozen_string_literal: true

require "csv"
require "yaml"
require_relative "../../base"

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

    attr_reader :site_name

    attr_reader :new_categories
    attr_reader :category_mappings
    attr_reader :tag_mappings
    attr_reader :rank_mapping

    attr_reader :import_anonymous_users
    attr_reader :import_attachments
    attr_reader :import_private_messages
    attr_reader :import_polls
    attr_reader :import_bookmarks
    attr_reader :import_passwords
    attr_reader :import_likes

    attr_reader :import_uploaded_avatars
    attr_reader :import_remote_avatars
    attr_reader :import_gallery_avatars

    attr_reader :use_bbcode_to_md

    attr_reader :original_site_prefix
    attr_reader :new_site_prefix
    attr_reader :base_dir
    attr_reader :permalinks

    attr_reader :username_as_name
    attr_reader :emojis
    attr_reader :custom_fields

    attr_reader :database

    def initialize(yaml)
      import_settings = yaml["import"]

      @site_name = import_settings["site_name"]

      @new_categories = import_settings["new_categories"]
      @category_mappings =
        import_settings.fetch("category_mappings", []).to_h { |m| [m[:source_category_id].to_s, m] }
      @tag_mappings = import_settings["tag_mappings"]
      @rank_mapping = import_settings["rank_mapping"]

      @import_anonymous_users = import_settings["anonymous_users"]
      @import_attachments = import_settings["attachments"]
      @import_private_messages = import_settings["private_messages"]
      @import_polls = import_settings["polls"]
      @import_bookmarks = import_settings["bookmarks"]
      @import_passwords = import_settings["passwords"]
      @import_likes = import_settings["likes"]

      avatar_settings = import_settings["avatars"]
      @import_uploaded_avatars = avatar_settings["uploaded"]
      @import_remote_avatars = avatar_settings["remote"]
      @import_gallery_avatars = avatar_settings["gallery"]

      @use_bbcode_to_md = import_settings["use_bbcode_to_md"]

      @original_site_prefix = import_settings["site_prefix"]["original"]
      @new_site_prefix = import_settings["site_prefix"]["new"]
      @base_dir = import_settings["phpbb_base_dir"]
      @permalinks = PermalinkSettings.new(import_settings["permalinks"])

      @username_as_name = import_settings["username_as_name"]
      @emojis = import_settings.fetch("emojis", [])
      @custom_fields = import_settings.fetch("custom_fields", [])

      @database = DatabaseSettings.new(yaml["database"])
    end

    def prefix(val)
      @site_name.present? && val.present? ? "#{@site_name}:#{val}" : val
    end

    def trust_level_for_posts(rank, trust_level: 0)
      if @rank_mapping.present?
        @rank_mapping.each do |key, value|
          trust_level = [trust_level, key.gsub("trust_level_", "").to_i].max if rank >= value
        end
      end

      trust_level
    end
  end

  class DatabaseSettings
    attr_reader :type
    attr_reader :host
    attr_reader :port
    attr_reader :username
    attr_reader :password
    attr_reader :schema
    attr_reader :table_prefix
    attr_reader :batch_size

    def initialize(yaml)
      @type = yaml["type"]
      @host = yaml["host"]
      @port = yaml["port"]
      @username = yaml["username"]
      @password = yaml["password"]
      @schema = yaml["schema"]
      @table_prefix = yaml["table_prefix"]
      @batch_size = yaml["batch_size"]
    end
  end

  class PermalinkSettings
    attr_reader :create_category_links
    attr_reader :create_topic_links
    attr_reader :create_post_links
    attr_reader :normalization_prefix

    def initialize(yaml)
      @create_category_links = yaml["categories"]
      @create_topic_links = yaml["topics"]
      @create_post_links = yaml["posts"]
      @normalization_prefix = yaml["prefix"]
    end
  end
end