mirror of
https://github.com/discourse/discourse.git
synced 2025-02-22 18:43:11 +08:00

Creates a mixin for the ensure_* functions and creates seperate mixins for functions dealing with posts, categories, and topics.
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 |