TwilightCoders/active_record-mti

require 'active_record/mti/core_extension'
require 'active_record/mti/postgresql/schema_statements'
require 'active_record/mti/postgresql/schema_creation'
require 'active_record/mti/postgresql/adapter'
require 'active_record/mti/schema_dumper'

module ActiveRecord
  module MTI
    class Railtie < Rails::Railtie
      initializer 'active_record-mti.load' do |_app|
        ActiveSupport.on_load(:active_record) do
          ::ActiveRecord::Base.prepend(::ActiveRecord::MTI::CoreExtension)

          ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.prepend(PostgreSQL::Adapter)
          ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.prepend(PostgreSQL::SchemaStatements)
          ::ActiveRecord::SchemaDumper.prepend(SchemaDumper)
          ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter::SchemaCreation.prepend(PostgreSQL::SchemaCreation)
        end
      end
    end
  end
end