mongoid-pre 2.0.0.beta1

data/lib/mongoid/errors.rb

@@ -0,0 +1,77 @@
+# encoding: utf-8
+module Mongoid #:nodoc
+  module Errors #:nodoc
+
+    # Raised when querying the database for a document by a specific id which
+    # does not exist. If multiple ids were passed then it will display all of
+    # those.
+    #
+    # Example:
+    #
+    # <tt>DocumentNotFound.new(Person, ["1", "2"])</tt>
+    class DocumentNotFound < RuntimeError
+      def initialize(klass, ids)
+        @klass, @identifier = klass, ids.is_a?(Array) ? ids.join(", ") : ids
+      end
+      def message
+        "Document not found for class #{@klass} and id(s) #{@identifier}"
+      end
+    end
+
+    # Raised when invalid options are passed into a constructor or method.
+    #
+    # Example:
+    #
+    # <tt>InvalidOptions.new</tt>
+    class InvalidOptions < RuntimeError; end
+
+    # Raised when the database connection has not been set up properly, either
+    # by attempting to set an object on the db that is not a +Mongo::DB+, or
+    # not setting anything at all.
+    #
+    # Example:
+    #
+    # <tt>InvalidDatabase.new("Not a DB")</tt>
+    class InvalidDatabase < RuntimeError
+      def initialize(database)
+        @database = database
+      end
+      def message
+        "Database should be a Mongo::DB, not #{@database.class.name}"
+      end
+    end
+
+    # Raised when a persisence method ending in ! fails validation. The message
+    # will contain the full error messages from the +Document+ in question.
+    #
+    # Example:
+    #
+    # <tt>Validations.new(person.errors)</tt>
+    class Validations < RuntimeError
+      def initialize(errors)
+        @errors = errors
+      end
+      def message
+        "Validation failed: #{@errors.full_messages}"
+      end
+    end
+
+    # This error is raised when trying to access a Mongo::Collection from an
+    # embedded document.
+    #
+    # Example:
+    #
+    # <tt>InvalidCollection.new(Address)</tt>
+    class InvalidCollection < RuntimeError
+      def initialize(klass)
+        @klass = klass
+      end
+      def message
+        "Access to the collection for #{@klass.name} is not allowed " +
+          "since it is an embedded document, please access a collection from " +
+          "the root document"
+      end
+    end
+
+  end
+end

data/lib/mongoid/extensions.rb

@@ -0,0 +1,99 @@
+# encoding: utf-8
+require "mongoid/extensions/array/accessors"
+require "mongoid/extensions/array/aliasing"
+require "mongoid/extensions/array/assimilation"
+require "mongoid/extensions/array/conversions"
+require "mongoid/extensions/array/parentization"
+require "mongoid/extensions/big_decimal/conversions"
+require "mongoid/extensions/binary/conversions"
+require "mongoid/extensions/boolean/conversions"
+require "mongoid/extensions/date/conversions"
+require "mongoid/extensions/datetime/conversions"
+require "mongoid/extensions/float/conversions"
+require "mongoid/extensions/hash/accessors"
+require "mongoid/extensions/hash/assimilation"
+require "mongoid/extensions/hash/conversions"
+require "mongoid/extensions/hash/criteria_helpers"
+require "mongoid/extensions/hash/scoping"
+require "mongoid/extensions/integer/conversions"
+require "mongoid/extensions/nil/assimilation"
+require "mongoid/extensions/object/conversions"
+require "mongoid/extensions/proc/scoping"
+require "mongoid/extensions/string/conversions"
+require "mongoid/extensions/string/inflections"
+require "mongoid/extensions/symbol/inflections"
+require "mongoid/extensions/time/conversions"
+require "mongoid/extensions/objectid/conversions"
+
+class Array #:nodoc
+  include Mongoid::Extensions::Array::Accessors
+  include Mongoid::Extensions::Array::Assimilation
+  include Mongoid::Extensions::Array::Conversions
+  include Mongoid::Extensions::Array::Parentization
+end
+
+class BigDecimal #:nodoc
+  extend Mongoid::Extensions::BigDecimal::Conversions
+end
+
+class Binary #:nodoc
+  extend Mongoid::Extensions::Binary::Conversions
+end
+
+class Boolean #:nodoc
+  extend Mongoid::Extensions::Boolean::Conversions
+end
+
+class Date #:nodoc
+  extend Mongoid::Extensions::Date::Conversions
+end
+
+class DateTime #:nodoc
+  extend Mongoid::Extensions::DateTime::Conversions
+end
+
+class Float #:nodoc
+  extend Mongoid::Extensions::Float::Conversions
+end
+
+class Hash #:nodoc
+  include Mongoid::Extensions::Hash::Accessors
+  include Mongoid::Extensions::Hash::Assimilation
+  include Mongoid::Extensions::Hash::CriteriaHelpers
+  include Mongoid::Extensions::Hash::Scoping
+  extend Mongoid::Extensions::Hash::Conversions
+end
+
+class Integer #:nodoc
+  extend Mongoid::Extensions::Integer::Conversions
+end
+
+class NilClass #:nodoc
+  include Mongoid::Extensions::Nil::Assimilation
+end
+
+class Object #:nodoc:
+  include Mongoid::Extensions::Object::Conversions
+end
+
+class Proc #:nodoc:
+  include Mongoid::Extensions::Proc::Scoping
+end
+
+class String #:nodoc
+  include Mongoid::Extensions::String::Inflections
+  extend Mongoid::Extensions::String::Conversions
+end
+
+class Symbol #:nodoc
+  remove_method :size if instance_methods.include? :size # temporal fix for ruby 1.9
+  include Mongoid::Extensions::Symbol::Inflections
+end
+
+class Time #:nodoc
+  extend Mongoid::Extensions::Time::Conversions
+end
+
+class Mongo::ObjectID #:nodoc
+  extend Mongoid::Extensions::ObjectID::Conversions
+end

data/lib/mongoid/extensions/array/accessors.rb

@@ -0,0 +1,17 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Array #:nodoc:
+      module Accessors #:nodoc:
+        # If the attributes already exists in the array then they will be
+        # updated, otherwise they will be appended.
+        def update(attributes)
+          delete_if { |e| attributes["_id"] && (e["_id"] == attributes["_id"]) }
+          self.<< attributes
+        end
+
+        alias :merge! :update
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/array/aliasing.rb

@@ -0,0 +1,4 @@
+# encoding: utf-8
+class Array
+  alias :count :size
+end

data/lib/mongoid/extensions/array/assimilation.rb

@@ -0,0 +1,26 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Array #:nodoc:
+      module Assimilation #:nodoc:
+        # Introduces a child object into the +Document+ object graph. This will
+        # set up the relationships between the parent and child and update the
+        # attributes of the parent +Document+.
+        #
+        # Options:
+        #
+        # parent: The +Document+ to assimilate into.
+        # options: The association +Options+ for the child.
+        #
+        # Example:
+        #
+        # <tt>[{:street => "Queen St."}, {:street => "King St."}].assimilate(person, options)</tt>
+        #
+        # Returns: The child +Document+.
+        def assimilate(parent, options)
+          each { |child| child.assimilate(parent, options) }
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/array/conversions.rb

@@ -0,0 +1,27 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Array #:nodoc:
+      # This module converts arrays into mongoid related objects.
+      module Conversions #:nodoc:
+        extend ActiveSupport::Concern
+
+        module InstanceMethods #:nodoc:
+          # Converts this array into an array of hashes.
+          def mongoidize
+            collect { |obj| obj.attributes }
+          end
+        end
+
+        module ClassMethods #:nodoc:
+          def get(value)
+            value
+          end
+          def set(value)
+            value
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/array/parentization.rb

@@ -0,0 +1,13 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Array #:nodoc:
+      module Parentization #:nodoc:
+        # Adds the parent document to each element in the array.
+        def parentize(parent, association_name)
+          each { |obj| obj.parentize(parent, association_name) }
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/big_decimal/conversions.rb

@@ -0,0 +1,19 @@
+# encoding: utf-8
+require "bigdecimal"
+
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module BigDecimal #:nodoc:
+      module Conversions #:nodoc:
+        # Get the string as a +BigDecimal+
+        def get(value)
+          ::BigDecimal.new(value)
+        end
+        # Set the value in the hash as a string.
+        def set(value)
+          value.to_s
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/binary/conversions.rb

@@ -0,0 +1,17 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Binary #:nodoc:
+      module Conversions #:nodoc:
+        # Get the value from the db hash.
+        def get(value)
+          value
+        end
+        # Set the value in the db hash.
+        def set(value)
+          value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/boolean/conversions.rb

@@ -0,0 +1,16 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Boolean #:nodoc:
+      module Conversions #:nodoc:
+        def set(value)
+          val = value.to_s
+          val == "true" || val == "1"
+        end
+        def get(value)
+          value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/date/conversions.rb

@@ -0,0 +1,15 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Date #:nodoc:
+      module Conversions #:nodoc:
+        def set(value)
+          value.to_date.at_midnight.to_time unless value.blank?
+        end
+        def get(value)
+          value ? value.getlocal.to_date : value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/datetime/conversions.rb

@@ -0,0 +1,17 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module DateTime #:nodoc:
+      module Conversions #:nodoc:
+        def set(value)
+          return nil if value.blank?
+          ::DateTime.parse(value.to_s).utc
+        end
+        def get(value)
+          return nil if value.blank?
+          ::Time.zone ? ::Time.parse(value.to_s).getlocal.to_datetime : value.to_datetime
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/float/conversions.rb

@@ -0,0 +1,16 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Float #:nodoc:
+      module Conversions #:nodoc:
+        def set(value)
+          return nil if value.blank?
+          value =~ /\d/ ? value.to_f : value
+        end
+        def get(value)
+          value
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/hash/accessors.rb

@@ -0,0 +1,38 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Hash #:nodoc:
+      module Accessors #:nodoc:
+
+        # Remove a set of attributes from a hash. If the attributes are
+        # contained in an array it will remove it from the array, otherwise it
+        # will delete the child attribute completely.
+        def remove(key, attrs)
+          elements = self[key]
+          if elements
+            key.singular? ? self[key] = nil : elements.delete(attrs)
+          end
+        end
+
+        # Inserts new attributes into the hash. If the elements are present in
+        # the hash it will update them, otherwise it will add the new
+        # attributes into the hash as either a child hash or child array of
+        # hashes.
+        def insert(key, attrs)
+          elements = self[key]
+          if elements
+            elements.merge!(attrs)
+          else
+            self[key] = key.singular? ? attrs : [attrs]
+          end
+        end
+
+        # If a _type key exists in the hash, return the class for the value.
+        def klass
+          class_name = self["_type"]
+          class_name ? class_name.constantize : nil
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/extensions/hash/assimilation.rb

@@ -0,0 +1,30 @@
+# encoding: utf-8
+module Mongoid #:nodoc:
+  module Extensions #:nodoc:
+    module Hash #:nodoc:
+      module Assimilation #:nodoc:
+        # Introduces a child object into the +Document+ object graph. This will
+        # set up the relationships between the parent and child and update the
+        # attributes of the parent +Document+.
+        #
+        # Options:
+        #
+        # parent: The +Document+ to assimilate into.
+        # options: The association +Options+ for the child.
+        #
+        # Example:
+        #
+        # <tt>{:first_name => "Hank", :last_name => "Moody"}.assimilate(name, options)</tt>
+        #
+        # Returns: The child +Document+.
+        def assimilate(parent, options, type = nil)
+          klass = type ? type : options.klass
+          child = klass.instantiate(:_parent => parent)
+          child.write_attributes(self.merge("_type" => klass.name))
+          child.identify
+          child.assimilate(parent, options)
+        end
+      end
+    end
+  end
+end