mm_es_search 0.0.1

data/lib/mm_es_search/api/query/custom_filters_score_query.rb

@@ -0,0 +1,52 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class CustomFiltersScoreQuery < AbstractQuery
+        
+        one  :query,          :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        many :scored_filters, :class_name => 'MmEsSearch::Api::Query::ScoredFilter'
+        key  :score_mode,     Symbol
+        
+        def to_mongo_query(options = {})
+          query = self.query.is_a?(MatchAllQuery) ? nil : self.query
+          filters = (scored_filters.map(&:filter) << query).compact
+          AndFilter.new(:filters => filters).to_mongo_query(options)
+        end
+        
+        def to_es_query
+          query_params = {
+            :query   => query.to_es_query,
+            :filters => scored_filters.map(&:to_es_query)
+          }
+          query_params[:score_mode] = score_mode unless score_mode.nil?
+          {:custom_filters_score => query_params}
+        end
+
+      end
+      
+      class ScoredFilter
+        
+        include MongoMapper::EmbeddedDocument
+        plugin  MmUsesNoId
+        
+        one :filter, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key :boost,  Float
+        key :script, String
+        
+        def to_mongo_query
+          filter.to_mongo_query
+        end
+        
+        def to_es_query
+          filter_params = attributes.except('filter', '_type')
+          filter_params[:filter] = filter.to_es_query
+          filter_params
+        end
+        
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/custom_score_query.rb

@@ -0,0 +1,31 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class CustomScoreQuery < AbstractQuery
+      
+        one :query, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key :script, String
+      
+        def to_mongo_query(options = {})
+          
+          return query.to_mongo_query(options)
+          
+        end
+        
+        def to_es_query
+          
+          custom_score_params = {
+            :query => query.to_es_query,
+            :script => script
+          }
+      
+          return {:custom_score => custom_score_params}
+          
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/dismax_query.rb

@@ -0,0 +1,29 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class DismaxQuery < AbstractQuery
+      
+        many :queries, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key :tie_breaker, Float
+        key :boost, Float
+
+        def to_mongo_query(options = {})
+          raise 'Dis Max Query cannot be run as a mongo query'
+        end
+        
+        def to_es_query
+          
+          dismax_params = {:queries => queries.map(&:to_es_query)}
+          dismax_params.merge!(:tie_breaker => tie_breaker) if tie_breaker?
+          dismax_params.merge!(:boost => boost) if boost?
+          
+          return {:dis_max => dismax_params}
+      
+        end
+      
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/filtered_query.rb

@@ -0,0 +1,30 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class FilteredQuery < AbstractQuery
+        
+        one :query, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        one :filter, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        
+        def to_mongo_query(options = {})
+          if query.is_a?(MatchAllQuery) or query.nil?
+            filter.to_mongo_query(options)
+          else
+            AndFilter.new(:filters => [query, filter]).to_mongo_query(options)
+          end
+        end
+        
+        def to_es_query
+          if query.is_a?(MatchAllQuery) or query.nil?
+            {:filtered => {:query => MatchAllQuery.new.to_es_query, :filter => filter.to_es_query}}
+          else
+            {:filtered => {:query => query.to_es_query, :filter => filter.to_es_query}}
+          end
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/has_child_filter.rb

@@ -0,0 +1,11 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class HasChildFilter < HasChildQuery
+        plugin AbstractFilter  
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/has_child_query.rb

@@ -0,0 +1,25 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class HasChildQuery < AbstractQuery
+        
+        key :type, Symbol
+        one :query, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key :scope, String
+        
+        def to_mongo_query(options = {})
+          raise NotImplementedError
+        end
+        
+        def to_es_query
+          params = {:type => type, :query => query.to_es_query}
+          params[:_scope] = scope if scope?
+          {:has_child => params}
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/has_parent_filter.rb

@@ -0,0 +1,11 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class HasParentFilter < HasParentQuery
+        plugin AbstractFilter  
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/has_parent_query.rb

@@ -0,0 +1,25 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class HasParentQuery < AbstractQuery
+        
+        key :parent_type, Symbol
+        one :query, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key :scope, String
+        
+        def to_mongo_query(options = {})
+          raise NotImplementedError
+        end
+        
+        def to_es_query
+          params = {:parent_type => parent_type, :query => query.to_es_query}
+          params[:_scope] = scope if scope?
+          {:has_parent => params}
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/match_all_filter.rb

@@ -0,0 +1,11 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class MatchAllFilter < MatchAllQuery
+        plugin AbstractFilter  
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/match_all_query.rb

@@ -0,0 +1,19 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class MatchAllQuery < AbstractQuery
+        
+        def to_mongo_query(options = {})
+          return {}
+        end
+        
+        def to_es_query
+          return {:match_all => {}}
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/nested_filter.rb

@@ -0,0 +1,22 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class NestedFilter < NestedQuery
+
+        plugin AbstractFilter  
+        private
+        
+        def es_api_keyword
+          :filter
+        end
+        
+        def anded_query(query_array)
+          AndFilter.new(:filters => query_array)
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/nested_query.rb

@@ -0,0 +1,62 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class NestedQuery < AbstractQuery
+      
+        one :query, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key :path, String
+        key :score_mode, String
+        key :_scope, String
+        key :array_index_name, String
+        
+        def to_mongo_query(options = {})
+          
+          mod_path, array_index = path_and_index
+          if array_index.nil?
+            #NOTE: we assume here that earlier terms in a dot-joined path has been used in an enclosing nested query
+            {path.split('.').last => {'$elemMatch' => query.to_mongo_query(options)}}
+          else
+            query.to_mongo_query(options) #don't need elemMatch as index means we're addressing one document only
+          end
+          
+        end
+        
+        def to_es_query
+          
+          mod_path, array_index = path_and_index
+          mod_query = if array_index.nil?
+            query
+          else
+            query_array = [
+              query,
+              TermQuery.new(:path => mod_path, :field => array_index_name || '_array_index', :value => array_index)]
+            anded_query(query_array)
+          end
+
+          nested_params = {
+            es_api_keyword => mod_query.to_es_query,
+            :path => mod_path
+          }
+          nested_params.merge!({:score_mode => score_mode}) unless score_mode.nil?
+          nested_params.merge!({:_scope => _scope}) unless _scope.nil?
+      
+          return {:nested => nested_params}
+      
+        end
+        
+        private
+        
+        def es_api_keyword
+          :query
+        end
+        
+        def anded_query(query_array)
+          BoolQuery.new(:musts => query_array)
+        end
+      
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/not_filter.rb

@@ -0,0 +1,9 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class NotFilter < SingleBoolFilter; end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/or_filter.rb

@@ -0,0 +1,9 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class OrFilter < SingleBoolFilter; end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/prefix_filter.rb

@@ -0,0 +1,11 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class PrefixFilter < PrefixQuery
+        plugin AbstractFilter  
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/prefix_query.rb

@@ -0,0 +1,34 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class PrefixQuery < AbstractQuery
+        
+        key :field, String
+        key :path, String
+        
+        key :value, String
+        key :boost, Float
+        
+        def to_mongo_query(options = {})
+          prefix_regex = /^#{value}/
+          if options[:negated]
+            {mongo_abs_field => {'$ne' => prefix_regex}}
+          else
+            {mongo_abs_field => prefix_regex}
+          end
+        end
+        
+        def to_es_query
+          if boost
+            {:prefix => {es_abs_field => {:value => value, :boost => boost}}} 
+          else
+            {:prefix => {es_abs_field => value}}
+          end
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/query_filter.rb

@@ -0,0 +1,28 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class QueryFilter < AbstractQuery
+        plugin AbstractFilter
+        
+        one :query, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key :cache, Boolean
+        
+        def to_mongo_query
+          query.to_mongo_query
+        end
+        
+        def to_es_query
+          query_params = {
+            :query => query.to_es_query
+          }
+          query_params[:_cache] = cache unless cache.nil?
+          {:fquery => query_params}
+        end
+        
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/query_string_query.rb

@@ -0,0 +1,37 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class QueryStringQuery < AbstractQuery
+        
+        key :query, String
+        key :default_field, String
+        key :boost, Float
+        key :default_operator, String
+        key :analyzer, String
+        key :allow_leading_wildcard, Boolean
+        key :lowercase_expanded_terms, Boolean
+        key :enable_position_increments, Boolean
+        key :fuzzy_prefix_length, Integer
+        key :fuzzy_min_sim, Float
+        key :phrase_slop, Integer
+        key :analyze_wildcard, Boolean
+        key :auto_generate_phrase_queries, Boolean
+        
+        def to_mongo_query(options = {})
+          
+          raise "QueryStringQuery doesn't support mongo execution"
+          
+        end
+        
+        def to_es_query
+  
+          return {:query_string => self.attributes.except("_type")}
+
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/range_filter.rb

@@ -0,0 +1,11 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class RangeFilter < RangeQuery
+        plugin AbstractFilter  
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/range_query.rb

@@ -0,0 +1,57 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class RangeQuery < AbstractQuery
+        
+        key :field, String
+        key :path, String
+        
+        key :from
+        key :to
+        key :include_lower, :default => true
+        key :include_upper, :default => true
+        key :boost, Float
+        
+        def to_mongo_query(options = {})
+          
+          range_params = {}
+          
+          if from
+            cmd = include_lower ? '$gte' : '$gt'
+            range_params.merge!({cmd => from})
+          end
+          
+          if to
+            cmd = include_upper ? '$lte' : '$lt'
+            range_params.merge!({cmd => to})
+          end
+          
+          if options[:negated]
+            return {mongo_abs_field => {"$not" => range_params}}
+          else
+            return {mongo_abs_field => range_params}
+          end
+          
+        end
+        
+        def to_es_query
+          
+          raise "must have either :from or :to" if from.nil? and to.nil?
+          
+          range_params = {}
+          range_params.merge!({:from => from}) unless from.nil?
+          range_params.merge!({:to => to}) unless to.nil?
+          range_params.merge!({:include_lower => include_lower}) unless (from.nil? or include_lower == true)
+          range_params.merge!({:include_upper => include_upper}) unless (to.nil? or include_upper == true)
+          range_params.merge!({:boost => boost}) unless boost.nil?
+      
+          return {:range => {es_abs_field => range_params}}
+          
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/scored_filter.rb

@@ -0,0 +1,29 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class ScoredFilter
+        
+        include MongoMapper::EmbeddedDocument
+        plugin  MmUsesNoId
+        
+        one :filter, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key :boost,  Float
+        key :script, String
+        
+        def to_mongo_query
+          filter.to_mongo_query
+        end
+        
+        def to_es_query
+          filter_params = attributes.except('filter', '_type')
+          filter_params[:filter] = filter.to_es_query
+          filter_params
+        end
+        
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/single_bool_filter.rb

@@ -0,0 +1,66 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class SingleBoolFilter < AbstractQuery
+        
+        plugin AbstractFilter
+        
+        many :filters, :class_name => 'MmEsSearch::Api::Query::AbstractQuery'
+        key  :_cache, Boolean
+      
+        def operator_name
+          self.class.to_s[0..-7].split('::').last.downcase #strip off "Filter" and downcase e.g. AndFilter => filter
+        end
+        
+        def optimize_filters
+          opt_filters = []
+          filters.each do |child_filter|
+            if child_filter.is_a?(self.class)
+              opt_filters += child_filter.filters
+            else
+              opt_filters << child_filter
+            end
+          end
+          opt_filters
+        end
+        
+        def to_mongo_query(options = {})
+          case self
+          when NotFilter
+            negated_options = options.merge({:negated => !options[:negated]})
+            return AndFilter.new(:filters => filters).to_mongo_query(negated_options)
+          else
+            opt_filters = optimize_filters
+            filter_array = opt_filters.map {|filter| filter.to_mongo_query(options)}
+            if filter_array.length == 1
+              return filter_array.first
+            else
+              return {"$#{operator_name}" => filter_array}
+            end
+          end
+        end
+        
+        def to_es_query
+          opt_filters = optimize_filters
+          filter_array = opt_filters.map {|query| query.to_es_query}
+          params = {}
+          case self
+          when NotFilter
+            if filter_array.length == 1
+              return {operator_name => {:filter => filter_array.first}}
+            else
+              return {operator_name => {:filter => {:or => filter_array}}}
+            end
+          else
+            params = {operator_name => filter_array}
+          end
+          params.merge!({"_cache" => _cache}) unless _cache.nil?
+          return params
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/term_filter.rb

@@ -0,0 +1,11 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class TermFilter < TermQuery
+        plugin AbstractFilter  
+      end
+      
+    end
+  end
+end

data/lib/mm_es_search/api/query/term_query.rb

@@ -0,0 +1,34 @@
+module MmEsSearch
+  module Api
+    module Query
+      
+      class TermQuery < AbstractQuery
+        
+        key :field, String
+        key :path, String
+        
+        key :value #string or number
+        key :boost, Float
+        
+        def to_mongo_query(options = {})
+          
+          if options[:negated]
+            {mongo_abs_field => {'$ne' => value}}
+          else
+            {mongo_abs_field => value}
+          end
+        end
+        
+        def to_es_query
+          if boost
+            {:term => {es_abs_field => {:value => value, :boost => boost}}} 
+          else
+            {:term => {es_abs_field => value}}
+          end
+        end
+        
+      end
+      
+    end
+  end
+end