tidus 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 14edabf3b34700e5d9ab484afd54af8348604833
+  data.tar.gz: 7df5e0603faff6661bebeb09ead1dd3442c1c8f2
+SHA512:
+  metadata.gz: 02cde5036ec4facd91e9ac0b4dc1e3dc97394a25ffd745ea31d070475b33223be0d9268fd5fb40ecfd6d5fe3a8dc059cb6a0d5042660fe6434b7de2b43c26480
+  data.tar.gz: c5c883a7aeefd19482c566cdaf367956362f17e013856320b9db3066cba8c03b8ba9fed314169b8689a973218912ff04088ff9e0a2ce3075a5e1730067a88e55

data/lib/tasks/views.rake

@@ -0,0 +1,43 @@
+namespace :db do
+  desc "Clears all the views which are currently existing"
+  task :clear_views do
+    Rails.application.eager_load! if defined?(Rails)
+    ActiveRecord::Base.descendants.each do |c|
+      next if c.table_name == "schema_migrations"
+      puts "Clearing view '#{c.view_name}' for table '#{c.table_name}'"
+
+      ActiveRecord::Base.connection.execute(
+        "DROP VIEW IF EXISTS #{c.view_name}"
+      )
+    end
+  end
+
+  desc "Generates all the views for the models"
+  task :generate_views do
+    Rails.application.eager_load! if defined?(Rails)
+    ActiveRecord::Base.descendants.each do |c|
+      next if c.table_name == "schema_migrations"
+
+      if ActiveRecord::Base.connection.table_exists? c.table_name
+        puts "Generating view '#{c.view_name}' for table '#{c.table_name}'"
+
+        ActiveRecord::Base.connection.execute(
+          "CREATE VIEW #{c.view_name} AS " +
+          "(SELECT #{c.view_columns.join(', ')} " +
+          "FROM #{c.table_name})"
+        )
+      end
+    end
+  end
+end
+
+Rake::Task["db:migrate"].enhance ["db:clear_views"]
+Rake::Task["db:rollback"].enhance ["db:clear_views"]
+
+Rake::Task["db:migrate"].enhance do
+  Rake::Task["db:generate_views"].invoke
+end
+
+Rake::Task["db:rollback"].enhance do
+  Rake::Task["db:generate_views"].invoke
+end

data/lib/tidus/anonymization.rb

@@ -0,0 +1,63 @@
+module Tidus
+	module Anonymization
+    def view_postfix
+      @view_postfix || "anonymized"
+    end
+
+    def view_postfix=(val)
+      @view_postfix = val
+    end
+
+    def view_name
+      @view_name || "#{table_name}_#{view_postfix}"
+    end
+
+    def view_name=(val)
+      @view_name = val
+    end
+
+    def view_columns
+      @view_columns ||= {}
+      default_view_columns.merge(@view_columns)
+                          .map{ |k,v| ["#{v} AS #{k}"] }
+                          .flatten
+    end
+
+    def default_view_columns
+      defaults = {}
+      column_names.each do |column|
+        defaults[column.to_sym] = "#{table_name}.#{column}"
+      end
+      defaults
+    end
+
+    def anonymizes(*attributes)
+      @view_columns ||= {}
+
+      options = attributes.extract_options!.dup
+      columns = attributes - [options]
+
+      raise ArgumentError, "You need to supply at least one attribute" if attributes.empty?
+      raise ArgumentError, "You need to supply a strategy" if options[:strategy].blank?
+
+      columns.each do |column|
+        key = options[:strategy].to_s.camelize
+
+        begin
+          if key.include?('::')
+            klass = key.constantize
+          else
+            klass = const_get("ActiveRecordAnonymize::#{key}Anonymizer")
+          end
+        rescue NameError
+          raise ArgumentError, "Unknown anonymizer: '#{key}'"
+        end
+
+        @view_columns[column.to_sym] = klass.anonymize(table_name, column, options)
+      end
+    end
+
+	end
+end
+
+ActiveRecord::Base.extend Tidus::Anonymization

data/lib/tidus/strategies/cond_anonymizer.rb

@@ -0,0 +1,44 @@
+module Tidus
+  class CondAnonymizer
+    def self.anonymize(table_name, column_name, options = {})
+      adapter = ActiveRecord::Base.connection.instance_values["config"][:adapter]
+      case adapter
+      when "postgresql"
+        name = "#{table_name}.#{column_name}"
+
+        type = options[:result_type] || "text"
+        default = options[:default].nil? ? "#{name}::#{type}" : "'#{options[:default]}'::#{type}"
+
+        if options[:conditions].blank?
+          raise "Missing option :conditions for CondAnonymizer on #{name}"
+        elsif options[:conditions].kind_of?(Array)
+          conditions = options[:conditions]
+        else
+          conditions = [options[:conditions]]
+        end
+
+        command = "CASE "
+
+        conditions.each do |cond|
+          raise ":column for condition must be set" if cond[:column].blank?
+          raise ":value for condition must be set" if cond[:value].nil?
+          raise ":result for condition must be set" if cond[:result].nil?
+          cond_column = cond[:column]
+          cond_value  = cond[:value]
+          cond_type   = cond[:type] || "text"
+          comparator  = cond[:comparator] || "="
+          cond_result = cond[:result]
+
+          command += "WHEN ((#{table_name}.#{cond_column})::#{cond_type} #{comparator} " +
+                     "'#{cond_value}'::#{cond_type}) THEN '#{cond_result}'::#{type} "
+        end
+
+        command += "ELSE #{default} END"
+
+        return command
+      else
+        raise "#{self.name} not implemented for #{adapter}"
+      end
+    end
+  end
+end

data/lib/tidus/strategies/email_anonymizer.rb

@@ -0,0 +1,21 @@
+module Tidus
+  class EmailAnonymizer
+    def self.anonymize(table_name, column_name, options = {})
+      adapter = ActiveRecord::Base.connection.instance_values["config"][:adapter]
+      case adapter
+      when "postgresql"
+        name = "#{table_name}.#{column_name}"
+        options[:length] ||= 15
+
+        return "CASE WHEN ((#{name})::text ~~ '%@%'::text) " +
+               "THEN (((\"left\"(md5((#{name})::text), #{options[:length]}) || '@'::text) " +
+                "|| split_part((#{name})::text, '@'::text, 2)))::character varying " +
+                "ELSE #{name} END"
+      else
+        raise "#{self.name} not implemented for #{adapter}"
+      end
+    end
+  end
+end
+
+

data/lib/tidus/strategies/null_anonymizer.rb

@@ -0,0 +1,13 @@
+module Tidus
+  class NullAnonymizer
+    def self.anonymize(table_name, column_name, options = {})
+      adapter = ActiveRecord::Base.connection.instance_values["config"][:adapter]
+      case adapter
+      when "postgresql"
+        return "NULL::unknown"
+      else
+        raise "#{self.name} not implemented for #{adapter}"
+      end
+    end
+  end
+end

data/lib/tidus/strategies/overlay_anonymizer.rb

@@ -0,0 +1,21 @@
+module Tidus
+  class OverlayAnonymizer
+    def self.anonymize(table_name, column_name, options = {})
+      adapter = ActiveRecord::Base.connection.instance_values["config"][:adapter]
+      case adapter
+      when "postgresql"
+        name = "#{table_name}.#{column_name}"
+
+        raise "Missing option :start for OverlayAnonymizer on #{name}" if options[:start].blank?
+        raise "Missing option :length for OverlayAnonymizer on #{name}" if options[:length].blank?
+
+        overlay_char = options[:char] || "X"
+        overlay = overlay_char * options[:length]
+        return "\"overlay\"((#{name})::text, " +
+               "'#{overlay}'::text, #{options[:start]})"
+      else
+        raise "#{self.name} not implemented for #{adapter}"
+      end
+    end
+  end
+end

data/lib/tidus/strategies/static_anonymizer.rb

@@ -0,0 +1,17 @@
+module Tidus
+  class StaticAnonymizer
+    def self.anonymize(table_name, column_name, options = {})
+      adapter = ActiveRecord::Base.connection.instance_values["config"][:adapter]
+      case adapter
+      when "postgresql"
+        raise "Missing option :value for StaticAnonymizer on #{table_name}.#{column_name}" if options[:value].blank?
+
+        return "'#{options[:value]}'"
+      else
+        raise "#{self.name} not implemented for #{adapter}"
+      end
+    end
+  end
+end
+
+

data/lib/tidus/strategies/text_anonymizer.rb

@@ -0,0 +1,28 @@
+module Tidus
+  class TextAnonymizer
+    def self.anonymize(table_name, column_name, options = {})
+      base = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZCßüäöÜÄÖ"
+      adapter = ActiveRecord::Base.connection.instance_values["config"][:adapter]
+      case adapter
+      when "postgresql"
+        return "translate((#{table_name}.#{column_name})::text, " +
+               "'#{base}'::text, '#{generate_mapping(base)}'::text)"
+      else
+        raise "#{self.name} not implemented for #{adapter}"
+      end
+    end
+
+    private
+      def self.generate_mapping(base)
+        result = ""
+        base.split("").each do |letter|
+          if letter == letter.upcase
+            result += ("A".."Z").to_a.shuffle.first
+          else
+            result += ("a".."z").to_a.shuffle.first
+          end
+        end
+        result
+      end
+  end
+end

data/lib/tidus/version.rb

@@ -0,0 +1,3 @@
+module Tidus
+  VERSION = "1.0.0"
+end

data/lib/tidus.rb

@@ -0,0 +1,14 @@
+require "rake"
+require "active_record"
+
+require "tidus/version"
+require "tidus/anonymization"
+require "tidus/strategies/cond_anonymizer.rb"
+require "tidus/strategies/email_anonymizer.rb"
+require "tidus/strategies/null_anonymizer.rb"
+require "tidus/strategies/overlay_anonymizer.rb"
+require "tidus/strategies/static_anonymizer.rb"
+require "tidus/strategies/text_anonymizer.rb"
+
+load "active_record/railties/databases.rake"
+load "tasks/views.rake"

metadata

@@ -0,0 +1,110 @@
+--- !ruby/object:Gem::Specification
+name: tidus
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Tobias Schoknecht
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-02-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 10.0.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 10.0.4
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.14.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.14.1
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.3.10
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.3.10
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.2'
+description: Creates views which allow anonymization of database tables.
+email:
+- tobias.schoknecht@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/tasks/views.rake
+- lib/tidus.rb
+- lib/tidus/anonymization.rb
+- lib/tidus/strategies/cond_anonymizer.rb
+- lib/tidus/strategies/email_anonymizer.rb
+- lib/tidus/strategies/null_anonymizer.rb
+- lib/tidus/strategies/overlay_anonymizer.rb
+- lib/tidus/strategies/static_anonymizer.rb
+- lib/tidus/strategies/text_anonymizer.rb
+- lib/tidus/version.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Gem for creating anonymization views.
+test_files: []