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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
# File 'lib/uidable.rb', line 29
def uidable(
uid_name: "uid",
uid_size: DEFAULT_UID_SIZE,
read_only: true,
presence: true,
uniqueness: true,
set_to_param: false,
scope: false)
unless uidable_cols.include?(uid_name.to_sym)
uidable_cols << uid_name.to_sym
mod = Module.new
mod.module_eval <<-RUBY, __FILE__, __LINE__ + 1
def self.included(base)
if defined?(::ActiveRecord::Base) && base < ::ActiveRecord::Base
base.before_validation :uidable_assign_#{uid_name}, on: :create
#{ scope ? "base.scope :'with_#{uid_name}', -> (uid) { base.where(:'#{uid_name}' => uid) }" : "" }
#{ set_to_param ? "base.include SetToParam#{uid_name}" : "" }
#{ read_only ? "base.attr_readonly :'#{uid_name}'" : "" }
else
#{ read_only ? "attr_reader :'#{uid_name}'" : "attr_accessor :'#{uid_name}'" }
end
if base.respond_to?(:validates)
#{ presence ? "base.validates :'#{uid_name}', presence: true" : "" }
#{ uniqueness ? "base.validates :'#{uid_name}', uniqueness: true" : "" }
end
end
module SetToParam#{uid_name}
def to_param
self.#{uid_name}
end
end
private
def uidable_assign_#{uid_name}
self.#{uid_name} = gen_#{uid_name}
end
def gen_#{uid_name}
Array.new(#{uid_size}){[*'a'..'z', *'0'..'9'].sample}.join
end
RUBY
include mod
end
end
|