eson-dsl 0.7.0

data/lib/eson/search/filter.rb

@@ -0,0 +1,28 @@
+module Eson
+  module Search
+    module Filter
+      module ClassMethods
+        def short_name(name)
+          define_method :name do
+            name
+          end
+          Filters.register name, self
+          
+          super if defined? super
+        end
+      end
+
+      def self.included(base)
+        base.class_eval do
+          extend ClassMethods
+          include Node unless base.kind_of? Node
+        end
+      end
+
+      def method_missing(name, arg)
+        warn("#{name} called as a filter option")
+        self.options[name] = arg
+      end
+    end
+  end
+end

data/lib/eson/search/filter_facet.rb

@@ -0,0 +1,13 @@
+module Eson
+  module Search
+    class FilterFacet
+      include Facet
+      include Filters
+      include FilterMethods
+
+      # @macro eson.facet
+      short_name :filter
+
+    end
+  end
+end

data/lib/eson/search/filters.rb

@@ -0,0 +1,69 @@
+module Eson
+  module Search
+    module Filters
+      module FilterMethods
+
+      end
+
+      class FilterArray < Array
+        include FilterMethods
+        include Parametrized
+        
+        def filters
+          self
+        end
+
+        def define_filters
+          instance_exec(self, &Proc.new) if block_given?
+        end
+
+        def to_query_hash
+          if length == 1
+            first.to_query_hash
+          else
+            map(&:to_query_hash)
+          end
+        end
+      end
+
+      attr_accessor :filters
+
+      def filters
+        @filters ||= FilterArray.new(args)
+
+        if block_given?
+          @filters.define_filters(&Proc.new)
+        end
+
+        @filters
+      end
+      alias :filter :filters
+
+      def filters?
+        @filters && !(@filters.empty?)
+      end
+
+      def self.register(name, klass)
+        FilterMethods.__send__(:define_method, name) do |*args, &block|
+          o = klass.new(*args)
+          o.context = :filter
+          o.args = self.args
+          o.instance_exec(o, &block) if block
+          filters << o
+        end
+      end
+
+      def to_query_hash
+        others = super if defined?(super)
+
+        if filters?
+          { :filter => filters.to_query_hash }.tap do |h|
+            h.merge!(others) if others
+          end
+        else
+          others
+        end
+      end
+    end
+  end
+end

data/lib/eson/search/fquery.rb

@@ -0,0 +1,22 @@
+module Eson
+  module Search
+    class FQuery
+      include Filter
+      include Queries
+      include QueryMethods
+
+      # @macro eson.filter
+      short_name :fquery
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+      end
+
+      def to_query_hash
+        {name => {:query => queries.to_query_hash}.merge!(options)}
+      end
+    end
+  end
+end

data/lib/eson/search/fuzzy.rb

@@ -0,0 +1,29 @@
+module Eson
+  module Search
+    class Fuzzy
+      include Query
+
+      # @macro eson.query
+      short_name :fuzzy
+
+      attr_accessor :field
+      attr_accessor :options
+
+      def initialize(*args)
+        if args.length == 1
+          self.options = args.first
+        else
+          self.field = QueryField.new(*args)
+        end
+      end
+
+      def to_query_hash
+        if field
+          {name => field.to_query_hash}
+        else
+          {name => options}
+        end
+      end
+    end
+  end
+end

data/lib/eson/search/fuzzy_like_this.rb

@@ -0,0 +1,16 @@
+module Eson
+  module Search
+    class FuzzyLikeThis
+      include Query
+
+      # @macro eson.query
+      short_name :flt
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+      end
+    end
+  end
+end

data/lib/eson/search/fuzzy_like_this_field.rb

@@ -0,0 +1,20 @@
+module Eson
+  module Search
+    class FuzzyLikeThisField
+      include Query
+
+      # @macro eson.query
+      short_name :flt_field
+
+      attr_accessor :field
+
+      def initialize(field, options = {})
+        self.field = QueryField.new(field, options)
+      end
+
+      def to_query_hash
+        {name => field.to_query_hash}
+      end
+    end
+  end
+end

data/lib/eson/search/geo_bounding_box.rb

@@ -0,0 +1,35 @@
+module Eson
+  module Search
+    class GeoBoundingBox
+      include Filter
+
+      # @macro eson.filter
+      short_name :geo_bounding_box
+
+      attr_accessor :field, :options
+
+      def initialize(field, options = {})
+        self.field = field
+        self.options = options
+      end
+
+      def top_left(location, *args)
+        unless args.empty?
+          location + args
+        end
+        options[:top_left] = location
+      end
+
+      def bottom_right(location, *args)
+        unless args.empty?
+          location + args
+        end
+        options[:bottom_right] = location
+      end
+
+      def to_query_hash
+        { name => {field => options } }
+      end
+    end
+  end
+end

data/lib/eson/search/geo_distance.rb

@@ -0,0 +1,34 @@
+module Eson
+  module Search
+    class GeoDistance
+      include Filter
+      include Facet
+
+      # @macro geo_distance
+      #   @yield a block defining options
+      #   Generates a `$1` filter or facet, depending on context.
+      #   {include:file:eson-dsl/examples/filters/$1.rb}
+      #   {include:file:eson-dsl/examples/facets/$1.rb}
+      short_name :geo_distance
+
+      attr_accessor :field, :options
+
+      def initialize(field, options = {})
+        self.field = field
+        self.options = options
+      end
+
+      def lat(lat)
+        @lat = lat
+      end
+
+      def lon(lon)
+        @lon = lon
+      end
+
+      def to_query_hash
+        { name => options.merge({ field => {:lat => @lat, :lon => @lon } }) }
+      end
+    end
+  end
+end

data/lib/eson/search/geo_distance_range.rb

@@ -0,0 +1,29 @@
+module Eson
+  module Search
+    class GeoDistanceRange
+      include Filter
+
+      # @macro eson.filter
+      short_name :geo_distance_range
+
+      attr_accessor :field, :options
+
+      def initialize(field, options = {})
+        self.field = field
+        self.options = options
+      end
+
+      def lat(lat)
+        @lat = lat
+      end
+
+      def lon(lon)
+        @lon = lon
+      end
+
+      def to_query_hash
+        { name => options.merge({ field => {:lat => @lat, :lon => @lon } }) }
+      end
+    end
+  end
+end

data/lib/eson/search/geo_polygon.rb

@@ -0,0 +1,32 @@
+module Eson
+  module Search
+    class GeoPolygon
+      include Filter
+      
+      # @macro eson.filter
+      short_name :geo_polygon
+      attr_accessor :field, :options
+
+      def initialize(field, options = {})
+        self.field = field
+        self.options = options
+      end
+
+      def points(points = nil)
+        if points
+          @points = points
+        else
+          @points ||= []
+        end
+      end
+
+      def point(point)
+        points << point
+      end
+
+      def to_query_hash
+        { name => options.merge({ field => {:points => points} }) }
+      end
+    end
+  end
+end

data/lib/eson/search/has_child.rb

@@ -0,0 +1,29 @@
+module Eson
+  module Search
+    class HasChild
+      include Query
+      include Filter
+
+      include QueryGroup
+
+      # @macro has_child
+      #   Generates a `$1` filter or query, depending on context.
+      #   {include:file:eson-dsl/examples/filters/$1.rb}
+      #   {include:file:eson-dsl/examples/queries/$1.rb}
+      short_name :has_child
+
+      attr_accessor :options
+
+      def initialize(type, options = {})
+        self.options = options.merge(:type => type)
+      end
+
+      def to_query_hash
+        {
+          name => options.merge(:query => super)
+        }
+      end
+
+    end
+  end
+end

data/lib/eson/search/histogram.rb

@@ -0,0 +1,20 @@
+module Eson
+  module Search
+    class Histogram
+      include Facet
+
+      # @macro eson.facet
+      short_name :histogram
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+      end
+
+      def to_query_hash
+        {name => options}
+      end
+    end
+  end
+end

data/lib/eson/search/ids.rb

@@ -0,0 +1,27 @@
+module Eson
+  module Search
+    class Ids
+      include Filter
+      include Query
+
+      # @macro ids
+      #   Generates a `$1` filter or query, depending on context.
+      #   {include:file:eson-dsl/examples/filters/$1.rb}
+      #   {include:file:eson-dsl/examples/queries/$1.rb}
+      short_name :ids
+
+      attr_accessor :options
+
+      def initialize(*args)
+        if args.length == 2
+          self.options = {
+            :type => args.first,
+            :values => args.last
+          }
+        else
+          self.options = args.first
+        end
+      end
+    end
+  end
+end

data/lib/eson/search/match_all.rb

@@ -0,0 +1,21 @@
+module Eson
+  module Search
+    class MatchAll
+      include Query
+
+      # @macro eson.query
+      short_name :match_all
+
+      attr_accessor :boost
+
+      def initialize(params = {})
+        self.boost = params.delete(:boost)
+      end
+
+      def to_query_hash
+        param = boost ? {:boost => self.boost} : {}
+        {:match_all => param }
+      end
+    end
+  end
+end

data/lib/eson/search/missing.rb

@@ -0,0 +1,21 @@
+module Eson
+  module Search
+    class Missing
+      include Filter
+
+      # @macro eson.filter
+      short_name :missing
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+      end
+
+      def to_query_hash
+        { name => options }
+      end
+
+    end
+  end
+end

data/lib/eson/search/more_like_this.rb

@@ -0,0 +1,16 @@
+module Eson
+  module Search
+    class MoreLikeThis
+      include Query
+
+      # @macro eson.query
+      short_name :mlt
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+      end
+    end
+  end
+end

data/lib/eson/search/more_like_this_field.rb

@@ -0,0 +1,20 @@
+module Eson
+  module Search
+    class MoreLikeThisField
+      include Query
+
+      # @macro eson.query
+      short_name :mlt_field
+
+      attr_accessor :field
+
+      def initialize(field, options = {})
+        self.field = QueryField.new(field, options)
+      end
+
+      def to_query_hash
+        {name => field.to_query_hash}
+      end
+    end
+  end
+end

data/lib/eson/search/nested.rb

@@ -0,0 +1,24 @@
+module Eson
+  module Search
+    class Nested
+      include Query
+      include QueryGroup
+
+      # @macro eson.query
+      short_name :nested
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+        instance_exec(self, &Proc.new) if block_given?
+      end
+
+      def to_query_hash
+        {
+          name => {:query => super}.merge(options)
+        }
+      end
+    end
+  end
+end

data/lib/eson/search/node.rb

@@ -0,0 +1,28 @@
+module Eson
+  module Search
+    module Node
+      attr_accessor :context
+      attr_accessor :args
+      attr_accessor :options
+      
+      def param(name)
+        args[name] || (raise "Parameter #{name} not given")
+      end
+      
+      def options
+        @options ||= {}
+      end
+      
+      def to_query_hash
+        values = options
+
+        if defined?(super)
+          values = values.merge(super)
+        end
+
+        { name => values }
+      end
+
+    end
+  end
+end

data/lib/eson/search/not.rb

@@ -0,0 +1,23 @@
+module Eson
+  module Search
+    class Not
+      include Filters
+      include FilterMethods
+      include Filter
+
+      # @macro eson.filter
+      short_name 'not' # yard chokes on :not
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+      end
+
+      def to_query_hash
+        { name => { :filter => filters.to_query_hash } }
+      end
+
+    end
+  end
+end

data/lib/eson/search/numeric_range.rb

@@ -0,0 +1,20 @@
+module Eson
+  module Search
+    class NumericRange
+      include Filter
+
+      # @macro eson.filter
+      short_name :numeric_range
+
+      attr_accessor :field
+
+      def initialize(field, options = {})
+        self.field = QueryField.new(field, options)
+      end
+
+      def to_query_hash
+        {name => field.to_query_hash}
+      end
+    end
+  end
+end

data/lib/eson/search/or.rb

@@ -0,0 +1,23 @@
+module Eson
+  module Search
+    class Or
+      include Filters
+      include FilterMethods
+      include Filter
+
+      # @macro eson.filter
+      short_name 'or' # yard chokes on :or
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+      end
+
+      def to_query_hash
+        { name => filters.to_query_hash }
+      end
+
+    end
+  end
+end

data/lib/eson/search/parametrized.rb

@@ -0,0 +1,15 @@
+module Eson
+  module Search
+    module Parametrized
+      attr_accessor :args
+      
+      def initialize(args)
+        self.args = args
+      end
+      
+      def param(name)
+        args[name] || (raise "Parameter #{name} not given")
+      end
+    end
+  end
+end

data/lib/eson/search/prefix.rb

@@ -0,0 +1,22 @@
+module Eson
+  module Search
+    class Prefix < FieldBased
+      include Query
+      include Filter
+
+      # @macro prefix
+      #   Generates a `$1` filter or query, depending on context.
+      #   {include:file:eson-dsl/examples/filters/$1.rb}
+      #   {include:file:eson-dsl/examples/queries/$1.rb}
+      short_name :prefix
+
+      def to_query_hash
+        if field
+          {name => field.to_query_hash}
+        else
+          {name => options}
+        end
+      end
+    end
+  end
+end