gitlab-org--gitlab-foss/app/models/ability.rb

322 lines
7.7 KiB
Ruby
Raw Normal View History

2011-10-08 17:36:38 -04:00
class Ability
2012-10-08 20:10:04 -04:00
class << self
2013-01-25 04:30:49 -05:00
def allowed(user, subject)
return not_auth_abilities(user, subject) if user.nil?
2013-01-25 04:30:49 -05:00
return [] unless user.kind_of?(User)
return [] if user.blocked?
2013-01-25 04:30:49 -05:00
2012-10-08 20:10:04 -04:00
case subject.class.name
2013-01-25 04:30:49 -05:00
when "Project" then project_abilities(user, subject)
when "Issue" then issue_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)
2013-01-25 04:30:49 -05:00
when "MergeRequest" then merge_request_abilities(user, subject)
2013-06-21 15:44:40 -04:00
when "Group" then group_abilities(user, subject)
when "Namespace" then namespace_abilities(user, subject)
when "GroupMember" then group_member_abilities(user, subject)
2012-10-08 20:10:04 -04:00
else []
2013-01-25 04:30:49 -05:00
end.concat(global_abilities(user))
end
# List of possible abilities
# for non-authenticated user
def not_auth_abilities(user, subject)
project = if subject.kind_of?(Project)
subject
elsif subject.respond_to?(:project)
subject.project
else
nil
end
if project && project.public?
[
:read_project,
:read_wiki,
:read_issue,
:read_milestone,
:read_project_snippet,
:read_project_member,
:read_merge_request,
:read_note,
:download_code
]
else
group = if subject.kind_of?(Group)
subject
elsif subject.respond_to?(:group)
subject.group
else
nil
end
if group && group.public_profile?
[:read_group]
else
[]
end
end
end
2013-01-25 04:30:49 -05:00
def global_abilities(user)
rules = []
rules << :create_group if user.can_create_group
rules
2011-10-08 17:36:38 -04:00
end
2012-10-08 20:10:04 -04:00
def project_abilities(user, project)
rules = []
2014-06-14 06:05:25 -04:00
key = "/user/#{user.id}/project/#{project.id}"
2014-06-14 06:05:25 -04:00
RequestStore.store[key] ||= begin
team = project.team
2011-10-08 17:36:38 -04:00
2014-06-14 06:05:25 -04:00
# Rules based on role in project
if team.master?(user)
2014-10-09 03:47:47 -04:00
rules.push(*project_master_rules)
2013-01-03 14:09:18 -05:00
2014-06-14 06:05:25 -04:00
elsif team.developer?(user)
2014-10-09 03:47:47 -04:00
rules.push(*project_dev_rules)
2014-06-14 06:05:25 -04:00
elsif team.reporter?(user)
2014-10-09 03:47:47 -04:00
rules.push(*project_report_rules)
2014-06-14 06:05:25 -04:00
elsif team.guest?(user)
2014-10-09 03:47:47 -04:00
rules.push(*project_guest_rules)
2014-06-14 06:05:25 -04:00
end
2014-06-14 06:05:25 -04:00
if project.public? || project.internal?
2014-10-09 03:47:47 -04:00
rules.push(*public_project_rules)
2014-06-14 06:05:25 -04:00
end
2014-06-14 06:05:25 -04:00
if project.owner == user || user.admin?
2014-10-09 03:47:47 -04:00
rules.push(*project_admin_rules)
2014-06-14 06:05:25 -04:00
end
2014-06-14 06:05:25 -04:00
if project.group && project.group.has_owner?(user)
2014-10-09 03:47:47 -04:00
rules.push(*project_admin_rules)
2014-06-14 06:05:25 -04:00
end
2014-06-14 06:05:25 -04:00
if project.archived?
rules -= project_archived_rules
end
unless project.issues_enabled
rules -= named_abilities('issue')
end
unless project.merge_requests_enabled
rules -= named_abilities('merge_request')
end
unless project.issues_enabled or project.merge_requests_enabled
rules -= named_abilities('label')
rules -= named_abilities('milestone')
end
unless project.snippets_enabled
rules -= named_abilities('project_snippet')
end
unless project.wiki_enabled
rules -= named_abilities('wiki')
end
2014-06-14 06:05:25 -04:00
rules
end
end
def public_project_rules
project_guest_rules + [
:download_code,
:fork_project
]
end
def project_guest_rules
[
2012-10-08 20:10:04 -04:00
:read_project,
:read_wiki,
:read_issue,
:read_label,
2012-10-08 20:10:04 -04:00
:read_milestone,
2013-03-25 03:20:14 -04:00
:read_project_snippet,
:read_project_member,
2012-10-08 20:10:04 -04:00
:read_merge_request,
:read_note,
:create_project,
:create_issue,
:create_note
]
end
2012-02-20 13:16:55 -05:00
def project_report_rules
project_guest_rules + [
2012-10-08 20:10:04 -04:00
:download_code,
2013-06-04 11:50:42 -04:00
:fork_project,
:create_project_snippet,
:update_issue,
:admin_issue,
:admin_label,
]
end
2012-02-20 13:16:55 -05:00
def project_dev_rules
project_report_rules + [
:create_merge_request,
:create_wiki,
:push_code
]
end
def project_archived_rules
[
:create_merge_request,
:push_code,
:push_code_to_protected_branches,
:update_merge_request,
:admin_merge_request
]
end
def project_master_rules
project_dev_rules + [
:push_code_to_protected_branches,
:update_project_snippet,
:update_merge_request,
2012-10-08 20:10:04 -04:00
:admin_milestone,
2013-03-25 03:20:14 -04:00
:admin_project_snippet,
:admin_project_member,
2012-10-08 20:10:04 -04:00
:admin_merge_request,
:admin_note,
:admin_wiki,
:admin_project
]
end
2011-10-08 17:36:38 -04:00
def project_admin_rules
project_master_rules + [
:change_namespace,
:change_visibility_level,
:rename_project,
:remove_project,
:archive_project
]
2012-10-08 20:10:04 -04:00
end
2011-10-17 06:39:03 -04:00
def group_abilities(user, group)
rules = []
if user.admin? || group.users.include?(user) || ProjectsFinder.new.execute(user, group: group).any?
rules << :read_group
end
# Only group masters and group owners can create new projects in group
if group.has_master?(user) || group.has_owner?(user) || user.admin?
2014-10-09 03:47:47 -04:00
rules.push(*[
:create_projects,
2014-10-09 03:47:47 -04:00
])
end
# Only group owner and administrators can admin group
if group.has_owner?(user) || user.admin?
2014-10-09 03:47:47 -04:00
rules.push(*[
:admin_group,
:admin_namespace
2014-10-09 03:47:47 -04:00
])
end
rules.flatten
end
def namespace_abilities(user, namespace)
2013-06-21 15:44:40 -04:00
rules = []
# Only namespace owner and administrators can admin it
2013-06-21 15:44:40 -04:00
if namespace.owner == user || user.admin?
2014-10-09 03:47:47 -04:00
rules.push(*[
:create_projects,
:admin_namespace
2014-10-09 03:47:47 -04:00
])
2013-06-21 15:44:40 -04:00
end
rules.flatten
end
[:issue, :merge_request].each do |name|
2011-10-17 06:39:03 -04:00
define_method "#{name}_abilities" do |user, subject|
rules = []
if subject.author == user || (subject.respond_to?(:assignee) && subject.assignee == user)
rules += [
2011-10-17 06:39:03 -04:00
:"read_#{name}",
:"update_#{name}",
2011-10-17 06:39:03 -04:00
]
end
rules += project_abilities(user, subject.project)
rules
end
end
[:note, :project_snippet, :personal_snippet].each do |name|
define_method "#{name}_abilities" do |user, subject|
rules = []
if subject.author == user
rules += [
:"read_#{name}",
:"update_#{name}",
:"admin_#{name}"
]
2011-10-17 06:39:03 -04:00
end
if subject.respond_to?(:project) && subject.project
rules += project_abilities(user, subject.project)
end
rules
2011-10-17 06:39:03 -04:00
end
end
2014-02-07 11:59:55 -05:00
def group_member_abilities(user, subject)
2014-02-07 11:59:55 -05:00
rules = []
target_user = subject.user
group = subject.group
can_manage = group_abilities(user, group).include?(:admin_group)
2014-02-07 11:59:55 -05:00
if can_manage && (user != target_user)
rules << :update_group_member
rules << :destroy_group_member
2014-02-07 11:59:55 -05:00
end
2014-02-07 11:59:55 -05:00
if !group.last_owner?(user) && (can_manage || (user == target_user))
rules << :destroy_group_member
2014-02-07 11:59:55 -05:00
end
2014-02-07 11:59:55 -05:00
rules
end
def abilities
@abilities ||= begin
abilities = Six.new
abilities << self
abilities
end
end
private
def named_abilities(name)
[
:"read_#{name}",
:"create_#{name}",
:"update_#{name}",
:"admin_#{name}"
]
end
2011-10-17 06:39:03 -04:00
end
2011-10-08 17:36:38 -04:00
end