mongoid 1.2.14 → 1.9.0

data/lib/mongoid/attributes.rb

@@ -46,6 +46,7 @@ module Mongoid #:nodoc:
             send("#{key}=", value)
           end
         end
+        setup_modifications
       end
 
       # Read a value from the +Document+ attributes. If the value does not exist
@@ -60,7 +61,7 @@ module Mongoid #:nodoc:
       # <tt>person.read_attribute(:title)</tt>
       def read_attribute(name)
         access = name.to_s
-        fields[access].get(@attributes[access])
+        accessed(access, fields[access].get(@attributes[access]))
       end
 
       # Remove a value from the +Document+ attributes. If the value does not exist
@@ -74,7 +75,8 @@ module Mongoid #:nodoc:
       #
       # <tt>person.remove_attribute(:title)</tt>
       def remove_attribute(name)
-        @attributes.delete(name.to_s)
+        access = name.to_s
+        modify(access, @attributes.delete(access), nil)
       end
 
       # Returns the object type. This corresponds to the name of the class that
@@ -106,8 +108,8 @@ module Mongoid #:nodoc:
       # there is any.
       def write_attribute(name, value)
         access = name.to_s
-        @attributes[access] = fields[access].set(value)
-        notify unless id.blank?
+        modify(access, @attributes[access], fields[access].set(value))
+        notify if !id.blank? && new_record?
       end
 
       # Writes the supplied attributes +Hash+ to the +Document+. This will only
@@ -126,13 +128,24 @@ module Mongoid #:nodoc:
       # there is any.
       def write_attributes(attrs = nil)
         process(attrs || {})
-        identify if id.blank?
-        notify
+        identified = !id.blank?
+        if new_record? && !identified
+          identify; notify
+        end
       end
       alias :attributes= write_attributes
 
       protected
-      # Return true is dynamic field setting is enabled.
+      # apply default values to attributes - calling procs as required
+      def default_attributes
+        default_values = defaults
+        default_values.each_pair do |key, val|
+          default_values[key] = val.call if val.respond_to?(:call)
+        end
+        default_values || {}
+      end
+
+      # Return true if dynamic field setting is enabled.
       def set_allowed?(key)
         Mongoid.allow_dynamic_fields && !respond_to?("#{key}=")
       end
@@ -150,8 +163,10 @@ module Mongoid #:nodoc:
 
       # Return true if writing to the given field is allowed
       def write_allowed?(key)
-        return true unless fields[key.to_s]
-        fields[key.to_s].accessible?
+        name = key.to_s
+        existing = fields[name]
+        return true unless existing
+        existing.accessible?
       end
     end
 
@@ -163,8 +178,8 @@ module Mongoid #:nodoc:
       #
       #   class Person
       #     include Mongoid::Document
-      #     has_one :name
-      #     has_many :addresses
+      #     embeds_one :name
+      #     embeds_many :addresses
       #
       #     accepts_nested_attributes_for :name, :addresses
       #   end
@@ -176,7 +191,7 @@ module Mongoid #:nodoc:
             reject(attrs, options)
             association = send(name)
             if association
-              update(association, true)
+              observe(association, true)
               association.nested_build(attrs)
             else
               send("build_#{name}", attrs)

data/lib/mongoid/collection.rb

@@ -1,13 +1,12 @@
 # encoding: utf-8
 require "mongoid/collections/operations"
 require "mongoid/collections/cyclic_iterator"
-require "mongoid/collections/mimic"
 require "mongoid/collections/master"
 require "mongoid/collections/slaves"
 
 module Mongoid #:nodoc
+  # The Mongoid wrapper to the Mongo Ruby driver's collection object.
   class Collection
-    include Collections::Mimic
     attr_reader :counter, :name
 
     # All write operations should delegate to the master connection. These
@@ -16,7 +15,9 @@ module Mongoid #:nodoc
     # Example:
     #
     # <tt>collection.save({ :name => "Al" })</tt>
-    proxy(:master, Collections::Operations::PROXIED)
+    Collections::Operations::PROXIED.each do |name|
+      define_method(name) { |*args| master.send(name, *args) }
+    end
 
     # Determines where to send the next read query. If the slaves are not
     # defined then send to master. If the read counter is under the configured

data/lib/mongoid/collections.rb

@@ -0,0 +1,41 @@
+# encoding: utf-8
+module Mongoid #:nodoc
+  # The collections module is used for providing functionality around setting
+  # up and updating collections.
+  module Collections
+    extend ActiveSupport::Concern
+    included do
+      cattr_accessor :_collection, :collection_name
+      self.collection_name = self.name.collectionize
+      delegate :collection, :to => "self.class"
+    end
+
+    module ClassMethods #:nodoc:
+      # Returns the collection associated with this +Document+. If the
+      # document is embedded, there will be no collection associated
+      # with it.
+      #
+      # Returns: <tt>Mongo::Collection</tt>
+      def collection
+        raise Errors::InvalidCollection.new(self) if embedded?
+        self._collection || set_collection
+        add_indexes; self._collection
+      end
+
+      # Macro for setting the collection name to store in.
+      #
+      # Example:
+      #
+      # <tt>Person.store_in :populdation</tt>
+      def store_in(name)
+        self.collection_name = name.to_s
+        set_collection
+      end
+
+      protected
+      def set_collection
+        self._collection = Mongoid::Collection.new(self, self.collection_name)
+      end
+    end
+  end
+end

data/lib/mongoid/collections/master.rb

@@ -2,7 +2,6 @@
 module Mongoid #:nodoc:
   module Collections #:nodoc:
     class Master
-      include Mimic
 
       attr_reader :collection
 
@@ -12,7 +11,9 @@ module Mongoid #:nodoc:
       # Example:
       #
       # <tt>collection.save({ :name => "Al" })</tt>
-      proxy(:collection, Operations::ALL)
+      Operations::ALL.each do |name|
+        define_method(name) { |*args| collection.send(name, *args) }
+      end
 
       # Create the new database writer. Will create a collection from the
       # master database.

data/lib/mongoid/collections/slaves.rb

@@ -2,7 +2,6 @@
 module Mongoid #:nodoc:
   module Collections #:nodoc:
     class Slaves
-      include Mimic
 
       attr_reader :iterator
 
@@ -12,7 +11,9 @@ module Mongoid #:nodoc:
       # Example:
       #
       # <tt>collection.save({ :name => "Al" })</tt>
-      proxy(:collection, Operations::READ)
+      Operations::READ.each do |name|
+        define_method(name) { |*args| collection.send(name, *args) }
+      end
 
       # Is the collection of slaves empty or not?
       #

data/lib/mongoid/components.rb

@@ -1,25 +1,27 @@
 # encoding: utf-8
 module Mongoid #:nodoc
   module Components #:nodoc
-    def self.included(base)
-      base.class_eval do
-        # All modules that a +Document+ is composed of are defined in this
-        # module, to keep the document class from getting too cluttered.
-        include Mongoid::Associations
-        include Mongoid::Attributes
-        include Mongoid::Callbacks
-        include Mongoid::Commands
-        include Mongoid::Extras
-        include Mongoid::Fields
-        include Mongoid::Indexes
-        include Mongoid::Matchers
-        include Mongoid::Memoization
-        include Mongoid::State
-        include Observable
-        include Validatable
-        extend Mongoid::Finders
-        extend Mongoid::NamedScope
-      end
+    extend ActiveSupport::Concern
+    included do
+      # All modules that a +Document+ is composed of are defined in this
+      # module, to keep the document class from getting too cluttered.
+      include Mongoid::Associations
+      include Mongoid::Attributes
+      include Mongoid::Callbacks
+      include Mongoid::Collections
+      include Mongoid::Dirty
+      include Mongoid::Extras
+      include Mongoid::Fields
+      include Mongoid::Indexes
+      include Mongoid::Matchers
+      include Mongoid::Memoization
+      include Mongoid::Observable
+      include Mongoid::Paths
+      include Mongoid::Persistence
+      include Mongoid::State
+      include Validatable
+      extend Mongoid::Finders
+      extend Mongoid::NamedScope
     end
   end
 end

data/lib/mongoid/concern.rb

@@ -0,0 +1,31 @@
+unless defined?(ActiveSupport::Concern)
+  module ActiveSupport
+    module Concern
+      def self.extended(base)
+        base.instance_variable_set("@_dependencies", [])
+      end
+
+      def append_features(base)
+        if base.instance_variable_defined?("@_dependencies")
+          base.instance_variable_get("@_dependencies") << self
+          return false
+        else
+          return false if base < self
+          @_dependencies.each { |dep| base.send(:include, dep) }
+          super
+          base.extend const_get("ClassMethods") if const_defined?("ClassMethods")
+          base.send :include, const_get("InstanceMethods") if const_defined?("InstanceMethods")
+          base.class_eval(&@_included_block) if instance_variable_defined?("@_included_block")
+        end
+      end
+
+      def included(base = nil, &block)
+        if base.nil?
+          @_included_block = block
+        else
+          super
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/config.rb

@@ -1,4 +1,6 @@
 # encoding: utf-8
+require "uri"
+
 module Mongoid #:nodoc
   class Config #:nodoc
     include Singleton
@@ -8,23 +10,44 @@ module Mongoid #:nodoc
       :reconnect_time,
       :parameterize_keys,
       :persist_in_safe_mode,
-      :persist_types,
       :raise_not_found_error,
       :use_object_ids
 
     # Defaults the configuration options to true.
     def initialize
-      @allow_dynamic_fields = true
-      @parameterize_keys = true
-      @persist_in_safe_mode = true
-      @persist_types = true
-      @raise_not_found_error = true
-      @reconnect_time = 3
-      @use_object_ids = false
+      reset
+    end
+
+    # Sets whether the times returned from the database are in UTC or local time.
+    # If you omit this setting, then times will be returned in
+    # the local time zone.
+    #
+    # Example:
+    #
+    # <tt>Config.use_utc = true</tt>
+    #
+    # Returns:
+    #
+    # A boolean
+    def use_utc=(value)
+      @use_utc = value || false
     end
 
-    # Sets the Mongo::DB master database to be used. If the object trying to me
-    # set is not a valid +Mongo::DB+, then an error will be raise.
+    # Returns whether times are return from the database in UTC. If
+    # this setting is false, then times will be returned in the local time zone.
+    #
+    # Example:
+    #
+    # <tt>Config.use_utc</tt>
+    #
+    # Returns:
+    #
+    # A boolean
+    attr_reader :use_utc
+    alias_method :use_utc?, :use_utc
+
+    # Sets the Mongo::DB master database to be used. If the object trying to be
+    # set is not a valid +Mongo::DB+, then an error will be raised.
     #
     # Example:
     #
@@ -34,7 +57,7 @@ module Mongoid #:nodoc
     #
     # The Master DB instance.
     def master=(db)
-      raise Errors::InvalidDatabase.new(db) unless db.kind_of?(Mongo::DB)
+      check_database!(db)
       @master = db
     end
 
@@ -66,7 +89,9 @@ module Mongoid #:nodoc
     #
     # The slaves DB instances.
     def slaves=(dbs)
-      dbs.each { |db| raise Errors::InvalidDatabase.new(db) unless db.kind_of?(Mongo::DB) }
+      dbs.each do |db|
+        check_database!(db)
+      end
       @slaves = dbs
     end
 
@@ -82,5 +107,85 @@ module Mongoid #:nodoc
     def slaves
       @slaves
     end
+
+    # Confiure mongoid from a hash that was usually parsed out of yml.
+    #
+    # Example:
+    #
+    # <tt>Mongoid::Config.instance.from_hash({})</tt>
+    def from_hash(settings)
+      _master(settings)
+      _slaves(settings)
+      settings.except("database").each_pair do |name, value|
+        send("#{name}=", value) if respond_to?(name)
+      end
+    end
+
+    # Reset the configuration options to the defaults.
+    #
+    # Example:
+    #
+    # <tt>config.reset</tt>
+    def reset
+      @allow_dynamic_fields = true
+      @parameterize_keys = true
+      @persist_in_safe_mode = true
+      @raise_not_found_error = true
+      @reconnect_time = 3
+      @use_object_ids = false
+      @time_zone = nil
+    end
+
+    protected
+
+    # Check if the database is valid and the correct version.
+    #
+    # Example:
+    #
+    # <tt>config.check_database!</tt>
+    def check_database!(database)
+      raise Errors::InvalidDatabase.new(database) unless database.kind_of?(Mongo::DB)
+      version = database.connection.server_version
+      raise Errors::UnsupportedVersion.new(version) if version < Mongoid::MONGODB_VERSION
+    end
+
+    # Get a Rails logger or stdout logger.
+    #
+    # Example:
+    #
+    # <tt>config.logger</tt>
+    def logger
+      defined?(Rails) ? Rails.logger : Logger.new($stdout)
+    end
+
+    # Get a master database from settings.
+    #
+    # Example:
+    #
+    # <tt>config._master({}, "test")</tt>
+    def _master(settings)
+      name = settings["database"]
+      host = settings.delete("host") || "localhost"
+      port = settings.delete("port") || 27017
+      self.master = Mongo::Connection.new(host, port, :logger => logger).db(name)
+    end
+
+    # Get a bunch-o-slaves from settings and names.
+    #
+    # Example:
+    #
+    # <tt>config._slaves({}, "test")</tt>
+    def _slaves(settings)
+      name = settings["database"]
+      self.slaves = []
+      slaves = settings.delete("slaves")
+      slaves.to_a.each do |slave|
+        self.slaves << Mongo::Connection.new(
+          slave["host"],
+          slave["port"],
+          :slave_ok => true
+        ).db(name)
+      end
+    end
   end
 end

data/lib/mongoid/contexts.rb

@@ -15,7 +15,7 @@ module Mongoid
     #
     # <tt>Contexts.context_for(criteria)</tt>
     def self.context_for(criteria)
-      if criteria.klass.embedded
+      if criteria.klass.embedded?
         return Contexts::Enumerable.new(criteria)
       end
       Contexts::Mongo.new(criteria)