chewy 0.0.1 → 0.1.0

data/lib/chewy/query/nodes/bool.rb

@@ -0,0 +1,32 @@
+module Chewy
+  class Query
+    module Nodes
+      class Bool < Expr
+        METHODS = %w(must must_not should)
+
+        def initialize options = {}
+          @options = options
+          @must, @must_not, @should = [], [], []
+        end
+
+        METHODS.each do |method|
+          define_method method do |*exprs|
+            instance_variable_get("@#{method}").push(*exprs)
+            self
+          end
+        end
+
+        def __render__
+          bool = {
+            bool: Hash[METHODS.map do |method|
+              value = instance_variable_get("@#{method}")
+              [method.to_sym, value.map(&:__render__)] if value.any?
+            end.compact]
+          }
+          bool[:bool][:_cache] = !!@options[:cache] if @options.key?(:cache)
+          bool
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/equal.rb

@@ -0,0 +1,34 @@
+module Chewy
+  class Query
+    module Nodes
+      class Equal < Expr
+        EXECUTION = {
+          :| => :or,
+          :or => :or,
+          :& => :and,
+          :and => :and,
+          :b => :bool,
+          :bool => :bool,
+          :f => :fielddata,
+          :fielddata => :fielddata,
+        }
+
+        def initialize name, value, *args
+          @name = name.to_s
+          @value = value
+          @options = args.extract_options!
+          execution = EXECUTION[args.first.to_sym] if args.first
+          @options[:execution] = execution if execution
+        end
+
+        def __render__
+          filter = (@value.is_a?(Array) ? :terms : :term)
+          body = {@name => @value}
+          body.merge!(@options.slice(:execution)) if filter == :terms
+          body.merge!(_cache: !!@options[:cache]) if @options.key?(:cache)
+          {filter => body}
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/exists.rb

@@ -0,0 +1,20 @@
+module Chewy
+  class Query
+    module Nodes
+      class Exists < Expr
+        def initialize name, options = {}
+          @name = name.to_s
+          @options = options
+        end
+
+        def !
+          Nodes::Missing.new @name, null_value: true
+        end
+
+        def __render__
+          {exists: {term: @name}}
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/expr.rb

@@ -0,0 +1,28 @@
+module Chewy
+  class Query
+    module Nodes
+      class Expr < Base
+        def & other
+          Nodes::And.new self, other
+        end
+
+        def | other
+          Nodes::Or.new self, other
+        end
+
+        def !
+          Nodes::Not.new self
+        end
+
+        def ~
+          @options[:cache] = true
+          self
+        end
+
+        def __render__
+          raise NotImplementedError
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/field.rb

@@ -0,0 +1,106 @@
+module Chewy
+  class Query
+    module Nodes
+      class Field < Base
+        def initialize name, *args
+          @name = name.to_s
+          @args = args
+        end
+
+        def !
+          Nodes::Missing.new @name
+        end
+
+        def ~
+          __options_merge__!(cache: true)
+          self
+        end
+
+        def > value
+          Nodes::Range.new @name, *__options_merge__(gt: value)
+        end
+
+        def < value
+          Nodes::Range.new @name, *__options_merge__(lt: value)
+        end
+
+        def >= value
+          Nodes::Range.new @name, *__options_merge__(gt: value, left_closed: true)
+        end
+
+        def <= value
+          Nodes::Range.new @name, *__options_merge__(lt: value, right_closed: true)
+        end
+
+        def == value
+          case value
+          when nil
+            Nodes::Missing.new @name, existence: false, null_value: true
+          when ::Regexp
+            Nodes::Regexp.new @name, value, *@args
+          when ::Range
+            Nodes::Range.new @name, *__options_merge__(gt: value.first, lt: value.last)
+          else
+            if value.is_a?(Array) && value.first.is_a?(::Range)
+              Nodes::Range.new @name, *__options_merge__(
+                gt: value.first.first, lt: value.first.last,
+                left_closed: true, right_closed: true
+              )
+            else
+              Nodes::Equal.new @name, value, *@args
+            end
+          end
+        end
+
+        def != value
+          case value
+          when nil
+            Nodes::Exists.new @name
+          else
+            Nodes::Not.new self == value
+          end
+        end
+
+        def =~ value
+          case value
+          when ::Regexp
+            Nodes::Regexp.new @name, value, *@args
+          else
+            Nodes::Prefix.new @name, value, @args.extract_options!
+          end
+        end
+
+        def !~ value
+          Not.new(self =~ value)
+        end
+
+        def method_missing method, *args, &block
+          method = method.to_s
+          if method =~ /\?\Z/
+            Nodes::Exists.new [@name, method.gsub(/\?\Z/, '')].join(?.)
+          else
+            self.class.new [@name, method].join(?.), *args
+          end
+        end
+
+        def to_ary
+          nil
+        end
+
+      private
+
+        def __options_merge__! additional = {}
+          options = @args.extract_options!
+          options = options.merge(additional)
+          @args.push(options)
+        end
+
+        def __options_merge__ additional = {}
+          options = @args.extract_options!
+          options = options.merge(additional)
+          @args + [options]
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/missing.rb

@@ -0,0 +1,20 @@
+module Chewy
+  class Query
+    module Nodes
+      class Missing < Expr
+        def initialize name, options = {}
+          @name = name.to_s
+          @options = options.reverse_merge(existence: true, null_value: false)
+        end
+
+        def !
+          Nodes::Exists.new @name
+        end
+
+        def __render__
+          {missing: {term: @name}.merge(@options.slice(:existence, :null_value))}
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/not.rb

@@ -0,0 +1,25 @@
+module Chewy
+  class Query
+    module Nodes
+      class Not < Expr
+        def initialize expr, options = {}
+          @expr = expr
+          @options = options
+        end
+
+        def !
+          @expr
+        end
+
+        def __render__
+          expr = @expr.__render__
+          if @options.key?(:cache)
+            {not: {filter: expr, _cache: !!@options[:cache]}}
+          else
+            {not: expr}
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/or.rb

@@ -0,0 +1,25 @@
+module Chewy
+  class Query
+    module Nodes
+      class Or < Expr
+        def initialize *nodes
+          @options = nodes.extract_options!
+          @nodes = nodes.flatten.map { |node| node.is_a?(self.class) ? node.__nodes__ : node }.flatten
+        end
+
+        def __nodes__
+          @nodes
+        end
+
+        def __render__
+          nodes = @nodes.map(&:__render__)
+          if @options.key?(:cache)
+            {or: {filters: nodes, _cache: !!@options[:cache]}}
+          else
+            {or: nodes}
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/prefix.rb

@@ -0,0 +1,18 @@
+module Chewy
+  class Query
+    module Nodes
+      class Prefix < Expr
+        def initialize name, value, options = {}
+          @name = name.to_s
+          @value, @options = value, options
+        end
+
+        def __render__
+          filter = {prefix: {@name => @value}}
+          filter[:prefix][:_cache] = !!@options[:cache] if @options.key?(:cache)
+          filter
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/query.rb

@@ -0,0 +1,20 @@
+module Chewy
+  class Query
+    module Nodes
+      class Query < Expr
+        def initialize query, options = {}
+          @query = query
+          @options = options
+        end
+
+        def __render__
+          if @options.key?(:cache)
+            {fquery: {query: @query, _cache: !!@options[:cache]}}
+          else
+            {query: @query}
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/range.rb

@@ -0,0 +1,63 @@
+module Chewy
+  class Query
+    module Nodes
+      class Range < Expr
+        EXECUTION = {
+          :i => :index,
+          :index => :index,
+          :f => :fielddata,
+          :fielddata => :fielddata,
+        }
+
+        def initialize name, *args
+          @name = name.to_s
+          @options = args.extract_options!
+          @range = @options.extract!(:gt, :lt)
+          @bounds = @options.extract!(:left_closed, :right_closed)
+          execution = EXECUTION[args.first.to_sym] if args.first
+          @options[:execution] = execution if execution
+        end
+
+        def & other
+          if other.is_a?(self.class) && other.__name__ == @name
+            state = __state__.merge(other.__state__)
+
+            cache = other.__options__[:cache] || @options[:cache]
+            state[:cache] = cache unless cache.nil?
+
+            execution = other.__options__[:execution] || @options[:execution]
+            state[:execution] = execution unless execution.nil?
+
+            self.class.new(@name, state)
+          else
+            super
+          end
+        end
+
+        def __name__
+          @name
+        end
+
+        def __state__
+          @range.merge(@bounds)
+        end
+
+        def __options__
+          @options
+        end
+
+        def __render__
+          body = {}
+          body[@bounds[:left_closed] ? :gte : :gt] = @range[:gt] if @range.key?(:gt)
+          body[@bounds[:right_closed] ? :lte : :lt] = @range[:lt] if @range.key?(:lt)
+
+          filter = {@name => body}
+          filter[:_cache] = !!@options[:cache] if @options.key?(:cache)
+          filter.merge!(@options.slice(:execution))
+
+          {range: filter}
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/raw.rb

@@ -0,0 +1,15 @@
+module Chewy
+  class Query
+    module Nodes
+      class Raw < Expr
+        def initialize raw
+          @raw = raw
+        end
+
+        def __render__
+          @raw
+        end
+      end
+    end
+  end
+end

data/lib/chewy/query/nodes/regexp.rb

@@ -0,0 +1,31 @@
+module Chewy
+  class Query
+    module Nodes
+      class Regexp < Expr
+        FLAGS = %w(all anystring automaton complement empty intersection interval none)
+
+        def initialize name, regexp, *args
+          @name = name.to_s
+          @regexp = regexp.respond_to?(:source) ? regexp.source : regexp.to_s
+          @options = args.extract_options!
+          if args.any? || @options[:flags].present?
+            @options[:flags] = FLAGS & (args.any? ? args.flatten : @options[:flags]).map(&:to_s).map(&:downcase)
+          end
+        end
+
+        def __render__
+          body = @options[:flags] ?
+            {value: @regexp, flags: @options[:flags].map(&:to_s).map(&:upcase).uniq.join('|')} :
+            @regexp
+          filter = {@name => body}
+          if @options.key?(:cache)
+            filter[:_cache] = !!@options[:cache]
+            filter[:_cache_key] = @options[:cache].is_a?(TrueClass) || @options[:cache].is_a?(FalseClass) ?
+              @regexp.underscore : @options[:cache]
+          end
+          {regexp: filter}
+        end
+      end
+    end
+  end
+end