mongoid 6.4.8 → 7.0.0.beta

data/lib/mongoid/association/referenced/has_and_belongs_to_many/proxy.rb

@@ -0,0 +1,310 @@
+# encoding: utf-8
+module Mongoid
+  module Association
+    module Referenced
+      class HasAndBelongsToMany
+
+        # This class defines the behaviour for all relations that are a
+        # many-to-many between documents in different collections.
+        class Proxy < Referenced::HasMany::Proxy
+
+          # Appends a document or array of documents to the relation. Will set
+          # the parent and update the index in the process.
+          #
+          # @example Append a document.
+          #   person.posts << post
+          #
+          # @example Push a document.
+          #   person.posts.push(post)
+          #
+          # @example Concat with other documents.
+          #   person.posts.concat([ post_one, post_two ])
+          #
+          # @param [ Document, Array<Document> ] args Any number of documents.
+          #
+          # @return [ Array<Document> ] The loaded docs.
+          #
+          # @since 2.0.0.beta.1
+          def <<(*args)
+            docs = args.flatten
+            return concat(docs) if docs.size > 1
+            if doc = docs.first
+              append(doc)
+              _base.add_to_set(foreign_key => doc.send(_association.primary_key))
+              if child_persistable?(doc)
+                doc.save
+              end
+            end
+            unsynced(_base, foreign_key) and self
+          end
+
+          alias :push :<<
+
+          # Appends an array of documents to the relation. Performs a batch
+          # insert of the documents instead of persisting one at a time.
+          #
+          # @example Concat with other documents.
+          #   person.posts.concat([ post_one, post_two ])
+          #
+          # @param [ Array<Document> ] documents The docs to add.
+          #
+          # @return [ Array<Document> ] The documents.
+          #
+          # @since 2.4.0
+          def concat(documents)
+            ids, docs, inserts = {}, [], []
+            documents.each do |doc|
+              next unless doc
+              append(doc)
+              if persistable? || _creating?
+                ids[doc._id] = true
+                save_or_delay(doc, docs, inserts)
+              else
+                existing = _base.send(foreign_key)
+                unless existing.include?(doc._id)
+                  existing.push(doc._id) and unsynced(_base, foreign_key)
+                end
+              end
+            end
+            if persistable? || _creating?
+              _base.push(foreign_key => ids.keys)
+            end
+            persist_delayed(docs, inserts)
+            self
+          end
+
+          # Build a new document from the attributes and append it to this
+          # relation without saving.
+          #
+          # @example Build a new document on the relation.
+          #   person.posts.build(:title => "A new post")
+          #
+          # @param [ Hash ] attributes The attributes of the new document.
+          # @param [ Class ] type The optional subclass to build.
+          #
+          # @return [ Document ] The new document.
+          #
+          # @since 2.0.0.beta.1
+          def build(attributes = {}, type = nil)
+            doc = Factory.build(type || klass, attributes)
+            _base.send(foreign_key).push(doc._id)
+            append(doc)
+            doc.apply_post_processed_defaults
+            unsynced(doc, inverse_foreign_key)
+            yield(doc) if block_given?
+            doc
+          end
+
+          alias :new :build
+
+          # Delete the document from the relation. This will set the foreign key
+          # on the document to nil. If the dependent options on the relation are
+          # :delete or :destroy the appropriate removal will occur.
+          #
+          # @example Delete the document.
+          #   person.posts.delete(post)
+          #
+          # @param [ Document ] document The document to remove.
+          #
+          # @return [ Document ] The matching document.
+          #
+          # @since 2.1.0
+          def delete(document)
+            doc = super
+            if doc && persistable?
+              _base.pull(foreign_key => doc.send(_association.primary_key))
+              _target._unloaded = criteria
+              unsynced(_base, foreign_key)
+            end
+            doc
+          end
+
+          # Removes all associations between the base document and the target
+          # documents by deleting the foreign keys and the references, orphaning
+          # the target documents in the process.
+          #
+          # @example Nullify the relation.
+          #   person.preferences.nullify
+          #
+          # @param [ Array<Document> ] replacement The replacement documents.
+          #
+          # @since 2.0.0.rc.1
+          def nullify(replacement = [])
+            _target.each do |doc|
+              execute_callback :before_remove, doc
+            end
+            unless _association.forced_nil_inverse?
+              if replacement
+                objects_to_clear = _base.send(foreign_key) - replacement.collect(&:id)
+                criteria(objects_to_clear).pull(inverse_foreign_key => _base._id)
+              else
+                criteria.pull(inverse_foreign_key => _base._id)
+              end
+            end
+            if persistable?
+              _base.set(foreign_key => _base.send(foreign_key).clear)
+            end
+            after_remove_error = nil
+            many_to_many = _target.clear do |doc|
+              unbind_one(doc)
+              unless _association.forced_nil_inverse?
+                doc.changed_attributes.delete(inverse_foreign_key)
+              end
+              begin
+                execute_callback :after_remove, doc
+              rescue => e
+                after_remove_error = e
+              end
+            end
+            raise after_remove_error if after_remove_error
+            many_to_many
+          end
+
+          alias :nullify_all :nullify
+          alias :clear :nullify
+          alias :purge :nullify
+
+          # Substitutes the supplied target documents for the existing documents
+          # in the relation. If the new target is nil, perform the necessary
+          # deletion.
+          #
+          # @example Replace the relation.
+          # person.preferences.substitute([ new_post ])
+          #
+          # @param [ Array<Document> ] replacement The replacement target.
+          #
+          # @return [ Many ] The relation.
+          #
+          # @since 2.0.0.rc.1
+          def substitute(replacement)
+            purge(replacement)
+            unless replacement.blank?
+              push(replacement.compact.uniq)
+            else
+              reset_unloaded
+            end
+            self
+          end
+
+          # Get a criteria for the documents without the default scoping
+          # applied.
+          #
+          # @example Get the unscoped criteria.
+          #   person.preferences.unscoped
+          #
+          # @return [ Criteria ] The unscoped criteria.
+          #
+          # @since 2.4.0
+          def unscoped
+            klass.unscoped.any_in(_id: _base.send(foreign_key))
+          end
+
+          private
+
+          # Appends the document to the target array, updating the index on the
+          # document at the same time.
+          #
+          # @example Append the document to the relation.
+          #   relation.append(document)
+          #
+          # @param [ Document ] document The document to append to the target.
+          #
+          # @since 2.0.0.rc.1
+          def append(document)
+            execute_callback :before_add, document
+            _target.push(document)
+            characterize_one(document)
+            bind_one(document)
+            execute_callback :after_add, document
+          end
+
+          # Instantiate the binding associated with this relation.
+          #
+          # @example Get the binding.
+          #   relation.binding([ address ])
+          #
+          # @return [ Binding ] The binding.
+          #
+          # @since 2.0.0.rc.1
+          def binding
+            HasAndBelongsToMany::Binding.new(_base, _target, _association)
+          end
+
+          # Determine if the child document should be persisted.
+          #
+          # @api private
+          #
+          # @example Is the child persistable?
+          #   relation.child_persistable?(doc)
+          #
+          # @param [ Document ] doc The document.
+          #
+          # @return [ true, false ] If the document can be persisted.
+          #
+          # @since 3.0.20
+          def child_persistable?(doc)
+            (persistable? || _creating?) &&
+                !(doc.persisted? && _association.forced_nil_inverse?)
+          end
+
+          # Returns the criteria object for the target class with its documents set
+          # to target.
+          #
+          # @example Get a criteria for the relation.
+          #   relation.criteria
+          #
+          # @return [ Criteria ] A new criteria.
+          def criteria(id_list = nil)
+            _association.criteria(_base, id_list)
+          end
+
+          # Flag the base as unsynced with respect to the foreign key.
+          #
+          # @api private
+          #
+          # @example Flag as unsynced.
+          #   relation.unsynced(doc, :preference_ids)
+          #
+          # @param [ Document ] doc The document to flag.
+          # @param [ Symbol ] key The key to flag on the document.
+          #
+          # @return [ true ] true.
+          #
+          # @since 3.0.0
+          def unsynced(doc, key)
+            doc._synced[key] = false
+            true
+          end
+
+          class << self
+
+            # Get the Eager object for this type of association.
+            #
+            # @example Get the eager loader object
+            #
+            # @param [ Association ] association The association object.
+            # @param [ Array<Document> ] docs The array of documents.
+            #
+            # @since 7.0
+            def eager_loader(association, docs)
+              Eager.new(association, docs)
+            end
+
+            # Returns true if the relation is an embedded one. In this case
+            # always false.
+            #
+            # @example Is this relation embedded?
+            #   Referenced::ManyToMany.embedded?
+            #
+            # @return [ false ] Always false.
+            #
+            # @since 2.0.0.rc.1
+            def embedded?
+              false
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongoid/association/referenced/has_many.rb

@@ -0,0 +1,273 @@
+require 'mongoid/association/referenced/has_many/binding'
+require 'mongoid/association/referenced/has_many/buildable'
+require 'mongoid/association/referenced/has_many/proxy'
+require 'mongoid/association/referenced/has_many/enumerable'
+require 'mongoid/association/referenced/has_many/eager'
+
+module Mongoid
+  module Association
+    module Referenced
+
+      # The has_many association.
+      #
+      # @since 7.0
+      class HasMany
+        include Relatable
+        include Buildable
+
+        # The options available for this type of association, in addition to the
+        # common ones.
+        #
+        # @return [ Array<Symbol> ] The extra valid options.
+        #
+        # @since 7.0
+        ASSOCIATION_OPTIONS = [
+            :after_add,
+            :after_remove,
+            :as,
+            :autosave,
+            :before_add,
+            :before_remove,
+            :dependent,
+            :foreign_key,
+            :order,
+            :primary_key
+        ].freeze
+
+        # The complete list of valid options for this association, including
+        # the shared ones.
+        #
+        # @return [ Array<Symbol> ] The valid options.
+        #
+        # @since 7.0
+        VALID_OPTIONS = (ASSOCIATION_OPTIONS + SHARED_OPTIONS).freeze
+
+        # The default foreign key suffix.
+        #
+        # @return [ String ] '_id'
+        #
+        # @since 7.0
+        FOREIGN_KEY_SUFFIX = '_id'.freeze
+
+        # The list of association complements.
+        #
+        # @return [ Array<Association> ] The association complements.
+        #
+        # @since 7.0
+        def relation_complements
+          @relation_complements ||= [ Referenced::BelongsTo ].freeze
+        end
+
+        # Setup the instance methods, fields, etc. on the association owning class.
+        #
+        # @return [ self ]
+        #
+        # @since 7.0
+        def setup!
+          setup_instance_methods!
+          self
+        end
+
+        # Setup the instance methods on the class having this association type.
+        #
+        # @return [ self ]
+        #
+        # @since 7.0
+        def setup_instance_methods!
+          define_getter!
+          define_ids_getter!
+          define_setter!
+          define_ids_setter!
+          define_existence_check!
+          define_autosaver!
+          polymorph!
+          define_dependency!
+          @owner_class.validates_associated(name) if validate?
+          self
+        end
+
+
+        # Get the foreign key field on the inverse for saving the association reference.
+        #
+        # @return [ String ] The foreign key field on the inverse for saving the
+        #   association reference.
+        #
+        # @since 7.0
+        def foreign_key
+          @foreign_key ||= @options[:foreign_key] ? @options[:foreign_key].to_s :
+                             default_foreign_key_field
+        end
+
+        # Is this association type embedded?
+        #
+        # @return [ false ] Always false.
+        #
+        # @since 7.0
+        def embedded?; false; end
+
+        # The default for validation the association object.
+        #
+        # @return [ true ] Always true.
+        #
+        # @since 7.0
+        def validation_default; true; end
+
+        # Does this association type store the foreign key?
+        #
+        # @return [ true ] Always true.
+        #
+        # @since 7.0
+        def stores_foreign_key?; false; end
+
+        # Get the relation proxy class for this association type.
+        #
+        # @return [ Association::HasMany::Proxy ] The proxy class.
+        #
+        # @since 7.0
+        def relation
+          Proxy
+        end
+
+        # The criteria used for querying this relation.
+        #
+        # @return [ Mongoid::Criteria ] The criteria used for querying this relation.
+        #
+        # @since 7.0
+        def criteria(base)
+          query_criteria(base.send(primary_key), base)
+        end
+
+        # The type of this association if it's polymorphic.
+        #
+        # @note Only relevant for polymorphic relations.
+        #
+        # @return [ String, nil ] The type field.
+        #
+        # @since 7.0
+        def type
+          @type ||= "#{as}_type" if polymorphic?
+        end
+
+        # Add polymorphic query criteria to a Criteria object, if this association is
+        #  polymorphic.
+        #
+        # @param [ Mongoid::Criteria ] criteria The criteria object to add to.
+        # @param [ Class ] object_class The object class.
+        #
+        # @return [ Mongoid::Criteria ] The criteria object.
+        #
+        # @since 7.0
+        def add_polymorphic_criterion(criteria, object_class)
+          if polymorphic?
+            criteria.where(type => object_class.name)
+          else
+            criteria
+          end
+        end
+
+        # Is this association polymorphic?
+        #
+        # @return [ true, false ] Whether this association is polymorphic.
+        #
+        # @since 7.0
+        def polymorphic?
+          @polymorphic ||= !!as
+        end
+
+        # Whether trying to bind an object using this association should raise
+        # an error.
+        #
+        # @param [ Document ] doc The document to be bound.
+        #
+        # @return [ true, false ] Whether the document can be bound.
+        def bindable?(doc)
+          forced_nil_inverse? || (!!inverse && doc.fields.keys.include?(foreign_key))
+        end
+
+        # The nested builder object.
+        #
+        # @param [ Hash ] attributes The attributes to use to build the association object.
+        # @param [ Hash ] options The options for the association.
+        #
+        # @return [ Association::Nested::Many ] The Nested Builder object.
+        #
+        # @since 7.0
+        def nested_builder(attributes, options)
+          Nested::Many.new(self, attributes, options)
+        end
+
+        # Get the path calculator for the supplied document.
+        #
+        # @example Get the path calculator.
+        #   Proxy.path(document)
+        #
+        # @param [ Document ] document The document to calculate on.
+        #
+        # @return [ Root ] The root atomic path calculator.
+        #
+        # @since 2.1.0
+        def path(document)
+          Mongoid::Atomic::Paths::Root.new(document)
+        end
+
+        private
+
+        def default_foreign_key_field
+          @default_foreign_key_field ||= "#{inverse}#{FOREIGN_KEY_SUFFIX}"
+        end
+
+        def polymorphic_inverses(other)
+          [ as ]
+        end
+
+        def determine_inverses(other)
+          matches = (other || relation_class).relations.values.select do |rel|
+            relation_complements.include?(rel.class) &&
+                rel.relation_class_name == inverse_class_name
+
+          end
+          if matches.size > 1
+            return [ default_inverse.name ] if default_inverse
+            raise Errors::AmbiguousRelationship.new(relation_class, @owner_class, name, matches)
+          end
+          matches.collect { |m| m.name } unless matches.blank?
+        end
+
+        def default_primary_key
+          PRIMARY_KEY_DEFAULT
+        end
+
+        def query_criteria(object, base)
+          crit = klass.where(foreign_key => object)
+          crit = with_polymorphic_criterion(crit, base)
+          with_ordering(crit)
+        end
+
+        def with_polymorphic_criterion(criteria, base)
+          if polymorphic?
+            criteria.where(type => base.class.name)
+          else
+            criteria
+          end
+        end
+
+        def with_ordering(criteria)
+          if order
+            criteria.order_by(order)
+          else
+            criteria
+          end
+        end
+
+        def with_inverse_field_criterion(criteria)
+          inverse_association = inverse_association(klass)
+          if inverse_association.try(:inverse_of)
+            criteria.any_in(inverse_association.inverse_of => [name, nil])
+          else
+            criteria
+          end
+        end
+      end
+    end
+  end
+end