discourse/script/import_scripts/base/uploader.rb

72 lines
2.0 KiB
Ruby

# frozen_string_literal: true
module ImportScripts
class Uploader
# Creates an upload.
# Expects path to be the full path and filename of the source file.
# @return [Upload]
def create_upload(user_id, path, source_filename)
tmp = copy_to_tempfile(path)
UploadCreator.new(tmp, source_filename).create_for(user_id)
rescue => e
STDERR.puts "Failed to create upload: #{e}"
nil
ensure
begin
tmp.close
rescue StandardError
nil
end
begin
tmp.unlink
rescue StandardError
nil
end
end
def create_avatar(user, avatar_path)
tempfile = copy_to_tempfile(avatar_path)
filename = "avatar#{File.extname(avatar_path)}"
upload = UploadCreator.new(tempfile, filename, type: "avatar").create_for(user.id)
if upload.present? && upload.persisted?
user.create_user_avatar
user.user_avatar.update(custom_upload_id: upload.id)
user.update(uploaded_avatar_id: upload.id)
else
STDERR.puts "Failed to upload avatar for user #{user.username}: #{avatar_path}"
STDERR.puts upload.errors.inspect if upload
end
rescue StandardError
STDERR.puts "Failed to create avatar for user #{user.username}: #{avatar_path}"
ensure
tempfile.close! if tempfile
end
def html_for_upload(upload, display_filename)
UploadMarkdown.new(upload).to_markdown(display_name: display_filename)
end
def embedded_image_html(upload)
UploadMarkdown.new(upload).image_markdown
end
def attachment_html(upload, display_filename)
UploadMarkdown.new(upload).attachment_markdown(display_name: display_filename)
end
private
def copy_to_tempfile(source_path)
extension = File.extname(source_path)
tmp = Tempfile.new(["discourse-upload", extension])
File.open(source_path) { |source_stream| IO.copy_stream(source_stream, tmp) }
tmp.rewind
tmp
end
end
end