lib/builder/mongo_helpers.rb in sinatra_resource-0.1.0 vs lib/builder/mongo_helpers.rb in sinatra_resource-0.2.0

@@ -1,16 +1,33 @@
 module SinatraResource
   
   class Builder
 
     module MongoHelpers
+
+
+      # Make sure that +parent+ document is related to the +child+ document
+      # by way of +association+. If not, return 404 Not Found.
+      #
+      # @param [MongoMapper::Document] parent
+      #
+      # @param [Symbol] association
+      #
+      # @param [String] child_id
+      #
+      # @return [MongoMapper::Document]
+      def check_related?(parent, association, child_id)
+        unless parent.send(association).find { |x| x.id == child_id }
+          error 404, convert(body_for(:not_found))
+        end
+      end
       
       # Create a document from params. If not valid, returns 400.
       #
       # @return [MongoMapper::Document]
-      def create_document!
-        document = config[:model].new(params)
+      def create_document!(model)
+        document = model.new(params)
         unless document.valid?
           error 400, convert(body_for(:invalid_document, document))
         end
         unless document.save
           error 400, convert(body_for(:internal_server_error))
@@ -21,23 +38,23 @@
       # Delete a document with +id+.
       #
       # @param [String] id
       #
       # @return [MongoMapper::Document]
-      def delete_document!(id)
-        document = find_document!(id)
+      def delete_document!(model, id)
+        document = find_document!(model, id)
         document.destroy
         document
       end
 
       # Find a document with +id+. If not found, returns 404.
       #
       # @param [String] id
       #
       # @return [MongoMapper::Document]
-      def find_document!(id)
-        document = config[:model].find_by_id(id)
+      def find_document!(model, id)
+        document = model.find_by_id(id)
         unless document
           error 404, convert(body_for(:not_found))
         end
         document
       end
@@ -46,18 +63,52 @@
       #
       # @param [Class] model
       #   a class that includes MongoMapper::Document
       #
       # @return [Array<MongoMapper::Document>]
-      def find_documents!
-        config[:model].find(:all)
+      def find_documents!(model)
+        model.find(:all)
       end
+      
+      # Delegates to application, who should use custom logic to related
+      # +parent+ and +child+.
+      #
+      # @param [MongoMapper::Document] parent
+      #
+      # @param [MongoMapper::Document] child
+      #
+      # @param [Hash] resource_config
+      #
+      # @return [MongoMapper::Document] child document
+      def make_related(parent, child, resource_config)
+        proc = resource_config[:relation][:create]
+        proc.call(parent, child) if proc
+        child
+      end
+      
+      # Select only the +children+ that are related to the +parent+ by
+      # way of the +association+.
+      #
+      # @param [MongoMapper::Document] parent
+      #
+      # @param [Symbol] association
+      #
+      # @param [Array<MongoMapper::Document>] children
+      #
+      # @return [MongoMapper::Document]
+      def select_related(parent, association, children)
+        children.select do |child|
+          parent.send(association).find { |x| x.id == child.id }
+        end
+        # TODO: this has O^2 complexity because of the nesting.
+        # I think it is reducible to O.
+      end
 
       # Update a document with +id+ from params. If not valid, returns 400.
       #
       # @return [MongoMapper::Document]
-      def update_document!(id)
-        document = config[:model].update(id, params)
+      def update_document!(model, id)
+        document = model.update(id, params)
         unless document.valid?
           error 400, convert(body_for(:invalid_document, document))
         end
         document
       end