107 lines
2.5 KiB
Ruby
107 lines
2.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class FileMover
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
attr_reader :secret, :file_name, :from_model, :to_model, :update_field
|
|
|
|
def initialize(file_path, update_field = :description, from_model:, to_model:)
|
|
@secret = File.split(File.dirname(file_path)).last
|
|
@file_name = File.basename(file_path)
|
|
@from_model = from_model
|
|
@to_model = to_model
|
|
@update_field = update_field
|
|
end
|
|
|
|
def execute
|
|
temp_file_uploader.retrieve_from_store!(file_name)
|
|
|
|
return unless valid?
|
|
|
|
uploader.retrieve_from_store!(file_name)
|
|
|
|
move
|
|
|
|
if update_markdown
|
|
update_upload_model
|
|
uploader.schedule_background_upload
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def valid?
|
|
if temp_file_uploader.file_storage?
|
|
Pathname.new(temp_file_path).realpath.to_path.start_with?(
|
|
(Pathname(temp_file_uploader.root) + temp_file_uploader.base_dir).to_path
|
|
)
|
|
else
|
|
temp_file_uploader.exists?
|
|
end
|
|
end
|
|
|
|
def move
|
|
if temp_file_uploader.file_storage?
|
|
FileUtils.mkdir_p(File.dirname(file_path))
|
|
FileUtils.move(temp_file_path, file_path)
|
|
else
|
|
uploader.copy_file(temp_file_uploader.file)
|
|
temp_file_uploader.upload.destroy!
|
|
end
|
|
end
|
|
|
|
def update_markdown
|
|
updated_text = to_model.read_attribute(update_field)
|
|
.gsub(temp_file_uploader.markdown_link, uploader.markdown_link)
|
|
to_model.update_attribute(update_field, updated_text)
|
|
rescue StandardError
|
|
revert
|
|
false
|
|
end
|
|
|
|
def update_upload_model
|
|
return unless upload = temp_file_uploader.upload
|
|
return if upload.destroyed?
|
|
|
|
upload.update!(model: to_model)
|
|
end
|
|
|
|
def temp_file_path
|
|
strong_memoize(:temp_file_path) do
|
|
temp_file_uploader.file.path
|
|
end
|
|
end
|
|
|
|
def file_path
|
|
strong_memoize(:file_path) do
|
|
uploader.file.path
|
|
end
|
|
end
|
|
|
|
def uploader
|
|
@uploader ||=
|
|
begin
|
|
uploader = PersonalFileUploader.new(to_model, secret: secret)
|
|
|
|
# Enforcing a REMOTE object storage given FileUploader#retrieve_from_store! won't do it
|
|
# (there's no upload at the target yet).
|
|
if uploader.class.object_store_enabled?
|
|
uploader.object_store = ::ObjectStorage::Store::REMOTE
|
|
end
|
|
|
|
uploader
|
|
end
|
|
end
|
|
|
|
def temp_file_uploader
|
|
@temp_file_uploader ||= PersonalFileUploader.new(from_model, secret: secret)
|
|
end
|
|
|
|
def revert
|
|
Gitlab::AppLogger.warn("Markdown not updated, file move reverted for #{to_model}")
|
|
|
|
if temp_file_uploader.file_storage?
|
|
FileUtils.move(file_path, temp_file_path)
|
|
end
|
|
end
|
|
end
|