mwmitchell-rsolr-ext 0.5.2 → 0.5.3

data/lib/rsolr-ext/hash_methodizer.rb

@@ -0,0 +1,43 @@
+#
+# A hash modifier that creates method readers from key names.
+# NOTE: reader methods are created recursively.
+# The method names are the same as the key names,
+# except that the values are snake-cased, for example:
+#   - QTime -> q_time
+#   - debugQuery -> debug_query
+#
+class RSolr::Ext::HashMethodizer
+  
+  class << self
+    
+    def snake_case(v)
+      v = v.to_s
+      return v.downcase if v =~ /^[A-Z]+$/
+      v.gsub(/([A-Z]+)(?=[A-Z][a-z]?)|\B[A-Z]/, '_\&') =~ /_*(.*)/
+      return $+.downcase
+    end
+    
+    def methodize!(h)
+      h.keys.each do |k|
+        meth = snake_case(k)
+        h.instance_variable_set("@#{k}", h[k])
+        h.instance_eval <<-RUBY
+        def #{meth}
+          val = @#{k}
+          if val.respond_to?(:each_pair)
+            RSolr::Ext::HashMethodizer.methodize!(val)
+          elsif val.is_a?(Array)
+            val.each do |item|
+              RSolr::Ext::HashMethodizer.methodize!(item) if item.respond_to?(:each_pair)
+            end
+          end
+          val
+        end
+        RUBY
+      end
+
+    end
+  
+  end
+  
+end

data/lib/rsolr-ext/mapable.rb

@@ -0,0 +1,44 @@
+# A module that provides method mapping capabilities.
+# The basic idea is to pass in a hash to the #map method,
+# the map method then goes through a list of keys (MAPPED_PARAMS) to
+# be processed. Each key name can match a key in the input hash.
+# If there is a match, a method by the name of "map_#{key}" is
+# called with the following args: input[key], output_hash
+# The method is responsible for processing the value.
+# The return value from the method is not used.
+#
+# For example: if the mapped params list has :query,
+# there should be a method like: map_query(input_value, output_hash)
+# The output_hash is the final hash the the #map method returns,
+# so whatever you do to output_hash gets returned in the end.
+module RSolr::Ext::Mapable
+  
+  # accepts an input hash.
+  # prepares a return hash by copying the input.
+  # runs through all of the keys in MAPPED_PARAMS.
+  # calls any mapper methods that match the current key in MAPPED_PARAMS.
+  # The mapped keys from the input hash are deleted.
+  # returns a new hash.
+  def map(input)
+    result = input.dup
+    self.class::MAPPED_PARAMS.each do |meth|
+      input_value = result.delete(meth)
+      next if input_value.to_s.empty?
+      send("map_#{meth}", input_value, result)
+    end
+    result
+  end
+  
+  # creates an array where the "existing_value" param is first
+  # and the "new_value" is the last.
+  # All empty/nil items are removed.
+  # the return result is either the result of the
+  # array being joined on a space, or the array itself.
+  # "auto_join" should be true or false.
+  def append_to_param(existing_value, new_value, auto_join=true)
+    values = [existing_value, new_value]
+    values.delete_if{|v|v.nil?}
+    auto_join ? values.join(' ') : values
+  end
+  
+end

data/lib/rsolr-ext/request/queryable.rb

@@ -0,0 +1,42 @@
+# a module to help the creation of solr queries.
+module RSolr::Ext::Request::Queryable
+  
+  # Wraps a string around double quotes
+  def quote(value)
+    %("#{value}")
+  end
+  
+  # builds a solr range query from a Range object
+  def build_range(r)
+    "[#{r.min} TO #{r.max}]"
+  end
+  
+  # builds a solr query fragment
+  # if "quote_string" is true, the values will be quoted.
+  # if "value" is a string/symbol, the #to_s method is called
+  # if the "value" is an array, each item in the array is 
+  # send to build_query (recursive)
+  # if the "value" is a Hash, a fielded query is built
+  # where the keys are used as the field names and
+  # the values are either processed as a Range or
+  # passed back into build_query (recursive)
+  def build_query(value, quote_string=false)
+    case value
+    when String,Symbol
+      return quote_string ? quote(value.to_s) : value.to_s
+    when Array
+      value.collect do |v|
+        build_query(v, quote_string)
+      end.flatten
+    when Hash
+      return value.collect do |(k,v)|
+        if v.is_a?(Range)
+          "#{k}:#{build_range(v)}"
+        else
+          "#{k}:#{build_query(v, quote_string)}"
+        end
+      end.flatten
+    end
+  end
+  
+end

data/lib/rsolr-ext.rb

@@ -15,7 +15,7 @@ module RSolr
   
   module Ext
     
-    VERSION = '0.5.2'
+    VERSION = '0.5.3'
     
     autoload :Request, 'rsolr-ext/request.rb'
     autoload :Response, 'rsolr-ext/response.rb'

data/rsolr-ext.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
 	s.name = "rsolr-ext"
-	s.version = "0.5.2"
+	s.version = "0.5.3"
 	s.date = "2009-03-17"
 	s.summary = "An extension lib for RSolr"
 	s.email = "goodieboy@gmail.com"
@@ -13,10 +13,10 @@ Gem::Specification.new do |s|
     "lib/core_ext.rb",
     "lib/mash.rb",
     
-    "lib/rsolr-ext/hash_methodizer",
+    "lib/rsolr-ext/hash_methodizer.rb",
+    "lib/rsolr-ext/mapable.rb",
     
-    "lib/rsolr-ext/request/dismax.rb",
-    "lib/rsolr-ext/request/standard.rb",
+    "lib/rsolr-ext/request/queryable.rb",
     "lib/rsolr-ext/request.rb",
     
     "lib/rsolr-ext/response/doc_ext.rb",

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: mwmitchell-rsolr-ext
 version: !ruby/object:Gem::Version 
-  version: 0.5.2
+  version: 0.5.3
 platform: ruby
 authors: 
 - Matt Mitchell
@@ -25,9 +25,9 @@ extra_rdoc_files:
 files: 
 - lib/core_ext.rb
 - lib/mash.rb
-- lib/rsolr-ext/hash_methodizer
-- lib/rsolr-ext/request/dismax.rb
-- lib/rsolr-ext/request/standard.rb
+- lib/rsolr-ext/hash_methodizer.rb
+- lib/rsolr-ext/mapable.rb
+- lib/rsolr-ext/request/queryable.rb
 - lib/rsolr-ext/request.rb
 - lib/rsolr-ext/response/doc_ext.rb
 - lib/rsolr-ext/response/facet_paginator.rb