tenant.rake in mongoid-tenant-0.3.7

- old
+ new

@@ -1,6 +1,7 @@
 Rake::Task['db:mongoid:create_indexes'].clear
+Rake::Task['db:mongoid:remove_indexes'].clear
 Rake::Task['db:mongoid:remove_undefined_indexes'].clear
 
 namespace :db do
   namespace :mongoid do
     def tenancy_env
@@ -13,43 +14,46 @@
 
     def non_tenancy_models
       ::Mongoid.models.reject { |k| k.include?(::Mongoid::Tenant) }
     end
 
+    def puts_warning
+      puts 'Not running tenants... Provide a tenancy model:'
+      puts "`TENANCY=Model #{ARGV.join(' ')}`"
+    end
+
     desc 'Create Mongoid indexes, tenant aware'
-    task create_indexes: [:environment, :load_models] do
+    task create_indexes: %i[environment load_models] do
       # Run once, for tables outside tenancy
       ::Mongoid::Tasks::Database.create_indexes(non_tenancy_models)
       if tenancy_env
         Object.const_get(tenancy_env).active_tenants.each do |t|
           puts "Tenant #{t}"
           t.tenancy!
           ::Mongoid::Tasks::Database.create_indexes(tenancy_models)
         end
       else
-        puts 'Not running tenants: Provide a tenancy model:'
-        puts "`TENANCY=Model #{ARGV.join(' ')}`"
+        puts_warning
       end
     end
 
     desc 'Remove Mongoid indexes, tenant aware'
-    task remove_indexes: [:environment, :load_models] do
+    task remove_indexes: %i[environment load_models] do
       # Run once, for tables outside tenancy
       ::Mongoid::Tasks::Database.remove_indexes(non_tenancy_models)
       if tenancy_env
         Object.const_get(tenancy_env).active_tenants.each do |t|
           puts "Tenant #{t}"
           t.tenancy!
           ::Mongoid::Tasks::Database.remove_indexes(tenancy_models)
         end
       else
-        puts 'Not running tenants: Provide a tenancy model:'
-        puts "`TENANCY=Model #{ARGV.join(' ')}`"
+        puts_warning
       end
     end
 
     desc 'Removes undefined Mongoid indexes, tenant aware'
-    task remove_undefined_indexes: [:environment, :load_models] do
+    task remove_undefined_indexes: %i[environment load_models] do
       # Run once, for tables outside tenancy
       ::Mongoid::Tasks::Database.remove_undefined_indexes(non_tenancy_models)
       if tenancy_env
         Object.const_get(tenancy_env).active_tenants.each do |t|
           puts "Tenant #{t}"