Contents

module RedHillConsulting::AutomaticForeignKey::ActiveRecord::ConnectionAdapters
  module TableDefinition
    def self.included(base)
      base.class_eval do
        alias_method_chain :column, :automatic_foreign_key
        alias_method_chain :primary_key, :automatic_foreign_key
      end
    end

    def primary_key_with_automatic_foreign_key(name, options = {})
      column(name, :primary_key, options)
    end

    def column_with_automatic_foreign_key(name, type, options = {})
      column_without_automatic_foreign_key(name, type, options)
      references = ActiveRecord::Base.references(self.name, name, options)
      foreign_key(name, references.first, references.last, options) if references
      self
    end

    # Some people liked this; personally I've decided against using it but I'll keep it nonetheless
    def belongs_to(table, options = {})
      options = options.merge(:references => table)
      options[:on_delete] = options.delete(:dependent) if options.has_key?(:dependent)
      column("#{table.to_s.singularize}_id".to_sym, :integer, options)
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
automatic_foreign_key-1.0.0	lib/red_hill_consulting/automatic_foreign_key/active_record/connection_adapters/table_definition.rb