153 lines
2.8 KiB
Ruby
153 lines
2.8 KiB
Ruby
class ProjectTeam
|
|
attr_accessor :project
|
|
|
|
def initialize(project)
|
|
@project = project
|
|
end
|
|
|
|
# Shortcut to add users
|
|
#
|
|
# Use:
|
|
# @team << [@user, :master]
|
|
# @team << [@users, :master]
|
|
#
|
|
def << args
|
|
users = args.first
|
|
|
|
if users.respond_to?(:each)
|
|
add_users(users, args.second)
|
|
else
|
|
add_user(users, args.second)
|
|
end
|
|
end
|
|
|
|
def find(user_id)
|
|
user = project.users.find_by(id: user_id)
|
|
|
|
if group
|
|
user ||= group.users.find_by(id: user_id)
|
|
end
|
|
|
|
user
|
|
end
|
|
|
|
def find_tm(user_id)
|
|
tm = project.users_projects.find_by(user_id: user_id)
|
|
|
|
# If user is not in project members
|
|
# we should check for group membership
|
|
if group && !tm
|
|
tm = group.users_groups.find_by(user_id: user_id)
|
|
end
|
|
|
|
tm
|
|
end
|
|
|
|
def add_user(user, access)
|
|
add_users_ids([user.id], access)
|
|
end
|
|
|
|
def add_users(users, access)
|
|
add_users_ids(users.map(&:id), access)
|
|
end
|
|
|
|
def add_users_ids(user_ids, access)
|
|
UsersProject.add_users_into_projects(
|
|
[project.id],
|
|
user_ids,
|
|
access
|
|
)
|
|
end
|
|
|
|
# Remove all users from project team
|
|
def truncate
|
|
UsersProject.truncate_team(project)
|
|
end
|
|
|
|
def users
|
|
members
|
|
end
|
|
|
|
def members
|
|
@members ||= fetch_members
|
|
end
|
|
|
|
def guests
|
|
@guests ||= fetch_members(:guests)
|
|
end
|
|
|
|
def reporters
|
|
@reporters ||= fetch_members(:reporters)
|
|
end
|
|
|
|
def developers
|
|
@developers ||= fetch_members(:developers)
|
|
end
|
|
|
|
def masters
|
|
@masters ||= fetch_members(:masters)
|
|
end
|
|
|
|
def import(source_project)
|
|
target_project = project
|
|
|
|
source_team = source_project.users_projects.to_a
|
|
target_user_ids = target_project.users_projects.pluck(:user_id)
|
|
|
|
source_team.reject! do |tm|
|
|
# Skip if user already present in team
|
|
target_user_ids.include?(tm.user_id)
|
|
end
|
|
|
|
source_team.map! do |tm|
|
|
new_tm = tm.dup
|
|
new_tm.id = nil
|
|
new_tm.project_id = target_project.id
|
|
new_tm
|
|
end
|
|
|
|
UsersProject.transaction do
|
|
source_team.each do |tm|
|
|
tm.save
|
|
end
|
|
end
|
|
|
|
true
|
|
rescue
|
|
false
|
|
end
|
|
|
|
def guest?(user)
|
|
find_tm(user.id).try(:access_field) == Gitlab::Access::GUEST
|
|
end
|
|
|
|
def reporter?(user)
|
|
find_tm(user.id).try(:access_field) == Gitlab::Access::REPORTER
|
|
end
|
|
|
|
def developer?(user)
|
|
find_tm(user.id).try(:access_field) == Gitlab::Access::DEVELOPER
|
|
end
|
|
|
|
def master?(user)
|
|
find_tm(user.id).try(:access_field) == Gitlab::Access::MASTER
|
|
end
|
|
|
|
private
|
|
|
|
def fetch_members(level = nil)
|
|
project_members = project.users_projects
|
|
group_members = group ? group.users_groups : []
|
|
|
|
if level
|
|
project_members = project_members.send(level)
|
|
group_members = group_members.send(level) if group
|
|
end
|
|
|
|
(project_members + group_members).map(&:user).uniq
|
|
end
|
|
|
|
def group
|
|
project.group
|
|
end
|
|
end
|