Extract similar.

floere · Jun 30, 2015 · b183151 · b183151
1 parent f159cc7
commit b183151
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/server/lib/picky/bundle.rb b/server/lib/picky/bundle.rb
@@ -131,10 +131,11 @@ def delete
     #
     # Note: Also checks for itself.
     #
-    def similar text
-      code = similarity_strategy.encode text
+    def similar str_or_sym
+      code = similarity_strategy.encode str_or_sym
       return [] unless code
       @similarity[code] || []
+
       # similar_codes = @similarity[code]
       # if similar_codes.blank?
       #   [] # Return a simple array.

diff --git a/server/lib/picky/category_indexed.rb b/server/lib/picky/category_indexed.rb
@@ -58,6 +58,13 @@ def ids token
       end
     end
 
+    # Gets the similars for this token's text.
+    #
+    def similar token
+      bundle = bundle_for token
+      bundle.similar str_or_sym(token.text)
+    end
+
     def str_or_sym text
       if @symbol_keys
         text.to_sym

diff --git a/server/lib/picky/query/token.rb b/server/lib/picky/query/token.rb
@@ -281,7 +281,7 @@ def combination_for category
       # Returns all similar tokens for the token.
       #
       def similar_tokens_for category
-        similars = category.bundle_for(self).similar @text
+        similars = category.similar self
         similars.map do |similar|
           # The array describes all possible categories. There is only one here.
           #