bard-rake 0.16.1 → 0.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: '013248985335f720029ac9533d99d4ca8656dba9'
-  data.tar.gz: 7d9d4065d2f69a1b59640617f61912d7c98d41b2
+SHA256:
+  metadata.gz: 6ff9efd86b5af9fe0cd76d0040ff9c8c1ee9f4ac44d6f734d2088208ecb540e8
+  data.tar.gz: 3aa80505be2130df0af16819812d7c7585a7437732e0b9676ced2d1a993f40f6
 SHA512:
-  metadata.gz: 7e45654df1e8d7c583772d800780a344faf918d5673292c107346c59ee6016b9b99eb4bffa223748c5d555eb50e9f3c23f86392e510fbcfc3bd64b5dbc9dd44c
-  data.tar.gz: ee347ce758d635e22d462638ae26cb4dec2320132115182627057f11b5e948a1e483ae0a3bc0eaa67bb0a626ea711effb0caf1101de00f484aab06b7f246ae9c
+  metadata.gz: 25ce15d5a69099f141387adba4cedb6375842eb9b8c3a798d1e6d1fe1a1c777fcdf4f51d029bb18577300f68b7ff697cb367f15ca5c69f41701ed087f11168ce
+  data.tar.gz: 31d9e860c78f90e836787a386eff659c56ab2a098cc5fc63d779edf15094e4a3abd14b60f2b906bac1f63b8595110efa2030ef76b980d78c0316cd0d4340ccf8

data/lib/bard/rake/db_multiple_environment_sanity.rb

@@ -1,59 +1,87 @@
 if defined?(ActiveRecord)
   namespace :db do
     namespace :create do
-      task :all => [:load_config] do
-        invoke_task_if_exists :rails_env
-        run_in_all_environments do |config|
-          ActiveRecord::Base.establish_connection(config)
-          ActiveRecord::Tasks::DatabaseTasks.create config
-        end
-
-        if includes_test_environment? && !parallel?
-          invoke_task_if_exists "parallel:create"
+      task :all => :load_config do
+        run_in_all_environments do |config, env|
+          if env == "test" && Rake::Task.task_defined?("parallel:create")
+            Rake::Task["parallel:create"].invoke
+          else
+            ActiveRecord::Tasks::DatabaseTasks.create config
+          end
         end
       end
     end
 
     namespace :drop do
-      task :all => [:load_config] do
-        invoke_task_if_exists :rails_env
-        run_in_all_environments do |config|
-          ActiveRecord::Base.establish_connection(config)
-          ActiveRecord::Tasks::DatabaseTasks.drop config
-        end
-
-        if includes_test_environment? && !parallel?
-          invoke_task_if_exists "parallel:drop"
+      task :all => :load_config do
+        run_in_all_environments do |config, env|
+          if env == "test" && Rake::Task.task_defined?("parallel:drop")
+            Rake::Task["parallel:drop"].invoke
+          else
+            ActiveRecord::Tasks::DatabaseTasks.drop config
+          end
         end
       end
     end
 
     namespace :migrate do
-      task :all => [:load_config] do
-        invoke_task_if_exists :rails_env
-        run_in_all_environments do |config|
-          ActiveRecord::Base.establish_connection(config)
-          ActiveRecord::Tasks::DatabaseTasks.migrate
+      task :all => :load_config do
+        run_in_all_environments do |config, env|
+          if env == "test" && Rake::Task.task_defined?("parallel:migrate")
+            Rake::Task["parallel:migrate"].invoke
+          else
+            ActiveRecord::Tasks::DatabaseTasks.migrate
+          end
         end
+      end
+    end
 
-        if includes_test_environment? && !parallel?
-          invoke_task_if_exists "parallel:migrate"
+    namespace :rollback do
+      task :all => :load_config do
+        run_in_all_environments do |config, env|
+          if env == "test" && Rake::Task.task_defined?("parallel:rollback")
+            Rake::Task["parallel:rollback"].invoke
+          else
+            ActiveRecord::Base.connection.migration_context.rollback(1)
+          end
         end
-
-        Rake::Task["db:_dump"].invoke
       end
     end
 
     def run_in_all_environments &block
-      if defined?(ActiveRecord::Tasks::DatabaseTasks)
-        ActiveRecord::Tasks::DatabaseTasks.send :each_local_configuration, &block
-      else
-        configs_for_environment.each &block
+      invoke_task_if_exists :rails_env
+
+      whitelist = %w(development test staging production)
+      modified_databases = Set.new
+
+      ActiveRecord::Base.configurations.configs_for.each do |db_config|
+        env = db_config.env_name
+        configuration = db_config.config
+        database = configuration["database"]
+
+        next unless database
+        next unless whitelist.include?(env)
+        next if modified_databases.include?(database)
+
+        if local_database?(configuration)
+          ActiveRecord::Base.establish_connection configuration
+          block.call configuration, env
+        else
+          $stderr.puts "This task only modifies local databases. #{database} is on a remote host."
+        end
+
+        modified_databases << database
       end
+
+      Rake::Task["db:_dump"].invoke
     end
 
-    def includes_test_environment?
-      %w(development test).include?(rails_env)
+    def local_database?(configuration)
+      configuration["host"].blank? || LOCAL_HOSTS.include?(configuration["host"])
+    end
+
+    def test_environment?
+      rails_env == "test"
     end
 
     def rails_env
@@ -63,10 +91,6 @@ if defined?(ActiveRecord)
         ENV["RAILS_ENV"]
       end
     end
-
-    def parallel?
-      !!ENV["TEST_ENV_NUMBER"]
-    end
   end
 end
 

data/lib/bard/rake/version.rb

@@ -1,6 +1,6 @@
 module Bard
   module Rake
-    VERSION = "0.16.1"
+    VERSION = "0.17.0"
   end
 end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bard-rake
 version: !ruby/object:Gem::Version
-  version: 0.16.1
+  version: 0.17.0
 platform: ruby
 authors:
 - Micah Geisel
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-07-31 00:00:00.000000000 Z
+date: 2019-09-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -112,8 +112,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.6.14
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: Rake tasks for all bard projects.