debian-mirror-gitlab/app/services/groups/import_export/import_service.rb
2020-05-24 23:13:21 +05:30

103 lines
2.4 KiB
Ruby

# frozen_string_literal: true
module Groups
module ImportExport
class ImportService
attr_reader :current_user, :group, :params
def initialize(group:, user:)
@group = group
@current_user = user
@shared = Gitlab::ImportExport::Shared.new(@group)
end
def execute
if valid_user_permissions? && import_file && restorer.restore
notify_success
@group
else
notify_error!
end
ensure
remove_import_file
end
private
def import_file
@import_file ||= Gitlab::ImportExport::FileImporter.import(
importable: @group,
archive_file: nil,
shared: @shared
)
end
def restorer
@restorer ||=
if ndjson?
Gitlab::ImportExport::Group::TreeRestorer.new(
user: @current_user,
shared: @shared,
group: @group
)
else
Gitlab::ImportExport::Group::LegacyTreeRestorer.new(
user: @current_user,
shared: @shared,
group: @group,
group_hash: nil
)
end
end
def ndjson?
::Feature.enabled?(:group_import_ndjson, @group&.parent, default_enabled: true) &&
File.exist?(File.join(@shared.export_path, 'tree/groups/_all.ndjson'))
end
def remove_import_file
upload = @group.import_export_upload
return unless upload&.import_file&.file
upload.remove_import_file!
upload.save!
end
def valid_user_permissions?
if current_user.can?(:admin_group, group)
true
else
@shared.error(::Gitlab::ImportExport::Error.permission_error(current_user, group))
false
end
end
def notify_success
@shared.logger.info(
group_id: @group.id,
group_name: @group.name,
message: 'Group Import/Export: Import succeeded'
)
end
def notify_error
@shared.logger.error(
group_id: @group.id,
group_name: @group.name,
message: "Group Import/Export: Errors occurred, see '#{Gitlab::ErrorTracking::Logger.file_name}' for details"
)
end
def notify_error!
notify_error
raise Gitlab::ImportExport::Error.new(@shared.errors.to_sentence)
end
end
end
end