add automatic detection of the policy class

This commit is contained in:
http://jneen.net/ 2016-08-16 12:55:44 -07:00
parent 16fe6dc7b1
commit 3656d3b88a
2 changed files with 18 additions and 16 deletions

View File

@ -61,6 +61,9 @@ class Ability
private
def uncached_allowed(user, subject)
policy_class = BasePolicy.class_for(subject) rescue nil
return policy_class.abilities(user, subject) if policy_class
return anonymous_abilities(subject) if user.nil?
return [] unless user.is_a?(User)
return [] if user.blocked?
@ -70,13 +73,6 @@ class Ability
def abilities_by_subject_class(user:, subject:)
case subject
when Project then ProjectPolicy.abilities(user, subject)
when Issue then IssuePolicy.abilities(user, subject)
when MergeRequest then MergeRequestPolicy.abilities(user, subject)
when Ci::Build then Ci::BuildPolicy.abilities(user, subject)
when CommitStatus then CommitStatus.abilities(user, subject)
when Note then note_abilities(user, subject)
when ProjectSnippet then project_snippet_abilities(user, subject)
when PersonalSnippet then personal_snippet_abilities(user, subject)
when Group then group_abilities(user, subject)
@ -96,14 +92,6 @@ class Ability
anonymous_personal_snippet_abilities(subject)
elsif subject.is_a?(ProjectSnippet)
anonymous_project_snippet_abilities(subject)
elsif subject.is_a?(CommitStatus)
anonymous_commit_status_abilities(subject)
elsif subject.is_a?(Project)
ProjectPolicy.abilities(nil, subject)
elsif subject.is_a?(Issue)
IssuePolicy.abilities(nil, subject)
elsif subject.is_a?(MergeRequest)
MergeRequestPolicy.abilities(nil, subject)
elsif subject.respond_to?(:project)
ProjectPolicy.abilities(nil, subject.project)
elsif subject.is_a?(Group) || subject.respond_to?(:group)

View File

@ -4,7 +4,21 @@ class BasePolicy
end
def self.class_for(subject)
"#{subject.class.name}Policy".constantize
subject.class.ancestors.each do |klass|
next unless klass.name
begin
policy_class = "#{klass.name}Policy".constantize
# NB: the < operator here tests whether policy_class
# inherits from BasePolicy
return policy_class if policy_class < BasePolicy
rescue NameError
nil
end
end
raise "no policy for #{subject.class.name}"
end
attr_reader :user, :subject