Module: Lockbox::Calculations

Defined in:
lib/lockbox/calculations.rb

Instance Method Summary collapse

Instance Method Details

#pluck(*column_names) ⇒ Object 



3
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
 
# File 'lib/lockbox/calculations.rb', line 3

def pluck(*column_names)
  return super unless model.respond_to?(:lockbox_attributes)

  lockbox_columns = column_names.map.with_index { |c, i| [model.lockbox_attributes[c.to_sym], i] }.select { |la, _i| la && !la[:migrating] }

  return super unless lockbox_columns.any?

  # replace column with ciphertext column
  lockbox_columns.each do |la, i|
    column_names[i] = la[:encrypted_attribute]
  end

  # pluck
  result = super(*column_names)

  # decrypt result
  # handle pluck to single columns and multiple
  #
  # we can't pass context to decrypt method
  # so this won't work if any options are a symbol or proc
  if column_names.size == 1
    la = lockbox_columns.first.first
    result.map! { |v| model.send("decrypt_#{la[:encrypted_attribute]}", v) }
  else
    lockbox_columns.each do |la, i|
      result.each do |v|
        v[i] = model.send("decrypt_#{la[:encrypted_attribute]}", v[i])
      end
    end
  end

  result
end