omelettes 0.2.0 → 0.5.0

data/Gemfile

@@ -1,3 +1,5 @@
 source :rubygems
 
 gemspec
+
+gem 'rake'

data/Rakefile

@@ -1,3 +1,4 @@
+require 'bundler/gem_tasks'
 require 'rubygems'
 require 'rake'
 require 'rspec/core/rake_task'

data/lib/omelettes/column.rb

@@ -22,7 +22,7 @@ module Omelettes
       case name
       when :hardened
         return value
-      when :name, :first_name, :last_name
+      when :first_name, :last_name
         return Faker::Name.send(name)
       when :city, :state, :country, :street_address, :street_name, :zip_code
         return Faker::Address.send(name)
@@ -32,8 +32,10 @@ module Omelettes
         return Faker::Internet.send(name)
       when :paragraph, :paragraphs, :sentence, :sentences, :words
         return Faker::Lorem.send(name)
-      when :phone
+      when :phone, :contact_phone, :fax
         return Faker::PhoneNumber.phone_number
+      when :url, :website
+        return Faker::Internet.domain_name
       else
         return Omelettes::Obfuscate.obfuscate(value)
       end

data/lib/omelettes/model_additions.rb

@@ -23,15 +23,19 @@ module Omelettes
       alias :harden :ignore
     end
 
-    def obfuscate(column_name)
-      column = self.class.column_config(column_name)
-      original_value = self.send(column_name)
-      if column
-        value = column.process(original_value)
-      else
-        value = Column.default(column_name, original_value)
+    def obfuscate(column_names)
+      attributes = {}
+      column_names.each do |column_name|
+        column = self.class.column_config(column_name)
+        original_value = self.send(column_name)
+        if column
+          value = column.process(original_value)
+        else
+          value = Column.default(column_name, original_value)
+        end
+        attributes[column_name] = value
       end
-      self.update_attribute(column_name, value) if original_value != value
+      self.class.where(:id => self.id).update_all(attributes)
     end
 
     def self.included(base)

data/lib/omelettes/obfuscate.rb

@@ -5,20 +5,40 @@ module Omelettes
         total_tables = 0
         total_attributes = 0
         Words.load(word_list || "/usr/share/dict/words")
+        processed = []
         tables.each do |table|
           next if ignore_table?(table)
-          print "\nProcessing #{model(table).name}" unless silent
+          processed << table
+          pbar = ProgressBar.new(model(table).name, model(table).count) unless silent
           model(table).find_each do |object|
-            model(table).columns.each do |column|
-              next if ignore_column?(column.name) || column.type != :string
-              object.obfuscate(column.name)
+            begin
+              object.obfuscate(columns_for_table(table))
               total_attributes += 1
+            rescue => e
+              puts e.message
+              next
+            ensure
+              pbar.inc unless silent
             end
-            print "." unless silent
           end
+          pbar.finish unless silent
           total_tables += 1
         end
-        print "\n" unless silent
+        if @callback
+          @callback.call
+        end
+        unless silent
+          puts " Obfuscation Report (the following tables and columns were processed)"
+          puts ("----------" * 8)
+          processed.each do |table|
+            label = model(table).name
+            columns_for_table(table).join(',').scan(/.{0,60}/).each do |columns|
+              puts "%20.20s | %-60.60s" % [label, columns] unless columns.blank?
+              label = ""
+            end
+          end
+        end
+
         [total_tables, total_attributes]
       end
 
@@ -36,6 +56,8 @@ module Omelettes
         ignore_tables.each do |ignore|
           return true if table.match(ignore).to_s == table
         end
+        return true unless columns_for_table(table).any?
+        return true if model(table).count == 0
         false
       end
 
@@ -46,13 +68,20 @@ module Omelettes
         false
       end
 
+      def columns_for_table(table)
+        @columns_for_table ||= {}
+        @columns_for_table[table] ||= model(table).columns.select {|column| !ignore_column?(column.name) && (column.type == :string || column.type == :text)}.map(&:name)
+      end
+
+      def post_cook(&callback)
+        @callback = callback
+      end
+
       def obfuscate(string)
         return nil if string.nil?
-        result = []
-        string.split(/(\s+)|([[:punct:]])/).each do |word|
-          result << (word.match(/[a-zA-Z]+/).nil? ? word : Words.replace(word))
-        end
-        result.join("")
+        string.split(/(\s+)|([[:punct:]])/).map do |word|
+          word.match(/[a-zA-Z]+/).nil? ? word : Words.replace(word)
+        end.join("")
       end
 
       attr_accessor :ignore_tables

data/lib/omelettes/words.rb

@@ -7,16 +7,16 @@ module Omelettes
       end
 
       def add(word)
-        key = "#{word[0].downcase}#{word.length}"
+        key = "#{word[0,1].downcase}#{word.length}"
         @word_hash[key] ||= []
         @word_hash[key] << word
       end
 
       def replace(word)
-        key = "#{word[0].downcase}#{word.length}"
+        key = "#{word[0,1].downcase}#{word.length}"
         valid_words = (@word_hash[key] || [])
         new_word = valid_words[rand(valid_words.size)]
-        return new_word.send(word[0].upcase == word[0] ? :capitalize : :downcase) unless new_word.nil?
+        return new_word.send(word[0,1].upcase == word[0,1] ? :capitalize : :downcase) unless new_word.nil?
         word
       end
 

data/lib/omelettes.rb

@@ -4,6 +4,8 @@ rescue LoadError
   require 'activerecord' unless defined?(ActiveRecord)
 end
 
+require 'progressbar'
+
 require 'omelettes/column'
 require 'omelettes/model_additions'
 require 'omelettes/obfuscate'

data/lib/tasks/omelettes.rake

@@ -1,10 +1,6 @@
 namespace :db do
   desc "Obfuscate the database with Omelettes"
   task :cook => :environment do
-    print "Are you sure you want to scramble all strings in the database? (y/n): "
-    input = $stdin.gets.strip
-    if input == "y"
-      Omelettes::Obfuscate.cook
-    end
+    Omelettes::Obfuscate.cook
   end
 end

metadata

@@ -1,59 +1,67 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: omelettes
-version: !ruby/object:Gem::Version 
+version: !ruby/object:Gem::Version
+  version: 0.5.0
   prerelease: 
-  version: 0.2.0
 platform: ruby
-authors: 
+authors:
 - Mark Sim
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2011-07-27 00:00:00 Z
-dependencies: 
-- !ruby/object:Gem::Dependency 
+date: 2012-02-10 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
   name: rspec
-  prerelease: false
-  requirement: &id001 !ruby/object:Gem::Requirement 
+  requirement: &70280857854620 !ruby/object:Gem::Requirement
     none: false
-    requirements: 
+    requirements:
     - - ~>
-      - !ruby/object:Gem::Version 
+      - !ruby/object:Gem::Version
         version: 2.1.0
   type: :development
-  version_requirements: *id001
-- !ruby/object:Gem::Dependency 
-  name: faker
   prerelease: false
-  requirement: &id002 !ruby/object:Gem::Requirement 
+  version_requirements: *70280857854620
+- !ruby/object:Gem::Dependency
+  name: faker
+  requirement: &70280857854040 !ruby/object:Gem::Requirement
     none: false
-    requirements: 
+    requirements:
     - - ~>
-      - !ruby/object:Gem::Version 
+      - !ruby/object:Gem::Version
         version: 0.9.5
   type: :runtime
-  version_requirements: *id002
-- !ruby/object:Gem::Dependency 
-  name: activerecord
   prerelease: false
-  requirement: &id003 !ruby/object:Gem::Requirement 
+  version_requirements: *70280857854040
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: &70280857853440 !ruby/object:Gem::Requirement
     none: false
-    requirements: 
+    requirements:
     - - ~>
-      - !ruby/object:Gem::Version 
-        version: "3.0"
+      - !ruby/object:Gem::Version
+        version: '3.0'
   type: :runtime
-  version_requirements: *id003
-description: Low-to-no configuration solution for obfuscating sensitive data in your database.
+  prerelease: false
+  version_requirements: *70280857853440
+- !ruby/object:Gem::Dependency
+  name: progressbar
+  requirement: &70280857853000 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70280857853000
+description: Low-to-no configuration solution for obfuscating sensitive data in your
+  database.
 email: mark@quarternotecoda.com
 executables: []
-
 extensions: []
-
 extra_rdoc_files: []
-
-files: 
+files:
 - lib/generators/omelettes/config/config_generator.rb
 - lib/generators/omelettes/config/templates/omelettes.rb
 - lib/generators/omelettes/config/USAGE
@@ -78,30 +86,29 @@ files:
 - init.rb
 homepage: http://github.com/marksim/omelettes
 licenses: []
-
 post_install_message: 
 rdoc_options: []
-
-require_paths: 
+require_paths:
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
+required_ruby_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      version: "0"
-required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -4130218374050120156
+required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
       version: 1.3.4
 requirements: []
-
 rubyforge_project: omelettes
-rubygems_version: 1.8.4
+rubygems_version: 1.8.10
 signing_key: 
 specification_version: 3
 summary: Database obfuscation.
 test_files: []
-