2013-01-19 18:52:55 +00:00
|
|
|
class ProjectTeam
|
2013-01-03 19:09:18 +00:00
|
|
|
attr_accessor :project
|
|
|
|
|
|
|
|
def initialize(project)
|
|
|
|
@project = project
|
2013-01-04 06:43:25 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
# Shortcut to add users
|
|
|
|
#
|
|
|
|
# Use:
|
|
|
|
# @team << [@user, :master]
|
|
|
|
# @team << [@users, :master]
|
|
|
|
#
|
2014-09-25 22:07:40 +00:00
|
|
|
def <<(args)
|
2015-04-10 12:46:09 +00:00
|
|
|
users, access, current_user = *args
|
2013-01-04 06:43:25 +00:00
|
|
|
|
|
|
|
if users.respond_to?(:each)
|
2015-04-10 12:46:09 +00:00
|
|
|
add_users(users, access, current_user)
|
2013-01-04 06:43:25 +00:00
|
|
|
else
|
2015-04-10 12:46:09 +00:00
|
|
|
add_user(users, access, current_user)
|
2013-01-04 06:43:25 +00:00
|
|
|
end
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
2013-06-26 12:57:02 +00:00
|
|
|
def find(user_id)
|
2014-01-19 18:55:59 +00:00
|
|
|
user = project.users.find_by(id: user_id)
|
2013-06-17 12:38:19 +00:00
|
|
|
|
|
|
|
if group
|
2014-01-19 18:55:59 +00:00
|
|
|
user ||= group.users.find_by(id: user_id)
|
2013-06-17 12:38:19 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
user
|
2013-06-17 11:55:41 +00:00
|
|
|
end
|
|
|
|
|
2015-03-13 15:23:45 +00:00
|
|
|
def find_member(user_id)
|
|
|
|
member = project.project_members.find_by(user_id: user_id)
|
2013-08-27 18:35:41 +00:00
|
|
|
|
|
|
|
# If user is not in project members
|
|
|
|
# we should check for group membership
|
2015-03-13 15:23:45 +00:00
|
|
|
if group && !member
|
|
|
|
member = group.group_members.find_by(user_id: user_id)
|
2013-08-27 18:35:41 +00:00
|
|
|
end
|
|
|
|
|
2015-03-13 15:23:45 +00:00
|
|
|
member
|
2013-01-22 17:45:13 +00:00
|
|
|
end
|
|
|
|
|
2015-04-10 12:46:09 +00:00
|
|
|
def add_users(users, access, current_user = nil)
|
2014-09-14 16:32:51 +00:00
|
|
|
ProjectMember.add_users_into_projects(
|
2013-01-03 19:09:18 +00:00
|
|
|
[project.id],
|
2015-04-10 12:46:09 +00:00
|
|
|
users,
|
|
|
|
access,
|
|
|
|
current_user
|
2013-01-03 19:09:18 +00:00
|
|
|
)
|
|
|
|
end
|
|
|
|
|
2015-04-10 12:46:09 +00:00
|
|
|
def add_user(user, access, current_user = nil)
|
|
|
|
add_users([user], access, current_user)
|
|
|
|
end
|
|
|
|
|
2013-01-03 19:09:18 +00:00
|
|
|
# Remove all users from project team
|
|
|
|
def truncate
|
2014-09-14 16:32:51 +00:00
|
|
|
ProjectMember.truncate_team(project)
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
2013-12-25 07:56:39 +00:00
|
|
|
def users
|
|
|
|
members
|
|
|
|
end
|
|
|
|
|
2013-01-03 19:09:18 +00:00
|
|
|
def members
|
2013-06-22 09:57:05 +00:00
|
|
|
@members ||= fetch_members
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def guests
|
2013-06-17 11:17:32 +00:00
|
|
|
@guests ||= fetch_members(:guests)
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def reporters
|
2013-06-17 11:17:32 +00:00
|
|
|
@reporters ||= fetch_members(:reporters)
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def developers
|
2013-06-17 11:17:32 +00:00
|
|
|
@developers ||= fetch_members(:developers)
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def masters
|
2013-06-17 11:17:32 +00:00
|
|
|
@masters ||= fetch_members(:masters)
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
2013-01-04 06:43:25 +00:00
|
|
|
|
2015-04-10 12:46:09 +00:00
|
|
|
def import(source_project, current_user = nil)
|
2013-01-04 06:43:25 +00:00
|
|
|
target_project = project
|
|
|
|
|
2015-03-13 15:23:45 +00:00
|
|
|
source_members = source_project.project_members.to_a
|
2014-09-14 16:32:51 +00:00
|
|
|
target_user_ids = target_project.project_members.pluck(:user_id)
|
2013-01-04 06:43:25 +00:00
|
|
|
|
2015-03-13 15:23:45 +00:00
|
|
|
source_members.reject! do |member|
|
2013-01-04 06:43:25 +00:00
|
|
|
# Skip if user already present in team
|
2015-04-10 13:26:53 +00:00
|
|
|
!member.invite? && target_user_ids.include?(member.user_id)
|
2013-01-04 06:43:25 +00:00
|
|
|
end
|
|
|
|
|
2015-03-13 15:23:45 +00:00
|
|
|
source_members.map! do |member|
|
|
|
|
new_member = member.dup
|
|
|
|
new_member.id = nil
|
|
|
|
new_member.source = target_project
|
2015-04-10 12:46:09 +00:00
|
|
|
new_member.created_by = current_user
|
2015-03-13 15:23:45 +00:00
|
|
|
new_member
|
2013-01-04 06:43:25 +00:00
|
|
|
end
|
|
|
|
|
2014-09-14 16:32:51 +00:00
|
|
|
ProjectMember.transaction do
|
2015-03-13 15:23:45 +00:00
|
|
|
source_members.each do |member|
|
|
|
|
member.save
|
2013-01-04 06:43:25 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
true
|
|
|
|
rescue
|
|
|
|
false
|
|
|
|
end
|
2013-06-17 11:17:32 +00:00
|
|
|
|
2016-03-14 13:13:35 +00:00
|
|
|
def pending?(user)
|
|
|
|
project.project_members.each do |member|
|
|
|
|
return member.pending? if member.user_id == user.id
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2014-06-04 08:52:17 +00:00
|
|
|
def guest?(user)
|
2015-03-13 15:23:45 +00:00
|
|
|
max_member_access(user.id) == Gitlab::Access::GUEST
|
2014-06-04 08:52:17 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def reporter?(user)
|
2015-03-13 15:23:45 +00:00
|
|
|
max_member_access(user.id) == Gitlab::Access::REPORTER
|
2014-06-04 08:52:17 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def developer?(user)
|
2015-03-13 15:23:45 +00:00
|
|
|
max_member_access(user.id) == Gitlab::Access::DEVELOPER
|
2014-06-04 08:52:17 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def master?(user)
|
2015-03-13 15:23:45 +00:00
|
|
|
max_member_access(user.id) == Gitlab::Access::MASTER
|
2014-06-20 09:54:03 +00:00
|
|
|
end
|
|
|
|
|
2016-06-06 19:13:31 +00:00
|
|
|
def member?(user, min_member_access = nil)
|
|
|
|
member = !!find_member(user.id)
|
|
|
|
|
|
|
|
if min_member_access
|
|
|
|
member && max_member_access(user.id) >= min_member_access
|
|
|
|
else
|
|
|
|
member
|
|
|
|
end
|
2014-10-08 13:44:25 +00:00
|
|
|
end
|
|
|
|
|
2015-10-02 12:10:33 +00:00
|
|
|
def human_max_access(user_id)
|
2016-02-18 21:39:59 +00:00
|
|
|
Gitlab::Access.options_with_owner.key(max_member_access(user_id))
|
2015-10-02 12:10:33 +00:00
|
|
|
end
|
|
|
|
|
2015-10-14 12:54:04 +00:00
|
|
|
# This method assumes project and group members are eager loaded for optimal
|
|
|
|
# performance.
|
2015-03-13 15:23:45 +00:00
|
|
|
def max_member_access(user_id)
|
2014-06-20 09:54:03 +00:00
|
|
|
access = []
|
2015-10-14 12:54:04 +00:00
|
|
|
|
|
|
|
project.project_members.each do |member|
|
|
|
|
if member.user_id == user_id
|
|
|
|
access << member.access_field if member.access_field
|
|
|
|
break
|
|
|
|
end
|
|
|
|
end
|
2014-06-20 09:54:03 +00:00
|
|
|
|
|
|
|
if group
|
2015-10-14 12:54:04 +00:00
|
|
|
group.group_members.each do |member|
|
|
|
|
if member.user_id == user_id
|
|
|
|
access << member.access_field if member.access_field
|
|
|
|
break
|
|
|
|
end
|
|
|
|
end
|
2014-06-20 09:54:03 +00:00
|
|
|
end
|
|
|
|
|
2016-03-11 17:46:01 +00:00
|
|
|
if project.invited_groups.any? && project.allowed_to_share_with_group?
|
|
|
|
access << max_invited_level(user_id)
|
|
|
|
end
|
|
|
|
|
|
|
|
access.compact.max
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
def max_invited_level(user_id)
|
|
|
|
project.project_group_links.map do |group_link|
|
|
|
|
invited_group = group_link.group
|
|
|
|
access = invited_group.group_members.find_by(user_id: user_id).try(:access_field)
|
|
|
|
|
|
|
|
# If group member has higher access level we should restrict it
|
|
|
|
# to max allowed access level
|
|
|
|
if access && access > group_link.group_access
|
|
|
|
access = group_link.group_access
|
|
|
|
end
|
|
|
|
|
|
|
|
access
|
|
|
|
end.compact.max
|
2014-06-04 08:52:17 +00:00
|
|
|
end
|
|
|
|
|
2013-06-17 11:17:32 +00:00
|
|
|
private
|
|
|
|
|
|
|
|
def fetch_members(level = nil)
|
2014-09-14 16:32:51 +00:00
|
|
|
project_members = project.project_members
|
|
|
|
group_members = group ? group.group_members : []
|
2016-03-11 17:46:01 +00:00
|
|
|
invited_members = []
|
|
|
|
|
|
|
|
if project.invited_groups.any? && project.allowed_to_share_with_group?
|
|
|
|
project.project_group_links.each do |group_link|
|
|
|
|
invited_group = group_link.group
|
|
|
|
im = invited_group.group_members
|
|
|
|
|
|
|
|
if level
|
|
|
|
int_level = GroupMember.access_level_roles[level.to_s.singularize.titleize]
|
|
|
|
|
|
|
|
# Skip group members if we ask for masters
|
|
|
|
# but max group access is developers
|
|
|
|
next if int_level > group_link.group_access
|
|
|
|
|
|
|
|
# If we ask for developers and max
|
|
|
|
# group access is developers we need to provide
|
|
|
|
# both group master, developers as devs
|
|
|
|
if int_level == group_link.group_access
|
|
|
|
im.where("access_level >= ?)", group_link.group_access)
|
|
|
|
else
|
|
|
|
im.send(level)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
invited_members << im
|
|
|
|
end
|
|
|
|
|
|
|
|
invited_members = invited_members.flatten.compact
|
|
|
|
end
|
2013-06-17 11:17:32 +00:00
|
|
|
|
|
|
|
if level
|
|
|
|
project_members = project_members.send(level)
|
2013-06-17 12:38:19 +00:00
|
|
|
group_members = group_members.send(level) if group
|
2013-06-17 11:17:32 +00:00
|
|
|
end
|
|
|
|
|
2014-06-04 16:20:25 +00:00
|
|
|
user_ids = project_members.pluck(:user_id)
|
2016-03-11 17:46:01 +00:00
|
|
|
user_ids.push(*invited_members.map(&:user_id)) if invited_members.any?
|
2014-10-09 07:47:47 +00:00
|
|
|
user_ids.push(*group_members.pluck(:user_id)) if group
|
2014-06-04 16:20:25 +00:00
|
|
|
|
|
|
|
User.where(id: user_ids)
|
2013-06-17 11:17:32 +00:00
|
|
|
end
|
2013-06-17 12:38:19 +00:00
|
|
|
|
|
|
|
def group
|
|
|
|
project.group
|
|
|
|
end
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|