summaryrefslogtreecommitdiff
path: root/lib/declarative_policy/policy_dsl.rb
blob: 96741c0478ef9c5ab395e84b6994ff62f04f6a89 (plain)
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
# frozen_string_literal: true

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

    def policy(&block)
      instance_eval(&block)
    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

    def method_missing(msg, *args, &block)
      return super unless @context_class.respond_to?(msg)

      @context_class.__send__(msg, *args, &block) # rubocop:disable GitlabSecurity/PublicSend
    end

    def respond_to_missing?(msg)
      @context_class.respond_to?(msg) || super
    end
  end
end