86 lines
2.4 KiB
Ruby
86 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "fileutils"
|
|
require "securerandom"
|
|
require "zip/filesystem"
|
|
|
|
# require all data export use cases via Zeitwerk
|
|
# rubocop:disable Lint/Void
|
|
UseCase::DataExport::Answers
|
|
UseCase::DataExport::Reactions
|
|
UseCase::DataExport::Comments
|
|
UseCase::DataExport::MuteRules
|
|
UseCase::DataExport::Questions
|
|
UseCase::DataExport::Relationships
|
|
UseCase::DataExport::Theme
|
|
UseCase::DataExport::User
|
|
# rubocop:enable all
|
|
|
|
# the justask data exporter, now with 200% less shelling out to system tools!
|
|
#
|
|
# the data export can be easily extended by subclassing `UseCase::DataExport::Base`
|
|
# and letting Zeitwerk know about the subclasses above
|
|
class Exporter
|
|
def initialize(user)
|
|
@user = user
|
|
|
|
@export_name = "export-#{@user.id}-#{SecureRandom.base36(32)}"
|
|
FileUtils.mkdir_p(Rails.public_path.join("export")) # ensure the public export path exists
|
|
export_zipfile_path = Rails.public_path.join("export", "#{@export_name}.zip")
|
|
@zipfile = Zip::File.open(export_zipfile_path, Zip::File::CREATE)
|
|
end
|
|
|
|
def export
|
|
@user.export_processing = true
|
|
@user.save validate: false
|
|
|
|
prepare_zipfile
|
|
write_files
|
|
publish
|
|
rescue => e
|
|
Sentry.capture_exception(e)
|
|
@user.export_processing = false
|
|
@user.save validate: false
|
|
|
|
# delete zipfile on errors
|
|
@zipfile.close # so it gets written to disk first
|
|
File.delete(@zipfile.name)
|
|
|
|
raise # let e.g. the sidekiq job fail so it's retryable later
|
|
ensure
|
|
@zipfile.close
|
|
end
|
|
|
|
private
|
|
|
|
# creates some directories we want to exist and sets a nice comment
|
|
def prepare_zipfile
|
|
@zipfile.mkdir(@export_name)
|
|
@zipfile.mkdir("#{@export_name}/pictures")
|
|
|
|
@zipfile.comment = <<~COMMENT
|
|
#{APP_CONFIG.fetch(:site_name)} export done for #{@user.screen_name} on #{Time.now.utc.iso8601}
|
|
COMMENT
|
|
end
|
|
|
|
# writes the files to the zip file
|
|
def write_files
|
|
UseCase::DataExport::Base.descendants.each do |export_klass|
|
|
export_klass.call(user: @user).each do |file_name, contents|
|
|
@zipfile.file.open("#{@export_name}/#{file_name}", "wb".dup) do |file| # .dup because of %(can't modify frozen String: "wb")
|
|
file.write contents
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def publish
|
|
url = "#{APP_CONFIG['https'] ? 'https' : 'http'}://#{APP_CONFIG['hostname']}/export/#{@export_name}.zip"
|
|
@user.export_processing = false
|
|
@user.export_url = url
|
|
@user.export_created_at = Time.now.utc
|
|
@user.save validate: false
|
|
url
|
|
end
|
|
end
|