module StrongMigrations
  module SchemaDumper
    def initialize(connection, *args, **options)
      return super unless StrongMigrations.alphabetize_schema

      super(WrappedConnection.new(connection), *args, **options)
    end
  end

  class WrappedConnection
    delegate_missing_to :@connection

    def initialize(connection)
      @connection = connection
    end

    def columns(*args, **options)
      @connection.columns(*args, **options).sort_by(&:name)
    end
  end
end