spambox 0.1 → 0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 45fbd091608fd08f1bb429828cf6dcd81c88b294
-  data.tar.gz: 2f3230af01f1ec27ac9e2449cfed2c4d24508ba6
+  metadata.gz: 7538f0fdf33dd4c140ecad13748399982c733556
+  data.tar.gz: ce640a122f6ef847dab35a1047759824c8c58593
 SHA512:
-  metadata.gz: c6025360f8a0312d3330ff9842095721ebe1d5d5437f294c8174ebc0f3dada9d6418bb6779da5123164f79826948704febbcf06548cbeed5bd43be5a53d16704
-  data.tar.gz: da398dd530a974a36bef8c2bb4d44bf8e7e0de17f2e9d27f48238d11618c50471681fb7e9cc57d7dd621a891100304ba648eb23855a8bc77b4557b0358e5bc71
+  metadata.gz: 3c5ad784eb07a7ea22e23389f42b5b739007be09173fd57d322589144bcd9e096c6b022d8d18a56347c3db2b5a739ebd841f3dd3d60a0fe45565efcee654bdc7
+  data.tar.gz: 636a4d87c4361b4f33e9e33505b89e81fc91416b54797752c4f1dc81f1bc7c8dc1db3ef0ddbcd30b1ff26de216059b605eb0994125e00a1d41ec26bf99a8171f

data/lib/spambox.rb

@@ -1,45 +1,58 @@
 require "json"
-require 'open-uri'
+require "open-uri"
+require "sanitize"
 
 class Spambox
-  attr_accessor :object
-  attr_reader :blacklist
-
   FORMBOX_BLACKLIST_URL = "https://formbox.es/spam-keywords.json"
 
-  def initialize(object)
+  def initialize(object, options = { allow_html: false })
     @object = object
-    @occurences = 0
-    @total_words = 0
-    @blacklist = blacklist
-
-    process(object)
+    @options = options
   end
 
   def spam_score
-    (@occurences.to_f / @total_words.to_f * 100).round
+    return 100 if @options[:allow_html] == false && contains_html?
+    return 0 if (string_length = sanitized_string.split.size.to_f) == 0
+    (count_occurences.to_f / string_length * 100).round
+  end
+
+  def update_blacklist
+    request = open(FORMBOX_BLACKLIST_URL)
+
+    Dir.mkdir("tmp") unless Dir.exists?("tmp")
+    File.open("tmp/spam_triggers.json", 'w') { |f| f.write(request.read) }
   end
 
   private
 
-  def process(object)
-    if object.is_a? String
-      @occurences = check(object)
-      @total_words = object.split.size
-    elsif object.is_a? ActiveRecord::Base
-      object.class.column_names.each do |column_name|
-        attribute = object.send(column_name).to_s
-        @occurences += check(attribute)
-        @total_words += attribute.split.size
-      end
+  def sanitized_string
+    Sanitize.fragment(flat_string)
+  end
+
+  def flat_string
+    case @object
+    when String
+      return @object
+    when Array
+      return @object.join(" ")
+    when Hash
+      return flatten_hash(h).values.join(" ")
+    when ActiveRecord::Base
+      return @object.class.column_names.map do |column|
+        @object.send(column).to_s
+      end.join(" ")
     else
-      raise ArgumentError, "SpamFilter only supports String- or ActiveRecord objects, got #{object.class}."
+      raise ArgumentError, "SpamFilter only supports Array-, String- or ActiveRecord objects, got #{@object.class}."
     end
   end
 
-  def check(attribute)
-    @blacklist.map { |s|
-      attribute.to_s.downcase.scan(s.downcase).count * s.split.size
+  def contains_html?
+    return true if sanitized_string != flat_string
+  end
+
+  def count_occurences
+    blacklist.map { |s|
+      sanitized_string.downcase.scan(s.downcase).count * s.split.size
     }.inject(0, :+)
   end
 
@@ -57,10 +70,8 @@ class Spambox
     "{}"
   end
 
-  def update_blacklist
-    request = open(FORMBOX_BLACKLIST_URL)
-
-    Dir.mkdir("tmp") unless Dir.exists?("tmp")
-    File.open("tmp/spam_triggers.json", 'w') { |f| f.write(request.read) }
+  def flatten_hash(h)
+    return { [] => h } unless h.is_a?(Hash)
+    Hash[h.map { |a,v1| flatten_hash(v1).map { |b,v2| [[a] + b, v2] } }.flatten(1)]
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: spambox
 version: !ruby/object:Gem::Version
-  version: '0.1'
+  version: '0.2'
 platform: ruby
 authors:
 - Joshua Jansen
@@ -9,7 +9,21 @@ autorequire:
 bindir: bin
 cert_chain: []
 date: 2015-12-04 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sanitize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
 description: Spambox uses Formbox.es API to check for spam.
 email:
 - joshuajansen88@gmail.com