couchrest_model/lib/couchrest/model/designs/view.rb

module CouchRest
  module Model
    module Designs

      #
      # A proxy class that allows view queries to be created using
      # chained method calls. After each call a new instance of the method
      # is created based on the original in a similar fashion to ruby's sequel 
      # library, or Rails 3's Arel.
      #
      # CouchDB views have inherent limitations, so joins and filters as used in 
      # a normal relational database are not possible. At least not yet!
      #
      class View

        attr_accessor :model, :name, :query, :result

        # Initialize a new View object. This method should not be called from outside CouchRest Model.
        def initialize(parent, new_query = {}, name = nil)
          if parent.is_a?(Class) && parent < CouchRest::Model::Base
            raise "Name must be provided for view to be initialized" if name.nil?
            self.model    = parent
            self.name     = name
            # Default options:
            self.query    = { :reduce => false }
          elsif parent.is_a?(self.class)
            self.model    = parent.model
            self.name     = parent.name
            self.query    = parent.query.dup
          else
            raise "View cannot be initialized without a parent Model or View"
          end
          query.update(new_query)
          super
        end


        # == View Execution Methods
        #
        # Send a request to the CouchDB database using the current query values.

        # Inmediatly send a request to the database for all documents provided by the query.
        #
        def all(&block)
          include_docs.rows.map{|r| r.doc}
        end

        # Inmediatly send a request for the first result of the dataset.
        # This will override any limit set in the view previously.
        def first
          limit(1).all.first
        end


        def info
          
        end

        def offset
          execute['offset']
        end

        def total_rows
          execute['total_rows']
        end

        def rows
          return @rows if @rows
          if execute && result['rows']
            @rows ||= result['rows'].map{|v| ViewRow.new(v, model)}
          else 
            [ ]
          end
        end

        def keys
          rows.map{|r| r.key}
        end


        # == View Filter Methods
        # 
        # View filters return an copy of the view instance with the query 
        # modified appropriatly. Errors will be raised if the methods
        # are combined in an incorrect fashion.
        #
       
        def database(value)
          update_query(:database => value)
        end

        # Find all entries in the index whose key matches the value provided.
        #
        # Cannot be used when the +#startkey+ or +#endkey+ have been set.
        def key(value)
          raise "View#key cannot be used when startkey or endkey have been set" unless query[:startkey].nil? && query[:endkey].nil?
          update_query(:key => value)
        end

        # Find all index keys that start with the value provided. May or may not be used in
        # conjunction with the +endkey+ option.
        #
        # When the +#descending+ option is used (not the default), the start and end keys should
        # be reversed.
        #
        # Cannot be used if the key has been set.
        def startkey(value)
          raise "View#startkey cannot be used when key has been set" unless query[:key].nil?
          update_query(:startkey => value)
        end

        # The result set should start from the position of the provided document. 
        # The value may be provided as an object that responds to the +#id+ call
        # or a string.
        def startkey_doc(value)
          update_query(:startkey_docid => value.is_a?(String) ? value : value.id)
        end

        # The opposite of +#startkey+, finds all index entries whose key is before the value specified.
        #
        # See the +#startkey+ method for more details and the +#inclusive_end+ option.
        def endkey(value)
          raise "View#endkey cannot be used when key has been set" unless query[:key].nil?
          update_query(:endkey => value)
        end

        # The result set should end at the position of the provided document. 
        # The value may be provided as an object that responds to the +#id+ call
        # or a string.
        def endkey_doc(value)
          update_query(:endkey_docid => value.is_a?(String) ? value : value.id)
        end


        # The results should be provided in descending order.
        #
        # Descending is false by default, this method will enable it and cannot be undone.
        def descending
          update_query(:descending => true)
        end

        # Limit the result set to the value supplied.
        def limit(value)
          update_query(:limit => value)
        end

        # Skip the number of entries in the index specified by value. This would be
        # the equivilent of an offset in SQL.
        #
        # The CouchDB documentation states that the skip option should not be used
        # with large data sets as it is inefficient. Use the +startkey_doc+ method
        # instead to skip ranges efficiently.
        def skip(value = 0)
          update_query(:skip => value)
        end

        # Use the reduce function on the view. If none is available this method will fail. 
        def reduce
          update_query(:reduce => true)
        end

        # Control whether the reduce function reduces to a set of distinct keys or to a single
        # result row.
        #
        # By default the value is false, and can only be set when the view's +#reduce+ option
        # has been set.
        def group
          raise "View#reduce must have been set before grouping is permitted" unless query[:reduce]
          update_query(:group => true)
        end

        def group_level(value)
          raise "View#reduce and View#group must have been set before group_level is called" unless query[:reduce] && query[:group]
          update_query(:group_level => value.to_i)
        end


        protected

        def update_query(new_query = {})
          self.class.new(self, new_query)
        end

        def database
          query[:database] || model.database
        end

        # Used internally to ensure that docs are provided. Should not be used outside of 
        # the view class under normal circumstances.
        def include_docs
          raise "Documents cannot be returned from a view that is prepared for a reduce" if query[:reduce]
          query.delete(:reduce)
          update_query(:include_docs => true)
        end
        
        def execute(&block)
          return self.result if result
          raise "Database must be defined in model or view!" if database.nil?
          retryable = true
          # Remove the reduce value if its not needed
          query.delete(:reduce) if !query[:reduce] && model.design_doc['views'][name.to_s]['reduce'].blank?
          begin
            self.result = model.design_doc.view_on(database, name, query, &block)
          rescue RestClient::ResourceNotFound => e
            if retryable
              model.save_design_doc(database)
              retryable = false
              retry
            else
              raise e
            end
          end
        end

        # Class Methods
        class << self
          
          # Simplified view creation. A new view will be added to the 
          # provided model's design document using the name and options.
          #
          # If the view name starts with "by_" and +:by+ is not provided in 
          # the options, the new view's map method will be interpretted and
          # generated automatically. For example:
          #
          #   View.create(Meeting, "by_date_and_name")
          #
          # Will create a view that searches by the date and name properties. 
          # Explicity setting the attributes to use is possible using the 
          # +:by+ option. For example:
          #
          #   View.create(Meeting, "by_date_and_name", :by => [:date, :firstname, :lastname])
          #
          # The view name is the same, but three keys would be used in the
          # subsecuent index.
          #
          def create(model, name, opts = {})

            unless opts[:map]
              if opts[:by].nil? && name =~ /^by_(.+)/
                opts[:by] = $1.split(/_and_/)
              end
              raise "View cannot be created without recognised name, :map or :by options" if opts[:by].nil?

              opts[:guards] ||= []
              opts[:guards].push "(doc['#{model.model_type_key}'] == '#{model.to_s}')"

              keys = opts[:by].map{|o| "doc['#{o}']"}
              emit = keys.length == 1 ? keys.first : "[#{keys.join(', ')}]"
              opts[:guards] += keys.map{|k| "(#{k} != null)"}
              opts[:map] = <<-EOF
function(doc) {
  if (#{opts[:guards].join(' && ')}) {
    emit(#{emit}, null);
  }
}
EOF
            end

            model.design_doc['views'] ||= {}
            model.design_doc['views'][name.to_s] = {
              'map'    => opts[:map],
              'reduce' => opts[:reduce]
            }
          end

        end

      end

      # A special wrapper class that provides easy access to the key
      # fields in a result row.
      class ViewRow < Hash
        attr_accessor :model
        def initialize(hash, model)
          self.model = model
          replace(hash)
        end
        def id
          self["id"]
        end
        def key
          self["key"]
        end
        def value
          self['value']
        end
        def raw_doc
          self['doc']
        end
        # Send a request for the linked document either using the "id" field's
        # value, or the ["value"]["_id"] used for linked documents.
        def doc
          return model.create_from_database(self['doc']) if self['doc']
          doc_id = (value && value['_id']) ? value['_id'] : self.id
          model.get(doc_id)
        end
      end

    end
  end
end