aloucas · October 31, 2021 15:16 · Sep 20, 2017 · Sep 14, 2017
diff --git a/encryptor.rb b/encryptor.rb
@@ -5,7 +5,7 @@ def has_encrypted_attributes(*attrs)
     attrs.each do |attr|
       # Define the dynamic getter
       define_method(attr) do
-        Encryptor.crypt.decrypt_and_verify(self[attr])
+        self[attr].present? ? Encryptor.crypt.decrypt_and_verify(self[attr]) : self[attr]
       end
       # Define the dynamic setter
       define_method("#{attr.to_s}=") do |val|

diff --git a/encryptor.rb b/encryptor.rb
@@ -0,0 +1,23 @@
+# lib/encryptor.rb
+# Module to dynamic encrypt attributes per model
+module Encryptor
+  def has_encrypted_attributes(*attrs)
+    attrs.each do |attr|
+      # Define the dynamic getter
+      define_method(attr) do
+        Encryptor.crypt.decrypt_and_verify(self[attr])
+      end
+      # Define the dynamic setter
+      define_method("#{attr.to_s}=") do |val|
+        super(Encryptor.crypt.encrypt_and_sign(val))
+      end
+    end
+  end
+
+  protected
+
+  # Create the base MessageEncryptor based on Rails secret_key_base
+  def self.crypt
+    ActiveSupport::MessageEncryptor.new(Rails.application.secrets.secret_key_base)
+  end
+end
diff --git a/model_with_encrypted_atts.rb b/model_with_encrypted_atts.rb
@@ -0,0 +1,10 @@
+# app/models/model_with_encrypted_atts.rb
+class ModelWithEncryptedAttrs < ApplicationRecord
+  # Modules
+  extend Encryptor
+
+  # Abilities
+  has_encrypted_attributes :attribute_1,
+                           :attribute_2
+
+end
No results found