Merge pull request #43 from ghiculescu/do_not_remove_extensions

Add an option to not remove extensions

Author: andyatkinson

README.md

@@ -87,6 +87,17 @@ INSERT INTO "schema_migrations" (version) VALUES
 ;
 ```
 
+By default the gem will remove [some extensions](https://github.com/ghiculescu/activerecord-clean-db-structure/blob/c9551391476a5e7a08ff314501af89baddcf669a/lib/activerecord-clean-db-structure/clean_dump.rb#L24) that typically aren't needed in structure dumps. You can choose to keep all, or just some, of those extensions:
+
+```ruby
+Rails.application.configure do
+  config.activerecord_clean_db_structure.keep_extensions = :all
+
+  # This does the same thing as :all. You can choose which optional extensions to keep.
+  config.activerecord_clean_db_structure.keep_extensions = ["pg_stat_statements", "pg_buffercache"]
+end
+```
+
 ## Authors
 
 * [Lukas Fittl](https://github.com/lfittl)

lib/activerecord-clean-db-structure/clean_dump.rb

@@ -21,13 +21,16 @@ def run
       dump.gsub!(/^SET xmloption = content;\n/m, '') # 12
       dump.gsub!(/^SET default_table_access_method = heap;\n/m, '') # 12
 
-      # Remove pg_stat_statements extension (its not relevant to the code)
-      dump.gsub!(/^CREATE EXTENSION IF NOT EXISTS pg_stat_statements.*/, '')
-      dump.gsub!(/^-- Name: (EXTENSION )?pg_stat_statements;.*/, '')
-
-      # Remove pg_buffercache extension (its not relevant to the code)
-      dump.gsub!(/^CREATE EXTENSION IF NOT EXISTS pg_buffercache.*/, '')
-      dump.gsub!(/^-- Name: (EXTENSION )?pg_buffercache;.*/, '')
+      extensions_to_remove = ["pg_stat_statements", "pg_buffercache"]
+      if options[:keep_extensions] == :all
+        extensions_to_remove = [] 
+      elsif options[:keep_extensions]
+        extensions_to_remove -= Array(options[:keep_extensions])
+      end
+      extensions_to_remove.each do |ext|
+        dump.gsub!(/^CREATE EXTENSION IF NOT EXISTS #{ext}.*/, '')
+        dump.gsub!(/^-- Name: (EXTENSION )?#{ext};.*/, '')
+      end
 
       # Remove comments on extensions, they create problems if the extension is owned by another user
       dump.gsub!(/^COMMENT ON EXTENSION .*/, '')