Contents

module QueryAssociation
  include RecordAssociation
  
  def initialize(name, options={})
    super
    if @options['where']
      @options['where'] = Plucky::CriteriaHash.new(@options['where']).to_hash
    end
  end
  
  def validate(record, errors)
    # noop
  end
  
  def default
    nil
  end
  
  def strip_nil?
    true
  end
  
  def associate(associated_record, store, record)
    associated_record.set_meta(foreign_key(record), record.id)
    associated_record.save_without_validation
  end
  
  def unassociate(associated_record, store, record)
    return unless associated_record.get_meta(foreign_key) == record.id
    associated_record.set_meta(foreign_key(record), nil)
    associated_record.save_without_validation
  end
  
  protected
    def clear(store, record)
      all(store, record).each {|associated_record| unassociate(associated_record, store, record)}
    end
    
    def all(store, record)
      scope(record).all
    end
    
    def scope(record)
      if @options['through']
        through_query = record.field(@options['through'].to_s).scope(record).only('_id')
        cursor = Record.collection.find(through_query.criteria.to_hash, through_query.options.to_hash)
        ids = cursor.to_a.collect {|doc| doc['_id']}
        scope = record.site.model(model_name).where(foreign_key(record) => ids)
      elsif @options['extends']
        scope = record.field(@options['extends'].to_s).scope(record)
      else
        scope = record.site.model(model_name).where(foreign_key(record) => record.id)
      end
      scope = scope.where(@options['where'].to_hash) if @options['where']
      scope = scope.sort(@options['order'].to_s) if @options['order']
      scope = scope.limit(@options['limit'].to_i) if @options['limit']
      scope = scope.skip(@options['skip'].to_i) if @options['skip']
      scope
    end
    
    def associated(store, record)
      record.site.model(model_name).first(foreign_key(record) => record.id)
    end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
yodel-0.0.1	lib/yodel/models/core/associations/query/query_association.rb