tidus 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9ef15fa25eadfd7fd12896aa48d5106c10c2d743
-  data.tar.gz: 2e9e30f4a43c97913a518b13f0efaa987bb30dbf
+  metadata.gz: 16d22e9b2bf400af04c3c38e0d230a19134edf46
+  data.tar.gz: c8c0389b398d87c9b440efc68eb2586263137241
 SHA512:
-  metadata.gz: 6359526ca259d737809d0161f1800f067ea626e7231302b0f892114f7d011496f0e807040cf55c7a775bacba1c3d276d4d66e3adacd92663fba7821921683175
-  data.tar.gz: 41dd5744e5290b3fd8752710f664a7afeba92f6dec2ba8bfe03b301b9c1a1a9f6e413be15c4764af51ff39e575ce8738001bfea4d1768f78c5cc6fadd8bb5699
+  metadata.gz: 2633f5e90d829abb8e8a598302f54f5dc22def128e1b74d48b94300992fb3e80f85008f456d1906b40c36edef3e5400bb13035eeffca2c6e873854745d59eb48
+  data.tar.gz: 11cdfd7de96c1d7ad031d452b8c5c9f100e6dfbaeba35a5524bc709f86b083a6e9c7720f04b70ff931fc22f7ff87f2c5950d260e95d581a735689f062b8fe242

data/lib/tasks/views.rake

@@ -6,9 +6,7 @@ namespace :db do
       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}"
-      )
+      c.clear_view
     end
   end
 
@@ -21,11 +19,7 @@ namespace :db do
       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})"
-        )
+        c.create_view
       end
     end
   end

data/lib/tidus.rb

@@ -1,14 +1,13 @@
+# encoding: utf-8
+
 require "rake"
 require "active_record"
 
 require "tidus/version"
+require "tidus/query"
 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"
+require "tidus/strategies/base_selector"
+Dir["#{File.dirname(__FILE__)}/tidus/strategies/**/*.rb"].each { |f| require f }
 
 load "active_record/railties/databases.rake"
 load "tasks/views.rake"

data/lib/tidus/anonymization.rb

@@ -1,5 +1,9 @@
+# encoding: utf-8
+
 module Tidus
 	module Anonymization
+    include Tidus::Query
+
     def view_postfix
       @view_postfix || "anonymized"
     end
@@ -37,8 +41,8 @@ module Tidus
       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?
+      raise ArgumentError, "Must have at least one attribute" if attributes.empty?
+      raise ArgumentError, "Must have a strategy" if options[:strategy].blank?
 
       columns.each do |column|
         key = options[:strategy].to_s.camelize

data/lib/tidus/query.rb

@@ -0,0 +1,23 @@
+# encoding: utf-8
+
+module Tidus
+  module Query
+    def create_query
+      "CREATE VIEW #{view_name} AS " +
+      "SELECT #{view_columns.join(', ')} " +
+      "FROM #{table_name}"
+    end
+
+    def create_view
+      connection.execute(create_query)
+    end
+
+    def clear_query
+      "DROP VIEW IF EXISTS #{view_name}"
+    end
+
+    def clear_view
+      connection.execute(clear_query)
+    end
+  end
+end

data/lib/tidus/strategies/base_selector.rb

@@ -0,0 +1,17 @@
+# encoding: utf-8
+
+module Tidus
+  class BaseSelector
+    def self.anonymize(table_name, column_name, options = {})
+      adapter = ActiveRecord::Base.connection.instance_values["config"][:adapter]
+
+      begin
+        klass = Kernel.const_get("Tidus::#{adapter.camelize}::#{self.name.demodulize}")
+        klass.anonymize(table_name, column_name, options)
+      rescue NameError
+        raise "#{self.name} not implemented for #{adapter}"
+      end
+
+    end
+  end
+end

data/lib/tidus/strategies/cond_anonymizer.rb

@@ -1,44 +1,6 @@
-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 "
+# encoding: utf-8
 
-        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
+module Tidus
+  class CondAnonymizer < Tidus::BaseSelector
   end
 end

data/lib/tidus/strategies/email_anonymizer.rb

@@ -1,20 +1,7 @@
-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
+# encoding: utf-8
 
-        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
+module Tidus
+  class EmailAnonymizer < Tidus::BaseSelector
   end
 end
 

data/lib/tidus/strategies/null_anonymizer.rb

@@ -1,13 +1,6 @@
+# encoding: utf-8
+
 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
+  class NullAnonymizer < Tidus::BaseSelector
   end
 end

data/lib/tidus/strategies/overlay_anonymizer.rb

@@ -1,21 +1,6 @@
-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?
+# encoding: utf-8
 
-        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
+module Tidus
+  class OverlayAnonymizer < Tidus::BaseSelector
   end
 end

data/lib/tidus/strategies/postgresql/cond_anonymizer.rb

@@ -0,0 +1,40 @@
+module Tidus
+  module Postgresql
+    class CondAnonymizer
+      def self.anonymize(table_name, column_name, options = {})
+        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
+      end
+    end
+  end
+end

data/lib/tidus/strategies/postgresql/email_anonymizer.rb

@@ -0,0 +1,15 @@
+module Tidus
+  module Postgresql
+    class EmailAnonymizer
+      def self.anonymize(table_name, column_name, options = {})
+        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"
+      end
+    end
+  end
+end

data/lib/tidus/strategies/postgresql/null_anonymizer.rb

@@ -0,0 +1,9 @@
+module Tidus
+  module Postgresql
+    class NullAnonymizer
+      def self.anonymize(table_name, column_name, options = {})
+        return "NULL::unknown"
+      end
+    end
+  end
+end

data/lib/tidus/strategies/postgresql/overlay_anonymizer.rb

@@ -0,0 +1,17 @@
+module Tidus
+  module Postgresql
+    class OverlayAnonymizer
+      def self.anonymize(table_name, column_name, options = {})
+        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]})"
+      end
+    end
+  end
+end

data/lib/tidus/strategies/postgresql/text_anonymizer.rb

@@ -0,0 +1,25 @@
+module Tidus
+  module Postgresql
+    class TextAnonymizer
+      def self.anonymize(table_name, column_name, options = {})
+        base = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZCßüäöÜÄÖ"
+        return "translate((#{table_name}.#{column_name})::text, " +
+               "'#{base}'::text, '#{generate_mapping(base)}'::text)"
+      end
+
+      private
+        def self.generate_mapping(base)
+          upper = ("A".."Z").to_a
+          lower = ("a".."z").to_a
+          result = base.split("").map do |letter|
+            if letter == letter.upcase
+              upper.sample
+            else
+              lower.sample
+            end
+          end
+          result.join("")
+        end
+    end
+  end
+end

data/lib/tidus/strategies/sqlite3/null_anonymizer.rb

@@ -0,0 +1,9 @@
+module Tidus
+  module Sqlite3
+    class NullAnonymizer
+      def self.anonymize(table_name, column_name, options = {})
+        return "NULL"
+      end
+    end
+  end
+end

data/lib/tidus/strategies/static_anonymizer.rb

@@ -1,15 +1,12 @@
+# encoding: utf-8
+
 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?
+      raise "Missing option :value for StaticAnonymizer on #{table_name}.#{column_name}" if options[:value].blank?
+      type = options[:type] || "unknown"
 
-        return "'#{options[:value]}'"
-      else
-        raise "#{self.name} not implemented for #{adapter}"
-      end
+      return "'#{options[:value]}'::#{type}"
     end
   end
 end

data/lib/tidus/strategies/text_anonymizer.rb

@@ -1,28 +1,6 @@
-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
+# encoding: utf-8
 
-    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
+module Tidus
+  class TextAnonymizer < Tidus::BaseSelector
   end
 end

data/lib/tidus/version.rb

@@ -1,3 +1,3 @@
 module Tidus
-  VERSION = "1.0.1"
+  VERSION = "1.0.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tidus
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Tobias Schoknecht
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-02-09 00:00:00.000000000 Z
+date: 2015-04-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -76,10 +76,18 @@ files:
 - lib/tasks/views.rake
 - lib/tidus.rb
 - lib/tidus/anonymization.rb
+- lib/tidus/query.rb
+- lib/tidus/strategies/base_selector.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/postgresql/cond_anonymizer.rb
+- lib/tidus/strategies/postgresql/email_anonymizer.rb
+- lib/tidus/strategies/postgresql/null_anonymizer.rb
+- lib/tidus/strategies/postgresql/overlay_anonymizer.rb
+- lib/tidus/strategies/postgresql/text_anonymizer.rb
+- lib/tidus/strategies/sqlite3/null_anonymizer.rb
 - lib/tidus/strategies/static_anonymizer.rb
 - lib/tidus/strategies/text_anonymizer.rb
 - lib/tidus/version.rb
@@ -103,7 +111,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.5
+rubygems_version: 2.4.6
 signing_key: 
 specification_version: 4
 summary: Gem for creating anonymization views.