Merge pull request ilyakatz#321 from jgmontoya/fix/db-prepare-withdat…

…a-task

fix(db:prepare:with_data): add missing prepare_all_with_data and task definition

Changelog.md

@@ -1,5 +1,8 @@
 # Changelog
 
+## 9.4.2
+  - Fix db:prepare:with_data task
+
 ## 9.4.1
   - Add db:prepare task
 

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    data_migrate (9.4.1)
+    data_migrate (9.4.2)
       activerecord (>= 6.1)
       railties (>= 6.1)
 

gemfiles/rails_6.1.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    data_migrate (9.4.1)
+    data_migrate (9.4.2)
       activerecord (>= 6.1)
       railties (>= 6.1)
 

gemfiles/rails_7.0.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    data_migrate (9.4.1)
+    data_migrate (9.4.2)
       activerecord (>= 6.1)
       railties (>= 6.1)
 

gemfiles/rails_7.1.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    data_migrate (9.4.1)
+    data_migrate (9.4.2)
       activerecord (>= 6.1)
       railties (>= 6.1)
 

lib/data_migrate/database_tasks.rb

@@ -204,5 +204,42 @@ def self.migrate_with_data
         end
       end
     end
+
+    def self.prepare_all_with_data
+      seed = false
+
+      each_current_configuration(env) do |db_config|
+        next unless db_config.primary?
+
+        with_temporary_pool(db_config) do
+          begin
+            database_initialized = migration_connection.schema_migration.table_exists?
+          rescue ActiveRecord::NoDatabaseError
+            create(db_config)
+            retry
+          end
+
+          unless database_initialized
+            if File.exist?(schema_dump_path(db_config))
+              load_schema(db_config, ActiveRecord.schema_format, nil)
+              load_schema_current(
+                :ruby,
+                ENV["DATA_SCHEMA"]
+              )
+            end
+
+            seed = true
+          end
+
+          migrate_with_data
+          if ActiveRecord.dump_schema_after_migration
+            dump_schema(db_config)
+            DataMigrate::Tasks::DataMigrateTasks.dump
+          end
+        end
+      end
+
+      load_seed if seed
+    end
   end
 end

lib/data_migrate/version.rb

@@ -1,3 +1,3 @@
 module DataMigrate
-  VERSION = "9.4.1".freeze
+  VERSION = "9.4.2".freeze
 end

tasks/databases.rake

@@ -146,6 +146,13 @@ namespace :db do
       end
     end
   end
+
+  namespace :prepare do
+    desc "Runs setup if database does not exist, or runs data and schema migrations if it does"
+    task with_data: :environment do
+      DataMigrate::DatabaseTasks.prepare_all_with_data
+    end
+  end
 end
 
 namespace :data do