gitlab-org--gitlab-foss/lib/gitlab/import_export/avatar_restorer.rb

33 lines
675 B
Ruby

# frozen_string_literal: true
module Gitlab
module ImportExport
class AvatarRestorer
def initialize(project:, shared:)
@project = project
@shared = shared
end
def restore
return true unless avatar_export_file
@project.avatar = File.open(avatar_export_file)
@project.save!
rescue StandardError => e
@shared.error(e)
false
end
private
def avatar_export_file
@avatar_export_file ||= Dir["#{avatar_export_path}/**/*"].find { |f| File.file?(f) }
end
def avatar_export_path
File.join(@shared.export_path, 'avatar')
end
end
end
end