eson-dsl 0.7.0

data/examples/queries/fuzzy.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    fuzzy :user => "ki"
+  end
+end

data/examples/queries/fuzzy_complex.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    fuzzy :user, :value => "ki", :boost => 1.0
+  end
+end

data/examples/queries/has_child.rb

@@ -0,0 +1,9 @@
+Eson::Search::BaseQuery.new do
+  query do
+    has_child :blog_tag do
+      query do
+        term :tag => "something"
+      end
+    end
+  end
+end

data/examples/queries/has_child_complex.rb

@@ -0,0 +1,12 @@
+Eson::Search::BaseQuery.new do
+  query do
+    has_child :blog_tag do
+      query do
+        term :tag => "something"
+      end
+      filter do
+        term :tag => "other"
+      end
+    end
+  end
+end

data/examples/queries/ids.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    ids("user", [1,2,3,4,5,6,7])
+  end
+end

data/examples/queries/match_all.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    match_all :boost => 2.0
+  end
+end

data/examples/queries/mlt.rb

@@ -0,0 +1,7 @@
+Eson::Search::BaseQuery.new do
+  query do
+    mlt :fields => ["name.first", "name.last"],
+        :like_text => "text like this one",
+        :max_query_terms => 12
+  end
+end

data/examples/queries/mlt_field.rb

@@ -0,0 +1,7 @@
+Eson::Search::BaseQuery.new do
+  query do
+    mlt_field "name.first",
+              :like_text => "text like this one",
+              :max_query_terms => 12
+  end
+end

data/examples/queries/nested.rb

@@ -0,0 +1,12 @@
+Eson::Search::BaseQuery.new do
+  query do
+    nested :path => :obj1, :score_mode => "avg" do
+      query do
+        match_all
+      end
+      filters do
+        exists :field => "user"
+      end
+    end
+  end
+end

data/examples/queries/prefix.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    prefix "user", :value => "sh", :boost => 2.0
+  end
+end

data/examples/queries/prefix_short.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    prefix "user" => "sh"
+  end
+end

data/examples/queries/query_string.rb

@@ -0,0 +1,6 @@
+Eson::Search::BaseQuery.new do
+  query do
+    query_string :default_field => "content",
+                 :query => "this AND that OR thus"
+  end
+end

data/examples/queries/query_string_with_multiple_fields.rb

@@ -0,0 +1,7 @@
+Eson::Search::BaseQuery.new do
+  query do
+    query_string :fields => ["content", "name^5"],
+                 :query => "this AND that OR thus",
+                 :use_dis_max => true
+  end
+end

data/examples/queries/query_string_without_default_field.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    query_string "this AND that OR thus"
+  end
+end

data/examples/queries/range.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    range(:age, :from => 10, :to => 20)
+  end
+end

data/examples/queries/span_first.rb

@@ -0,0 +1,7 @@
+Eson::Search::BaseQuery.new do
+  query do
+    span_first :end => 3 do
+      span_term :field => "value1"
+    end
+  end
+end

data/examples/queries/span_near.rb

@@ -0,0 +1,11 @@
+Eson::Search::BaseQuery.new do
+  query do
+    span_near :slop => 12, :in_order => false, :collect_payloads => false do
+      clauses do
+        span_term :field => "value1"
+        span_term :field => "value2"
+        span_term :field => "value3"
+      end
+    end
+  end
+end

data/examples/queries/span_not.rb

@@ -0,0 +1,8 @@
+Eson::Search::BaseQuery.new do
+  query do
+    span_not do
+      include { span_term :field => "value1" }
+      exclude { span_term :field => "value2" }
+    end
+  end
+end

data/examples/queries/span_or.rb

@@ -0,0 +1,11 @@
+Eson::Search::BaseQuery.new do
+  query do
+    span_or do
+      clauses do
+        span_term :field => "value1"
+        span_term :field => "value2"
+        span_term :field => "value3"
+      end
+    end
+  end
+end

data/examples/queries/span_term.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    span_term :user => "kimchy"
+  end
+end

data/examples/queries/term.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    term "user", :value => "kimchy", :boost => 2.0
+  end
+end

data/examples/queries/term_with_parameters.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new(:term_boost => 2.0, :term => "kimchy") do
+  query do
+    term "user", :value => param(:term), :boost => param(:term_boost)
+  end
+end

data/examples/queries/terms.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    terms({:tags => ['blue', 'pill'], :minimum_match => 2})
+  end
+end

data/examples/queries/text.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    text :type => :phrase_prefix, :message => "this is a text"
+  end
+end

data/examples/queries/top_children.rb

@@ -0,0 +1,7 @@
+Eson::Search::BaseQuery.new do
+  query do
+    top_children :blog_tag, :score => "max" do
+      query { term :tag => "something" }
+    end
+  end
+end

data/examples/queries/wildcard.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    wildcard "user", :value => "kim*y", :boost => 2.0
+  end
+end

data/examples/queries/wildcard_short.rb

@@ -0,0 +1,5 @@
+Eson::Search::BaseQuery.new do
+  query do
+    wildcard "user" => "kim*y"
+  end
+end

data/lib/eson/modules/query_plugin.rb

@@ -0,0 +1,44 @@
+module Eson
+  module QueryPlugin
+    def query
+      if block_given?
+        @query ||= Eson::Search::BaseQuery.new
+        @query.instance_exec(@query, &Proc.new)
+      end
+
+      @query
+    end
+
+    def call(*args)
+      if query.respond_to?(:to_query_hash)
+        self.params = query.to_query_hash
+      end
+
+      super(*args)
+    end
+
+    def method_missing(sym, *args, &block)
+      if query.query.respond_to?(sym)
+        query.query.send(sym, *args, &block)
+      else
+        super
+      end
+    end
+
+    def self.plugin_for(protocol)
+      [protocol::Search,
+       protocol::Count,
+       protocol::DeleteByQuery,
+       protocol::Percolate]
+    end
+    
+    def handle_block(&block)
+      # in ruby 1.8.7, `lambda {}.arity == -1`
+      if (block.arity == 0) || (block.arity == -1)
+        query(&block)
+      else
+        super
+      end
+    end
+  end
+end

data/lib/eson/search/and.rb

@@ -0,0 +1,27 @@
+module Eson
+  module Search
+    class And
+      include Filters
+      include FilterMethods
+      include Filter
+
+      ##
+      # @macro and
+      #   @yield the block describing all subfilters
+      #   Generates a `$1` filter.
+      #   {include:file:eson-dsl/examples/filters/$1.rb}
+      short_name 'and'
+
+      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/base_query.rb

@@ -0,0 +1,39 @@
+module Eson
+  module Search
+    class BaseQuery
+      include QueryGroup
+      include Facets
+      include Parametrized
+      
+      def initialize(args = {})
+        self.args = args
+        instance_exec(self, &Proc.new) if block_given?
+      end
+
+      def to_query_hash
+        if (!queries || queries.empty?)
+          hash = {:query => {:match_all => {}}}
+        else
+          hash = {:query => queries.to_query_hash }
+        end
+
+        if filters && !filters.empty?
+          hash = {
+            :query => {
+              :filtered => {
+                :query => hash[:query],
+                :filter => filters.to_query_hash
+              }
+            }
+          }
+        end
+
+        if facets.length > 0
+          hash[:facets] = facets.to_query_hash
+        end
+
+        hash
+      end
+    end
+  end
+end

data/lib/eson/search/bool.rb

@@ -0,0 +1,51 @@
+module Eson
+  module Search
+    class Bool
+      include Filter
+      include Query
+      
+      # @macro bool
+      #   @yield the block containing all subqueries
+      #   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 :bool
+
+      attr_accessor :options, :conditions
+
+      def initialize(options = {})
+        self.options = options
+      end
+
+      def conditions
+        @conditions ||= {}
+      end
+
+      def must(&block)
+        q = (conditions[:must] ||= SubQuery.new(args))
+        q.send(context, &block)
+      end
+
+      def must_not(&block)
+        q = (conditions[:must_not] ||= SubQuery.new(args))
+        q.send(context, &block)
+      end
+
+      def should(&block)
+        q = (conditions[:should] ||= SubQuery.new(args))
+        q.send(context, &block)
+      end
+
+      def to_query_hash
+        result = {}
+
+        conditions.each do |k,v|
+          result[k] = v.to_query_hash
+        end
+
+        {name => result.merge(options)}
+      end
+
+    end
+  end
+end

data/lib/eson/search/boosting.rb

@@ -0,0 +1,45 @@
+module Eson
+  module Search
+    class Boosting
+      include Query
+
+      ##
+      # @macro eson.query
+      #   @yield the block describing all subqueries 
+      #   Generates a `$1` query.
+      #   {include:file:eson-dsl/examples/queries/$1.rb}
+      short_name :boosting
+
+      attr_accessor :options, :conditions
+
+      def initialize(options = {})
+        self.options = options
+      end
+
+      def boosts
+        @boosts ||= {}
+      end
+
+      def positive(&block)
+        q = (boosts[:positive] ||= SubQuery.new(args))
+        q.query(&block)
+      end
+
+      def negative(&block)
+        q = (boosts[:negative] ||= SubQuery.new(args))
+        q.query(&block)
+      end
+
+      def to_query_hash
+        result = {}
+
+        boosts.each do |k,v|
+          result[k] = v.to_query_hash
+        end
+
+        {name => result.merge(options)}
+      end
+
+    end
+  end
+end

data/lib/eson/search/constant_score.rb

@@ -0,0 +1,21 @@
+module Eson
+  module Search
+    class ConstantScore
+
+      include Filters
+      include Queries
+
+      include Query
+
+      # @macro eson.query
+      short_name :constant_score
+
+      attr_accessor :options
+
+      def initialize(options = {})
+        self.options = options
+      end
+
+    end
+  end
+end

data/lib/eson/search/custom_score.rb

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

data/lib/eson/search/date_histogram.rb

@@ -0,0 +1,13 @@
+module Eson
+  module Search
+    class DateHistogram < Histogram
+
+      ##
+      # @macro eson.facet
+      #   Generates a `$1` facet.
+      #   {include:file:eson-dsl/examples/facets/$1.rb}
+      short_name :date_histogram
+
+    end
+  end
+end

data/lib/eson/search/dis_max.rb

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

data/lib/eson/search/exists.rb

@@ -0,0 +1,18 @@
+module Eson
+  module Search
+    class Exists < FieldBased
+      include Filter
+
+      # @macro eson.filter
+      short_name :exists
+
+      def to_query_hash
+        if field
+          {name => field.to_query_hash}
+        else
+          {name => options}
+        end
+      end
+    end
+  end
+end

data/lib/eson/search/facet.rb

@@ -0,0 +1,39 @@
+module Eson
+  module Search
+    module Facet
+      module ClassMethods
+        def short_name(name)
+          define_method :name do
+            name
+          end
+          Facets.register name, self
+
+          super if defined? super
+        end
+      end
+      
+      def self.included(base)
+        base.class_eval do
+          extend ClassMethods
+        end
+      end
+      
+      attr_accessor :args
+      
+      def param(name)
+        args[name] || (raise "Parameter #{name} not given")
+      end
+      
+      def scope(query, name)
+        raise "Cannot scope unless Query is a NestedQuery" unless Nested === query
+        query.options[:_scope] = name
+        self.options[:scope] = name
+      end
+
+      def method_missing(name, arg)
+        warn("#{name} called as a facet option")
+        self.options[name] = arg
+      end
+    end
+  end
+end

data/lib/eson/search/facets.rb

@@ -0,0 +1,37 @@
+module Eson
+  module Search
+    module Facets
+      class FacetArray < Array
+        include Parametrized
+        
+        def initialize(args)
+          self.args = args
+          instance_exec(self, &Proc.new) if block_given?
+        end
+
+        def to_query_hash
+          h = {}
+          self.each do |k,v|
+            h[k] = v.to_query_hash
+          end
+          h
+        end
+      end
+
+      def self.register(name, klass)
+        FacetArray.__send__(:define_method, name) do |facet_name, *args, &block|
+          o = klass.new(*args)
+          o.args = self.args
+          o.instance_exec(o, &block) if block
+          self << [facet_name, o]
+          o
+        end
+      end
+
+      def facets(&block)
+        @facets ||= FacetArray.new(args, &block)
+      end
+      alias :facet :facets
+    end
+  end
+end

data/lib/eson/search/field.rb

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

data/lib/eson/search/field_based.rb

@@ -0,0 +1,24 @@
+module Eson
+  module Search
+    class FieldBased
+      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