obscenity 1.0.0

data/lib/obscenity.rb

@@ -0,0 +1,36 @@
+require 'obscenity/error'
+require 'obscenity/config'
+require 'obscenity/base'
+require 'obscenity/version'
+
+module Obscenity extend self
+  
+  attr_accessor :config
+  
+  def configure(&block)
+    @config = Config.new(&block)
+  end
+  
+  def config
+    @config ||= Config.new
+  end
+  
+  def profane?(word)
+    Obscenity::Base.profane?(word)
+  end
+  
+  def sanitize(text)
+    Obscenity::Base.sanitize(text)
+  end
+  
+  def replacement(chars)
+    Obscenity::Base.replacement(chars)
+  end
+
+  def offensive(text)
+    Obscenity::Base.offensive(text)
+  end
+  
+  
+end
+  

data/lib/obscenity/active_model.rb

@@ -0,0 +1,18 @@
+if defined?(ActiveModel)
+  module ActiveModel
+    module Validations
+      class ObscenityValidator < ActiveModel::EachValidator
+        
+        def validate_each(record, attribute, value)
+          if options.present? && options.has_key?(:sanitize)
+            object = record.respond_to?(:[]) ? record[attribute] : record.send(attribute)
+            object = Obscenity.replacement(options[:replacement]).sanitize(object)
+          else
+            record.errors.add(attribute, options[:message] || 'cannot be profane') if Obscenity.profane?(value)
+          end
+        end
+        
+      end
+    end
+  end
+end

data/lib/obscenity/base.rb

@@ -0,0 +1,73 @@
+module Obscenity
+  class Base
+    class << self
+      
+      def blacklist
+        @blacklist ||= set_list_content(Obscenity.config.blacklist)
+      end
+      
+      def blacklist=(value)
+        @blacklist = value == :default ? set_list_content(Obscenity::Config.new.blacklist) : value
+      end
+      
+      def whitelist
+        @whitelist ||= set_list_content(Obscenity.config.whitelist)
+      end
+      
+      def whitelist=(value)
+        @whitelist = value == :default ? set_list_content(Obscenity::Config.new.whitelist) : value
+      end
+    
+      def profane?(text)
+        return(false) unless text.to_s.size >= 3
+        blacklist.each do |foul|
+          return(true) if text =~ /\b#{foul}\b/i && !whitelist.include?(foul)
+        end
+        false
+      end
+      
+      def sanitize(text)
+        return(text) unless text.to_s.size >= 3
+        blacklist.each do |foul|
+          text.gsub!(/\b#{foul}\b/i, replace(foul)) unless whitelist.include?(foul)
+        end
+        @scoped_replacement = nil
+        text
+      end
+      
+      def replacement(chars)
+        @scoped_replacement = chars
+        self
+      end
+      
+      def offensive(text)
+        words = []
+        return(words) unless text.to_s.size >= 3
+        blacklist.each do |foul|
+          words << foul if text =~ /\b#{foul}\b/i && !whitelist.include?(foul)
+        end
+        words.uniq
+      end
+      
+      def replace(word)
+        content = @scoped_replacement || Obscenity.config.replacement
+        case content
+        when :vowels then word.gsub(/[aeiou]/i, '*')
+        when :stars  then '*' * word.size
+        when :default, :garbled then '$@!#%'
+        else content
+        end
+      end
+      
+      private
+      def set_list_content(list)
+        case list
+        when Array then list
+        when String, Pathname then YAML.load_file( list.to_s )
+        else []
+        end
+      end
+      
+    end
+  end
+end

data/lib/obscenity/config.rb

@@ -0,0 +1,51 @@
+module Obscenity
+  class Config
+    
+    attr_accessor :replacement
+    
+    DEFAULT_WHITELIST = []
+    DEFAULT_BLACKLIST = File.dirname(__FILE__) + "/../../config/blacklist.yml"
+    
+    def initialize
+      yield(self) if block_given?
+      validate_config_options
+    end
+    
+    def replacement
+      @replacement ||= :garbled
+    end
+    
+    def blacklist
+      @blacklist ||= DEFAULT_BLACKLIST
+    end
+    
+    def blacklist=(value)
+      @blacklist = value == :default ? DEFAULT_BLACKLIST : value
+    end
+    
+    def whitelist
+      @whitelist ||= DEFAULT_WHITELIST
+    end
+    
+    def whitelist=(value)
+      @whitelist = value == :default ? DEFAULT_WHITELIST : value
+    end
+    
+    private
+    def validate_config_options
+      [@blacklist, @whitelist].each{ |content| validate_list_content(content) if content }
+    end
+    
+    def validate_list_content(content)
+      case content
+      when Array    then !content.empty?       || raise(Obscenity::EmptyContentList.new('Content array is empty.'))
+      when String   then File.exists?(content) || raise(Obscenity::UnkownContentFile.new("Content file can't be found."))
+      when Pathname then content.exist?        || raise(Obscenity::UnkownContentFile.new("Content file can't be found."))
+      when Symbol   then content == :default   || raise(Obscenity::UnkownContent.new("The only accepted symbol is :default."))
+      else
+        raise Obscenity::UnkownContent.new("The content can be either an Array, Pathname, or String path to a .yml file.")
+      end
+    end
+    
+  end
+end

data/lib/obscenity/error.rb

@@ -0,0 +1,7 @@
+module Obscenity
+  class Error < RuntimeError; end
+  
+  class UnkownContent     < Error; end
+  class UnkownContentFile < Error; end
+  class EmptyContentList  < Error; end
+end

data/lib/obscenity/rack.rb

@@ -0,0 +1,89 @@
+module Rack
+  class Obscenity
+        
+    def initialize(app, options = {})  
+      @app, @options = app, options
+    end  
+    
+    def call(env)
+      rejectable  = false
+      post_params = Rack::Utils.parse_query(env['rack.input'].read, "&")
+      get_params  = Rack::Utils.parse_query(env['QUERY_STRING'],    "&")
+      
+      if @options.has_key?(:reject)
+        rejactable = validate_rejectability_of( select_params(:reject, get_params.update(post_params)) )
+      
+      elsif @options.has_key?(:sanitize)
+        get_params  = sanitize_contents_of(get_params)
+        post_params = sanitize_contents_of(post_params)
+        
+        env['QUERY_STRING'] = Rack::Utils.build_query(get_params)
+        env['rack.input']   = StringIO.new(Rack::Utils.build_query(post_params))
+      end
+      
+      rejactable ? reject : continue(env)
+    end
+    
+    private
+    def continue(env)
+      @app.call(env)
+    end
+    
+    def reject
+      length, content = 0, ''
+      if @options[:reject].is_a?(Hash) 
+        if (message = @options[:reject][:message]).present?
+          content = message
+          length  = message.size
+        elsif (path = @options[:reject][:path]).present?
+          if (path = ::File.expand_path(path)) && ::File.exists?(path)
+            content = ::File.read(path)
+            length  = content.size
+          end
+        end
+      end
+
+      [422, {'Content-Type' => 'text/html', 'Content-Length' => length.to_s}, [content]]
+    end
+
+    def validate_rejectability_of(params = {})
+      should_reject_request = false
+      params.each_pair do |param, value|
+        if value.is_a?(Hash)
+          validates_rejectability_of(value)
+        elsif value.is_a?(String)
+          next unless value.size >= 3
+          if ::Obscenity.profane?(value)
+            should_reject_request = true
+            break
+          end
+        else
+          next
+        end
+      end
+      should_reject_request
+    end
+        
+    def sanitize_contents_of(params)
+      sanitized_params   = {}
+      replacement_method = @options[:sanitize].is_a?(Hash) && @options[:sanitize][:replacement]
+      select_params(:sanitize, params).each{|param, value|
+        if value.is_a?(String)
+          next unless value.size >= 3
+          sanitized_params[param] = ::Obscenity.replacement(replacement_method).sanitize(value)
+        else
+          next
+        end
+      }
+      params.update(sanitized_params)
+    end
+    
+    def select_params(key, params = {})
+      if @options[key].is_a?(Hash) && @options[key][:params].is_a?(Array)
+        params.select{ |param, vvalue| @options[key][:params].include?(param.to_sym) } 
+      else
+        params
+      end
+    end
+  end
+end

data/lib/obscenity/version.rb

@@ -0,0 +1,5 @@
+module Obscenity
+
+  VERSION = '1.0.0'
+  
+end

data/obscenity.gemspec

@@ -0,0 +1,84 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "obscenity"
+  s.version = "1.0.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Thiago Jackiw"]
+  s.date = "2012-05-28"
+  s.description = " Obscenity is a profanity filter gem for Ruby/Rubinius, Rails (through ActiveModel), and Rack middleware "
+  s.email = "tjackiw@gmail.com"
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.md"
+  ]
+  s.files = [
+    ".document",
+    ".gitignore",
+    ".travis.yml",
+    "Gemfile",
+    "Gemfile.lock",
+    "LICENSE.txt",
+    "README.md",
+    "Rakefile",
+    "config/blacklist.yml",
+    "config/international.yml",
+    "lib/obscenity.rb",
+    "lib/obscenity/active_model.rb",
+    "lib/obscenity/base.rb",
+    "lib/obscenity/config.rb",
+    "lib/obscenity/error.rb",
+    "lib/obscenity/rack.rb",
+    "lib/obscenity/version.rb",
+    "obscenity.gemspec",
+    "test/helper.rb",
+    "test/static/422.html",
+    "test/test_active_model.rb",
+    "test/test_base.rb",
+    "test/test_config.rb",
+    "test/test_obscenity.rb",
+    "test/test_rack.rb",
+    "test/test_version.rb"
+  ]
+  s.homepage = "http://github.com/tjackiw/obscenity"
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.24"
+  s.summary = "Obscenity is a profanity filter gem for Ruby/Rubinius, Rails (through ActiveModel), and Rack middleware"
+  s.test_files = ["test/helper.rb", "test/static/422.html", "test/test_active_model.rb", "test/test_base.rb", "test/test_config.rb", "test/test_obscenity.rb", "test/test_rack.rb", "test/test_version.rb"]
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<shoulda>, [">= 0"])
+      s.add_development_dependency(%q<rdoc>, ["~> 3.12"])
+      s.add_development_dependency(%q<bundler>, [">= 0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.8.3"])
+      s.add_development_dependency(%q<activemodel>, ["~> 3.0"])
+      s.add_development_dependency(%q<rack>, [">= 0"])
+      s.add_development_dependency(%q<rake>, [">= 0"])
+    else
+      s.add_dependency(%q<shoulda>, [">= 0"])
+      s.add_dependency(%q<rdoc>, ["~> 3.12"])
+      s.add_dependency(%q<bundler>, [">= 0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.8.3"])
+      s.add_dependency(%q<activemodel>, ["~> 3.0"])
+      s.add_dependency(%q<rack>, [">= 0"])
+      s.add_dependency(%q<rake>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<shoulda>, [">= 0"])
+    s.add_dependency(%q<rdoc>, ["~> 3.12"])
+    s.add_dependency(%q<bundler>, [">= 0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.8.3"])
+    s.add_dependency(%q<activemodel>, ["~> 3.0"])
+    s.add_dependency(%q<rack>, [">= 0"])
+    s.add_dependency(%q<rake>, [">= 0"])
+  end
+end
+

data/test/helper.rb

@@ -0,0 +1,33 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'active_model'
+require 'obscenity'
+require 'obscenity/active_model'
+
+module Dummy
+  class BaseModel
+    include ActiveModel::Validations
+  
+    attr_accessor :title
+
+    def initialize(attr_names)
+      attr_names.each{ |k,v| send("#{k}=", v) }
+    end
+  end
+end
+
+class Test::Unit::TestCase
+end

data/test/static/422.html

@@ -0,0 +1 @@
+We don't accept profanity

data/test/test_active_model.rb

@@ -0,0 +1,69 @@
+require 'helper'
+
+class TestActiveModel < Test::Unit::TestCase
+  
+  def generate_new_class(name, options = {})
+    Dummy.send(:remove_const, name) if Dummy.const_defined?(name)
+    klass = Class.new(Dummy::BaseModel) do
+      validates :title, options
+    end
+    Dummy.const_set(name, klass)
+  end
+
+  should "be invalid when title is profane" do
+    klass = generate_new_class("Post", obscenity: true)
+    post  = klass.new(title: "He who poops, shits itself")
+    assert !post.valid?
+    assert post.errors.has_key?(:title)
+    assert_equal ['cannot be profane'], post.errors[:title]
+  end
+
+  should "be invalid when title is profane and should include a custom error message" do
+    klass = generate_new_class("Post", obscenity: { message: "can't be profane!" })
+    post  = klass.new(title: "He who poops, shits itself")
+    assert !post.valid?
+    assert post.errors.has_key?(:title)
+    assert_equal ["can't be profane!"], post.errors[:title]
+  end
+
+  should "sanitize the title using the default replacement" do
+    klass = generate_new_class("Post", obscenity: { sanitize: true })
+    post  = klass.new(title: "He who poops, shits itself")
+    assert post.valid?
+    assert !post.errors.has_key?(:title)
+    assert_equal "He who poops, $@!#% itself", post.title
+  end
+  
+  should "sanitize the title using the :garbled replacement" do
+    klass = generate_new_class("Post", obscenity: { sanitize: true, replacement: :garbled })
+    post  = klass.new(title: "He who poops, shits itself")
+    assert post.valid?
+    assert !post.errors.has_key?(:title)
+    assert_equal "He who poops, $@!#% itself", post.title
+  end
+
+  should "sanitize the title using the :stars replacement" do
+    klass = generate_new_class("Post", obscenity: { sanitize: true, replacement: :stars })
+    post  = klass.new(title: "He who poops, shits itself")
+    assert post.valid?
+    assert !post.errors.has_key?(:title)
+    assert_equal "He who poops, ***** itself", post.title
+  end
+
+  should "sanitize the title using the :vowels replacement" do
+    klass = generate_new_class("Post", obscenity: { sanitize: true, replacement: :vowels })
+    post  = klass.new(title: "He who poops, shits itself")
+    assert post.valid?
+    assert !post.errors.has_key?(:title)
+    assert_equal "He who poops, sh*ts itself", post.title
+  end
+
+  should "sanitize the title using a custom replacement" do
+    klass = generate_new_class("Post", obscenity: { sanitize: true, replacement: '[censored]' })
+    post  = klass.new(title: "He who poops, shits itself")
+    assert post.valid?
+    assert !post.errors.has_key?(:title)
+    assert_equal "He who poops, [censored] itself", post.title
+  end
+
+end