tidus 1.2.3 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b51c5f1d85405f2bba2691de20b1dd7cfe696d37569cada6b5a1d4864bedda4b
-  data.tar.gz: 0f4e3bb64adb66fb3bbe13838cdd29c67eabb9838a4ff03b3bbb9ef6f8552796
+  metadata.gz: e856d718f2b4408d674dbd77073588f3da9f9c26dd8f9f61f9d4c1bbb674d525
+  data.tar.gz: ed491dc4694a926936fe3d8c879a48894ae6ef77685730004c63e1cd6e1e1493
 SHA512:
-  metadata.gz: 929142964abaa2025dd83cb11712b299e927e102b1ad7b5d36573d25ad32c8d7371504fcbe55965da1b8f0e3f03aebdd0542b182442f226811021b07df295b1f
-  data.tar.gz: 0c55226cd2527f613209eabf9df8d70554eef1a68e2188ea6ae5c90a615cf63f2518c10be5b9d72fb034a915f58f1216c0e08bdf48f5e9c1c92aecfd2eb505e5
+  metadata.gz: f7ceb7506c37c5c9fd66fc5f417e8318b566ad8fcf900341c61a0cc3423bf69fe8066194c8cb1610296ecfe36fc48fd3ded8361904ab29d449e99828c17a54c8
+  data.tar.gz: 885933234f25620860632d23ea8e1567034ee9455840b1ebde57e5a88a25c4cb4803ed66a45b2ae9ffdcac93bb95eeb03ddc052557d116097207a638451282d8

data/lib/tasks/views.rake

@@ -4,7 +4,7 @@ skip_tables = [
 ]
 
 namespace :db do
-  desc "Clears all the views which are currently existing"
+  desc 'Clears all the views which are currently existing'
   task :clear_views do
     Rails.application.eager_load! if defined?(Rails)
 
@@ -17,7 +17,7 @@ namespace :db do
     end
   end
 
-  desc "Generates all the views for the models"
+  desc 'Generates all the views for the models'
   task :generate_views do
     Rails.application.eager_load! if defined?(Rails)
 
@@ -33,13 +33,13 @@ namespace :db do
   end
 end
 
-Rake::Task["db:migrate"].enhance ["db:clear_views"]
-Rake::Task["db:rollback"].enhance ["db:clear_views"]
+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
+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
+Rake::Task['db:rollback'].enhance do
+  Rake::Task['db:generate_views'].invoke
 end

data/lib/tidus/anonymization.rb

@@ -2,6 +2,7 @@
 
 module Tidus
   module Anonymization
+
     include Tidus::Query
 
     def skip_anonymization
@@ -13,7 +14,7 @@ module Tidus
     end
 
     def view_postfix
-      @view_postfix || "anonymized"
+      @view_postfix || 'anonymized'
     end
 
     def view_postfix=(val)
@@ -30,9 +31,10 @@ module Tidus
 
     def view_columns
       @view_columns ||= {}
-      default_view_columns.merge(@view_columns)
-                          .map{ |k,v| ["#{v} AS #{k}"] }
-                          .flatten
+      default_view_columns
+        .merge(@view_columns)
+        .map { |k, v| ["#{v} AS #{k}"] }
+        .flatten
     end
 
     def default_view_columns
@@ -49,8 +51,8 @@ module Tidus
       options = attributes.extract_options!.dup
       columns = attributes - [options]
 
-      raise ArgumentError, "Must have at least one attribute" if attributes.empty?
-      raise ArgumentError, "Must have a strategy" if options[:strategy].blank?
+      raise ArgumentError.new('Must have at least one attribute') if attributes.empty?
+      raise ArgumentError.new('Must have a strategy') if options[:strategy].blank?
 
       columns.each do |column|
         key = options[:strategy].to_s.camelize
@@ -62,14 +64,14 @@ module Tidus
             klass = const_get("Tidus::#{key}Anonymizer")
           end
         rescue NameError
-          raise ArgumentError, "Unknown anonymizer: '#{key}'"
+          raise ArgumentError.new("Unknown anonymizer: '#{key}'")
         end
 
         @view_columns[column.to_sym] = klass.anonymize(table_name, column, options)
       end
     end
 
-	end
+  end
 end
 
-ActiveRecord::Base.extend Tidus::Anonymization
+ActiveRecord::Base.extend Tidus::Anonymization

data/lib/tidus/query.rb

@@ -2,6 +2,7 @@
 
 module Tidus
   module Query
+
     def create_view_query_part
       case connection.instance_values['config'][:adapter].to_s.downcase
       when 'postgresql'
@@ -15,8 +16,8 @@ module Tidus
 
     def create_query
       "#{create_view_query_part} #{view_name} AS " +
-      "SELECT #{view_columns.join(', ')} " +
-      "FROM #{table_name}"
+        "SELECT #{view_columns.join(', ')} " +
+        "FROM #{table_name}"
     end
 
     def create_view
@@ -34,5 +35,6 @@ module Tidus
     def clear_view
       connection.execute(clear_query)
     end
+
   end
 end

data/lib/tidus/strategies/base_selector.rb

@@ -2,8 +2,9 @@
 
 module Tidus
   class BaseSelector
+
     def self.anonymize(table_name, column_name, options = {})
-      adapter = ActiveRecord::Base.connection.instance_values["config"][:adapter]
+      adapter = ActiveRecord::Base.connection.instance_values['config'][:adapter]
 
       begin
         klass = Kernel.const_get("Tidus::#{adapter.camelize}::#{self.name.demodulize}")
@@ -11,7 +12,12 @@ module Tidus
       rescue NameError
         raise "#{self.name} not implemented for #{adapter}"
       end
-
+    rescue ActiveRecord::ConnectionNotEstablished
+      # Do nothing here.
+      # If the connection cannot be established this cannot be run in a migration
+      # and will fail there.
+      # For all other cases it does not matter, since the return value of anonymize is not used.
     end
+
   end
 end

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

@@ -1,15 +1,27 @@
 module Tidus
   module Postgresql
     class EmailAnonymizer
+
       def self.anonymize(table_name, column_name, options = {})
         name = "#{table_name}.#{column_name}"
-        options[:length] ||= 15
+        length = options[:length] || 15
+        anonymize_domain = options[:anonymize_domain] || false
 
         return "CASE WHEN ((#{name})::text ~~ '%@%'::text) " +
-               "THEN (((\"left\"(md5((#{name})::text), #{options[:length]}) || '@'::text) " +
-                "|| split_part((#{name})::text, '@'::text, 2)))::character varying " +
+               "THEN (((\"left\"(md5((#{name})::text), #{length}) || '@'::text) " +
+                "|| #{domain_part(name, anonymize_domain, length)}))::character varying " +
                 "ELSE #{name} END"
       end
+
+      def self.domain_part(name, anonymize_domain, length)
+        if anonymize_domain
+          return "(\"left\"(md5(split_part((#{name})::text, '@'::text, 2)::text), #{length}) " +
+                 "|| '.com')"
+        end
+
+        return "split_part((#{name})::text, '@'::text, 2)"
+      end
+
     end
   end
 end

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

@@ -0,0 +1,15 @@
+module Tidus
+  module Postgresql
+    class Sha256Anonymizer
+
+      def self.anonymize(table_name, column_name, options = {})
+        name = "#{table_name}.#{column_name}"
+
+        length = options[:length] || 64
+
+        return "SUBSTR(ENCODE(DIGEST(#{name}, 'sha256')::TEXT, 'HEX'), 0, #{length.to_i + 1})"
+      end
+
+    end
+  end
+end

data/lib/tidus/strategies/sha256_anonymizer.rb

@@ -0,0 +1,6 @@
+# encoding: utf-8
+
+module Tidus
+  class Sha256Anonymizer < Tidus::BaseSelector
+  end
+end

data/lib/tidus/version.rb

@@ -1,3 +1,3 @@
 module Tidus
-  VERSION = "1.2.3"
+  VERSION = "1.3.0"
 end

data/lib/tidus.rb

@@ -1,13 +1,13 @@
 # encoding: utf-8
 
-require "rake"
-require "active_record"
+require 'rake'
+require 'active_record'
 
-require "tidus/version"
-require "tidus/query"
-require "tidus/anonymization"
-require "tidus/strategies/base_selector"
+require 'tidus/version'
+require 'tidus/query'
+require 'tidus/anonymization'
+require 'tidus/strategies/base_selector'
 Dir["#{File.dirname(__FILE__)}/tidus/strategies/**/*.rb"].each { |f| require f }
 
-load "active_record/railties/databases.rake" if defined?(Rails)
-load "tasks/views.rake"
+load 'active_record/railties/databases.rake' if defined?(Rails)
+load 'tasks/views.rake'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tidus
 version: !ruby/object:Gem::Version
-  version: 1.2.3
+  version: 1.3.0
 platform: ruby
 authors:
 - Tobias Schoknecht
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-01-19 00:00:00.000000000 Z
+date: 2023-07-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -90,9 +90,11 @@ files:
 - lib/tidus/strategies/postgresql/overlay_anonymizer.rb
 - lib/tidus/strategies/postgresql/regex_replace_anonymizer.rb
 - lib/tidus/strategies/postgresql/remove_json_keys_anonymizer.rb
+- lib/tidus/strategies/postgresql/sha256_anonymizer.rb
 - lib/tidus/strategies/postgresql/text_anonymizer.rb
 - lib/tidus/strategies/regex_replace_anonymizer.rb
 - lib/tidus/strategies/remove_json_keys_anonymizer.rb
+- lib/tidus/strategies/sha256_anonymizer.rb
 - lib/tidus/strategies/sqlite3/null_anonymizer.rb
 - lib/tidus/strategies/static_anonymizer.rb
 - lib/tidus/strategies/text_anonymizer.rb
@@ -116,7 +118,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.26
+rubygems_version: 3.4.13
 signing_key:
 specification_version: 4
 summary: Gem for creating anonymization views.