gitlab-org--gitlab-foss/app/policies/base_policy.rb

56 lines
936 B
Ruby
Raw Normal View History

class BasePolicy
2016-08-12 18:36:16 +00:00
def self.abilities(user, subject)
new(user, subject).abilities
end
2016-08-16 18:10:34 +00:00
def self.class_for(subject)
"#{subject.class.name}Policy".constantize
end
2016-08-12 18:36:16 +00:00
attr_reader :user, :subject
def initialize(user, subject)
@user = user
@subject = subject
end
def abilities
2016-08-12 18:36:16 +00:00
return anonymous_abilities if @user.nil?
collect_rules { rules }
end
def anonymous_abilities
collect_rules { anonymous_rules }
end
2016-08-16 18:10:34 +00:00
def anonymous_rules
rules
end
def delegate!(new_subject)
@can.merge(BasePolicy.class_for(new_subject).abilities(@user, new_subject))
end
2016-08-16 19:05:44 +00:00
def can?(rule)
@can.include?(rule) && !@cannot.include?(rule)
end
def can!(*rules)
@can.merge(rules)
end
def cannot!(*rules)
@cannot.merge(rules)
end
2016-08-12 18:36:16 +00:00
private
def collect_rules(&b)
return Set.new if @subject.nil?
@can = Set.new
@cannot = Set.new
yield
@can - @cannot
end
end