Froze rails gems

[depot.git] / vendor / rails / railties / lib / rails_generator / generators / components / migration / migration_generator.rb

diff --git a/vendor/rails/railties/lib/rails_generator/generators/components/migration/migration_generator.rb b/vendor/rails/railties/lib/rails_generator/generators/components/migration/migration_generator.rb
new file mode 100644 (file)
index 0000000..acf41e0
--- /dev/null
+++ b/vendor/rails/railties/lib/rails_generator/generators/components/migration/migration_generator.rb
@@ -0,0 +1,20 @@
+class MigrationGenerator < Rails::Generator::NamedBase  
+  def manifest
+    record do |m|
+      m.migration_template 'migration.rb', 'db/migrate', :assigns => get_local_assigns
+    end
+  end
+
+  
+  private  
+    def get_local_assigns
+      returning(assigns = {}) do
+        if class_name.underscore =~ /^(add|remove)_.*_(?:to|from)_(.*)/
+          assigns[:migration_action] = $1
+          assigns[:table_name]       = $2.pluralize
+        else
+          assigns[:attributes] = []
+        end
+      end
+    end
+end