Class: ActiveRecord::Base

Inherits:
Object
  • Object
show all
Defined in:
lib/encrypted_search_attributes/extensions/active_record/base.rb

Class Method Summary collapse

Class Method Details

.attr_encrypted_search(*params) ⇒ Object



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
# File 'lib/encrypted_search_attributes/extensions/active_record/base.rb', line 4

def attr_encrypted_search(*params)
  define_attribute_methods rescue nil

  options = params.last.is_a?(Hash) ? params.pop.dup : {}
  compress = options.delete(:compress) || false
  type = options.delete(:type) || :string
  normalize = options.delete(:normalize)

  raise "Invalid type: #{type.inspect}. Valid types: #{SymmetricEncryption::COERCION_TYPES.inspect}" unless SymmetricEncryption::COERCION_TYPES.include?(type)

  options.each {|option| warn "Ignoring unknown option #{option.inspect} supplied to attr_encrypted_search with #{params.inspect}"}

  if const_defined?(:EncryptedSearchAttributes, _search_ancestors = false)
    mod = const_get(:EncryptedSearchAttributes)
  else
    mod = const_set(:EncryptedSearchAttributes, Module.new)
    include mod
  end

  params.each do |attribute|
    mod.module_eval do
      define_method("#{attribute}=") do |value|
        if value
          send(
            "encrypted_search_#{attribute}=",
            ::SymmetricEncryption.encrypt(
              send("normalize_#{attribute}", value),
              false,
              compress,
              type
            )
          )
        end
        super(value)
      end

      define_method("normalize_#{attribute}") do |value|
        if normalize
          if normalize.respond_to?(:call)
            normalize.call(value)
          else
            send(normalize, value)
          end
        else
          value.downcase
        end
      end
    end
  end

end