data-anonymization 0.1.2

data/lib/strategy/field/random_float_delta.rb

@@ -0,0 +1,24 @@
+module DataAnon
+  module Strategy
+    module Field
+      class RandomFloatDelta
+
+        DEFAULT_DELTA = 10.0
+
+        def initialize delta = nil
+          @delta = delta || DEFAULT_DELTA
+        end
+
+        def anonymize field
+           return range(field.value-@delta,field.value+@delta)
+        end
+
+        def range (min, max)
+          Random.new.rand * (max-min) + min
+        end
+
+
+      end
+    end
+  end
+end

data/lib/strategy/field/random_full_name.rb

@@ -0,0 +1,28 @@
+module DataAnon
+  module Strategy
+    module Field
+
+      class RandomFullName
+
+        def initialize first_names = nil, last_names = nil
+          @first_name_anonymizer = DataAnon::Strategy::Field::RandomFirstName.new(first_names)
+          @last_name_anonymizer = DataAnon::Strategy::Field::RandomLastName.new(last_names)
+        end
+
+        def anonymize field
+
+          name_words = field.value.split(' ')
+
+          anonymized_first_name = @first_name_anonymizer.anonymize(name_words[0])
+          anonymized_last_name = ""
+          for counter in (1..name_words.size-1)
+            anonymized_last_name = anonymized_last_name + " " + @last_name_anonymizer.anonymize(name_words[counter])
+          end
+
+          return anonymized_first_name + anonymized_last_name
+
+        end
+      end
+    end
+  end
+end

data/lib/strategy/field/random_int.rb

@@ -0,0 +1,23 @@
+module DataAnon
+  module Strategy
+    module Field
+
+
+      class RandomInt
+
+        def initialize from = 0, to = 100
+          @from = from
+          @to = to
+
+        end
+
+        def anonymize field
+          DataAnon::Utils::RandomInt.generate(@from,@to)
+        end
+
+      end
+
+
+    end
+  end
+end

data/lib/strategy/field/random_integer_delta.rb

@@ -0,0 +1,21 @@
+module DataAnon
+  module Strategy
+    module Field
+      class RandomIntegerDelta
+
+        DEFAULT_DELTA = 10
+
+        def initialize delta = nil
+          @delta = delta || DEFAULT_DELTA
+
+        end
+
+        def anonymize field
+          adjustment = DataAnon::Utils::RandomInt.generate(-@delta,@delta)
+          return field.value + adjustment
+        end
+      end
+
+    end
+  end
+end

data/lib/strategy/field/random_last_name.rb

@@ -0,0 +1,19 @@
+module DataAnon
+  module Strategy
+    module Field
+
+      class RandomLastName
+
+        def initialize file_path = nil
+          file = file_path || DataAnon::Utils::Resource.file('last_names.txt')
+          @names = File.read(file).split
+        end
+
+        def anonymize field
+          return @names[rand(@names.size)]
+        end
+
+      end
+    end
+  end
+end

data/lib/strategy/field/random_mailinator_email.rb

@@ -0,0 +1,20 @@
+module DataAnon
+  module Strategy
+    module Field
+
+      class RandomMailinatorEmail
+
+        def initialize
+          @email_anonymizer = DataAnon::Strategy::Field::RandomEmail.new("mailinator","com")
+        end
+
+        def anonymize field
+
+          return @email_anonymizer.anonymize(field)
+
+        end
+      end
+    end
+
+  end
+end

data/lib/strategy/field/random_phone_number.rb

@@ -0,0 +1,24 @@
+module DataAnon
+  module Strategy
+    module Field
+
+
+      class RandomPhoneNumber
+
+        def anonymize field
+          @original_phone_number = field.value
+          @anonymized_phone_number = ""
+          @original_phone_number.each_char do |char|
+            if /\d/.match(char).nil?
+              @anonymized_phone_number += char
+            else
+              @anonymized_phone_number += DataAnon::Utils::RandomInt.generate(0,9).to_s
+            end
+          end
+
+          @anonymized_phone_number
+        end
+      end
+    end
+  end
+end

data/lib/strategy/field/random_selection.rb

@@ -0,0 +1,23 @@
+module DataAnon
+  module Strategy
+    module Field
+
+
+      class RandomSelection
+
+        def initialize values
+          @values = values.class == Array ? values : [values]
+
+        end
+
+        def anonymize field
+          return @values[0] if @values.length == 1
+          @values[DataAnon::Utils::RandomInt.generate(0,(@values.length - 1))]
+        end
+
+      end
+
+
+    end
+  end
+end

data/lib/strategy/field/random_string.rb

@@ -0,0 +1,22 @@
+module DataAnon
+  module Strategy
+    module Field
+      class RandomString
+
+        def anonymize field
+
+          original_string = field.value
+          string_words = original_string.split(' ')
+          anonymized_string = ""
+
+          string_words.each do |word|
+            anonymized_string = anonymized_string + DataAnon::Utils::RandomString.generate(word.length) + " "
+          end
+
+          anonymized_string.strip
+
+        end
+      end
+    end
+  end
+end

data/lib/strategy/field/random_user_name.rb

@@ -0,0 +1,23 @@
+module DataAnon
+  module Strategy
+    module Field
+
+      class RandomUserName
+
+        DEFAULT_MIN_LENGTH = 5
+        DEFAULT_MAX_LENGTH = 10
+
+        def initialize min_length = DEFAULT_MIN_LENGTH, max_length = DEFAULT_MAX_LENGTH
+          @min_length = min_length
+          @max_length = max_length
+        end
+
+        def anonymize field
+          username_length = DataAnon::Utils::RandomInt.generate(@min_length,@max_length)
+          return DataAnon::Utils::RandomString.generate(username_length)
+
+        end
+      end
+    end
+  end
+end

data/lib/strategy/field/string_template.rb

@@ -0,0 +1,22 @@
+module DataAnon
+  module Strategy
+    module Field
+
+
+      class StringTemplate
+
+        def initialize template
+          @template = template
+        end
+
+        def anonymize field
+          context = field.instance_eval { binding }
+          eval ('"' + @template + '"'), context
+        end
+
+      end
+
+
+    end
+  end
+end

data/lib/strategy/field/user_name_template.rb

@@ -0,0 +1,22 @@
+module DataAnon
+  module Strategy
+    module Field
+
+
+      class UserNameTemplate
+
+        def initialize template
+          @template = template
+        end
+
+        def anonymize field
+          context = field.instance_eval { binding }
+          eval ('"' + @template + '"'), context
+        end
+
+      end
+
+
+    end
+  end
+end

data/lib/strategy/field/whitelist.rb

@@ -0,0 +1,17 @@
+module DataAnon
+  module Strategy
+    module Field
+
+
+      class Whitelist
+
+        def anonymize field
+          field.value
+        end
+
+      end
+
+
+    end
+  end
+end

data/lib/strategy/strategies.rb

@@ -0,0 +1,4 @@
+require 'strategy/base'
+require 'strategy/whitelist'
+require 'strategy/blacklist'
+require 'strategy/field/fields'

data/lib/strategy/whitelist.rb

@@ -0,0 +1,21 @@
+module DataAnon
+  module Strategy
+    class Whitelist < DataAnon::Strategy::Base
+
+      def process_record(index, record)
+        dest_record_map = {}
+        record.attributes.each do |field_name, field_value|
+          unless field_value.nil? || field_name.downcase == @primary_key.downcase
+            field = DataAnon::Core::Field.new(field_name, field_value, index, record)
+            field_strategy = @fields[field_name.downcase] || DataAnon::Strategy::Field::DefaultAnon.new(@user_strategies)
+            dest_record_map[field_name] = field_strategy.anonymize(field)
+          end
+        end
+        dest_record = dest_table.new dest_record_map
+        dest_record[@primary_key] = record[@primary_key]
+        dest_record.save!
+      end
+
+    end
+  end
+end

data/lib/tasks/rake_tasks.rb

@@ -0,0 +1,19 @@
+require 'rake'
+require 'rake/tasklib'
+
+module DataAnonymization
+  class RakeTasks
+    include Rake::DSL if defined? Rake::DSL
+
+    def initialize
+      desc "Task to build the clean empty destination database"
+      task :empty_dest do
+        system "rm sample-data/chinook-empty.sqlite"
+        system "sqlite3 sample-data/chinook-empty.sqlite < sample-data/chinook_schema.sql"
+      end
+    end
+
+
+  end
+end
+

data/lib/utils/database.rb

@@ -0,0 +1,53 @@
+require 'active_record'
+require 'logger'
+
+module DataAnon
+  module Utils
+
+    class MassAssignmentIgnoreSanitizer < ActiveModel::MassAssignmentSecurity::Sanitizer
+      def process_removed_attributes(attrs)
+      end
+    end
+
+    class TempDatabase < ActiveRecord::Base
+      self.abstract_class = true
+    end
+
+    class SourceDatabase < ActiveRecord::Base
+      self.abstract_class = true
+    end
+
+    class DestinationDatabase < ActiveRecord::Base
+      self.abstract_class = true
+    end
+
+    class BaseTable
+
+      def self.create_table table_name, primary_key, database
+        Class.new(database) do
+          self.table_name = table_name
+          self.primary_key = primary_key
+          self.mass_assignment_sanitizer = MassAssignmentIgnoreSanitizer.new(self)
+        end
+      end
+
+    end
+
+    class SourceTable < BaseTable
+
+      def self.create table_name, primary_key = nil
+        create_table table_name, primary_key, SourceDatabase
+      end
+
+    end
+
+    class DestinationTable < BaseTable
+
+      def self.create table_name, primary_key = nil
+        create_table table_name, primary_key, DestinationDatabase
+      end
+
+    end
+
+  end
+end

data/lib/utils/logging.rb

@@ -0,0 +1,29 @@
+require 'active_record'
+require 'logger'
+
+module DataAnon
+  module Utils
+    module Logging
+
+      def logger
+        @@logger ||= (self.logger = Logger.new(STDOUT) )
+      end
+
+      def logger= logger
+        @@logger = logger
+        ActiveRecord::Base.logger = logger
+        @@logger
+      end
+
+      def progress_logger
+        @@progress_logger ||= (self.progress_logger = Logger.new(STDOUT) )
+      end
+
+      def progress_logger= logger
+        logger.formatter = proc { |severity, datetime, progname, msg| msg }
+        @@progress_logger = logger
+      end
+
+    end
+  end
+end

data/lib/utils/random_int.rb

@@ -0,0 +1,15 @@
+module DataAnon
+  module Utils
+    class RandomInt
+
+      DEFAULT_MIN = 1
+      DEFAULT_MAX = 100
+
+      def self.generate min = nil, max = nil
+        @min = min || DEFAULT_MIN
+        @max = max || DEFAULT_MAX
+        Random.new.rand @min...@max
+      end
+    end
+  end
+end