redis_autocomplete 0.1.3 → 0.1.4

data/VERSION

@@ -1 +1 @@
-0.1.3
+0.1.4

data/lib/redis_autocomplete.rb

@@ -26,6 +26,21 @@ class RedisAutocomplete
   def add_words(words, set_name = nil)
     words.flatten.compact.uniq.each { |word| add_word word, set_name }
   end
+  
+  def remove_word(word, set_name = nil, remove_stems = true)
+    set_name ||= @set_name
+    @redis.zrem(set_name, "#{word}#{@terminal}")
+    # remove_word_stem is inefficient and is best done later on with a cron job
+    remove_word_stem(word, set_name) if remove_stems
+  end
+  
+  def remove_word_stem(stem, set_name)
+    if suggest(stem, 1, set_name).empty?
+      @redis.zrem(set_name, stem)
+      remove_word_stem(stem[0...-1], set_name)
+    end    
+  end
+  protected :remove_word_stem
 
   def suggest(prefix, count = 10, set_name = nil)
     set_name ||= @set_name

data/redis_autocomplete.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{redis_autocomplete}
-  s.version = "0.1.3"
+  s.version = "0.1.4"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Joe Johnston (simple10)", "Ben Woosley (Empact)", "Salvatore Sanfilippo (antirez)"]

data/spec/redis_autocomplete_spec.rb

@@ -78,6 +78,45 @@ describe RedisAutocomplete do
         end
       end
     end
+    
+    describe "#remove_word" do
+      context "with default options" do
+        before do
+          @r.remove_word('Catherine')
+        end
+      
+        it "should not include word after removing it" do
+          @r.suggest('Cath').should_not include('Catherine')
+        end
+      
+        it "should remove unique word stems" do
+          @r.redis.zrank(@set, 'Catherine').should be_nil
+          @r.redis.zrank(@set, 'Catheri').should == nil
+          @r.redis.zrank(@set, 'Cather').should == nil
+          @r.redis.zrank(@set, 'Cathe').should == nil
+          @r.redis.zrank(@set, 'Cath').should_not == nil
+        end
+      end
+      
+      context "when remove_stems is false" do
+        before do
+          @r.remove_word('Catherine', nil, false)
+        end
+
+        it "should not include word after removing it" do
+          @r.suggest('Cath').should_not include('Catherine')
+        end
+
+        it "should remove unique word stems" do
+          @r.redis.zrank(@set, 'Catherine+').should be_nil
+          @r.redis.zrank(@set, 'Catherine').should_not be_nil
+          @r.redis.zrank(@set, 'Catheri').should_not be_nil
+          @r.redis.zrank(@set, 'Cather').should_not be_nil
+          @r.redis.zrank(@set, 'Cathe').should_not be_nil
+          @r.redis.zrank(@set, 'Cath').should_not be_nil
+        end
+      end
+    end
   end
 
   context "with :case_sensitive => false" do

metadata

@@ -2,7 +2,7 @@
 name: redis_autocomplete
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors: 
 - Joe Johnston (simple10)
@@ -106,7 +106,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: -3325895771545261792
+      hash: -1638761389715983542
       segments: 
       - 0
       version: "0"