mirror of
https://github.com/discourse/discourse.git
synced 2024-11-26 19:03:48 +08:00
8b2580e20f
* FEATURE: admin/user exports are compressed using the zip format * Update translations. Theme exporter now exports .zip file. Theme importer supports .zip and .gz files * Fix controller test, updated locale and skip saving the csv export to disk
80 lines
2.3 KiB
Ruby
80 lines
2.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'zip'
|
|
|
|
module ThemeStore; end
|
|
|
|
class ThemeStore::TgzExporter
|
|
|
|
def initialize(theme)
|
|
@theme = theme
|
|
@temp_folder = "#{Pathname.new(Dir.tmpdir).realpath}/discourse_theme_#{SecureRandom.hex}"
|
|
@export_name = @theme.name.downcase.gsub(/[^0-9a-z.\-]/, '-')
|
|
@export_name = "discourse-#{@export_name}" unless @export_name.starts_with?("discourse")
|
|
end
|
|
|
|
def export_name
|
|
@export_name
|
|
end
|
|
|
|
def package_filename
|
|
export_package
|
|
end
|
|
|
|
def cleanup!
|
|
FileUtils.rm_rf(@temp_folder)
|
|
end
|
|
|
|
def export_to_folder
|
|
FileUtils.mkdir(@temp_folder)
|
|
|
|
Dir.chdir(@temp_folder) do
|
|
FileUtils.mkdir(@export_name)
|
|
|
|
@theme.theme_fields.each do |field|
|
|
next unless path = field.file_path
|
|
|
|
# Belt and braces approach here. All the user input should already be
|
|
# sanitized, but check for attempts to leave the temp directory anyway
|
|
pathname = Pathname.new("#{@export_name}/#{path}")
|
|
folder_path = pathname.parent.cleanpath
|
|
raise RuntimeError.new("Theme exporter tried to leave directory") unless folder_path.to_s.starts_with?("#{@export_name}")
|
|
pathname.parent.mkpath
|
|
path = pathname.realdirpath
|
|
raise RuntimeError.new("Theme exporter tried to leave directory") unless path.to_s.starts_with?("#{@temp_folder}/#{@export_name}")
|
|
|
|
if ThemeField.types[field.type_id] == :theme_upload_var
|
|
filename = Discourse.store.path_for(field.upload)
|
|
content = filename ? File.read(filename) : Discourse.store.download(field.upload).read
|
|
else
|
|
content = field.value
|
|
end
|
|
File.write(path, content)
|
|
end
|
|
|
|
File.write("#{@export_name}/about.json", JSON.pretty_generate(@theme.generate_metadata_hash))
|
|
end
|
|
@temp_folder
|
|
end
|
|
|
|
private
|
|
def export_package
|
|
export_to_folder
|
|
|
|
Dir.chdir(@temp_folder) do
|
|
tar_filename = "#{@export_name}.tar"
|
|
Discourse::Utils.execute_command('tar', '--create', '--file', tar_filename, @export_name, failure_message: "Failed to tar theme.")
|
|
|
|
zip_filename = "#{tar_filename}.zip"
|
|
absolute_path = "#{@temp_folder}/#{tar_filename}"
|
|
Zip::File.open(zip_filename, Zip::File::CREATE) do |zipfile|
|
|
zipfile.add(tar_filename, absolute_path)
|
|
zipfile.close
|
|
end
|
|
|
|
"#{absolute_path}.zip"
|
|
end
|
|
end
|
|
|
|
end
|