mongo_doc 0.3.0

data/lib/mongo_doc/collection.rb

@@ -0,0 +1,55 @@
+require 'mongo_doc/cursor'
+
+module MongoDoc
+  class Collection
+    attr_accessor :_collection
+
+    delegate :[], :clear, :count, :create_index, :db, :distinct, :drop, :drop_index, :drop_indexes, :group, :hint, :index_information, :map_reduce, :mapreduce, :name, :options, :pk_factory, :remove, :rename, :size, :to => :_collection
+
+    def initialize(name)
+      self._collection = self.class.mongo_collection(name)
+    end
+
+    def find(query = {}, options = {})
+      cursor = wrapped_cursor(query, options)
+      if block_given?
+        yield cursor
+        cursor.close
+      else
+        cursor
+      end
+    end
+
+    def find_one(spec_or_object_id = nil, options = {})
+      MongoDoc::BSON.decode(_collection.find_one(spec_or_object_id, options))
+    end
+
+    def insert(doc_or_docs, options = {})
+      _collection.insert(doc_or_docs.to_bson, options)
+    end
+    alias :<< :insert
+
+    def save(doc, options = {})
+      _collection.save(doc.to_bson, options)
+    end
+
+    def update(spec, doc, options = {})
+      _collection.update(spec, doc.to_bson, options)
+      (last_error || {})['updatedExisting'] || false
+    end
+
+    protected
+
+    def last_error
+      MongoDoc::Connection.database.command({'getlasterror' => 1})
+    end
+
+    def wrapped_cursor(query = {}, options = {})
+      MongoDoc::Cursor.new(self, _collection.find(query, options))
+    end
+
+    def self.mongo_collection(name)
+      MongoDoc::Connection.database.collection(name)
+    end
+  end
+end

data/lib/mongo_doc/connection.rb

@@ -0,0 +1,88 @@
+module MongoDoc
+  class NoConnectionError < RuntimeError; end
+  class UnsupportedServerVersionError < RuntimeError; end
+
+  module Connection
+
+    extend self
+
+    attr_writer :config_path, :env, :host, :name, :options, :port, :strict
+
+    def config_path
+      @config_path || default_path
+    end
+
+    def configuration
+      @configuration ||= File.exists?(config_path) ? YAML.load_file(config_path)[env] : {}
+    end
+
+    def connection
+      @connection ||= connect
+    end
+
+    def database
+      @database ||= connection.db(name, :strict => strict)
+    end
+
+    def env
+      if rails?
+        Rails.env
+      else
+        @env ||= 'development'
+      end
+    end
+
+    def host
+      @host ||= configuration['host']
+    end
+
+    def name
+      @name ||= configuration['name'] || default_name
+    end
+
+    def options
+      @options ||= configuration['options'] || {}
+    end
+
+    def port
+      @port ||= configuration['port']
+    end
+
+    def strict
+      @strict ||= configuration['strict'] || false
+    end
+
+    private
+
+    def connect
+      connection = Mongo::Connection.new(host, port, options)
+      raise NoConnectionError unless connection
+      verify_server_version(connection)
+      connection
+    end
+
+    def default_name
+      if rails?
+        "#{Rails.root.basename}_#{Rails.env}"
+      else
+        "mongo_doc"
+      end
+    end
+
+    def default_path
+      if rails?
+        Rails.root + 'config/mongodb.yml'
+      else
+        './mongodb.yml'
+      end
+    end
+
+    def rails?
+      Object.const_defined?("Rails")
+    end
+
+    def verify_server_version(connection)
+      raise UnsupportedServerVersionError.new('MongoDoc requires at least mongoDB version 1.3.2') unless connection.server_version >= "1.3.2"
+    end
+  end
+end

data/lib/mongo_doc/contexts/enumerable.rb

@@ -0,0 +1,128 @@
+# encoding: utf-8
+module MongoDoc #:nodoc:
+  module Contexts #:nodoc:
+    class Enumerable
+      include Mongoid::Contexts::Paging
+      include MongoDoc::Contexts::Ids
+
+      attr_reader :criteria
+
+      delegate :first, :last, :to => :execute
+      delegate :documents, :options, :selector, :to => :criteria
+
+      # Return aggregation counts of the grouped documents. This will count by
+      # the first field provided in the fields array.
+      #
+      # Returns:
+      #
+      # A +Hash+ with field values as keys, count as values
+      def aggregate
+        counts = {}
+        group.each_pair { |key, value| counts[key] = value.size }
+        counts
+      end
+
+      # Gets the number of documents in the array. Delegates to size.
+      def count
+        @count ||= documents.size
+      end
+
+      # Groups the documents by the first field supplied in the field options.
+      #
+      # Returns:
+      #
+      # A +Hash+ with field values as keys, arrays of documents as values.
+      def group
+        field = options[:fields].first
+        documents.group_by { |doc| doc.send(field) }
+      end
+
+      # Enumerable implementation of execute. Returns matching documents for
+      # the selector, and adds options if supplied.
+      #
+      # Returns:
+      #
+      # An +Array+ of documents that matched the selector.
+      def execute(paginating = false)
+        limit(documents.select { |document| document.matches?(selector) })
+      end
+
+      # Create the new enumerable context. This will need the selector and
+      # options from a +Criteria+ and a documents array that is the underlying
+      # array of embedded documents from a has many association.
+      #
+      # Example:
+      #
+      # <tt>MongoDoc::Contexts::Enumerable.new(criteria)</tt>
+      def initialize(criteria)
+        @criteria = criteria
+      end
+
+      # Iterate over each +Document+ in the results. This can take an optional
+      # block to pass to each argument in the results.
+      #
+      # Example:
+      #
+      # <tt>context.iterate { |doc| p doc }</tt>
+      def iterate(&block)
+        execute.each(&block)
+      end
+
+      # Get the largest value for the field in all the documents.
+      #
+      # Returns:
+      #
+      # The numerical largest value.
+      def max(field)
+        determine(field, :>=)
+      end
+
+      # Get the smallest value for the field in all the documents.
+      #
+      # Returns:
+      #
+      # The numerical smallest value.
+      def min(field)
+        determine(field, :<=)
+      end
+
+      # Get one document.
+      #
+      # Returns:
+      #
+      # The first document in the +Array+
+      alias :one :first
+
+      # Get the sum of the field values for all the documents.
+      #
+      # Returns:
+      #
+      # The numerical sum of all the document field values.
+      def sum(field)
+        sum = documents.inject(nil) do |memo, doc|
+          value = doc.send(field)
+          memo ? memo += value : value
+        end
+      end
+
+      protected
+      # If the field exists, perform the comparison and set if true.
+      def determine(field, operator)
+        matching = documents.inject(nil) do |memo, doc|
+          value = doc.send(field)
+          (memo && memo.send(operator, value)) ? memo : value
+        end
+      end
+
+      # Limits the result set if skip and limit options.
+      def limit(documents)
+        skip, limit = options[:skip], options[:limit]
+        if skip && limit
+          return documents.slice(skip, limit)
+        end
+        documents
+      end
+    end
+  end
+end
+

data/lib/mongo_doc/contexts/ids.rb

@@ -0,0 +1,41 @@
+module MongoDoc
+  module Contexts
+    module Ids
+      # Return documents based on an id search. Will handle if a single id has
+      # been passed or mulitple ids.
+      #
+      # Example:
+      #
+      #   context.id_criteria([1, 2, 3])
+      #
+      # Returns:
+      #
+      # The single or multiple documents.
+      def id_criteria(params)
+        criteria.id(strings_to_object_ids(params))
+        params.is_a?(Array) ? criteria.entries : one
+      end
+
+      protected
+
+      # Convert ids from strings to +Mongo::ObjectID+s
+      def strings_to_object_ids(ids)
+        if Array === ids
+          ids.map {|id| string_to_object_id(id) }
+        else
+          string_to_object_id(ids)
+        end
+
+      end
+
+      # Convert ids from strings to +Mongo::ObjectID+s
+      def string_to_object_id(id)
+        if String === id
+          ::Mongo::ObjectID.from_string(id)
+        else
+          id
+        end
+      end
+    end
+  end
+end

data/lib/mongo_doc/contexts/mongo.rb

@@ -0,0 +1,232 @@
+module MongoDoc
+  module Contexts
+    class Mongo
+      include Mongoid::Contexts::Paging
+      include MongoDoc::Contexts::Ids
+
+      attr_reader :criteria, :cache
+
+      delegate :klass, :options, :selector, :to => :criteria
+      delegate :collection, :to => :klass
+
+      AGGREGATE_REDUCE = "function(obj, prev) { prev.count++; }"
+      # Aggregate the context. This will take the internally built selector and options
+      # and pass them on to the Ruby driver's +group()+ method on the collection. The
+      # collection itself will be retrieved from the class provided, and once the
+      # query has returned it will provided a grouping of keys with counts.
+      #
+      # Example:
+      #
+      # <tt>context.aggregate</tt>
+      #
+      # Returns:
+      #
+      # A +Hash+ with field values as keys, counts as values
+      def aggregate
+        collection.group(options[:fields], selector, { :count => 0 }, AGGREGATE_REDUCE, true)
+      end
+
+      # Get the count of matching documents in the database for the context.
+      #
+      # Example:
+      #
+      # <tt>context.count</tt>
+      #
+      # Returns:
+      #
+      # An +Integer+ count of documents.
+      def count
+        @count ||= collection.find(selector, options).count
+      end
+
+      # Execute the context. This will take the selector and options
+      # and pass them on to the Ruby driver's +find()+ method on the collection. The
+      # collection itself will be retrieved from the class provided, and once the
+      # query has returned new documents of the type of class provided will be instantiated.
+      #
+      # Example:
+      #
+      # <tt>mongo.execute</tt>
+      #
+      # Returns:
+      #
+      # An enumerable +Cursor+.
+      def execute(paginating = false)
+        cursor = collection.find(selector, options)
+        if cursor
+          @count = cursor.count if paginating
+          cursor
+        else
+          []
+        end
+      end
+
+      GROUP_REDUCE = "function(obj, prev) { prev.group.push(obj); }"
+      # Groups the context. This will take the internally built selector and options
+      # and pass them on to the Ruby driver's +group()+ method on the collection. The
+      # collection itself will be retrieved from the class provided, and once the
+      # query has returned it will provided a grouping of keys with objects.
+      #
+      # Example:
+      #
+      # <tt>context.group</tt>
+      #
+      # Returns:
+      #
+      # A +Hash+ with field values as keys, arrays of documents as values.
+      def group
+        collection.group(
+          options[:fields],
+          selector,
+          { :group => [] },
+          GROUP_REDUCE,
+          true
+        ).collect {|docs| docs["group"] = MongoDoc::BSON.decode(docs["group"]); docs }
+      end
+
+      # Create the new mongo context. This will execute the queries given the
+      # selector and options against the database.
+      #
+      # Example:
+      #
+      # <tt>Mongoid::Contexts::Mongo.new(criteria)</tt>
+      def initialize(criteria)
+        @criteria = criteria
+      end
+
+      # Iterate over each +Document+ in the results. This can take an optional
+      # block to pass to each argument in the results.
+      #
+      # Example:
+      #
+      # <tt>context.iterate { |doc| p doc }</tt>
+      def iterate(&block)
+        return caching(&block) if criteria.cached?
+        if block_given?
+          execute.each do |doc|
+            yield doc
+          end
+        end
+      end
+
+      # Return the last result for the +Context+. Essentially does a find_one on
+      # the collection with the sorting reversed. If no sorting parameters have
+      # been provided it will default to ids.
+      #
+      # Example:
+      #
+      # <tt>context.last</tt>
+      #
+      # Returns:
+      #
+      # The last document in the collection.
+      def last
+        sorting = options[:sort] || [[:_id, :asc]]
+        options[:sort] = sorting.collect { |option| [ option[0], option[1].invert ] }
+        collection.find_one(selector, options)
+      end
+
+      MAX_REDUCE = "function(obj, prev) { if (prev.max == 'start') { prev.max = obj.[field]; } " +
+        "if (prev.max < obj.[field]) { prev.max = obj.[field]; } }"
+      # Return the max value for a field.
+      #
+      # This will take the internally built selector and options
+      # and pass them on to the Ruby driver's +group()+ method on the collection. The
+      # collection itself will be retrieved from the class provided, and once the
+      # query has returned it will provided a grouping of keys with sums.
+      #
+      # Example:
+      #
+      # <tt>context.max(:age)</tt>
+      #
+      # Returns:
+      #
+      # A numeric max value.
+      def max(field)
+        grouped(:max, field.to_s, MAX_REDUCE)
+      end
+
+      MIN_REDUCE = "function(obj, prev) { if (prev.min == 'start') { prev.min = obj.[field]; } " +
+        "if (prev.min > obj.[field]) { prev.min = obj.[field]; } }"
+      # Return the min value for a field.
+      #
+      # This will take the internally built selector and options
+      # and pass them on to the Ruby driver's +group()+ method on the collection. The
+      # collection itself will be retrieved from the class provided, and once the
+      # query has returned it will provided a grouping of keys with sums.
+      #
+      # Example:
+      #
+      # <tt>context.min(:age)</tt>
+      #
+      # Returns:
+      #
+      # A numeric minimum value.
+      def min(field)
+        grouped(:min, field.to_s, MIN_REDUCE)
+      end
+
+      # Return the first result for the +Context+.
+      #
+      # Example:
+      #
+      # <tt>context.one</tt>
+      #
+      # Return:
+      #
+      # The first document in the collection.
+      def one
+        collection.find_one(selector, options)
+      end
+
+      alias :first :one
+
+      SUM_REDUCE = "function(obj, prev) { if (prev.sum == 'start') { prev.sum = 0; } prev.sum += obj.[field]; }"
+      # Sum the context.
+      #
+      # This will take the internally built selector and options
+      # and pass them on to the Ruby driver's +group()+ method on the collection. The
+      # collection itself will be retrieved from the class provided, and once the
+      # query has returned it will provided a grouping of keys with sums.
+      #
+      # Example:
+      #
+      # <tt>context.sum(:age)</tt>
+      #
+      # Returns:
+      #
+      # A numeric value that is the sum.
+      def sum(field)
+        grouped(:sum, field.to_s, SUM_REDUCE)
+      end
+
+      # Common functionality for grouping operations. Currently used by min, max
+      # and sum. Will gsub the field name in the supplied reduce function.
+      def grouped(start, field, reduce)
+        result = collection.group(
+          nil,
+          selector,
+          { start => "start" },
+          reduce.gsub("[field]", field),
+          true
+        )
+        result.empty? ? nil : result.first[start.to_s]
+      end
+
+      protected
+
+      # Iterate and cache results from execute
+      def caching(&block)
+        if cache
+          cache.each(&block)
+        else
+          @cache = []
+          execute.each do |doc|
+            @cache << doc
+            yield doc if block_given?
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongo_doc/contexts.rb

@@ -0,0 +1,25 @@
+# encoding: utf-8
+require "mongoid/contexts/paging"
+require "mongo_doc/contexts/ids"
+require "mongo_doc/contexts/enumerable"
+require "mongo_doc/contexts/mongo"
+
+module Mongoid
+  module Contexts
+    # Determines the context to be used for this criteria. If the class is an
+    # embedded document, then the context will be the array in the has_many
+    # association it is in. If the class is a root, then the database itself
+    # will be the context.
+    #
+    # Example:
+    #
+    # <tt>Contexts.context_for(criteria)</tt>
+    def self.context_for(criteria)
+      if criteria.klass.respond_to?(:collection)
+        return MongoDoc::Contexts::Mongo.new(criteria)
+      end
+      return MongoDoc::Contexts::Enumerable.new(criteria)
+    end
+
+  end
+end

data/lib/mongo_doc/criteria.rb

@@ -0,0 +1,38 @@
+require 'mongoid/extensions/hash/criteria_helpers'
+require 'mongoid/extensions/symbol/inflections'
+require 'mongo_doc/matchers'
+require 'mongo_doc/contexts'
+require 'mongoid/criteria'
+
+module MongoDoc
+  module Criteria
+    # Create a criteria for this +Document+ class
+    #
+    # <tt>Person.criteria</tt>
+    def criteria
+      Mongoid::Criteria.new(self)
+    end
+
+    delegate \
+      :and,
+      :any_in,
+      :cache,
+      :enslave,
+      :excludes,
+      :extras,
+      :id,
+      :in,
+      :limit,
+      :not_in,
+      :offset,
+      :only,
+      :order_by,
+      :page,
+      :per_page,
+      :skip,
+      :where, :to => :criteria
+  end
+end
+
+Hash.send(:include, Mongoid::Extensions::Hash::CriteriaHelpers)
+Symbol.send(:include, Mongoid::Extensions::Symbol::Inflections)

data/lib/mongo_doc/cursor.rb

@@ -0,0 +1,32 @@
+module MongoDoc
+  class Cursor
+    include Enumerable
+
+    attr_accessor :_collection, :_cursor
+
+    delegate :admin, :close, :closed?, :count, :explain, :fields, :full_collection_name, :hint, :limit, :order, :query_options_hash, :query_opts, :selector, :skip, :snapshot, :sort, :timeout, :to => :_cursor
+
+    def initialize(mongo_doc_collection, cursor)
+      self._collection = mongo_doc_collection
+      self._cursor = cursor
+    end
+
+    def collection
+      _collection
+    end
+
+    def each
+      _cursor.each do |next_document|
+        yield MongoDoc::BSON.decode(next_document)
+      end
+    end
+
+    def next_document
+      MongoDoc::BSON.decode(_cursor.next_document)
+    end
+
+    def to_a
+      MongoDoc::BSON.decode(_cursor.to_a)
+    end
+  end
+end