Merge pull request #9203 from jaggederest/dumping_pgsql_extensions

Adding database extension support to schema.rb

activerecord/lib/active_record/connection_adapters/abstract_adapter.rb

@@ -177,6 +177,12 @@ def supports_extensions?
         false
       end
 
+      # A list of extensions, to be filled in by databases that
+      # support them (at the moment, postgresql)
+      def extensions
+        []
+      end
+
       # QUOTING ==================================================
 
       # Returns a bind substitution value given a +column+ and list of current

activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb

@@ -605,6 +605,15 @@ def extension_enabled?(name)
         end
       end
 
+      def extensions
+        if supports_extensions?
+          res = exec_query "SELECT extname from pg_extension", "SCHEMA"
+          res.rows.map { |r| res.column_types['extname'].type_cast r.first }
+        else
+          []
+        end
+      end
+
       # Returns the configured supported identifier length supported by PostgreSQL
       def table_alias_length
         @table_alias_length ||= query('SHOW max_identifier_length', 'SCHEMA')[0][0].to_i

activerecord/lib/active_record/schema_dumper.rb

@@ -24,6 +24,7 @@ def self.dump(connection=ActiveRecord::Base.connection, stream=STDOUT)
 
     def dump(stream)
       header(stream)
+      extensions(stream)
       tables(stream)
       trailer(stream)
       stream
@@ -66,6 +67,17 @@ def trailer(stream)
         stream.puts "end"
       end
 
+      def extensions(stream)
+        return unless @connection.supports_extensions?
+        extensions = @connection.extensions
+        stream.puts <<EXTENSIONS
+# These are extensions that must be enabled in order to support this database
+EXTENSIONS
+        extensions.each do |extension|
+          stream.puts " enable_extension #{extension.inspect}"
+        end
+      end
+
       def tables(stream)
         @connection.tables.sort.each do |tbl|
           next if ['schema_migrations', ignore_tables].flatten.any? do |ignored|

activerecord/test/cases/adapters/postgresql/hstore_test.rb

@@ -35,6 +35,11 @@ def teardown
     @connection.execute 'drop table if exists hstores'
   end
 
+  def test_hstore_included_in_extensions
+    assert @connection.respond_to?(:extensions), "connection should have a list of extensions"
+    assert @connection.extensions.include?('hstore'), "extension list should include hstore"
+  end
+
   def test_hstore_enabled
     assert @connection.extension_enabled?('hstore')
   end

activerecord/test/cases/schema_dumper_test.rb

@@ -231,6 +231,16 @@ def test_schema_dump_includes_decimal_options
   end
 
   if current_adapter?(:PostgreSQLAdapter)
+    def test_schema_dump_includes_extensions
+      connection = ActiveRecord::Base.connection
+      return skip unless connection.supports_extensions?
+      unless connection.extension_enabled?('hstore')
+        connection.enable_extension 'hstore'
+      end
+      output = standard_dump
+      assert_match %r{enable_extension "hstore"}, output
+    end
+
     def test_schema_dump_includes_xml_shorthand_definition
       output = standard_dump
       if %r{create_table "postgresql_xml_data_type"} =~ output