# encoding: utf-8
module Mongoid
  module Relations

    # This module contains the behaviour for auto-saving relations in
    # different collections.
    module AutoSave
      extend ActiveSupport::Concern

      included do
        class_attribute :autosaved_relations
        self.autosaved_relations = []
      end

      # Used to prevent infinite loops in associated autosaves.
      #
      # @example Is the document autosaved?
      #   document.autosaved?
      #
      # @return [ true, false ] Has the document already been autosaved?
      #
      # @since 3.0.0
      def autosaved?
        Threaded.autosaved?(self)
      end

      # Begin the associated autosave.
      #
      # @example Begin autosave.
      #   document.__autosaving__
      #
      # @since 3.1.3
      def __autosaving__
        Threaded.begin_autosave(self)
        yield
      ensure
        Threaded.exit_autosave(self)
      end

      # Check if there is changes for auto-saving
      #
      # @example Return true if there is changes on self or in
      #           autosaved relations.
      #   document.changed_for_autosave?
      #
      # @since 3.1.3
      def changed_for_autosave?
        new_record? || changed? || marked_for_destruction?
      end

      # Returns the relation, if it exists
      #
      # @since 3.1.3
      def relation_changed_for_autosave(metadata)
        ivar(metadata.name) if self.class.autosaved_relations.include?(metadata.name)
      end

      module ClassMethods

        # Set up the autosave behaviour for references many and references one
        # relations. When the option is set to true, these relations will get
        # saved automatically when the parent saved, if they are dirty.
        #
        # @example Set up autosave options.
        #   Person.autosave(metadata)
        #
        # @param [ Metadata ] metadata The relation metadata.
        #
        # @since 2.0.0.rc.1
        def autosave(metadata)
          if metadata.autosave? && autosavable?(metadata)
            autosaved_relations.push(metadata.name)
            set_callback :save, :after, unless: :autosaved? do |document|
              if before_callback_halted?
                self.before_callback_halted = false
              else
                __autosaving__ do
                  if document.changed_for_autosave? || relation = document.relation_changed_for_autosave(metadata)
                    relation = document.__send__(metadata.name) unless relation
                    (relation.do_or_do_not(:in_memory) || Array.wrap(relation)).each do |doc|
                      doc.save
                    end if relation
                  end
                end
              end
            end
          end
        end

        # Can the autosave be added?
        #
        # @example Can the autosave be added?
        #   Person.autosavable?(metadata)
        #
        # @param [ Metadata ] metadata The relation metadata.
        #
        # @return [ true, false ] If the autosave is able to be added.
        #
        # @since 3.0.0
        def autosavable?(metadata)
          !autosaved_relations.include?(metadata.name) && !metadata.embedded?
        end
      end
    end
  end
end