Rake task: rake db:dev:dump_schema

Just a utility task to get a dump of the database schema similar
to what you'd see from ActiveRecord

Author: Brian Butz

.gitignore

@@ -12,6 +12,7 @@ spec/examples.txt
 ci-artifacts-dir
 ci-working-dir
 coverage
+db/schema.rb
 doc
 *.rbc
 *.swp

lib/tasks/db.rake

@@ -207,6 +207,29 @@ namespace :db do
       number_to_rollback = (args[:number_to_rollback] || 1).to_i
       for_each_database { rollback(number_to_rollback) }
     end
+
+    desc 'Dump schema to file'
+    task :dump_schema do
+      require_relative '../../spec/support/bootstrap/db_config'
+      require_relative '../../spec/support/bootstrap/table_recreator'
+
+      db = DbConfig.new.connection
+
+      puts 'Recreating tables...'
+      TableRecreator.new(db).recreate_tables(without_fake_tables: true)
+
+      db.extension(:schema_dumper)
+      puts 'Dumping schema...'
+      schema = db.dump_schema_migration(indexes: true, foreign_keys: true)
+
+      File.open('db/schema.rb', 'w') { |f|
+        f.write("# rubocop:disable all\n")
+        f.write(schema)
+        f.write("# rubocop:enable all\n")
+      }
+
+      puts 'Wrote db/schema.rb'
+    end
   end
 
   namespace :parallel do

spec/support/bootstrap/table_recreator.rb

@@ -1,4 +1,4 @@
-require 'support/bootstrap/fake_model_tables'
+require_relative 'fake_model_tables'
 
 class TableRecreator
   SAFE_VIEWS = [:pg_stat_statements].freeze
@@ -7,7 +7,7 @@ def initialize(db)
     @db = db
   end
 
-  def recreate_tables
+  def recreate_tables(without_fake_tables: false)
     prepare_database
 
     (db.views - SAFE_VIEWS).each do |view|
@@ -20,8 +20,10 @@ def recreate_tables
 
     DBMigrator.new(db).apply_migrations
 
-    fake_model_tables = FakeModelTables.new(db)
-    fake_model_tables.create_tables
+    unless without_fake_tables
+      fake_model_tables = FakeModelTables.new(db)
+      fake_model_tables.create_tables
+    end
   end
 
   private