Merge pull request #2 from YodelTalk/config

Fixes problems with Rails Engines

Author: lfittl

lib/activerecord-clean-db-structure/tasks/clean_db_structure.rake

@@ -1,13 +1,14 @@
 require 'activerecord-clean-db-structure/clean_dump'
 
 Rake::Task['db:structure:dump'].enhance do
-  Rake::Task['clean_db_structure'].invoke
-end
-
-task :clean_db_structure do
-  filename = ENV['DB_STRUCTURE'] || File.join(ActiveRecord::Tasks::DatabaseTasks.db_dir, 'structure.sql')
+  filenames = ENV['DB_STRUCTURE']
+  filenames ||= Rails.application.config.paths['db'].map do |path|
+    File.join(path, 'structure.sql')
+  end
 
-  cleaner = ActiveRecordCleanDbStructure::CleanDump.new(File.read(filename))
-  cleaner.run
-  File.write(filename, cleaner.dump)
+  filenames.each do |filename|
+    cleaner = ActiveRecordCleanDbStructure::CleanDump.new(File.read(filename))
+    cleaner.run
+    File.write(filename, cleaner.dump)
+  end
 end