class Ability class << self def allowed(object, subject) case subject.class.name when "Project" then project_abilities(object, subject) when "Issue" then issue_abilities(object, subject) when "Note" then note_abilities(object, subject) when "Snippet" then snippet_abilities(object, subject) when "MergeRequest" then merge_request_abilities(object, subject) when "Group" then group_abilities(object, subject) else [] end end def project_abilities(user, project) rules = [] # Rules based on role in project if project.master_access_for?(user) # TODO: replace with master rules. # Only allow project administration for owners rules << project_admin_rules elsif project.dev_access_for?(user) rules << project_dev_rules elsif project.report_access_for?(user) rules << project_report_rules elsif project.guest_access_for?(user) rules << project_guest_rules end # If user own project namespace (Ex. group owner or account owner) if project.namespace && project.namespace.owner == user rules << project_admin_rules end # If user was set as direct project owner if project.owner == user rules << project_admin_rules end rules.flatten end def project_guest_rules [ :read_project, :read_wiki, :read_issue, :read_milestone, :read_snippet, :read_team_member, :read_merge_request, :read_note, :write_project, :write_issue, :write_note ] end def project_report_rules project_guest_rules + [ :download_code, :write_merge_request, :write_snippet ] end def project_dev_rules project_report_rules + [ :write_wiki, :push_code ] end def project_master_rules project_dev_rules + [ :push_code_to_protected_branches, :modify_issue, :modify_snippet, :modify_merge_request, :admin_issue, :admin_milestone, :admin_snippet, :admin_team_member, :admin_merge_request, :admin_note, :accept_mr, :admin_wiki ] end def project_admin_rules project_master_rules + [ :admin_project ] end def group_abilities user, group rules = [] rules << [ :manage_group ] if group.owner == user rules.flatten end [:issue, :note, :snippet, :merge_request].each do |name| define_method "#{name}_abilities" do |user, subject| if subject.author == user [ :"read_#{name}", :"write_#{name}", :"modify_#{name}", :"admin_#{name}" ] elsif subject.respond_to?(:assignee) && subject.assignee == user [ :"read_#{name}", :"write_#{name}", :"modify_#{name}", ] else subject.respond_to?(:project) ? project_abilities(user, subject.project) : [] end end end end end