Contents

module Mover
  module Migrator
    
    def method_missing_with_mover(method, *arguments, &block)
      args = Marshal.load(Marshal.dump(arguments))
      method_missing_without_mover(method, *arguments, &block)
      
      supported = [
        :add_column, :add_timestamps, :change_column,
        :change_column_default, :change_table,
        :drop_table, :remove_column, :remove_columns,
        :remove_timestamps, :rename_column, :rename_table
      ]
      
      # Don't change these columns
      %w(moved_at move_id).each do |column|
        return if args.include?(column) || args.include?(column.intern)
      end
      
      if !args.empty? && supported.include?(method)
        connection = ActiveRecord::Base.connection
        table_name = ActiveRecord::Migrator.proper_table_name(args[0])
        # Find model
        klass = Object.subclasses_of(ActiveRecord::Base).detect do |klass|
          if klass.respond_to?(:movable_types)
            klass.table_name.to_s == table_name
          end
        end
        # Run migration on movable table
        if klass
          klass.movable_types.each do |type|
            args[0] = [ table_name, type ].join('_')
            if method == :rename_table
              args[1] = [ args[1].to_s, type ].join('_')
            end
            if connection.table_exists?(args[0])
              connection.send(method, *args, &block)
            end
          end
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
mover-0.1.1	lib/mover/migrator.rb