-
Notifications
You must be signed in to change notification settings - Fork 402
/
role.rb
112 lines (94 loc) · 3.55 KB
/
role.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
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
require "rolify/finders"
require "rolify/utils"
module Rolify
module Role
extend Utils
def self.included(base)
base.extend Finders
end
def add_role(role_name, resource = nil)
role = self.class.adapter.find_or_create_by(role_name.to_s,
(resource.is_a?(Class) ? resource.to_s : resource.class.name if resource),
(resource.id if resource && !resource.is_a?(Class)))
if !roles.include?(role)
self.class.define_dynamic_method(role_name, resource) if Rolify.dynamic_shortcuts
self.class.adapter.add(self, role)
end
role
end
alias_method :grant, :add_role
def has_role?(role_name, resource = nil)
return has_strict_role?(role_name, resource) if self.class.strict_rolify and resource and resource != :any
if new_record?
role_array = self.roles.detect { |r|
r.name.to_s == role_name.to_s &&
(r.resource == resource ||
resource.nil? ||
(resource == :any && r.resource.present?))
}
else
role_array = self.class.adapter.where(self.roles, name: role_name, resource: resource)
end
return false if role_array.nil?
role_array != []
end
def has_strict_role?(role_name, resource)
self.class.adapter.where_strict(self.roles, name: role_name, resource: resource).any?
end
def has_cached_role?(role_name, resource = nil)
return has_strict_cached_role?(role_name, resource) if self.class.strict_rolify and resource and resource != :any
self.class.adapter.find_cached(self.roles, name: role_name, resource: resource).any?
end
def has_strict_cached_role?(role_name, resource = nil)
self.class.adapter.find_cached_strict(self.roles, name: role_name, resource: resource).any?
end
def has_all_roles?(*args)
args.each do |arg|
if arg.is_a? Hash
return false if !self.has_role?(arg[:name], arg[:resource])
elsif arg.is_a?(String) || arg.is_a?(Symbol)
return false if !self.has_role?(arg)
else
raise ArgumentError, "Invalid argument type: only hash or string or symbol allowed"
end
end
true
end
def has_any_role?(*args)
if new_record?
args.any? { |r| self.has_role?(r) }
else
self.class.adapter.where(self.roles, *args).size > 0
end
end
def only_has_role?(role_name, resource = nil)
return self.has_role?(role_name,resource) && self.roles.count == 1
end
def remove_role(role_name, resource = nil)
self.class.adapter.remove(self, role_name.to_s, resource)
end
alias_method :revoke, :remove_role
deprecate :has_no_role, :remove_role
def roles_name
self.roles.pluck(:name)
end
def method_missing(method, *args, &block)
if method.to_s.match(/^is_(\w+)_of[?]$/) || method.to_s.match(/^is_(\w+)[?]$/)
resource = args.first
self.class.define_dynamic_method $1, resource
return has_role?("#{$1}", resource)
end if Rolify.dynamic_shortcuts
super
end
def respond_to?(method, include_private = false)
if Rolify.dynamic_shortcuts && (method.to_s.match(/^is_(\w+)_of[?]$/) || method.to_s.match(/^is_(\w+)[?]$/))
query = self.class.role_class.where(:name => $1)
query = self.class.adapter.exists?(query, :resource_type) if method.to_s.match(/^is_(\w+)_of[?]$/)
return true if query.count > 0
false
else
super
end
end
end
end