also_migrate 0.2.0 → 0.2.1

data/README.md

@@ -15,14 +15,25 @@ Define the model
 
 <pre>
 class Article &lt; ActiveRecord::Base
-  also_migrate :article_archives, :ignore => 'moved_at', :indexes => 'id'
+  also_migrate(
+    :article_archives,
+    :add => [
+      # Parameters to ActiveRecord::ConnectionAdapters::SchemaStatements#add_column
+      [ 'deleted_at', :datetime, {} ]
+    ],
+    :subtract => 'restored_at',
+    :ignore => 'deleted_at',
+    :indexes => 'id'
+  )
 end
 </pre>
 
 Options:
 
+* <code>add</code> Create columns that the original table doesn't have (defaults to none)
+* <code>subtract</code> Exclude columns from the original table (defaults to none)
 * <code>ignore</code> Ignore migrations that apply to certain columns (defaults to none)
-* <code>indexes</code> Only index certain columns (defaults to all)
+* <code>indexes</code> Only index certain columns (duplicates all indexes by default)
 
 That's it!
 ----------

data/lib/also_migrate/base.rb

@@ -15,6 +15,8 @@ module AlsoMigrate
         @also_migrate_config << {
           :tables => args.collect(&:to_s),
           :options => {
+            :add => options[:add] ? options[:add] : [],
+            :subtract => [ options[:subtract] ].flatten.compact,
             :ignore => [ options[:ignore] ].flatten.compact,
             :indexes => options[:indexes] ? [ options[:indexes] ].flatten : nil
           }

data/lib/also_migrate/migrator.rb

@@ -44,17 +44,17 @@ module AlsoMigrate
               config[:tables].each do |new_table|
                 if !connection.table_exists?(new_table) && connection.table_exists?(old_table)
                   columns = connection.columns(old_table).collect(&:name)
-                  columns -= options[:ignore].collect(&:to_s)
+                  columns -= options[:subtract].collect(&:to_s)
                   columns.collect! { |col| connection.quote_column_name(col) }
-                  engine =
-                    if connection.class.to_s.include?('Mysql')
-                      'ENGINE=' + connection.select_one(<<-SQL)['Engine']
-                        SHOW TABLE STATUS
-                        WHERE Name = '#{old_table}'
-                      SQL
-                    end
                   indexes = options[:indexes]
                   if indexes
+                    engine =
+                      if connection.class.to_s.include?('Mysql')
+                        'ENGINE=' + connection.select_one(<<-SQL)['Engine']
+                          SHOW TABLE STATUS
+                          WHERE Name = '#{old_table}'
+                        SQL
+                      end
                     connection.execute(<<-SQL)
                       CREATE TABLE #{new_table} #{engine}
                       AS SELECT #{columns.join(',')}
@@ -71,6 +71,25 @@ module AlsoMigrate
                     SQL
                   end
                 end
+                if connection.table_exists?(new_table)
+                  if options[:add] || options[:subtract]
+                    columns = connection.columns(new_table).collect(&:name)
+                  end
+                  if options[:add]
+                    options[:add].each do |column|
+                      unless columns.include?(column[0])
+                        connection.add_column(*([ new_table ] + column))
+                      end
+                    end
+                  end
+                  if options[:subtract]
+                    options[:subtract].each do |column|
+                      if columns.include?(column)
+                        connection.remove_column(new_table, column)
+                      end
+                    end
+                  end
+                end
               end
             end
           end

data/lib/also_migrate/version.rb

@@ -1,3 +1,3 @@
 module AlsoMigrate
-  VERSION = "0.2.0" unless defined?(::AlsoMigrate::VERSION)
+  VERSION = "0.2.1" unless defined?(::AlsoMigrate::VERSION)
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 2
-  - 0
-  version: 0.2.0
+  - 1
+  version: 0.2.1
 platform: ruby
 authors: 
 - Winton Welsh
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-12-04 00:00:00 -08:00
+date: 2010-12-12 00:00:00 -08:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency