lib/named_seeds/railtie.rb in named_seeds-2.0.0 vs lib/named_seeds/railtie.rb in named_seeds-2.0.1

@@ -2,22 +2,44 @@
 
 module NamedSeeds
   class Railtie < Rails::Railtie
 
     config.named_seeds = ActiveSupport::OrderedOptions.new
+    config.named_seeds.load_app_seed_file     = true
+    config.named_seeds.custom_seed_file       = nil
     config.named_seeds.engines_with_load_seed = []
 
     config.before_initialize do |app|
       ActiveRecord::Tasks::DatabaseTasks.seed_loader = NamedSeeds::Railtie
     end
 
+    rake_tasks do
+      load 'named_seeds/databases.rake'
+    end
+
     def load_seed
-      Rails.application.load_seed
+      load_app_seed_file
+      load_custom_seed_file
       engines_load_seed
     end
 
+    def db_setup
+      load_custom_seed_file
+      engines_load_seed
+    end
 
+
     protected
+
+    def load_app_seed_file
+      Rails.application.load_seed if config.named_seeds.load_app_seed_file
+    end
+
+    def load_custom_seed_file
+      return unless config.named_seeds.custom_seed_file
+      custom_seed_file = Rails.root.join(config.named_seeds.custom_seed_file)
+      load(custom_seed_file) if File.exists?(custom_seed_file)
+    end
 
     def engines_load_seed
       config.named_seeds.engines_with_load_seed.each { |engine| engine.load_seed }
     end