Stanko K.R.
80 lines
1.8 KiB
Ruby
80 lines
1.8 KiB
Ruby
class Export < ApplicationRecord
|
|
belongs_to :account
|
|
belongs_to :user
|
|
|
|
has_one_attached :file
|
|
|
|
enum :status, %w[ pending processing completed failed ].index_by(&:itself), default: :pending
|
|
|
|
scope :current, -> { where(created_at: 24.hours.ago..) }
|
|
scope :expired, -> { where(completed_at: ...24.hours.ago) }
|
|
|
|
def self.cleanup
|
|
expired.destroy_all
|
|
end
|
|
|
|
def build_later
|
|
ExportDataJob.perform_later(self)
|
|
end
|
|
|
|
def build
|
|
processing!
|
|
|
|
Current.set(account: account) do
|
|
with_url_options do
|
|
zipfile = generate_zip { |zip| populate_zip(zip) }
|
|
|
|
file.attach io: File.open(zipfile.path), filename: "fizzy-export-#{id}.zip", content_type: "application/zip"
|
|
mark_completed
|
|
|
|
ExportMailer.completed(self).deliver_later
|
|
ensure
|
|
zipfile&.close
|
|
zipfile&.unlink
|
|
end
|
|
end
|
|
rescue => e
|
|
update!(status: :failed)
|
|
raise
|
|
end
|
|
|
|
def mark_completed
|
|
update!(status: :completed, completed_at: Time.current)
|
|
end
|
|
|
|
def accessible_to?(accessor)
|
|
accessor == user
|
|
end
|
|
|
|
private
|
|
def with_url_options
|
|
ActiveStorage::Current.set(url_options: { host: "localhost" }) { yield }
|
|
end
|
|
|
|
def populate_zip(zip)
|
|
raise NotImplementedError, "Subclasses must implement populate_zip"
|
|
end
|
|
|
|
def generate_zip
|
|
raise ArgumentError, "Block is required" unless block_given?
|
|
|
|
Tempfile.new([ "export", ".zip" ]).tap do |tempfile|
|
|
Zip::File.open(tempfile.path, create: true) do |zip|
|
|
yield zip
|
|
end
|
|
end
|
|
end
|
|
|
|
def add_file_to_zip(zip, path, content = nil, **options)
|
|
zip.get_output_stream(path, **options) do |f|
|
|
if block_given?
|
|
yield f
|
|
elsif content
|
|
f.write(content)
|
|
else
|
|
raise ArgumentError, "Either content or a block must be provided"
|
|
end
|
|
end
|
|
end
|
|
end
|