lib/couchrest/model/designs.rb in couchrest_model-1.1.2 vs lib/couchrest/model/designs.rb in couchrest_model-1.2.0.beta

@@ -17,21 +17,38 @@
     #   end
     #
     module Designs
       extend ActiveSupport::Concern
 
+
       module ClassMethods
 
         # Add views and other design document features
         # to the current model.
-        def design(*args, &block)
-          mapper = DesignMapper.new(self)
-          mapper.create_view_method(:all)
+        def design(prefix = nil, &block)
 
+          # Store ourselves a copy of this design spec incase any other model inherits.
+          (@_design_blocks ||= [ ]) << {:args => [prefix], :block => block}
+
+          mapper = DesignMapper.new(self, prefix)
           mapper.instance_eval(&block) if block_given?
+
+          # Create an 'all' view if no prefix and one has not been defined already
+          mapper.view(:all) if prefix.nil? and !mapper.design_doc.has_view?(:all)
         end
 
+        def inherited(model)
+          super
+
+          # Go through our design blocks and re-implement them in the child.
+          unless @_design_blocks.nil?
+            @_design_blocks.each do |row|
+              model.design(*row[:args], &row[:block])
+            end
+          end
+        end
+
         # Override the default page pagination value:
         #
         #   class Person < CouchRest::Model::Base
         #     paginates_per 10
         #   end
@@ -45,44 +62,80 @@
         # Returns 25 unless explicitly overridden via <tt>paginates_per</tt>
         def default_per_page
           @_default_per_page || 25
         end
 
+        def design_docs
+          @_design_docs ||= []
+        end
+
       end
 
-      # 
+      # Map method calls defined in a design block to actions
+      # in the Design Document.
       class DesignMapper
 
-        attr_accessor :model
+        # Basic mapper attributes
+        attr_accessor :model, :method, :prefix
 
-        def initialize(model)
-          self.model = model
+        # Temporary variable storing the design doc
+        attr_accessor :design_doc
+
+        def initialize(model, prefix = nil)
+          self.model  = model
+          self.prefix = prefix
+          self.method = Design.method_name(prefix)
+
+          create_model_design_doc_reader
+          self.design_doc = model.send(method) || assign_model_design_doc
         end
 
-        # Define a view and generate a method that will provide a new 
-        # View instance when requested.
+        def disable_auto_update
+          design_doc.auto_update = false
+        end
+
+        def enable_auto_update
+          design_doc.auto_update = true
+        end
+
+        # Add the specified view to the design doc the definition was made in
+        # and create quick access methods in the model.
         def view(name, opts = {})
-          View.create(model, name, opts)
-          create_view_method(name)
+          design_doc.create_view(name, opts)
         end
 
         # Really simple design function that allows a filter
         # to be added. Filters are simple functions used when listening
         # to the _changes feed.
         #
         # No methods are created here, the design is simply updated.
         # See the CouchDB API for more information on how to use this.
         def filter(name, function)
-          filters = (self.model.design_doc['filters'] ||= {})
-          filters[name.to_s] = function
+          design_doc.create_filter(name, function)
         end
 
-        def create_view_method(name)
-          model.class_eval <<-EOS, __FILE__, __LINE__ + 1
-            def self.#{name}(opts = {})
-              CouchRest::Model::Designs::View.new(self, opts, '#{name}')
-            end
-          EOS
+        # Convenience wrapper to access model's type key option.
+        def model_type_key
+          model.model_type_key
+        end
+
+        protected
+
+        # Create accessor in model and assign a new design doc.
+        # New design doc is returned ready to use.
+        def create_model_design_doc_reader
+          model.instance_eval "def #{method}; @#{method}; end"
+        end
+
+        def assign_model_design_doc
+          doc = Design.new(model, prefix)
+          model.instance_variable_set("@#{method}", doc)
+          model.design_docs << doc
+
+          # Set defaults
+          doc.auto_update = model.auto_update_design_doc
+
+          doc
         end
 
       end
     end
   end