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)
|
2016-08-02 18:37:22 +00:00
|
|
|
add_users(users, access, current_user: current_user)
|
2013-01-04 06:43:25 +00:00
|
|
|
else
|
2016-08-18 16:01:50 +00:00
|
|
|
add_user(users, access, current_user: current_user)
|
2013-01-04 06:43:25 +00:00
|
|
|
end
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
2016-11-18 12:52:39 +00:00
|
|
|
def add_guest(user, current_user: nil)
|
|
|
|
self << [user, :guest, current_user]
|
|
|
|
end
|
|
|
|
|
|
|
|
def add_reporter(user, current_user: nil)
|
|
|
|
self << [user, :reporter, current_user]
|
|
|
|
end
|
|
|
|
|
|
|
|
def add_developer(user, current_user: nil)
|
|
|
|
self << [user, :developer, current_user]
|
|
|
|
end
|
|
|
|
|
|
|
|
def add_master(user, current_user: nil)
|
|
|
|
self << [user, :master, current_user]
|
|
|
|
end
|
|
|
|
|
2015-03-13 15:23:45 +00:00
|
|
|
def find_member(user_id)
|
2016-06-27 14:20:57 +00:00
|
|
|
member = 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
|
2016-06-27 14:20:57 +00:00
|
|
|
member = 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
|
|
|
|
|
2016-09-16 15:54:21 +00:00
|
|
|
def add_users(users, access_level, current_user: nil, expires_at: nil)
|
2017-04-21 14:07:42 +00:00
|
|
|
ProjectMember.add_users(
|
|
|
|
project,
|
2015-04-10 12:46:09 +00:00
|
|
|
users,
|
2016-09-16 15:54:21 +00:00
|
|
|
access_level,
|
2016-08-02 18:37:22 +00:00
|
|
|
current_user: current_user,
|
|
|
|
expires_at: expires_at
|
2013-01-03 19:09:18 +00:00
|
|
|
)
|
|
|
|
end
|
|
|
|
|
2016-09-16 15:54:21 +00:00
|
|
|
def add_user(user, access_level, current_user: nil, expires_at: nil)
|
|
|
|
ProjectMember.add_user(
|
|
|
|
project,
|
|
|
|
user,
|
|
|
|
access_level,
|
|
|
|
current_user: current_user,
|
|
|
|
expires_at: expires_at
|
|
|
|
)
|
2015-04-10 12:46:09 +00:00
|
|
|
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
|
|
|
|
|
|
|
|
def members
|
2013-06-22 09:57:05 +00:00
|
|
|
@members ||= fetch_members
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
2016-04-18 16:53:32 +00:00
|
|
|
alias_method :users, :members
|
2013-01-03 19:09:18 +00:00
|
|
|
|
|
|
|
def guests
|
2016-11-21 14:26:12 +00:00
|
|
|
@guests ||= fetch_members(Gitlab::Access::GUEST)
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def reporters
|
2016-11-21 14:26:12 +00:00
|
|
|
@reporters ||= fetch_members(Gitlab::Access::REPORTER)
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def developers
|
2016-11-21 14:26:12 +00:00
|
|
|
@developers ||= fetch_members(Gitlab::Access::DEVELOPER)
|
2013-01-03 19:09:18 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def masters
|
2016-11-21 14:26:12 +00:00
|
|
|
@masters ||= fetch_members(Gitlab::Access::MASTER)
|
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
|
|
|
|
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-11-18 17:15:47 +00:00
|
|
|
# Checks if `user` is authorized for this project, with at least the
|
|
|
|
# `min_access_level` (if given).
|
|
|
|
def member?(user, min_access_level = Gitlab::Access::GUEST)
|
|
|
|
return false unless user
|
|
|
|
|
|
|
|
user.authorized_project?(project, min_access_level)
|
2014-10-08 13:44:25 +00:00
|
|
|
end
|
|
|
|
|
2016-06-02 16:05:06 +00:00
|
|
|
def human_max_access(user_id)
|
|
|
|
Gitlab::Access.options_with_owner.key(max_member_access(user_id))
|
|
|
|
end
|
|
|
|
|
2016-07-20 04:52:31 +00:00
|
|
|
# Determine the maximum access level for a group of users in bulk.
|
|
|
|
#
|
2016-07-27 00:07:51 +00:00
|
|
|
# Returns a Hash mapping user ID -> maximum access level.
|
2016-07-20 04:52:31 +00:00
|
|
|
def max_member_access_for_user_ids(user_ids)
|
|
|
|
user_ids = user_ids.uniq
|
|
|
|
key = "max_member_access:#{project.id}"
|
2016-08-01 20:11:45 +00:00
|
|
|
|
|
|
|
access = {}
|
|
|
|
|
|
|
|
if RequestStore.active?
|
|
|
|
RequestStore.store[key] ||= {}
|
|
|
|
access = RequestStore.store[key]
|
|
|
|
end
|
2015-10-14 12:54:04 +00:00
|
|
|
|
2016-07-20 04:52:31 +00:00
|
|
|
# Lookup only the IDs we need
|
|
|
|
user_ids = user_ids - access.keys
|
2017-03-31 06:00:00 +00:00
|
|
|
|
|
|
|
return access if user_ids.empty?
|
|
|
|
|
2016-11-21 13:36:40 +00:00
|
|
|
users_access = project.project_authorizations.
|
|
|
|
where(user: user_ids).
|
|
|
|
group(:user_id).
|
|
|
|
maximum(:access_level)
|
2014-06-20 09:54:03 +00:00
|
|
|
|
2016-11-21 13:36:40 +00:00
|
|
|
access.merge!(users_access)
|
2016-07-20 04:52:31 +00:00
|
|
|
access
|
|
|
|
end
|
|
|
|
|
|
|
|
def max_member_access(user_id)
|
2016-11-21 13:36:40 +00:00
|
|
|
max_member_access_for_user_ids([user_id])[user_id] || Gitlab::Access::NO_ACCESS
|
2016-03-11 17:46:01 +00:00
|
|
|
end
|
|
|
|
|
2016-04-18 16:53:32 +00:00
|
|
|
private
|
2016-03-11 17:46:01 +00:00
|
|
|
|
2013-06-17 11:17:32 +00:00
|
|
|
def fetch_members(level = nil)
|
2016-11-21 14:26:12 +00:00
|
|
|
members = project.authorized_users
|
|
|
|
members = members.where(project_authorizations: { access_level: level }) if level
|
2013-06-17 11:17:32 +00:00
|
|
|
|
2016-11-21 14:26:12 +00:00
|
|
|
members
|
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
|