lib/erd/migrator.rb in erd-0.1.3 vs lib/erd/migrator.rb in erd-0.1.4

@@ -10,23 +10,28 @@
         migrations = []
         ActiveRecord::Migrator.migrations_paths.each do |path|
           Dir.foreach(Rails.root.join(path)) do |file|
             if (version_and_name = /^(\d{3,})_(.+)\.rb$/.match(file))
               status = migrated_versions.delete(version_and_name[1]) ? 'up' : 'down'
-              migrations << {:status => status, :version => version_and_name[1], :name => version_and_name[2]}
+              migrations << {:status => status, :version => version_and_name[1], :name => version_and_name[2], :filename => file}
             end
           end
         end
-        migrations += migrated_versions.map {|v| {:status => 'up', :version => v, :name => '*** NO FILE ***'}}
+        migrations += migrated_versions.map {|v| {:status => 'up', :version => v, :name => '*** NO FILE ***', :filename => v}}
         migrations.sort_by {|m| m[:version]}
       end
 
       # `rake db:migrate`
+      # example:
+      #   run_migrations up: '/Users/a_matsuda/my_app/db/migrate/20120423023323_create_products.rb'
+      #   run_migrations up: '20120512020202', down: ...
+      #   run_migrations up: ['20120512020202', '20120609010203', ...]
       def run_migrations(migrations)
-        migrations.each do |direction, versions|
-          versions.each do |version|
+        migrations.each do |direction, version_or_filenames|
+          Array.wrap(version_or_filenames).each do |version_or_filename|
+            /^(?<version>\d{3,})/ =~ File.basename(version_or_filename)
             ActiveRecord::Migrator.run(direction, ActiveRecord::Migrator.migrations_path, version.to_i)
-          end if versions
+          end if version_or_filenames
         end
         if ActiveRecord::Base.schema_format == :ruby
           File.open(ENV['SCHEMA'] || "#{Rails.root}/db/schema.rb", 'w') do |file|
             ActiveRecord::SchemaDumper.dump(ActiveRecord::Base.connection, file)
           end