-
Notifications
You must be signed in to change notification settings - Fork 1
/
model_additions.rb
49 lines (44 loc) · 1.25 KB
/
model_additions.rb
1
2
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
module Omelettes
module ModelAdditions
module ClassMethods
def column_config(column_name)
@column_config ||= {}
@column_config[column_name.to_s]
end
def treat(column_name, style=nil, &block)
@column_config ||= {}
column = Column.new(column_name, style, &block)
@column_config[column_name.to_s] = column
column
end
alias :scramble :treat
def ignore(column_name)
@column_config ||= {}
column = Column.new(column_name, :hardened)
@column_config[column_name.to_s] = column
column
end
alias :harden :ignore
end
def obfuscate(column_names)
attributes = {}
column_names.each do |column_name|
column = self.class.column_config(column_name)
original_value = self.send(column_name)
if column
value = column.process(original_value)
else
value = Column.default(column_name, original_value)
end
attributes[column_name] = value
end
self.class.where(:id => self.id).update_all(attributes)
end
def self.included(base)
base.extend ClassMethods
end
end
end
ActiveRecord::Base.class_eval do
include Omelettes::ModelAdditions
end