48b17e991a
This makes accessing LFS Objects for a project easier project.lfs_storage_project.lfs_objects` becomes project.all_lfs_objects This will make the refactor in https://gitlab.com/gitlab-org/gitlab-ce/issues/39769 easier to deal with.
55 lines
1.3 KiB
Ruby
55 lines
1.3 KiB
Ruby
module Gitlab
|
|
module ImportExport
|
|
class LfsSaver
|
|
include Gitlab::ImportExport::CommandLineUtil
|
|
|
|
def initialize(project:, shared:)
|
|
@project = project
|
|
@shared = shared
|
|
end
|
|
|
|
def save
|
|
@project.all_lfs_objects.each do |lfs_object|
|
|
save_lfs_object(lfs_object)
|
|
end
|
|
|
|
true
|
|
rescue => e
|
|
@shared.error(e)
|
|
|
|
false
|
|
end
|
|
|
|
private
|
|
|
|
def save_lfs_object(lfs_object)
|
|
if lfs_object.local_store?
|
|
copy_file_for_lfs_object(lfs_object)
|
|
else
|
|
download_file_for_lfs_object(lfs_object)
|
|
end
|
|
end
|
|
|
|
def download_file_for_lfs_object(lfs_object)
|
|
destination = destination_path_for_object(lfs_object)
|
|
mkdir_p(File.dirname(destination))
|
|
|
|
File.open(destination, 'w') do |file|
|
|
IO.copy_stream(URI.parse(lfs_object.file.url).open, file)
|
|
end
|
|
end
|
|
|
|
def copy_file_for_lfs_object(lfs_object)
|
|
copy_files(lfs_object.file.path, destination_path_for_object(lfs_object))
|
|
end
|
|
|
|
def destination_path_for_object(lfs_object)
|
|
File.join(lfs_export_path, lfs_object.oid)
|
|
end
|
|
|
|
def lfs_export_path
|
|
File.join(@shared.export_path, 'lfs-objects')
|
|
end
|
|
end
|
|
end
|
|
end
|