/
power.rb
162 lines (139 loc) · 4.84 KB
/
power.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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
module Consul
module Power
include Consul::Power::DynamicAccess::InstanceMethods
def self.included(base)
base.extend ClassMethods
base.send :include, Memoizer
end
private
def default_include_power?(power_name, *context)
result = send(power_name, *context)
# Everything that is not nil is considered as included.
# We are short-circuiting for #scoped first since sometimes
# has_many associations (which behave scopish) trigger their query
# when you try to negate them, compare them or even retrieve their
# class. Unfortunately we can only reproduce this in live Rails
# apps, not in Consul tests. Might be some standard gem that is not
# loaded in Consul tests.
result.respond_to?(:load_target, true) || !!result
end
def default_include_object?(power_name, *args)
object = args.pop
context = args
power_value = send(power_name, *context)
if power_value.nil?
false
elsif Util.scope?(power_value)
if Util.scope_selects_all_records?(power_value)
true
else
power_ids_name = self.class.power_ids_name(power_name)
send(power_ids_name, *context).include?(object.id)
end
elsif Util.collection?(power_value)
power_value.include?(object)
else
raise Consul::NoCollection, "can only call #include_object? on a collection, but power was of type #{power_value.class.name}"
end
end
def default_power_ids(power_name, *args)
scope = send(power_name, *args)
database_touched
scope.collect_ids
end
def powerless!(*args)
raise Consul::Powerless.new("No power to #{[*args].inspect}")
end
def boolean_or_nil?(value)
[TrueClass, FalseClass, NilClass].include?(value.class)
end
def database_touched
# spy for tests
end
def singularize_power_name(name)
self.class.singularize_power_name(name)
end
module ClassMethods
include Consul::Power::DynamicAccess::ClassMethods
def power(*names, &block)
names.each do |name|
define_power(name, &block)
end
end
def power_ids_name(name)
"#{name.to_s.singularize}_ids"
end
def self.thread_key(klass)
"consul|#{klass.to_s}.current"
end
def current
Thread.current[ClassMethods.thread_key(self)]
end
def current=(power)
Thread.current[ClassMethods.thread_key(self)] = power
end
def with_power(inner_power, &block)
unless inner_power.is_a?(self) || inner_power.nil?
inner_power = new(inner_power)
end
old_power = current
self.current = inner_power
block.call
ensure
self.current = old_power
end
def without_power(&block)
with_power(nil, &block)
end
def define_query_and_bang_methods(name, &query)
query_method = "#{name}?"
bang_method = "#{name}!"
define_method(query_method, &query)
memoize query_method
define_method(bang_method) { |*args| send(query_method, *args) or powerless!(name, *args) }
# We don't memoize the bang method since memoizer can't memoize a thrown exception
end
def define_ids_method(name)
ids_method = power_ids_name(name)
define_method(ids_method) { |*args| default_power_ids(name, *args) }
# Memoize `ids_method` in addition to the collection method itself, since
# #default_include_object? directly accesses `ids_method`.
memoize ids_method
end
def define_main_method(name, &block)
define_method(name, &block)
memoize name
end
def define_power(name, &block)
name = name.to_s
if name.ends_with?('?')
# The developer is trying to register an optimized query method
# for singular object queries.
name_without_suffix = name.chop
define_query_and_bang_methods(name_without_suffix, &block)
else
define_main_method(name, &block)
define_ids_method(name)
define_query_and_bang_methods(name) { |*args| default_include_power?(name, *args) }
begin
singular = singularize_power_name(name)
define_query_and_bang_methods(singular) { |*args| default_include_object?(name, *args) }
rescue Consul::PowerNotSingularizable
# We do not define singularized power methods if it would
# override the collection method
end
end
name
end
def singularize_power_name(name)
name = name.to_s
singularized = name.singularize
if singularized == name
raise Consul::PowerNotSingularizable, "Power name can not have an singular form: #{name}"
else
singularized
end
end
end
end
end