discourse/migrations/lib/importer/shared_data.rb

# frozen_string_literal: true

module Migrations::Importer
  class SharedData
    def initialize(discourse_db)
      @discourse_db = discourse_db
    end

    def load_set(sql)
      result = @discourse_db.query_result(sql)
      depth = result.column_count - 1

      set = Migrations::SetStore.create(depth)
      set.bulk_add(result.rows)

      set
    end

    def load_mapping(sql)
      rows = @discourse_db.query_array(sql)

      # While rows is an enumerator, it's not fully compliant, it does not
      # rewind on #first, #peek, #any?, etc.
      # So we need to hold on to first_row for use later
      first_row = rows.first

      return {} if first_row.nil?

      has_multiple_values = first_row.size > 2
      result =
        if has_multiple_values
          rows.to_h { |key, *values| [key, values] }
        else
          rows.to_h
        end

      result[first_row[0]] = has_multiple_values ? first_row[1..] : first_row[1]
      result
    end

    def load(type)
      case type
      when :usernames
        @existing_usernames_lower ||= load_set <<~SQL
          SELECT username_lower
          FROM users
        SQL
      when :group_names
        @existing_group_names_lower ||= load_set <<~SQL
          SELECT LOWER(name)
          FROM groups
        SQL
      else
        raise "Unknown type: #{type}"
      end
    end

    def unload_shared_data(type)
      case type
      when :usernames
        @existing_usernames_lower = nil
      when :group_names
        @existing_group_names_lower = nil
      else
        raise "Unknown type: #{type}"
      end
    end
  end
end