gitlab-org--gitlab-foss/app/uploaders/file_mover.rb

66 lines
1.4 KiB
Ruby
Raw Normal View History

2017-05-03 11:26:49 -04:00
class FileMover
attr_reader :secret, :file_name, :model, :update_field
2017-05-03 11:26:49 -04:00
def initialize(file_path, model, update_field = :description)
@secret = File.split(File.dirname(file_path)).last
@file_name = File.basename(file_path)
@model = model
@update_field = update_field
2017-05-03 11:26:49 -04:00
end
def execute
move
if update_markdown
uploader.record_upload
uploader.schedule_background_upload
end
2017-05-03 11:26:49 -04:00
end
private
def move
FileUtils.mkdir_p(File.dirname(file_path))
2017-05-03 11:26:49 -04:00
FileUtils.move(temp_file_path, file_path)
end
def update_markdown
updated_text = model.read_attribute(update_field)
.gsub(temp_file_uploader.markdown_link, uploader.markdown_link)
model.update_attribute(update_field, updated_text)
rescue
revert
false
2017-05-03 11:26:49 -04:00
end
def temp_file_path
return @temp_file_path if @temp_file_path
2017-05-03 11:26:49 -04:00
temp_file_uploader.retrieve_from_store!(file_name)
@temp_file_path = temp_file_uploader.file.path
2017-05-03 11:26:49 -04:00
end
def file_path
return @file_path if @file_path
uploader.retrieve_from_store!(file_name)
@file_path = uploader.file.path
end
def uploader
@uploader ||= PersonalFileUploader.new(model, secret: secret)
2017-05-03 11:26:49 -04:00
end
def temp_file_uploader
@temp_file_uploader ||= PersonalFileUploader.new(nil, secret: secret)
2017-05-03 11:26:49 -04:00
end
def revert
Rails.logger.warn("Markdown not updated, file move reverted for #{model}")
FileUtils.move(file_path, temp_file_path)
end
2017-05-03 11:26:49 -04:00
end