module MongoModel
  module DocumentParent
    extend ActiveSupport::Concern
    
    def parent_document
      if @_parent_document.respond_to?(:call)
        case @_parent_document.arity
        when 0 then @_parent_document.call
        else        @_parent_document.call(self)
        end
      else
        @_parent_document
      end
    end
    
    def parent_document=(doc)
      @_parent_document = doc
    end
  end
end