BigW Consortium Gitlab

project_member.rb 3.71 KB
Newer Older
Valery Sizov committed
1 2 3 4 5 6 7 8
# == Schema Information
#
# Table name: members
#
#  id                 :integer          not null, primary key
#  access_level       :integer          not null
#  source_id          :integer          not null
#  source_type        :string(255)      not null
Stan Hu committed
9
#  user_id            :integer
Valery Sizov committed
10 11 12 13
#  notification_level :integer          not null
#  type               :string(255)
#  created_at         :datetime
#  updated_at         :datetime
Stan Hu committed
14 15 16 17
#  created_by_id      :integer
#  invite_email       :string(255)
#  invite_token       :string(255)
#  invite_accepted_at :datetime
Valery Sizov committed
18 19
#

20 21 22 23 24
class ProjectMember < Member
  SOURCE_TYPE = 'Project'

  include Gitlab::ShellAdapter

25 26
  belongs_to :project, class_name: 'Project', foreign_key: 'source_id'

27

28 29
  # Make sure project member points only to project as it source
  default_value_for :source_type, SOURCE_TYPE
30
  default_value_for :notification_level, Notification::N_GLOBAL
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
  validates_format_of :source_type, with: /\AProject\z/
  default_scope { where(source_type: SOURCE_TYPE) }

  scope :in_project, ->(project) { where(source_id: project.id) }
  scope :in_projects, ->(projects) { where(source_id: projects.pluck(:id)) }
  scope :with_user, ->(user) { where(user_id: user.id) }

  class << self

    # Add users to project teams with passed access option
    #
    # access can be an integer representing a access code
    # or symbol like :master representing role
    #
    # Ex.
    #   add_users_into_projects(
    #     project_ids,
    #     user_ids,
    #     ProjectMember::MASTER
    #   )
    #
    #   add_users_into_projects(
    #     project_ids,
    #     user_ids,
    #     :master
    #   )
    #
58
    def add_users_into_projects(project_ids, user_ids, access, current_user = nil)
59 60 61 62 63 64 65 66
      access_level = if roles_hash.has_key?(access)
                       roles_hash[access]
                     elsif roles_hash.values.include?(access.to_i)
                       access
                     else
                       raise "Non valid access"
                     end

67
      users = user_ids.map { |user_id| Member.user_for_id(user_id) }
68

69 70
      ProjectMember.transaction do
        project_ids.each do |project_id|
71 72 73
          project = Project.find(project_id)

          users.each do |user|
74
            Member.add_user(project.project_members, user, access_level, current_user)
75 76 77 78 79 80 81 82 83 84 85 86
          end
        end
      end

      true
    rescue
      false
    end

    def truncate_teams(project_ids)
      ProjectMember.transaction do
        members = ProjectMember.where(source_id: project_ids)
87
        
88 89 90 91 92 93 94 95 96 97
        members.each do |member|
          member.destroy
        end
      end

      true
    rescue
      false
    end

98
    def truncate_team(project)
99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
      truncate_teams [project.id]
    end

    def roles_hash
      Gitlab::Access.sym_options
    end

    def access_roles
      Gitlab::Access.options
    end
  end

  def access_field
    access_level
  end

115 116 117 118
  def project
    source
  end

119 120 121 122
  def owner?
    project.owner == user
  end

123 124
  private

125 126 127 128 129 130
  def send_invite
    notification_service.invite_project_member(self, @raw_invite_token)

    super
  end

131
  def post_create_hook
132 133
    unless owner?
      event_service.join_project(self.project, self.user)
134
      notification_service.new_project_member(self)
135 136
    end
    
137
    super
138 139 140
  end

  def post_update_hook
141 142 143 144 145
    if access_level_changed?
      notification_service.update_project_member(self) 
    end

    super
146 147 148
  end

  def post_destroy_hook
149
    event_service.leave_project(self.project, self.user)
150 151

    super
152 153
  end

154 155 156 157 158 159
  def after_accept_invite
    notification_service.accept_project_invite(self)

    super
  end

Douwe Maan committed
160 161 162 163 164 165
  def after_decline_invite
    notification_service.decline_project_invite(self)

    super
  end

166 167 168
  def event_service
    EventCreateService.new
  end
169
end