From 7de537dbf4cf99e2d85854fd6331a30604064656 Mon Sep 17 00:00:00 2001
From: Rick Olson <technoweenie@gmail.com>
Date: Tue, 24 Jul 2007 04:10:22 +0000
Subject: Automatically generate add/remove column commands in specially named
 migrations like AddLocationToEvent.  Closes #9006 [zenspider]

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@7216 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
---
 .../rails_generator/generators/components/migration/USAGE  | 14 ++++++++++++++
 .../generators/components/migration/migration_generator.rb | 13 +++++++++++++
 .../generators/components/migration/templates/migration.rb |  4 ++--
 3 files changed, 29 insertions(+), 2 deletions(-)

(limited to 'railties/lib/rails_generator')

diff --git a/railties/lib/rails_generator/generators/components/migration/USAGE b/railties/lib/rails_generator/generators/components/migration/USAGE
index 152e2abdd6..c3f8ce0987 100644
--- a/railties/lib/rails_generator/generators/components/migration/USAGE
+++ b/railties/lib/rails_generator/generators/components/migration/USAGE
@@ -3,8 +3,22 @@ Description:
     CamelCased or under_scored, as an argument. A migration class is generated
     in db/migrate prefixed by the latest migration number.
 
+    You can name your migration in either of these formats to generate add/remove
+    column lines: AddColumnToTable or RemoveColumnFromTable
+
 Example:
     `./script/generate migration AddSslFlag`
 
     With 4 existing migrations, this creates the AddSslFlag migration in
     db/migrate/005_add_ssl_flag.rb
+
+    `./script/generate migration AddSslFlagToAccount`
+    
+    This will create the AddSslFlagToAccount in db/migrate/005_add_ssl_flag_to_account.rb with
+    this in the Up migration:
+
+      add_column :accounts, :ssl_flag, :type, :null => :no?, :default => :maybe?
+
+    And this in the Down migration:
+    
+      remove_column :accounts, :ssl_flag
\ No newline at end of file
diff --git a/railties/lib/rails_generator/generators/components/migration/migration_generator.rb b/railties/lib/rails_generator/generators/components/migration/migration_generator.rb
index a0d0d472c2..2fd348e47b 100644
--- a/railties/lib/rails_generator/generators/components/migration/migration_generator.rb
+++ b/railties/lib/rails_generator/generators/components/migration/migration_generator.rb
@@ -4,4 +4,17 @@ class MigrationGenerator < Rails::Generator::NamedBase
       m.migration_template 'migration.rb', 'db/migrate'
     end
   end
+
+  def auto_migration direction
+    case class_name.underscore
+    when /^(add|remove)_(.*)_(?:to|from)_(.*)/ then
+      action, col, tbl = $1, $2, $3.pluralize
+
+      unless (action == "add") ^ (direction == :up) then
+        %(\n    add_column :#{tbl}, :#{col}, :type, :null => :no?, :default => :maybe?)
+      else
+        %(\n    remove_column :#{tbl}, :#{col})
+      end
+    end
+  end
 end
diff --git a/railties/lib/rails_generator/generators/components/migration/templates/migration.rb b/railties/lib/rails_generator/generators/components/migration/templates/migration.rb
index 2600610fc6..86edcac261 100644
--- a/railties/lib/rails_generator/generators/components/migration/templates/migration.rb
+++ b/railties/lib/rails_generator/generators/components/migration/templates/migration.rb
@@ -1,7 +1,7 @@
 class <%= class_name.underscore.camelize %> < ActiveRecord::Migration
-  def self.up
+  def self.up<%= auto_migration :up %>
   end
 
-  def self.down
+  def self.down<%= auto_migration :down %>
   end
 end
-- 
cgit v1.2.3