80 lines
2.4 KiB
Ruby
80 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module ImportExport
|
|
class ProjectTreeSaver
|
|
include Gitlab::ImportExport::CommandLineUtil
|
|
|
|
attr_reader :full_path
|
|
|
|
def initialize(project:, current_user:, shared:, params: {})
|
|
@params = params
|
|
@project = project
|
|
@current_user = current_user
|
|
@shared = shared
|
|
@full_path = File.join(@shared.export_path, ImportExport.project_filename)
|
|
end
|
|
|
|
def save
|
|
mkdir_p(@shared.export_path)
|
|
|
|
project_tree = serialize_project_tree
|
|
fix_project_tree(project_tree)
|
|
project_tree_json = JSON.generate(project_tree)
|
|
File.write(full_path, project_tree_json)
|
|
|
|
true
|
|
rescue => e
|
|
@shared.error(e)
|
|
false
|
|
end
|
|
|
|
private
|
|
|
|
# Aware that the resulting hash needs to be pure-hash and
|
|
# does not include any AR objects anymore, only objects that run `.to_json`
|
|
def fix_project_tree(project_tree)
|
|
if @params[:description].present?
|
|
project_tree['description'] = @params[:description]
|
|
end
|
|
|
|
project_tree['project_members'] += group_members_array
|
|
|
|
RelationRenameService.add_new_associations(project_tree)
|
|
end
|
|
|
|
def serialize_project_tree
|
|
if Feature.enabled?(:export_fast_serialize, default_enabled: true)
|
|
Gitlab::ImportExport::FastHashSerializer
|
|
.new(@project, reader.project_tree)
|
|
.execute
|
|
else
|
|
@project.as_json(reader.project_tree)
|
|
end
|
|
end
|
|
|
|
def reader
|
|
@reader ||= Gitlab::ImportExport::Reader.new(shared: @shared)
|
|
end
|
|
|
|
def group_members_array
|
|
group_members.as_json(reader.group_members_tree).each do |group_member|
|
|
group_member['source_type'] = 'Project' # Make group members project members of the future import
|
|
end
|
|
end
|
|
|
|
def group_members
|
|
return [] unless @current_user.can?(:admin_group, @project.group)
|
|
|
|
# We need `.where.not(user_id: nil)` here otherwise when a group has an
|
|
# invitee, it would make the following query return 0 rows since a NULL
|
|
# user_id would be present in the subquery
|
|
# See http://stackoverflow.com/questions/129077/not-in-clause-and-null-values
|
|
non_null_user_ids = @project.project_members.where.not(user_id: nil).select(:user_id)
|
|
|
|
GroupMembersFinder.new(@project.group).execute.where.not(user_id: non_null_user_ids)
|
|
end
|
|
end
|
|
end
|
|
end
|