Fixed remove. Added a functions to remove all documents with the same base

Author: Crenshinibon

packages/spomet/client.coffee

@@ -11,6 +11,9 @@ Spomet.update = (findable) ->
 Spomet.remove = (findable) ->
     Meteor.call 'spometRemove', findable, () ->
 
+Spomet.removeBase = (baseId) ->
+    Meteor.call 'spometRemoveBase', baseId, () ->
+
 class Spomet.Search
 
     constructor: () ->

packages/spomet/indexes/index.coffee

@@ -1,4 +1,4 @@
-index = (findable, tokens, collection) ->
+indexTokens = (findable, tokens, collection) ->
     tokens.forEach (token) ->
         doc = 
             docId: findable.docId
@@ -18,8 +18,8 @@ index = (findable, tokens, collection) ->
                 documentsCount: 1, 
                 documents: [doc]
 
-tokenizeWithIndex = (index, text) ->
-    tokenizer = new index.Tokenizer
+tokenizeWithIndex = (ind, text) ->
+    tokenizer = new ind.Tokenizer
     text.split('').forEach (c, i) ->
         tokenizer.parseCharacter c, i
     tokenizer.finalize()
@@ -61,8 +61,8 @@ Index.add = (findable, callback) ->
 
     unless Documents.exists findable
         #init indexer for each index
-        tokenizers = Spomet.options.indexes.map (index) ->
-            new index.Tokenizer
+        tokenizers = Spomet.options.indexes.map (i) ->
+            new i.Tokenizer
 
         #normalize and tokenize over all indexes in one go
         findable.text.split('').forEach (c, pos) ->
@@ -71,8 +71,8 @@ Index.add = (findable, callback) ->
 
         tokenizers.forEach (t) ->
             t.finalize()
-            index findable, t.tokens, t.collection
-        
+            indexTokens findable, t.tokens, t.collection
+            
         Documents.add findable, tokenizers.map((i) -> i.tokens).reduce (s, a) -> s.concat a
         callback? findable.docId, 'Document added to all indexes'
     else
@@ -92,8 +92,8 @@ Index.setup = () ->
 
 
 Index.remove = (docId, indexName, remToken) ->
-    index = i for i in Spomet.options.indexes when i.name is indexName
-    index.collection.update {token: remToken},
+    ind = i for i in Spomet.options.indexes when i.name is indexName
+    ind.collection.update {token: remToken},
         $pull: {documents: {docId: docId}}
         $inc: {documentsCount: -1}
 

packages/spomet/server.coffee

@@ -68,25 +68,33 @@ cleanupSearches = () ->
 Spomet.add = (findable, callback) ->
     cleanupSearches()
     Spomet.Index.add findable, callback
-    
+ 
+ 
+removeTokens = (indexTokens) ->
+    result = {}
+    indexTokens.forEach (e) ->
+        id = e.indexName + e.token
+        result[id] = 
+            token: e.token
+            indexName: e.indexName
+    _.values result
+               
 Spomet.remove = (docId) ->
     cleanupSearches()
     doc = Spomet.Documents.collection.findOne {docId: docId}
     if doc?
-        removeTokens = (indexTokens) ->
-            result = {}
-            indexTokens.forEach (e) ->
-                id = e.indexName + e.token
-                result[id] = 
-                    token: e.token
-                    indexName: e.indexName
-            _.values result
-        
         removeTokens(doc.indexTokens).forEach (rToken) ->
             Spomet.Index.remove docId, rToken.indexName, rToken.token
 
         Spomet.Documents.collection.remove {_id: doc._id}
 
+Spomet.removeBase = (baseId) ->
+    cleanupSearches()
+    c = Spomet.Documents.collection.find {'findable.base': baseId}
+    c.forEach (e) ->
+        removeTokens(e.indexTokens).forEach (rToken) ->
+            Spomet.Index.remove e.docId, rToken.indexName, rToken.token
+        Spomet.Documents.collection.remove {_id: e._id}
 
 Spomet.reset = () ->
     cleanupSearches()
@@ -107,6 +115,8 @@ Meteor.methods
             Spomet.remove findable.docId
         else if findable?
             Spomet.remove findable
+    spometRemoveBase: (baseId) ->
+        Spomet.removeBase baseId
     spometUpdate: (findable) ->
         Spomet.remove findable.previousVersionDocId
         Spomet.add findable