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
55
|
# File 'lib/bitroles.rb', line 9
def has_roles(*args)
if args.any?
roles_mask = 'roles_mask'
if args.last.is_a?(Hash)
options = args.pop if args.last.is_a?(Hash)
roles_mask = options[:mask_column].to_s
end
roles = args.map(&:to_s)
class_eval <<-EVAL, __FILE__, __LINE__
def self.roles
#{roles}
end
def roles=(roles)
roles = (roles.map(&:to_s) & #{roles}).map { |r| 2**#{roles}.index(r) }.sum
self.#{roles_mask} = roles
end
def roles
#{roles}.reject { |r| ((#{roles_mask} || 0) & 2**#{roles}.index(r)).zero? }
end
def has_role?(role)
roles.include? role.to_s
end
EVAL
roles.each do |role|
class_eval <<-EVAL, __FILE__, __LINE__
scope :#{role.pluralize}, -> { where(['#{roles_mask} & ? > 0', 2**#{roles}.index('#{role}')]) }
def is_#{role}?
has_role? '#{role}'
end
def #{role}=(val)
if val
self.roles += ['#{role}'] unless is_#{role}?
else
self.roles -= ['#{role}'] if is_#{role}?
end
end
EVAL
end
end
end
|