module Pupa
  class Processor
    # A proxy class to save plain old Ruby objects to MongoDB.
    class Persistence
      # @param [Object] object an object
      def initialize(object)
        @object = object
      end

      # Finds a document matching the selection criteria.
      #
      # The selection criteria *must* set a `_type` key in order to determine
      # the collection to query.
      #
      # @param [Hash] selector the selection criteria
      # @return [Hash,nil] the matched document, or nil
      # @raises [Pupa::Errors::TooManyMatches] if multiple documents are found
      def self.find(selector)
        collection_name = collection_name_from_class_name(selector[:_type].camelize)
        query = Pupa.session[collection_name].find(selector)
        case query.count
        when 0
          nil
        when 1
          query.first
        else
          raise Errors::TooManyMatches, "selector matches multiple documents during find: #{collection_name} #{JSON.dump(selector)}"
        end
      end

      # Inserts or replaces a document in MongoDB.
      #
      # @return [Array] whether the object was inserted and the object's database ID
      # @raises [Pupa::Errors::TooManyMatches] if multiple documents would be updated
      def save
        selector = @object.fingerprint
        query = collection.find(selector)

        # Run query before callbacks to avoid e.g. timestamps in the selector.
        case query.count
        when 0
          @object.run_callbacks(:save) do
            @object.run_callbacks(:create) do
              collection.insert(@object.to_h(persist: true))
              [true, @object._id.to_s]
            end
          end
        when 1
          # Make the document available to the callbacks.
          # @see https://github.com/opennorth/pupa-ruby/issues/17
          @object.document = query.first
          @object.run_callbacks(:save) do
            query.update(@object.to_h(persist: true).except(:_id))
            [false, @object.document['_id'].to_s]
          end
        else
          raise Errors::TooManyMatches, "selector matches multiple documents during save: #{collection_name} #{JSON.dump(selector)} for #{@object._id}"
        end
      end

    private

      # Returns the name of the collection in which to save the object.
      #
      # @return [String] the name of the object's class
      def self.collection_name_from_class_name(class_name)
        class_name.demodulize.underscore.pluralize.to_sym
      end

      # Returns the name of the collection in which to save the object.
      #
      # @return [String] the name of the collection in which to save the object
      def collection_name
        self.class.collection_name_from_class_name(@object.class.to_s)
      end

      # Returns the collection in which to save the object.
      #
      # @return [Moped::Collection] the collection in which to save the object
      def collection
        Pupa.session[collection_name]
      end
    end
  end
end