feat(reset): Add reset method

jojoee · Mar 7, 2017 · e2cb907 · e2cb907
1 parent e52efc6
commit e2cb907
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -79,6 +79,10 @@ filter.remove('b00b');
 filter.remove(['b00b', 'b@@b']);
 ```
 
+### filter.reset()
+
+Reset word list by using default dictionary (also remove word that manually add)
+
 ### filter.clearList()
 
 Clear all profanity words

diff --git a/src/index.js b/src/index.js
@@ -1,5 +1,7 @@
 var fs = require('fs');
-var words = JSON.parse(fs.readFileSync('./dictionary/default.json', 'utf8'));
+var wordDictionary = [];
+wordDictionary['default'] = JSON.parse(fs.readFileSync('./dictionary/default.json', 'utf8'));
+var words = JSON.parse(JSON.stringify(wordDictionary['default']));
 
 /**
  * LeoProfanity
@@ -63,6 +65,22 @@ var LeoProfanity = {
     return replacementWord;
   },
 
+  /**
+   * Get word dictionary
+   * Now, we only have default dictionary
+   * (private)
+   *
+   * @param {string} dictionaryName
+   */
+  getDictionary: function(dictionaryName) {
+    var result = [];
+    if (wordDictionary[dictionaryName] !== 'undefined') {
+      result = JSON.parse(JSON.stringify(wordDictionary[dictionaryName]));
+    }
+
+    return result;
+  },
+
   /**
    * Return all profanity words
    *
@@ -168,6 +186,15 @@ var LeoProfanity = {
     return this;
   },
 
+  /**
+   * Reset word list by using default dictionary (also remove word that manually add)
+   */
+  reset: function() {
+    words = this.getDictionary('default');
+
+    return this;
+  },
+
   /**
    * Clear word list
    */

diff --git a/test/index.spec.js b/test/index.spec.js
@@ -63,6 +63,20 @@ describe('Profanity filter', function() {
     expect(filter.list()).to.not.include('boobs');
   });
 
+  it('reset', function() {
+    // reset
+    filter.reset();
+
+    // prepare data to test by adding new 2 bad words
+    var numberOfCurrentWords = filter.list().length;
+    filter.add(['badword1', 'badword2']);
+    expect(filter.list().length).to.equal(numberOfCurrentWords + 2);
+
+    // reset
+    filter.reset();
+    expect(filter.list().length).to.equal(numberOfCurrentWords);
+  });
+
   it('clearList', function() {
     filter.clearList();
     expect(filter.list()).to.be.empty;