tire-erez 0.5.4

data/lib/tire/results/pagination.rb

@@ -0,0 +1,68 @@
+module Tire
+  module Results
+
+    # Adds support for WillPaginate and Kaminari
+    #
+    module Pagination
+
+      def default_per_page
+        10
+      end
+      module_function :default_per_page
+
+      def total_entries
+        @total
+      end
+
+      def per_page
+        (@options[:per_page] || @options[:size] || default_per_page ).to_i
+      end
+
+      def total_pages
+        ( @total.to_f / per_page ).ceil
+      end
+
+      def current_page
+        if @options[:page]
+          @options[:page].to_i
+        else
+          (per_page + @options[:from].to_i) / per_page
+        end
+      end
+
+      def previous_page
+        current_page > 1 ? (current_page - 1) : nil
+      end
+
+      def next_page
+        current_page < total_pages ? (current_page + 1) : nil
+      end
+
+      def offset
+        per_page * (current_page - 1)
+      end
+
+      def out_of_bounds?
+        current_page > total_pages
+      end
+
+      # Kaminari support
+      #
+      alias :limit_value  :per_page
+      alias :total_count  :total_entries
+      alias :num_pages    :total_pages
+      alias :offset_value :offset
+
+      def first_page?
+        current_page == 1
+      end
+
+      def last_page?
+        current_page == total_pages
+      end
+
+
+    end
+
+  end
+end

data/lib/tire/rubyext/hash.rb

@@ -0,0 +1,8 @@
+class Hash
+
+  def to_json(options=nil)
+    MultiJson.encode(self)
+  end unless respond_to?(:to_json)
+
+  alias_method :to_indexed_json, :to_json
+end

data/lib/tire/rubyext/ruby_1_8.rb

@@ -0,0 +1 @@
+require 'tire/rubyext/uri_escape'

data/lib/tire/rubyext/symbol.rb

@@ -0,0 +1,11 @@
+# ActiveModel::Serialization Ruby < 1.9.x compatibility
+
+class Symbol
+  def <=> other
+    self.to_s <=> other.to_s
+  end unless method_defined?(:'<=>')
+
+  def capitalize
+    to_s.capitalize
+  end unless method_defined?(:capitalize)
+end

data/lib/tire/rubyext/uri_escape.rb

@@ -0,0 +1,74 @@
+# Steal the URI escape/unescape compatibility layer from Rack
+#
+# See <http://www.ruby-doc.org/stdlib-1.9.3/libdoc/uri/rdoc/URI.html#method-c-encode_www_form_component>
+
+# :stopdoc:
+
+# Stolen from ruby core's uri/common.rb, with modifications to support 1.8.x
+#
+# https://github.com/ruby/ruby/blob/trunk/lib/uri/common.rb
+#
+#
+
+module URI
+  TBLENCWWWCOMP_ = {} # :nodoc:
+  TBLDECWWWCOMP_ = {} # :nodoc:
+
+  # Encode given +s+ to URL-encoded form data.
+  #
+  # This method doesn't convert *, -, ., 0-9, A-Z, _, a-z, but does convert SP
+  # (ASCII space) to + and converts others to %XX.
+  #
+  # This is an implementation of
+  # http://www.w3.org/TR/html5/forms.html#url-encoded-form-data
+  #
+  # See URI.decode_www_form_component, URI.encode_www_form
+  def self.encode_www_form_component(s)
+    str = s.to_s
+    if RUBY_VERSION < "1.9" && $KCODE =~ /u/i
+      str.gsub(/([^ a-zA-Z0-9_.-]+)/) do
+        '%' + $1.unpack('H2' * Rack::Utils.bytesize($1)).join('%').upcase
+      end.tr(' ', '+')
+    else
+      if TBLENCWWWCOMP_.empty?
+        tbl = {}
+        256.times do |i|
+          tbl[i.chr] = '%%%02X' % i
+        end
+        tbl[' '] = '+'
+        begin
+          TBLENCWWWCOMP_.replace(tbl)
+          TBLENCWWWCOMP_.freeze
+        rescue
+        end
+      end
+      str.gsub(/[^*\-.0-9A-Z_a-z]/) {|m| TBLENCWWWCOMP_[m]}
+    end
+  end
+
+  # Decode given +str+ of URL-encoded form data.
+  #
+  # This decods + to SP.
+  #
+  # See URI.encode_www_form_component, URI.decode_www_form
+  def self.decode_www_form_component(str, enc=nil)
+    if TBLDECWWWCOMP_.empty?
+      tbl = {}
+      256.times do |i|
+        h, l = i>>4, i&15
+        tbl['%%%X%X' % [h, l]] = i.chr
+        tbl['%%%x%X' % [h, l]] = i.chr
+        tbl['%%%X%x' % [h, l]] = i.chr
+        tbl['%%%x%x' % [h, l]] = i.chr
+      end
+      tbl['+'] = ' '
+      begin
+        TBLDECWWWCOMP_.replace(tbl)
+        TBLDECWWWCOMP_.freeze
+      rescue
+      end
+    end
+    raise ArgumentError, "invalid %-encoding (#{str})" unless /\A(?:%[0-9a-fA-F]{2}|[^%])*\z/ =~ str
+    str.gsub(/\+|%[0-9a-fA-F]{2}/) {|m| TBLDECWWWCOMP_[m]}
+  end
+end

data/lib/tire/search.rb

@@ -0,0 +1,211 @@
+module Tire
+  module Search
+    class SearchRequestFailed < StandardError; end
+
+    class Search
+
+      attr_reader :indices, :types, :query, :facets, :filters, :options, :explain, :script_fields
+
+      def initialize(indices=nil, options={}, &block)
+        if indices.is_a?(Hash)
+          set_indices_options(indices)
+          @indices = indices.keys
+        else
+          @indices = Array(indices)
+        end
+        @types   = Array(options.delete(:type)).map { |type| Utils.escape(type) }
+        @options = options
+
+        @path    = ['/', @indices.join(','), @types.join(','), '_search'].compact.join('/').squeeze('/')
+
+        block.arity < 1 ? instance_eval(&block) : block.call(self) if block_given?
+      end
+
+
+      def set_indices_options(indices)
+        indices.each do |index, index_options|
+          if index_options[:boost]
+            @indices_boost ||= {}
+            @indices_boost[index] = index_options[:boost]
+          end
+        end
+      end
+
+      def results
+        @results  || (perform; @results)
+      end
+
+      def response
+        @response || (perform; @response)
+      end
+
+      def json
+        @json     || (perform; @json)
+      end
+
+      def url
+        Configuration.url + @path
+      end
+
+      def params
+        options = @options.except(:wrapper)
+        options.empty? ? '' : '?' + options.to_param
+      end
+
+      def query(&block)
+        @query = Query.new
+        block.arity < 1 ? @query.instance_eval(&block) : block.call(@query)
+        self
+      end
+
+      def sort(&block)
+        @sort = Sort.new(&block).to_ary
+        self
+      end
+
+      def facet(name, options={}, &block)
+        @facets ||= {}
+        @facets.update Facet.new(name, options, &block).to_hash
+        self
+      end
+
+      def filter(type, *options)
+        @filters ||= []
+        @filters << Filter.new(type, *options).to_hash
+        self
+      end
+
+      def script_field(name, options={})
+        @script_fields ||= {}
+        @script_fields.merge! ScriptField.new(name, options).to_hash
+        self
+      end
+
+      def highlight(*args)
+        unless args.empty?
+          @highlight = Highlight.new(*args)
+          self
+        else
+          @highlight
+        end
+      end
+
+      def from(value)
+        @from = value
+        @options[:from] = value
+        self
+      end
+
+      def size(value)
+        @size = value
+        @options[:size] = value
+        self
+      end
+
+      def fields(*fields)
+        @fields = Array(fields.flatten)
+        self
+      end
+
+      def partial_field(name, options)
+        @partial_fields ||= {}
+        @partial_fields[name] = options
+      end
+
+      def explain(value)
+        @explain = value
+        self
+      end
+
+      def version(value)
+        @version = value
+        self
+      end
+
+      def min_score(value)
+        @min_score = value
+        self
+      end
+
+      def track_scores(value)
+        @track_scores = value
+        self
+      end
+
+      def perform
+        @response = Configuration.client.get(self.url + self.params, self.to_json)
+        if @response.failure?
+          STDERR.puts "[REQUEST FAILED] #{self.to_curl}\n"
+          raise SearchRequestFailed, @response.to_s
+        end
+        @json     = MultiJson.decode(@response.body)
+        @results  = Results::Collection.new(@json, @options)
+        return self
+      ensure
+        logged
+      end
+
+      def to_curl
+        to_json_escaped = to_json.gsub("'",'\u0027')
+        %Q|curl -X GET '#{url}#{params.empty? ? '?' : params.to_s + '&'}pretty' -d '#{to_json_escaped}'|
+      end
+
+      def to_hash
+        @options[:payload] || begin
+          request = {}
+          request.update( { :indices_boost => @indices_boost } ) if @indices_boost
+          request.update( { :query  => @query.to_hash } )    if @query
+          request.update( { :sort   => @sort.to_ary   } )    if @sort
+          request.update( { :facets => @facets.to_hash } )   if @facets
+          request.update( { :filter => @filters.first.to_hash } ) if @filters && @filters.size == 1
+          request.update( { :filter => { :and => @filters.map {|filter| filter.to_hash} } } ) if  @filters && @filters.size > 1
+          request.update( { :highlight => @highlight.to_hash } ) if @highlight
+          request.update( { :size => @size } )               if @size
+          request.update( { :from => @from } )               if @from
+          request.update( { :fields => @fields } )           if @fields
+          request.update( { :partial_fields => @partial_fields } ) if @partial_fields
+          request.update( { :script_fields => @script_fields } ) if @script_fields
+          request.update( { :version => @version } )         if @version
+          request.update( { :explain => @explain } )         if @explain
+          request.update( { :min_score => @min_score } )     if @min_score
+          request.update( { :track_scores => @track_scores } ) if @track_scores
+          request
+        end
+      end
+
+      def to_json(options={})
+        payload = to_hash
+        # TODO: Remove when deprecated interface is removed
+        if payload.is_a?(String)
+          payload
+        else
+          MultiJson.encode(payload, :pretty => Configuration.pretty)
+        end
+      end
+
+      def logged(endpoint='_search')
+        if Configuration.logger
+
+          Configuration.logger.log_request endpoint, indices, to_curl
+
+          took = @json['took']  rescue nil
+          code = @response.code rescue nil
+
+          if Configuration.logger.level.to_s == 'debug'
+            body = if @json
+              MultiJson.encode( @json, :pretty => Configuration.pretty)
+            else
+              MultiJson.encode( MultiJson.load(@response.body), :pretty => Configuration.pretty) rescue ''
+            end
+          else
+            body = ''
+          end
+
+          Configuration.logger.log_response code || 'N/A', took || 'N/A', body || 'N/A'
+        end
+      end
+
+    end
+
+  end
+end

data/lib/tire/search/facet.rb

@@ -0,0 +1,81 @@
+module Tire
+  module Search
+
+    #--
+    # TODO: Implement all elastic search facets (geo, histogram, range, etc)
+    # http://elasticsearch.org/guide/reference/api/search/facets/
+    #++
+
+    class Facet
+
+      def initialize(name, options={}, &block)
+        @name    = name
+        @options = options
+        @value   = {}
+        block.arity < 1 ? self.instance_eval(&block) : block.call(self) if block_given?
+      end
+
+      def terms(field, options={})
+        size      = options.delete(:size) || 10
+        all_terms = options.delete(:all_terms) || false
+        @value[:terms] = if field.is_a?(Enumerable) and not field.is_a?(String)
+          { :fields => field }.update({ :size => size, :all_terms => all_terms }).update(options)
+        else
+          { :field => field  }.update({ :size => size, :all_terms => all_terms }).update(options)
+        end
+        self
+      end
+
+      def date(field, options={})
+        interval = { :interval => options.delete(:interval) || 'day' }
+        fields   = options[:value_field] || options[:value_script] ? { :key_field => field } : { :field => field }
+        @value[:date_histogram] = {}.update(fields).update(interval).update(options)
+        self
+      end
+
+      def range(field, ranges=[], options={})
+        @value[:range] = { :field => field, :ranges => ranges }.update(options)
+        self
+      end
+
+      def histogram(field, options={})
+        @value[:histogram] = (options.delete(:histogram) || {:field => field}.update(options))
+        self
+      end
+
+      def statistical(field, options={})
+        @value[:statistical] = (options.delete(:statistical) || {:field => field}.update(options))
+        self
+      end
+
+      def terms_stats(key_field, value_field, options={})
+        @value[:terms_stats] = {:key_field => key_field, :value_field => value_field}.update(options)
+        self
+      end
+
+      def query(&block)
+        @value[:query] = Query.new(&block).to_hash
+      end
+
+      def filter(type, options={})
+        @value[:filter] = Filter.new(type, options)
+        self
+      end
+
+      def facet_filter(type, *options)
+        @value[:facet_filter] = Filter.new(type, *options).to_hash
+        self
+      end
+
+      def to_json(options={})
+        to_hash.to_json
+      end
+
+      def to_hash
+        @value.update @options
+        { @name => @value }
+      end
+    end
+
+  end
+end