gistfile1.builder

class Ohm::Model
  @@associations = ::Hash.new { |hash, key| hash[key] = [] }
  @@references = ::Hash.new { |hash, key| hash[key] = [] }

  def self.associations
    @@associations[self]
  end

  def associations
    self.class.associations
  end


  def self.references
    @@references[self]
  end

  def references
    self.class.references
  end

  class << self
    alias_method :orig_collection, :collection
    alias_method :orig_reference, :reference

    def collection(*args)
      orig_collection(*args)
      associations << args[0]
    end

    def reference(*args)
      orig_reference(*args)
      references << :"#{args[0]}_id"
    end
  end

end