Contents

module AutomaticForeignKey::ActiveRecord
  module Migration
    def self.included(base)
      base.extend(ClassMethods)
    end

    module ClassMethods
      def add_column(table_name, column_name, type, options = {})
        super
        references = ActiveRecord::Base.references(table_name, column_name, options)
        add_foreign_key(table_name, column_name, references.first, references.last, options) if references
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
automatic_foreign_key-1.0.2	lib/automatic_foreign_key/active_record/migration.rb
automatic_foreign_key-1.0.1	lib/automatic_foreign_key/active_record/migration.rb