acts_as_scrubbable 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4234c5c4fc17e4e56651d0af73d4d8618479722f
-  data.tar.gz: 1cced9cfe5e6c3a798326778370f10ca34332aab
+  metadata.gz: 2b879327a376524d60458ad12c6a7688b5d85baf
+  data.tar.gz: 52411d6ea51c0a3f4484ef09c96195cea42bae9f
 SHA512:
-  metadata.gz: 4ce7605fe9907554adcab36eec514ed28320062e778f012a321ecf6dece86ed8249d1401632156c615b21ef4e816677f04f76cd38a7929bb9036aa0d07700352
-  data.tar.gz: 5ec1de24a3bedf7497b6d33205811800a8a7736b0c23b03a98eb26d862971e15e5ba338a682019bb72b9d22c89b9e96642353117f51ba741080b252a1695fd38
+  metadata.gz: bb3958c928fc147ab33945e5e045db624715e584c72d15eb4b86718b9da3613730c0389ec17b567b24a7fcb0b200849cd1b7e4cfe923143bc508d2d4f88fe5db
+  data.tar.gz: e49d143b35922305445915b2bc88138399cdb10e6670ee801fd3b2e4d2f3824332cefcf1e6ca188caa5fd6f458f205152f7f457883803d6315295e5ce9c13a17

data/README.md

@@ -53,10 +53,15 @@ I, [2015-11-05T14:09:25.615155 #64194]  INFO -- : Scrub Complete!
 
 ### Extending
 
-You may find the need to extend or add additional generators
+You may find the need to extend or add additional generators or an after_hook
 
 ```ruby
 ActsAsScrubbable.configure do |c|
   c.add :email_with_prefix, -> { "prefix-#{Faker::Internet.email}" }
+
+  c.after_hook do
+    puts "Running after commit"
+    ActiveRecord::Base.connection.execute("SELECT * FROM FOO")
+  end
 end
 ```

data/lib/acts_as_scrubbable.rb

@@ -11,10 +11,19 @@ module ActsAsScrubbable
   autoload :Scrub
   autoload :VERSION
 
+
   def self.configure(&block)
     yield self
   end
 
+  def self.after_hook(&block)
+    @after_hook = block
+  end
+
+  def self.execute_after_hook
+    @after_hook.call if @after_hook
+  end
+
   def self.add(key, value)
     ActsAsScrubbable.scrub_map[key] = value
   end
@@ -26,15 +35,21 @@ module ActsAsScrubbable
       :first_name        => -> { Faker::Name.first_name },
       :last_name         => -> { Faker::Name.first_name },
       :middle_name       => -> { Faker::Name.name },
-      :full_name         => -> { Faker::Name.name },
+      :name              => -> { Faker::Name.name },
       :email             => -> { Faker::Internet.email },
       :name_title        => -> { Faker::Name.title },
       :company_name      => -> { Faker::Company.name },
       :street_address    => -> { Faker::Address.street_address },
       :secondary_address => -> { Faker::Address.secondary_address },
+      :zip_code          => -> { Faker::Address.zip_code },
+      :state_abbr        => -> { Faker::Address.state_abbr },
+      :state             => -> { Faker::Address.state },
       :city              => -> { Faker::Address.city },
       :latitude          => -> { Faker::Address.latitude },
-      :longitude         => -> { Faker::Address.longitude }
+      :longitude         => -> { Faker::Address.longitude },
+      :username          => -> { Faker::Internet.user_name },
+      :boolean           => -> { [true, false ].sample },
+      :school            => -> { Faker::University.name }
     }
   end
 end

data/lib/acts_as_scrubbable/scrub.rb

@@ -5,10 +5,17 @@ module ActsAsScrubbable
       if self.class.scrubbable?
         _updates = {}
 
-        scrubbable_fields.each do |key, value|
-          next if self.send(key).blank?
+        scrubbable_fields.each do |_field, value|
+          unless self.respond_to?(_field)
+            raise ArgumentError, "#{self.class} do not respond to #{_field}"
+          end
+          next if self.send(_field).blank?
 
-          _updates[key] = ActsAsScrubbable.scrub_map[value].call
+          if ActsAsScrubbable.scrub_map.keys.include?(value)
+            _updates[_field] = ActsAsScrubbable.scrub_map[value].call
+          else
+            puts "Undefined scrub: #{value} for #{self.class}#{_field}"
+           end
         end
 
         self.update_columns(_updates) unless _updates.empty?

data/lib/acts_as_scrubbable/tasks.rb

@@ -20,18 +20,26 @@ namespace :scrub do
     @logger.warn "Scrubbing classes".red
 
     Rails.application.eager_load! # make sure all the classes are loaded
+
+    @total_scrubbed = 0
+
     ActiveRecord::Base.descendants.sort_by{|d| d.to_s }.each do |ar_class|
       next unless ar_class.scrubbable?
 
-      @logger.info "Scrubbing #{ar_class}".green
-
+      scrubbed_count = 0
       ar_class.find_in_batches do |batch|
         batch.each do |obj|
           obj.scrub!
+          scrubbed_count += 1
         end
       end
-
+      @logger.info "Scrubbed #{scrubbed_count} #{ar_class} objects".green
+      @total_scrubbed += scrubbed_count
     end
+
+    ActsAsScrubbable.execute_after_hook
+
+    @logger.info "#{@total_scrubbed} scrubbed objects".blue
     @logger.info "Scrub Complete!".white
   end
 end

data/lib/acts_as_scrubbable/version.rb

@@ -1,3 +1,3 @@
 module ActsAsScrubbable
-  VERSION = '0.0.2'
+  VERSION = '0.0.3'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: acts_as_scrubbable
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Samer Masry
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-11-05 00:00:00.000000000 Z
+date: 2015-11-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport