ability.rb 2.3 KB
Newer Older
G
gitlabhq 已提交
1
class Ability
A
Andrey Kumanyaev 已提交
2 3 4 5 6 7 8 9
  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)
10
      when "Group" then group_abilities(object, subject)
A
Andrey Kumanyaev 已提交
11 12
      else []
      end
G
gitlabhq 已提交
13 14
    end

A
Andrey Kumanyaev 已提交
15 16
    def project_abilities(user, project)
      rules = []
G
gitlabhq 已提交
17

A
Andrey Kumanyaev 已提交
18 19 20 21 22 23 24 25 26 27 28 29 30
      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
      ] if project.guest_access_for?(user)
D
Dmitriy Zaporozhets 已提交
31

A
Andrey Kumanyaev 已提交
32 33 34 35 36
      rules << [
        :download_code,
        :write_merge_request,
        :write_snippet
      ] if project.report_access_for?(user)
D
Dmitriy Zaporozhets 已提交
37

A
Andrey Kumanyaev 已提交
38
      rules << [
39 40
        :write_wiki,
        :push_code
A
Andrey Kumanyaev 已提交
41
      ] if project.dev_access_for?(user)
G
gitlabhq 已提交
42

A
Andrey Kumanyaev 已提交
43
      rules << [
44 45 46 47
        :push_code_to_protected_branches
      ] if project.master_access_for?(user)

      rules << [
A
Andrey Kumanyaev 已提交
48 49 50 51 52 53 54 55 56 57 58 59 60
        :modify_issue,
        :modify_snippet,
        :modify_merge_request,
        :admin_project,
        :admin_issue,
        :admin_milestone,
        :admin_snippet,
        :admin_team_member,
        :admin_merge_request,
        :admin_note,
        :accept_mr,
        :admin_wiki
      ] if project.master_access_for?(user) || project.owner == user
G
gitlabhq 已提交
61

A
Andrey Kumanyaev 已提交
62 63
      rules.flatten
    end
G
gitlabhq 已提交
64

65 66 67 68 69 70 71 72 73 74
    def group_abilities user, group
      rules = []

      rules << [
        :manage_group
      ] if group.owner == user

      rules.flatten
    end

D
Dmitriy Zaporozhets 已提交
75
    [:issue, :note, :snippet, :merge_request].each do |name|
G
gitlabhq 已提交
76 77 78 79 80
      define_method "#{name}_abilities" do |user, subject|
        if subject.author == user
          [
            :"read_#{name}",
            :"write_#{name}",
D
Dmitriy Zaporozhets 已提交
81
            :"modify_#{name}",
G
gitlabhq 已提交
82 83
            :"admin_#{name}"
          ]
84 85 86 87 88 89
        elsif subject.respond_to?(:assignee) && subject.assignee == user
          [
            :"read_#{name}",
            :"write_#{name}",
            :"modify_#{name}",
          ]
G
gitlabhq 已提交
90
        else
A
Andrey Kumanyaev 已提交
91
          subject.respond_to?(:project) ? project_abilities(user, subject.project) : []
G
gitlabhq 已提交
92 93 94 95
        end
      end
    end
  end
G
gitlabhq 已提交
96
end