debian-mirror-gitlab/lib/declarative_policy/policy_dsl.rb

47 lines
1.2 KiB
Ruby
Raw Normal View History

2018-12-13 13:39:08 +05:30
# frozen_string_literal: true
2018-05-09 12:01:36 +05:30
module DeclarativePolicy
# The return value of a rule { ... } declaration.
# Can call back to register rules with the containing
# Policy class (context_class here). See Base.rule
#
# Note that the #policy method just performs an #instance_eval,
# which is useful for multiple #enable or #prevent callse.
#
# Also provides a #method_missing proxy to the context
# class's class methods, so that helper methods can be
# defined and used in a #policy { ... } block.
class PolicyDsl
def initialize(context_class, rule)
@context_class = context_class
@rule = rule
end
2018-11-18 11:00:15 +05:30
def policy(&block)
instance_eval(&block)
2018-05-09 12:01:36 +05:30
end
def enable(*abilities)
@context_class.enable_when(abilities, @rule)
end
def prevent(*abilities)
@context_class.prevent_when(abilities, @rule)
end
def prevent_all
@context_class.prevent_all_when(@rule)
end
2018-11-18 11:00:15 +05:30
def method_missing(msg, *args, &block)
return super unless @context_class.respond_to?(msg)
2018-05-09 12:01:36 +05:30
2018-11-18 11:00:15 +05:30
@context_class.__send__(msg, *args, &block) # rubocop:disable GitlabSecurity/PublicSend
2018-05-09 12:01:36 +05:30
end
2018-11-18 11:00:15 +05:30
def respond_to_missing?(msg)
@context_class.respond_to?(msg) || super
2018-05-09 12:01:36 +05:30
end
end
end