mirror of
https://github.com/discourse/discourse.git
synced 2024-11-22 14:03:22 +08:00
22 lines
611 B
Ruby
22 lines
611 B
Ruby
|
# Support for ensure_{blah}! methods.
|
||
|
|
module EnsureMagic
|
||
|
|
|
||
|
|
def method_missing(method, *args, &block)
|
||
|
|
if method.to_s =~ /^ensure_(.*)\!$/
|
||
|
|
can_method = :"#{Regexp.last_match[1]}?"
|
||
|
|
|
||
|
|
if respond_to?(can_method)
|
||
|
|
raise Discourse::InvalidAccess.new("#{can_method} failed") unless send(can_method, *args, &block)
|
||
|
|
return
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
super.method_missing(method, *args, &block)
|
||
|
|
end
|
||
|
|
|
||
|
|
# Make sure we can see the object. Will raise a NotFound if it's nil
|
||
|
|
def ensure_can_see!(obj)
|
||
|
|
raise Discourse::InvalidAccess.new("Can't see #{obj}") unless can_see?(obj)
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|