mongoid/mongoid

View on GitHub
lib/mongoid/relations/bindings/referenced/many_to_many.rb

Summary

Maintainability
A
25 mins
Test Coverage
# encoding: utf-8
module Mongoid
  module Relations
    module Bindings
      module Referenced

        # Binding class for all references_and_referenced_in_many relations.
        class ManyToMany < Binding

          # Binds a single document with the inverse relation. Used
          # specifically when appending to the proxy.
          #
          # @example Bind one document.
          #   person.preferences.bind_one(preference)
          #
          # @param [ Document ] doc The single document to bind.
          #
          # @since 2.0.0.rc.1
          def bind_one(doc)
            binding do
              inverse_keys = doc.you_must(metadata.inverse_foreign_key)
              if inverse_keys
                unless inverse_keys.include?(inverse_record_id(doc))
                  inverse_keys.push(inverse_record_id(doc))
                end
                doc.reset_relation_criteria(metadata.inverse)
              end
              base.synced[metadata.foreign_key] = true
              doc.synced[metadata.inverse_foreign_key] = true
            end
          end

          # Unbind a single document.
          #
          # @example Unbind the document.
          #   person.preferences.unbind_one(document)
          #
          # @since 2.0.0.rc.1
          def unbind_one(doc)
            binding do
              base.send(metadata.foreign_key).delete_one(record_id(doc))
              inverse_keys = doc.you_must(metadata.inverse_foreign_key)
              if inverse_keys
                inverse_keys.delete_one(inverse_record_id(doc))
                doc.reset_relation_criteria(metadata.inverse)
              end
              base.synced[metadata.foreign_key] = true
              doc.synced[metadata.inverse_foreign_key] = true
            end
          end

          # Find the inverse id referenced by inverse_keys
          def inverse_record_id(doc)
            inverse_metadata = determine_inverse_metadata(doc)
            if inverse_metadata
              base.__send__(inverse_metadata.primary_key)
            else
              base._id
            end
          end

          def determine_inverse_metadata(doc)
            doc.relations[base.class.name.demodulize.underscore.pluralize]
          end
        end
      end
    end
  end
end