bad_word_detector 0.0.2 → 0.0.3

data/README.md

@@ -49,6 +49,12 @@ You can also set own rules:
 finder = BadWordDetector.new rules, library, whitelist
 ```
 
+Or you can use string filenames for YAML files of same format:
+
+```ruby
+finder = BadWordDetector.new 'my_rules.yaml', 'my_library.yaml', 'my_whitelist.yaml'
+```
+
 ## Contributing
 
 1. Fork it

data/lib/bad_word_detector.rb

@@ -20,8 +20,30 @@ class BadWordDetector
   # 
   def initialize(rules = nil, library = nil,  whitelist = nil)
     confdir = File.expand_path(File.dirname(__FILE__) + "/conf")
-    rules ||= YAML.load_file("#{confdir}/rules.yaml")
-    library ||= YAML.load_file("#{confdir}/library.yaml")
+    rules_file = if rules.is_a? String 
+      rules 
+    else 
+      "#{confdir}/rules.yaml" 
+    end
+    library_file = if library.is_a? String 
+      library 
+    else 
+      "#{confdir}/library.yaml" 
+    end
+    whitelist_file = if whitelist.is_a? String 
+      whitelist 
+    else 
+      "#{confdir}/whitelist.yaml" 
+    end
+    unless rules.is_a? Hash
+      rules = YAML.load_file(rules_file)  
+    end
+    unless library.is_a? Array
+      library = YAML.load_file(library_file)
+    end
+    unless whitelist.is_a? Array
+      whitelist = YAML.load_file(whitelist_file)
+    end
 
     @rule_sets = rules.select do |key, _|
       key.to_s.length == 1
@@ -43,9 +65,8 @@ class BadWordDetector
       end
       [key, rule]
     end
-
     @library = PrefixTree.new library
-    @whitelist = Whitelist.new whitelist || YAML.load_file("#{confdir}/whitelist.yaml")
+    @whitelist = Whitelist.new whitelist
     true
   end
 

data/lib/bad_word_detector/version.rb

@@ -1,3 +1,3 @@
 class BadWordDetector
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/test/test_bad_word_detector.rb

@@ -67,6 +67,6 @@ end
 
 class Finder
   def self.finder
-    @finder ||= BadWordDetector.new YAML.load_file(File.dirname(__FILE__)+"/rules.yaml")
+    @finder ||= BadWordDetector.new File.dirname(__FILE__)+"/rules.yaml"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bad_word_detector
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: