mongo_doc_rails2 0.6.1

data/lib/mongo_doc/criteria.rb

@@ -0,0 +1,70 @@
+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
+      CriteriaWrapper.new(self)
+    end
+
+    delegate \
+      :aggregate,
+      :all,
+      :and,
+      :any_in,
+      :blank?,
+      :count,
+      :empty?,
+      :excludes,
+      :extras,
+      :first,
+      :group,
+      :id,
+      :in,
+      :last,
+      :limit,
+      :max,
+      :min,
+      :not_in,
+      :offset,
+      :one,
+      :only,
+      :order_by,
+      :page,
+      :paginate,
+      :per_page,
+      :skip,
+      :sum,
+      :where, :to => :criteria
+
+    class CriteriaWrapper < Mongoid::Criteria
+      %w(all and any_in cache enslave excludes extras fuse in limit not_in offset only order_by skip where).each do |method|
+        class_eval(<<-RUBY, __FILE__, __LINE__ + 1)
+          def #{method}_with_wrapping(*args, &block)                # def and(*args, &block)
+            new_criteria = CriteriaWrapper.new(klass)               #   new_criteria = CriteriaWrapper.new(klass)
+            new_criteria.merge(self)                                #   new_criteria.merge(criteria)
+            new_criteria.#{method}_without_wrapping(*args, &block)  #   new_criteria.and_without_wrapping(*args, &block)
+            new_criteria                                            #   new_criteria
+          end                                                       # end
+
+          alias_method_chain :#{method}, :wrapping
+          protected :#{method}_without_wrapping
+        RUBY
+      end
+
+      protected
+
+      attr_accessor :criteria
+
+    end
+  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

data/lib/mongo_doc/document.rb

@@ -0,0 +1,205 @@
+require 'mongo_doc/bson'
+require 'mongo_doc/attributes'
+require 'mongo_doc/root'
+require 'mongo_doc/associations'
+require 'mongo_doc/criteria'
+require 'mongo_doc/finders'
+require 'mongo_doc/index'
+require 'mongo_doc/scope'
+require 'mongo_doc/validations'
+
+module MongoDoc
+  class UnsupportedOperation < RuntimeError; end
+  class DocumentInvalidError < RuntimeError; end
+  class NotADocumentError < RuntimeError; end
+
+  module Document
+
+    def self.included(klass)
+      klass.class_eval do
+        include Attributes
+        include Root
+        extend Associations
+        extend ClassMethods
+        extend Criteria
+        extend Finders
+        extend Index
+        extend Scope
+        include Validations
+
+        alias id _id
+      end
+    end
+
+    def _collection
+      _root and _root._collection or self.class.collection
+    end
+
+    def initialize(attrs = {})
+      self.attributes = attrs
+    end
+
+    def ==(other)
+      return false unless self.class === other
+      self.class._attributes.all? {|var| self.send(var) == other.send(var)}
+    end
+
+    def new_record?
+      _id.nil?
+    end
+
+    def remove
+      raise UnsupportedOperation.new('Document#remove is not supported for embedded documents') if _root
+      remove_document
+    end
+
+    def remove_document
+      return _root.remove_document if _root
+      _remove
+    end
+
+    def save(validate = true)
+      return _root.save(validate) if _root
+      return _save(false) unless validate and not valid?
+      false
+    end
+
+    def save!
+      return _root.save! if _root
+      raise DocumentInvalidError unless valid?
+      _save(true)
+    end
+
+    def to_bson(*args)
+      {MongoDoc::BSON::CLASS_KEY => self.class.name}.tap do |bson_hash|
+        bson_hash['_id'] = _id unless new_record?
+        self.class._attributes.each do |name|
+          bson_hash[name.to_s] = send(name).to_bson(args)
+        end
+      end
+    end
+
+    def to_param
+      _id.to_s
+    end
+
+    def update_attributes(attrs)
+      self.attributes = attrs
+      return save if new_record?
+      return false unless valid?
+      _update({}, attrs, false)
+    end
+
+    def update_attributes!(attrs)
+      strict = attrs.delete(:__strict__)
+      self.attributes = attrs
+      return save! if new_record?
+      raise DocumentInvalidError unless valid?
+      _update({}, attrs, true)
+    end
+
+    module ClassMethods
+      def bson_create(bson_hash, options = {})
+        allocate.tap do |obj|
+          bson_hash.each do |name, value|
+            obj.send("#{name}=", MongoDoc::BSON.decode(value, options))
+          end
+        end
+      end
+
+      def collection
+        @collection ||= MongoDoc::Collection.new(collection_name)
+      end
+
+      def collection_name
+        self.to_s.tableize.gsub('/', '.')
+      end
+
+      def create(attrs = {})
+        instance = new(attrs)
+        instance.save(true)
+        instance
+      end
+
+      def create!(attrs = {})
+        instance = new(attrs)
+        instance.save!
+        instance
+      end
+    end
+
+    protected
+
+    def _remove
+      _collection.remove({'_id' => _id})
+    end
+
+    def _update(selector, data, safe)
+      return _root.send(:_update, {_selector_path + '._id' => _id}, hash_with_modifier_path_keys(data), safe) if _root
+      _collection.update({'_id' => _id}.merge(selector), {'$set' => data}, :safe => safe)
+    end
+
+    def _save(safe)
+      notify_before_save_observers
+      self._id = _collection.save(self, :safe => safe)
+      notify_save_success_observers
+      self._id
+    rescue Mongo::MongoDBError => e
+      notify_save_failed_observers
+      raise e
+    end
+
+    def hash_with_modifier_path_keys(hash)
+      hash.stringify_keys!
+      {}.tap do |dup|
+        hash.each do |key, value|
+          dup[_modifier_path + '.' + key] = value
+        end
+      end
+    end
+
+    def path(parent_path, child_path)
+      if parent_path.blank?
+        child_path
+      else
+        parent_path + '.' + child_path
+      end
+    end
+
+    def before_save_callback(root)
+      self._id = ::BSON::ObjectID.new if new_record?
+    end
+
+    def save_failed_callback(root)
+      self._id = nil
+    end
+
+    def save_success_callback(root)
+      root.unregister_save_observer(self)
+    end
+
+    def save_observers
+      @save_observers ||= []
+    end
+
+    def register_save_observer(child)
+      save_observers << child
+    end
+
+    def unregister_save_observer(child)
+      save_observers.delete(child)
+    end
+
+    def notify_before_save_observers
+      save_observers.each {|obs| obs.before_save_callback(self) }
+    end
+
+    def notify_save_success_observers
+      save_observers.each {|obs| obs.save_success_callback(self) }
+    end
+
+    def notify_save_failed_observers
+      save_observers.each {|obs| obs.save_failed_callback(self) }
+    end
+  end
+end

data/lib/mongo_doc/ext.rb

@@ -0,0 +1,16 @@
+require 'mongo_doc/ext/array'
+require 'mongo_doc/ext/binary'
+require 'mongo_doc/ext/boolean_class'
+require 'mongo_doc/ext/date'
+require 'mongo_doc/ext/date_time'
+require 'mongo_doc/ext/dbref'
+require 'mongo_doc/ext/hash'
+require 'mongo_doc/ext/min_max_keys'
+require 'mongo_doc/ext/nil_class'
+require 'mongo_doc/ext/numeric'
+require 'mongo_doc/ext/object'
+require 'mongo_doc/ext/object_id'
+require 'mongo_doc/ext/regexp'
+require 'mongo_doc/ext/string'
+require 'mongo_doc/ext/symbol'
+require 'mongo_doc/ext/time'

data/lib/mongo_doc/ext/array.rb

@@ -0,0 +1,5 @@
+class Array
+  def to_bson(*args)
+    map {|item| item.to_bson(args)}
+  end
+end

data/lib/mongo_doc/ext/binary.rb

@@ -0,0 +1,7 @@
+module BSON
+  class Binary
+    def to_bson(*args)
+      self
+    end
+  end
+end

data/lib/mongo_doc/ext/boolean_class.rb

@@ -0,0 +1,17 @@
+class Boolean
+  def self.cast_from_string(value)
+    value == '1' || value.downcase == 'true'
+  end
+end
+
+class FalseClass
+  def to_bson(*args)
+    self
+  end
+end
+
+class TrueClass
+  def to_bson(*args)
+    self
+  end
+end

data/lib/mongo_doc/ext/date.rb

@@ -0,0 +1,19 @@
+class Date
+  def to_bson(*args)
+    {
+      MongoDoc::BSON::CLASS_KEY => self.class.name,
+      'dt' => strftime,
+      'sg' => start
+    }
+  end
+
+  alias start sg unless method_defined?(:start)
+
+  def self.bson_create(bson_hash, options = nil)
+    Date.parse(*bson_hash.values_at('dt', 'sg'))
+  end
+
+  def self.cast_from_string(value)
+    Date.parse(value) unless value.blank?
+  end
+end

data/lib/mongo_doc/ext/date_time.rb

@@ -0,0 +1,17 @@
+class DateTime
+  def to_bson(*args)
+    {
+      MongoDoc::BSON::CLASS_KEY => self.class.name,
+      'dt' => strftime,
+      'sg' => start
+    }
+  end
+
+  def self.bson_create(bson_hash, options = nil)
+    DateTime.parse(*bson_hash.values_at('dt', 'sg'))
+  end
+
+  def self.cast_from_string(string)
+    DateTime.parse(string) unless string.blank?
+  end
+end

data/lib/mongo_doc/ext/dbref.rb

@@ -0,0 +1,7 @@
+module BSON
+  class DBRef
+    def to_bson(*args)
+      self
+    end
+  end
+end

data/lib/mongo_doc/ext/hash.rb

@@ -0,0 +1,7 @@
+class Hash
+  def to_bson(*args)
+    {}.tap do |hash|
+      each {|key, value| hash[key.to_s] = value.to_bson}
+    end
+  end  
+end

data/lib/mongo_doc/ext/min_max_keys.rb

@@ -0,0 +1,13 @@
+module BSON
+  class MaxKey
+    def to_bson(*args)
+      self
+    end
+  end
+
+  class MinKey
+    def to_bson(*args)
+      self
+    end
+  end
+end

data/lib/mongo_doc/ext/nil_class.rb

@@ -0,0 +1,5 @@
+class NilClass
+  def to_bson(*args)
+    self
+  end
+end

data/lib/mongo_doc/ext/numeric.rb

@@ -0,0 +1,17 @@
+class Numeric
+  def to_bson(*args)
+    self
+  end
+end
+
+class BigDecimal
+  def self.cast_from_string(string)
+    BigDecimal.new(string) unless string.blank?
+  end
+end
+
+class Integer
+  def self.cast_from_string(string)
+    string.to_i unless string.blank?
+  end
+end