Contents

require 'active_support/core_ext/class/subclasses'

module MongoModel
  class MongoOptions
    ValidKeys = [ :conditions, :select, :offset, :limit, :order ]
    
    attr_reader :selector, :options
    
    def initialize(model, options={})
      options.assert_valid_keys(ValidKeys)
      
      @model = model
      
      @selector = extract_conditions(options)
      @options  = extract_options(options)
      
      add_type_to_selector
    end
    
    def to_a
      [selector, options]
    end
  
  private
    def extract_conditions(options)
      result = {}
      
      (options[:conditions] || {}).each do |k, v|
        if k.is_a?(MongoOperator)
          key = k.field
          value = k.to_mongo_selector(v)
        else
          key = k
          value = v
        end
        
        property = @model.properties[key]
        
        result[property ? property.as : key] = value
      end
      
      result
    end
    
    def extract_options(options)
      result = {}
      
      result[:fields] = options[:select] if options[:select]
      result[:skip]   = options[:offset] if options[:offset]
      result[:limit]  = options[:limit]  if options[:limit]
      result[:sort]   = MongoOrder.parse(options[:order]).to_sort(@model) if options[:order]
      
      result
    end
    
    def convert_order(order)
      case order
      when Array
        order.map { |clause|
          key, sort = clause.split(/ /)
          
          property = @model.properties[key.to_sym]
          sort = (sort =~ /desc/i) ? :descending : :ascending
          
          [property ? property.as : key, sort]
        } if order.size > 0
      when String, Symbol
        convert_order(order.to_s.split(/,/).map { |c| c.strip })
      end
    end
    
    def add_type_to_selector
      unless selector['_type'] || @model.superclass.abstract_class?
        selector['_type'] = { '$in' => [@model.to_s] + @model.subclasses }
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
mongomodel-0.2.6	lib/mongomodel/support/mongo_options.rb
mongomodel-0.2.5	lib/mongomodel/support/mongo_options.rb
mongomodel-0.2.4	lib/mongomodel/support/mongo_options.rb
mongomodel-0.2.3	lib/mongomodel/support/mongo_options.rb
mongomodel-0.2.2	lib/mongomodel/support/mongo_options.rb
mongomodel-0.2.1	lib/mongomodel/support/mongo_options.rb
mongomodel-0.2.0	lib/mongomodel/support/mongo_options.rb