Contents

require 'rubygems'
require File.join(File.dirname(__FILE__),'mongo_scope','util')

module MongoScope
  module ScopeMethods
    def raw_scope(ops)
      ScopedCollection.new(ops,self)
    end
    def scope_eq(ops)
      scope_ops = {:field => ops.keys.first, :val => ops.values.first}
      ScopedCollection.new(EqScope.new(scope_ops),self)
    end
    %w(in nin gt lt gte lte ne mod all size exists).each do |op|
      define_method("scope_#{op}") do |ops|
        raw_scope(:op => "$#{op}", :field => ops.keys.first, :val => ops.values.first)
      end
    end
  end

  module CollMethods
    def count
      find.count
    end
  end

  class ScopedCollection
    include ScopeMethods
    include CollMethods
    attr_accessor :scope_obj, :coll
    def initialize(scope_ops,coll)
      self.scope_obj = Scope.new(scope_ops)
      self.coll = coll
    end
    def scoped_ops(ops)
      ops = {'_id' => ops} unless ops.kind_of?(Hash)
      ops.merge(scope_obj.find_ops)
    end
    def find(selector={},options={})
      coll.find(scoped_ops(selector),options)
    end
    def find_one(selector={},options={})
      coll.find_one(scoped_ops(selector),options)
    end
    def method_missing(sym,*args,&b)
      coll.send(sym,*args,&b)
    end
  end

  class Scope
    def self.new(ops)
      ops.respond_to?(:find_ops) ? ops : super(ops)
    end
    attr_accessor :field, :op, :val
    include FromHash
    def find_ops
      {field => {op => val}}
    end
  end

  class EqScope
    attr_accessor :field, :val
    include FromHash
    def find_ops
      {field => val}
    end
  end
  
  def self.use!
    [CollMethods,ScopeMethods].each { |mod| ::Mongo::Collection.send(:include,mod) }
  end
  
  use!
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
mongo_scope-0.0.3	lib/mongo_scope.rb