converter.rb

class Mysql2psql
  class Converter
    attr_reader :reader, :writer, :options
    attr_reader :exclude_tables, :only_tables, :suppress_data, :suppress_ddl, :force_truncate, :preserve_order, :clear_schema

    def initialize(reader, writer, options)
      @reader = reader
      @writer = writer
      @options = options
      @exclude_tables = options.exclude_tables([])
      @only_tables = options.only_tables(nil)
      @suppress_data = options.suppress_data(false)
      @suppress_ddl = options.suppress_ddl(false)
      @force_truncate = options.force_truncate(false)
      @preserve_order = options.preserve_order(false)
      @clear_schema = options.clear_schema(false)
    end

    def convert
      tables = reader.tables
               .reject { |table| @exclude_tables.include?(table.name) }
               .select { |table| @only_tables ? @only_tables.include?(table.name) : true }

      if @preserve_order

        reordered_tables = []

        @only_tables.each do |only_table|
          idx = tables.index { |table| table.name == only_table }
          reordered_tables << tables[idx]
        end

        tables = reordered_tables

      end

      tables.each do |table|
        writer.write_table(table)
      end unless @suppress_ddl

      # tables.each do |table|
      #   writer.truncate(table) if force_truncate && suppress_ddl
      #   writer.write_contents(table, reader)
      # end unless @suppress_data

      unless @suppress_data

        tables.each do |table|
          writer.truncate(table) if force_truncate && suppress_ddl
        end

        tables.each do |table|
          writer.write_contents(table, reader)
        end

      end

      tables.each do |table|
        writer.write_indexes(table)
      end unless @suppress_ddl
      tables.each do |table|
        writer.write_constraints(table)
      end unless @suppress_ddl

      writer.close

      if @clear_schema
        writer.clear_schema
      end

      writer.inload

      0
    end
  end
end