picky 4.0.4 → 4.0.5

data/lib/picky/index.rb

@@ -239,7 +239,7 @@ module Picky
     #
     # === Options
     # * precision: Default is 1 (20% error margin, very fast), up to 5 (5% error margin, slower) makes sense.
-    # * anchor: Where to anchor the geo grid.
+    # * anchor: Where to anchor the grid.
     # * ... all options of #category.
     #
     def ranged_category category_name, range, options = {}

data/lib/picky/query/allocation.rb

@@ -15,7 +15,7 @@ module Picky
       #
       #
       def initialize index, combinations
-        @combinations      = combinations
+        @combinations = combinations
 
         # Could this be rewritten?
         #

data/lib/picky/query/indexes.rb

@@ -25,16 +25,8 @@ module Picky
         IndexesCheck.check_backends indexes
 
         @indexes = indexes
-
-        map_categories
-      end
-      def map_categories
-        @mapper = Query::QualifierCategoryMapper.new
-        @indexes.each do |index|
-          index.each_category do |category|
-            @mapper.add category
-          end
-        end
+        
+        @mapper = QualifierCategoryMapper.new indexes # TODO Move out?
       end
 
       # Ignore the categories with these qualifiers.
@@ -51,6 +43,19 @@ module Picky
         @ignored_categories += qualifiers.map { |qualifier| @mapper.map qualifier }.compact
         @ignored_categories.uniq!
       end
+      
+      # Restrict categories to the given ones.
+      #
+      # Functionally equivalent as if indexes didn't
+      # have the categories at all.
+      #
+      # Note: Probably only makes sense when an index
+      # is used in multiple searches. If not, why even
+      # have the categories?
+      #
+      def only *qualifiers
+        @mapper.restrict_to *qualifiers
+      end
 
       # Returns a number of prepared (sorted, reduced etc.) allocations for the given tokens.
       #

data/lib/picky/query/qualifier_category_mapper.rb

@@ -9,11 +9,16 @@ module Picky
     class QualifierCategoryMapper # :nodoc:all
 
       attr_reader :mapping
-
+      
       #
       #
-      def initialize
+      def initialize indexes
         @mapping = {}
+        indexes.each do |index|
+          index.each_category do |category|
+            add category
+          end
+        end
       end
 
       #
@@ -35,6 +40,19 @@ module Picky
 
         @mapping[qualifier.intern]
       end
+      
+      # Restricts the given categories.
+      #
+      def restrict user_qualified
+        if @restricted
+          user_qualified ? @restricted & user_qualified : @restricted 
+        else
+          user_qualified
+        end
+      end
+      def restrict_to *qualifiers
+        @restricted = qualifiers.map { |qualifier| map qualifier }.compact
+      end
 
     end
 

data/lib/picky/query/token.rb

@@ -57,13 +57,20 @@ module Picky
 
       # Translates this token's qualifiers into actual categories.
       #
-      # Note:
-      # If this is not done, there is no mapping.
+      # Note: If this is not done, there is no mapping.
+      # Note: predefined is an Array of mapped categories.
       #
-      # THINK Can this be improved somehow?
+      # TODO Rename @user_defined_categories. It could now also be predefined by the query.
       #
       def categorize mapper
-        @user_defined_categories = @qualifiers && @qualifiers.map do |qualifier|
+        @user_defined_categories ||= extract_predefined mapper
+      end
+      def extract_predefined mapper
+        user_qualified = categorize_with mapper, @qualifiers
+        mapper.restrict user_qualified
+      end
+      def categorize_with mapper, qualifiers
+        qualifiers && qualifiers.map do |qualifier|
           mapper.map qualifier
         end.compact
       end

data/lib/picky/search.rb

@@ -8,10 +8,9 @@ module Picky
   # * holds one or more indexes
   # * offers an interface to query these indexes.
   #
-  # You connect URL paths to indexes via a Query.
-  #
-  # We recommend to not use this directly, but connect it to an URL and query through one of these
-  # (Protip: Use "curl 'localhost:8080/query/path?query=exampletext')" in a Terminal.
+  # Example:
+  #   search = Picky::Search.new index1, index2
+  #   search.search 'query'
   #
   class Search
 
@@ -25,6 +24,7 @@ module Picky
                   :boosts
 
     delegate :ignore,
+             :only,
              :to => :indexes
 
     # Takes:
@@ -38,8 +38,8 @@ module Picky
     #             [:title, :isbn] => +1
     #   end
     #
-    def initialize *index_definitions
-      @indexes = Query::Indexes.new *index_definitions
+    def initialize *indexes
+      @indexes = Query::Indexes.new *indexes
 
       instance_eval(&Proc.new) if block_given?
 

data/spec/functional/only_spec.rb

@@ -0,0 +1,52 @@
+# encoding: utf-8
+#
+require 'spec_helper'
+
+describe 'Search#only' do
+
+  it 'offers the option only' do
+    index = Picky::Index.new :only do
+      category :text1
+      category :text2
+      category :text3
+    end
+
+    index.add Struct.new(:id, :text1, :text2, :text3).new(1, 'text1', 'text2', 'text3')
+
+    try = Picky::Search.new index
+    try.search('text1').ids.should == [1]
+    try.search('text2').ids.should == [1]
+    try.search('text3').ids.should == [1]
+
+    try_again = Picky::Search.new index do
+      only :text1
+    end
+    try_again.search('text1').ids.should == [1]
+    try_again.search('text2').ids.should == []
+    try_again.search('text3').ids.should == []
+
+    try_again.only :text2, :text3
+    try_again.search('text1').ids.should == []
+    try_again.search('text2').ids.should == [1]
+    try_again.search('text3').ids.should == [1]
+    
+    try_again.search('text1:text1').ids.should == []
+    try_again.search('text2:text2').ids.should == [1]
+    try_again.search('text3:text3').ids.should == [1]
+    
+    try_again.search('text1,text2,text3:text1').ids.should == []
+    try_again.search('text1,text2:text1').ids.should == []
+    try_again.search('text1,text3:text1').ids.should == []
+    try_again.search('text2,text3:text1').ids.should == []
+    
+    try_again.search('text1,text2,text3:text2').ids.should == [1]
+    try_again.search('text1,text2:text2').ids.should == [1]
+    try_again.search('text1,text3:text2').ids.should == []
+    try_again.search('text2,text3:text2').ids.should == [1]
+    
+    try_again.search('text1,text2,text3:text3').ids.should == [1]
+    try_again.search('text1,text2:text3').ids.should == []
+    try_again.search('text1,text3:text3').ids.should == [1]
+    try_again.search('text2,text3:text3').ids.should == [1]
+  end
+end

data/spec/functional/regression_spec.rb

@@ -3,7 +3,7 @@
 require 'spec_helper'
 
 describe "Regression" do
-
+  
   it 'does not get confused' do
     index = Picky::Index.new :dynamic_weights do
       category :text1

data/spec/lib/query/indexes_spec.rb

@@ -7,10 +7,8 @@ describe Picky::Query::Indexes do
   end
 
   3.times do |i|
-    n       = i + 1
-    name    = :"index#{n}"
-    indexed = :"indexed#{n}"
-    let(name) { stub name, :internal_indexed => stub(indexed) }
+    name = :"index#{i+1}"
+    let(name) { Picky::Index.new(name) }
   end
 
   let(:indexes) do

data/spec/lib/query/qualifier_category_mapper_spec.rb

@@ -2,15 +2,15 @@ require 'spec_helper'
 
 describe Picky::Query::QualifierCategoryMapper do
   
-  let(:mapper) { described_class.new }
-  before(:each) do
-    @category1 = stub(:category1, :qualifiers => ['t1', 'tt1', 'ttt1'])
-    @category2 = stub(:category2, :qualifiers => [:t2, :tt2, :ttt2])
-    @category3 = stub(:category3, :qualifiers => [:t3, :tt3, :ttt3])
-    mapper.add @category1
-    mapper.add @category2
-    mapper.add @category3
+  let(:index) do
+    categories = Picky::Index.new :categories
+    @category1 = categories.category :category1, :qualifiers => ['t1', 'tt1', 'ttt1']
+    @category2 = categories.category :category2, :qualifiers => [:t2, :tt2, :ttt2]
+    @category3 = categories.category :category3, :qualifiers => [:t3, :tt3, :ttt3]
+    categories
   end
+  let(:indexes) { [index] }
+  let(:mapper) { described_class.new indexes }
   
   def self.it_should_map(qualifier, expected)
     it "should map #{qualifier} to #{expected}" do

data/spec/lib/query/token_spec.rb

@@ -11,6 +11,64 @@ describe Picky::Query::Token do
       described_class.processed('similar~', 'Similar~').should_not == described_class.processed('similar', 'Similar')
     end
   end
+  
+  describe 'categorize' do
+    let(:mapper) do
+      categories = Picky::Index.new :categories
+      @category1 = categories.category :category1
+      @category2 = categories.category :category2
+      @category3 = categories.category :category3
+      Picky::Query::QualifierCategoryMapper.new [categories]
+    end
+    context 'with qualifiers' do
+      let(:token) { described_class.processed 'category1:qualifier', 'category1:Qualifier' }
+      context 'unrestricted' do
+        it 'categorizes correctly' do
+          token.categorize(mapper).should == [@category1]
+        end
+      end
+      context 'restricted' do
+        before(:each) do
+          mapper.restrict_to :category1
+        end
+        it 'categorizes correctly' do
+          token.categorize(mapper).should == [@category1]
+        end
+      end
+      context 'restricted' do
+        before(:each) do
+          mapper.restrict_to :category2, :category3
+        end
+        it 'categorizes correctly' do
+          token.categorize(mapper).should == []
+        end
+      end
+    end
+    context 'without qualifiers' do
+      let(:token) { described_class.processed 'noqualifier', 'NoQualifier' }
+      context 'unrestricted' do
+        it 'categorizes correctly' do
+          token.categorize(mapper).should == nil
+        end
+      end
+      context 'restricted' do
+        before(:each) do
+          mapper.restrict_to :category1
+        end
+        it 'categorizes correctly' do
+          token.categorize(mapper).should == [@category1]
+        end
+      end
+      context 'restricted' do
+        before(:each) do
+          mapper.restrict_to :category2, :category3
+        end
+        it 'categorizes correctly' do
+          token.categorize(mapper).should == [@category2, @category3]
+        end
+      end
+    end
+  end
 
   describe 'similar_tokens_for' do
     let(:token) { described_class.processed 'similar~', 'Similar~' }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: picky
 version: !ruby/object:Gem::Version
-  version: 4.0.4
+  version: 4.0.5
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-12-22 00:00:00.000000000 Z
+date: 2011-12-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70357445983920 !ruby/object:Gem::Requirement
+  requirement: &70174977200100 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,21 +21,21 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70357445983920
+  version_requirements: *70174977200100
 - !ruby/object:Gem::Dependency
   name: picky-client
-  requirement: &70357445983240 !ruby/object:Gem::Requirement
+  requirement: &70174977199140 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: 4.0.4
+        version: 4.0.5
   type: :development
   prerelease: false
-  version_requirements: *70357445983240
+  version_requirements: *70174977199140
 - !ruby/object:Gem::Dependency
   name: text
-  requirement: &70357445982560 !ruby/object:Gem::Requirement
+  requirement: &70174977198140 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70357445982560
+  version_requirements: *70174977198140
 - !ruby/object:Gem::Dependency
   name: yajl-ruby
-  requirement: &70357445998020 !ruby/object:Gem::Requirement
+  requirement: &70174977197100 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +54,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70357445998020
+  version_requirements: *70174977197100
 - !ruby/object:Gem::Dependency
   name: activesupport
-  requirement: &70357445997240 !ruby/object:Gem::Requirement
+  requirement: &70174977195740 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -65,10 +65,10 @@ dependencies:
         version: '3.0'
   type: :runtime
   prerelease: false
-  version_requirements: *70357445997240
+  version_requirements: *70174977195740
 - !ruby/object:Gem::Dependency
   name: procrastinate
-  requirement: &70357445996260 !ruby/object:Gem::Requirement
+  requirement: &70174977194540 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -76,10 +76,10 @@ dependencies:
         version: '0.4'
   type: :runtime
   prerelease: false
-  version_requirements: *70357445996260
+  version_requirements: *70174977194540
 - !ruby/object:Gem::Dependency
   name: rack_fast_escape
-  requirement: &70357445995560 !ruby/object:Gem::Requirement
+  requirement: &70174977192780 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -87,7 +87,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70357445995560
+  version_requirements: *70174977192780
 description: Fast Ruby semantic text search engine with comfortable single field interface.
 email: florian.hanke+picky@gmail.com
 executables:
@@ -247,6 +247,7 @@ files:
 - spec/functional/dynamic_weights_spec.rb
 - spec/functional/exact_first_spec.rb
 - spec/functional/max_allocations_spec.rb
+- spec/functional/only_spec.rb
 - spec/functional/realtime_spec.rb
 - spec/functional/regression_spec.rb
 - spec/functional/speed_spec.rb
@@ -390,6 +391,7 @@ test_files:
 - spec/functional/dynamic_weights_spec.rb
 - spec/functional/exact_first_spec.rb
 - spec/functional/max_allocations_spec.rb
+- spec/functional/only_spec.rb
 - spec/functional/realtime_spec.rb
 - spec/functional/regression_spec.rb
 - spec/functional/speed_spec.rb