Module: ActiveEnquo::ActiveRecord::BaseExtension::ClassMethods

Includes:
QueryFilterMangler
Defined in:
lib/active_enquo.rb

Instance Method Summary collapse

Instance Method Details

#enquo(attr_name, value_or_meta_id, maybe_value = nil) ⇒ Object 



119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/active_enquo.rb', line 119

def enquo(attr_name, value_or_meta_id, maybe_value = nil)
  meta_id, value = if value_or_meta_id.is_a?(Symbol)
    [value_or_meta_id, maybe_value]
  else
    [nil, value_or_meta_id]
  end

  t = self.attribute_types[attr_name.to_s]
  if t.is_a?(::ActiveEnquo::Type)
    relation = self.arel_table.name.to_s
    field = ::ActiveEnquo.root.field(relation, attr_name.to_s)
    if meta_id.nil?
      t.encrypt(value, "", field, enable_reduced_security_operations: true)
    else
      t.encrypt_metadata_value(meta_id, value, field)
    end
  else
    raise ArgumentError, "Cannot produce encrypted value on a non-enquo attribute '#{attr_name}'"
  end
end

#enquo_attr(attr_name, opts) ⇒ Object 



159
160
161
162
163
164
165
166
167
168
169
170
 
# File 'lib/active_enquo.rb', line 159

def enquo_attr(attr_name, opts)
  if opts.key?(:default)
    default_value = opts.delete(:default)
    after_initialize do
      next if persisted?
      next unless self.send(attr_name).nil?
      self.send(:"#{attr_name}=", default_value.duplicable? ? default_value.dup : default_value)
    end
  end

  enquo_attribute_options[attr_name] = @enquo_attribute_options[attr_name].merge(opts)
end

#enquo_attr?(attr_name) ⇒ Boolean

Returns:

  • (Boolean) 


155
156
157
 
# File 'lib/active_enquo.rb', line 155

def enquo_attr?(attr_name)
  self.attribute_types[attr_name.to_s].is_a?(::ActiveEnquo::Type)
end

#enquo_attribute_options ⇒ Object 



172
173
174
 
# File 'lib/active_enquo.rb', line 172

def enquo_attribute_options
  @enquo_attribute_options ||= Hash.new({})
end

#enquo_encrypt_columns(column_map, batch_size: 10_000) ⇒ Object 



176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
 
# File 'lib/active_enquo.rb', line 176

def enquo_encrypt_columns(column_map, batch_size: 10_000)
  plaintext_columns = column_map.keys
  relation = self.arel_table.name
  in_progress = true
  self.reset_column_information

  while in_progress
    self.transaction do
      # The .where("0=1") here is a dummy condition so that the q.or in the .each will work properly
      q = self.select(self.primary_key).select(plaintext_columns).where("0=1")
      column_map.each do |pt_col, ct_col|
        q = q.or(self.where(ct_col => nil).where.not(pt_col => nil))
      end

      q = q.limit(batch_size).lock

      rows = ::ActiveRecord::Base.connection.exec_query(q.to_sql)
      if rows.length == 0
        in_progress = false
      else
        rows.each do |row|
          values = Hash[column_map.map do |pt_col, ct_col|
            field = ::ActiveEnquo.root.field(relation, ct_col)
            attr_opts = self.enquo_attribute_options.fetch(ct_col.to_sym, {})
            t = self.attribute_types[ct_col.to_s]
            db_value = t.encrypt(row[pt_col.to_s], row[self.primary_key].to_s, field, **attr_opts)

            [ct_col, db_value]
          end]

          self.where(self.primary_key => row[self.primary_key]).update_all(values)
        end
      end
    end
  end
end

#find_by(*a) ⇒ Object 



114
115
116
117
 
# File 'lib/active_enquo.rb', line 114

def find_by(*a)
  mangle_query_filter(a)
  super
end

#unenquo(attr_name, value, ctx) ⇒ Object 



140
141
142
143
144
145
146
147
148
149
150
151
152
153
 
# File 'lib/active_enquo.rb', line 140

def unenquo(attr_name, value, ctx)
  t = self.attribute_types[attr_name.to_s]
  if t.is_a?(::ActiveEnquo::Type)
    relation = self.arel_table.name.to_s
    field = ::ActiveEnquo.root.field(relation, attr_name.to_s)
    begin
      t.decrypt(value, ctx, field)
    rescue Enquo::Error
      t.decrypt(value, "", field)
    end
  else
    raise ArgumentError, "Cannot decrypt value on a non-enquo attribute '#{attr_name}'"
  end
end