tidus 1.0.3 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7ab60caab05fd3272ef551a3ef6268a98dcc5858
-  data.tar.gz: d0086aa189be407f46628569cd9dd7dc46b57d89
+  metadata.gz: 982c186ae247f2768f5836f664786d27e0746267
+  data.tar.gz: 47adfce0419345aa46db2da66954a22a13eeb706
 SHA512:
-  metadata.gz: 3e3377bd211a50ada2550c3d0aa27e020fb38b62f029e04aa0a99ec9e8c217550956c076142505d1628945a4a2eb30929c564d4e1a8a1796af7376e2246c5957
-  data.tar.gz: 1050fd7c7e127e3ea01c73dbeba0ae2a5be207c78510dc9d27a0ea4506e1e574b11a9e86b7816794b50396d05cc08012084cb339f283a47f27ecc8e7cb5eba14
+  metadata.gz: e64fd16e2f06d5238ef075e215a1eb97976c0153677f3ab96aa4cf3c7cefd5af86855d893bc4fbe08e398db5d3c5b13af3a30d8c53209c72f9df8bcb45c2c275
+  data.tar.gz: 7f77bbd00064728512bf6b999cf29b08817e7f74900582cd139592fc06800bae3f83b29df725cd3c6ae15866ea604773250b5682014319f8934017b3f37ea288

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

@@ -4,8 +4,14 @@ module Tidus
       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 is edited on a deeper level .dup has to be replaced by marshal dump/load
+        options = options.dup
+        options[:result_type] ||= "text"
+        type = options[:result_type]
+
+        if options[:default] && options[:default_strategy]
+          raise "CondAnonymizer: Only one of default and default_strategy can be used"
+        end
 
         if options[:conditions].blank?
           raise "Missing option :conditions for CondAnonymizer on #{name}"
@@ -31,10 +37,31 @@ module Tidus
                      "'#{cond_value}'::#{cond_type}) THEN '#{cond_result}'::#{type} "
         end
 
-        command += "ELSE #{default} END"
+        command += "ELSE #{self.default(table_name, column_name, options)} END"
 
         return command
       end
+
+      def self.default(table_name, column_name, options)
+        type = options[:result_type]
+
+        value = if options[:default_strategy]
+          klass = self.anonymizer(options[:default_strategy])
+          klass.anonymize(table_name, column_name, {})
+        elsif options[:default]
+          "'#{options[:default]}'::#{type}"
+        else
+          "#{table_name}.#{column_name}::#{type}"
+        end
+      end
+
+      def self.anonymizer(name)
+        begin
+          klass = Kernel.const_get("Tidus::Postgresql::#{name.to_s.camelize}Anonymizer")
+        rescue NameError
+          raise "default_strategy '#{name}' not implemented for Postgresql"
+        end
+      end
     end
   end
-end
+end

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

@@ -2,18 +2,18 @@ module Tidus
   module Postgresql
     class TextAnonymizer
       def self.anonymize(table_name, column_name, options = {})
-        base = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZCßüäöÜÄÖ"
+        base = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZCßüäöÜÄÖ0123456789"
         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
+          upper_and_nums = ("A".."Z").to_a + ('0'..'9').to_a
           lower = ("a".."z").to_a
           result = base.split("").map do |letter|
             if letter == letter.upcase
-              upper.sample
+              upper_and_nums.sample
             else
               lower.sample
             end
@@ -22,4 +22,4 @@ module Tidus
         end
     end
   end
-end
+end

data/lib/tidus/version.rb

@@ -1,3 +1,3 @@
 module Tidus
-  VERSION = "1.0.3"
+  VERSION = "1.0.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tidus
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.4
 platform: ruby
 authors:
 - Tobias Schoknecht
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-01 00:00:00.000000000 Z
+date: 2015-05-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -111,7 +111,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.6
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Gem for creating anonymization views.