class Spambox

# File lib/spambox.rb, line 5
def initialize(object, options = { allow_html: false })
  @object = object
  @options = options
end

# File lib/spambox.rb, line 10
def spam_score
  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

# File lib/spambox.rb, line 54
def blacklist
  path = File.expand_path(File.dirname(__FILE__))
  file = File.read("#{path}/spam-keywords.json")
  JSON.parse(file)
end

# File lib/spambox.rb, line 44
def contains_html?
  return true if sanitized_string != flat_string
end

# File lib/spambox.rb, line 48
def count_occurences
  blacklist.map do |s|
    sanitized_string.downcase.scan(s.downcase).count * s.split.size
  end.inject(0, :+)
end

# File lib/spambox.rb, line 22
def flat_string
  @flat_string ||= squish(object_to_string)
end

# File lib/spambox.rb, line 60
def flatten_hash_values(hash)
  return hash if hash.is_a? String
  hash.flat_map { |_, v| [*flatten_hash_values(v)] }
end

# File lib/spambox.rb, line 26
def object_to_string
  case @object
  when String
    return @object
  when Array
    return @object.join(" ")
  when Hash
    return flatten_hash_values(h).values.join(" ")
  when ActiveRecord::Base
    return @object.class.column_names.map do |column|
      @object.send(column).to_s
    end.join(" ")
  else
    fail ArgumentError, "SpamFilter only supports Array-,
      String- or ActiveRecord objects, got #{@object.class}."
  end
end

# File lib/spambox.rb, line 18
def sanitized_string
  Sanitize.fragment(flat_string)
end

# File lib/spambox.rb, line 65
def squish(string)
  string.gsub(/[[:space:]]+/, ' ').strip
end