elasticsearch-dsl 0.0.0 → 0.1.0

data/lib/elasticsearch/dsl/search/filters/bool.rb

@@ -0,0 +1,80 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A compound filter which matches documents based on combinations of filters
+        #
+        # @example Defining a bool filter with multiple conditions
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             bool do
+        #               must do
+        #                 term category: 'men'
+        #               end
+        #
+        #               must do
+        #                 term size:  'xxl'
+        #               end
+        #
+        #               should do
+        #                 term color: 'red'
+        #               end
+        #
+        #               must_not do
+        #                 term manufacturer: 'evil'
+        #               end
+        #             end
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # See the integration test for a working example.
+        #
+        # @see http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-bool-filter.html
+        #
+        class Bool
+          include BaseComponent
+
+          def must(*args, &block)
+            @hash[name][:must] ||= []
+            value = Filter.new(*args, &block).to_hash
+            @hash[name][:must].push(value).flatten! unless @hash[name][:must].include?(value)
+            self
+          end
+
+          def must_not(*args, &block)
+            @hash[name][:must_not] ||= []
+            value = Filter.new(*args, &block).to_hash
+            @hash[name][:must_not].push(value).flatten! unless @hash[name][:must_not].include?(value)
+            self
+          end
+
+          def should(*args, &block)
+            @hash[name][:should] ||= []
+            value = Filter.new(*args, &block).to_hash
+            @hash[name][:should].push(value).flatten! unless @hash[name][:should].include?(value)
+            self
+          end
+
+          def to_hash
+            @hash[name].update(@args.to_hash) if @args.respond_to?(:to_hash)
+
+            if @block
+              call
+            else
+              @hash[name] = @args unless @args.nil? || @args.empty?
+            end
+
+            @hash
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/elasticsearch/dsl/search/filters/exists.rb

@@ -0,0 +1,32 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A filter which returns documents which have a non-`null` value in the specified field
+        # (ie. the reverse of the `missing` filter)
+        #
+        # @example
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             exists field: 'occupation'
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # @see http://elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-exists-filter.html
+        #
+        class Exists
+          include BaseComponent
+
+          option_method :field
+        end
+
+      end
+    end
+  end
+end

data/lib/elasticsearch/dsl/search/filters/geo_bounding_box.rb

@@ -0,0 +1,45 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A filter which returns documents which fall into a "box" of the specified geographical coordinates
+        #
+        # @example
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             geo_bounding_box :location do
+        #               top_right   "50.1815123678,14.7149200439"
+        #               bottom_left "49.9415476869,14.2162566185"
+        #             end
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # See the integration test for a working example.
+        #
+        # Use eg. <http://boundingbox.klokantech.com> to visually define the bounding box.
+        #
+        # @see http://elasticsearch.org/guide/en/elasticsearch/guide/current/geo-bounding-box.html
+        #
+        class GeoBoundingBox
+          include BaseComponent
+
+          option_method :top_left
+          option_method :bottom_right
+          option_method :top_right
+          option_method :bottom_left
+          option_method :top
+          option_method :left
+          option_method :bottom
+          option_method :right
+        end
+
+      end
+    end
+  end
+end

data/lib/elasticsearch/dsl/search/filters/geo_distance.rb

@@ -0,0 +1,57 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A filter which returns documents which fall into a specified geographical distance
+        #
+        # @example Define the filter with a hash
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             geo_distance location: '50.090223,14.399590', distance: '5km'
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # @example Define the filter with a block
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             geo_distance :location do
+        #               lat '50.090223'
+        #               lon '14.399590'
+        #               distance '5km'
+        #             end
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # See the integration test for a working example.
+        #
+        # @see http://elasticsearch.org/guide/en/elasticsearch/guide/current/geo-distance.html
+        #
+        class GeoDistance
+          include BaseComponent
+
+          option_method :distance,      lambda { |*args| @hash[self.name.to_sym].update distance: args.pop }
+          option_method :distance_type, lambda { |*args| @hash[self.name.to_sym].update distance_type: args.pop }
+          option_method :lat,           lambda { |*args| @hash[self.name.to_sym][@args].update lat: args.pop }
+          option_method :lon,           lambda { |*args| @hash[self.name.to_sym][@args].update lon: args.pop }
+
+          def initialize(*args, &block)
+            super
+            @hash[self.name.to_sym] = { @args => {} } unless @args.empty?
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/elasticsearch/dsl/search/filters/geo_distance_range.rb

@@ -0,0 +1,32 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A filter which returns documents which fall into a specified geographical distance range
+        #
+        # @example Define the filter with a hash
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             geo_distance location: '50.090223,14.399590', gte: '2km', lte: '5km'
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # @see http://elasticsearch.org/guide/en/elasticsearch/guide/current/geo-distance.html
+        #
+        class GeoDistanceRange
+          include BaseComponent
+
+          option_method :lat
+          option_method :lon
+        end
+
+      end
+    end
+  end
+end

data/lib/elasticsearch/dsl/search/filters/geo_polygon.rb

@@ -0,0 +1,41 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A filter which returns documents which fall into a specified geographical polygon
+        #
+        # @example
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             geo_polygon :location do
+        #               points [
+        #                [14.2244355,49.9419006],
+        #                [14.2244355,50.1774301],
+        #                [14.7067869,50.1774301],
+        #                [14.7067869,49.9419006],
+        #                [14.2244355,49.9419006]
+        #               ]
+        #             end
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # See the integration test for a working example.
+        #
+        # @see http://elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-geo-polygon-filter.html
+        #
+        class GeoPolygon
+          include BaseComponent
+
+          option_method :points
+        end
+
+      end
+    end
+  end
+end

data/lib/elasticsearch/dsl/search/filters/geo_shape.rb

@@ -0,0 +1,35 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A filter which returns documents which fall into a specified geographical shape
+        #
+        # @example
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             geo_shape :location do
+        #               shape type: 'envelope',
+        #                     coordinates: [[14.2162566185,49.9415476869], [14.7149200439,50.1815123678]]
+        #             end
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # @see http://elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-geo-shape-filter.html
+        #
+        class GeoShape
+          include BaseComponent
+
+          option_method :shape
+          option_method :indexed_shape
+        end
+
+      end
+    end
+  end
+end

data/lib/elasticsearch/dsl/search/filters/geohash_cell.rb

@@ -0,0 +1,46 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A filter which translates lat/lon values into a geohash with the specified precision
+        # and returns all documents which fall into it
+        #
+        # @example
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             geohash_cell :location do
+        #               lat '50.090223'
+        #               lon '14.399590'
+        #               precision '5km'
+        #               neighbors true
+        #             end
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # See the integration test for a working example.
+        #
+        # @see http://elasticsearch.org/guide/en/elasticsearch/guide/current/geohash-cell-filter.html
+        #
+        class GeohashCell
+          include BaseComponent
+
+          option_method :precision, lambda { |*args| @hash[self.name.to_sym].update precision: args.pop }
+          option_method :lat,       lambda { |*args| @hash[self.name.to_sym][@args].update lat: args.pop }
+          option_method :lon,       lambda { |*args| @hash[self.name.to_sym][@args].update lon: args.pop }
+          option_method :neighbors, lambda { |*args| @hash[self.name.to_sym].update neighbors: args.pop }
+
+          def initialize(*args, &block)
+            super
+            @hash[self.name.to_sym] = { @args => {} } unless @args.empty?
+          end
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/dsl/search/filters/has_child.rb

@@ -0,0 +1,73 @@
+module Elasticsearch
+  module DSL
+    module Search
+      module Filters
+
+        # A filter which returns parent documents for children documents matching a query or a filter
+        #
+        # @example Return articles where John has commented
+        #
+        #     search do
+        #       query do
+        #         filtered do
+        #           filter do
+        #             has_child do
+        #               type 'comment'
+        #               query do
+        #                 match author: 'John'
+        #               end
+        #             end
+        #           end
+        #         end
+        #       end
+        #     end
+        #
+        # @see http://elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-has-child-filter.html
+        #
+        class HasChild
+          include BaseComponent
+
+          option_method :type
+          option_method :min_children
+          option_method :max_children
+
+          # DSL method for building the `query` part of the query definition
+          #
+          # @return [self]
+          #
+          def query(*args, &block)
+            @query = block ? Elasticsearch::DSL::Search::Query.new(*args, &block) : args.first
+            self
+          end
+
+          # DSL method for building the `filter` part of the query definition
+          #
+          # @return [self]
+          #
+          def filter(*args, &block)
+            @filter = block ? Elasticsearch::DSL::Search::Filter.new(*args, &block) : args.first
+            self
+          end
+
+          # Converts the query definition to a Hash
+          #
+          # @return [Hash]
+          #
+          def to_hash
+            hash = super
+            if @query
+              _query = @query.respond_to?(:to_hash) ? @query.to_hash : @query
+              hash[self.name].update(query: _query)
+            end
+            if @filter
+              _filter = @filter.respond_to?(:to_hash) ? @filter.to_hash : @filter
+              hash[self.name].update(filter: _filter)
+            end
+            hash
+          end
+        end
+
+      end
+    end
+  end
+end