# frozen_string_literal: true
module Reports
class PdfJob < ApplicationJob
extend InputSanitizeHelper
include InputSanitizeHelper
include ReportsHelper
queue_as :reports
discard_on StandardError do |job, error|
report = Report.find_by(id: job.arguments.first)
return if report.blank?
ActiveRecord::Base.no_touching do
report.pdf_error!
end
report_path =
if report.pdf_file.attached?
Rails.application.routes.url_helpers
.reports_path(team: report.team.id, preview_report_id: report.id, preview_type: :pdf)
else
Rails.application.routes.url_helpers.reports_path(team: report.team.id)
end
user = job.arguments.second
notification = Notification.create(
type_of: :deliver_error,
title: I18n.t('projects.reports.index.generation.error_pdf_notification_title'),
message: I18n.t('projects.reports.index.generation.error_notification_message',
report_link: "#{sanitize_input(report.name)}",
team_name: sanitize_input(report.team.name))
)
notification.create_user_notification(user)
Rails.logger.error("Couldn't generate PDF for Report with id: #{report.id}. Error:\n #{error}")
end
PREVIEW_EXTENSIONS = %w(docx pdf).freeze
def perform(report_id, user)
report = Report.find(report_id)
file = Tempfile.new(['report', '.pdf'], binmode: true)
begin
template =
if Extends::REPORT_TEMPLATES.key?(report.settings[:template]&.to_sym)
report.settings[:template]
else
Extends::REPORT_TEMPLATES.keys.first.to_s
end
raise StandardError, 'Report template not found!' if template.blank?
I18n.backend.date_format = user.settings[:date_format] || Constants::DEFAULT_DATE_FORMAT
ActionController::Renderer::RACK_KEY_TRANSLATION['warden'] ||= 'warden'
proxy = Warden::Proxy.new({}, Warden::Manager.new({}))
proxy.set_user(user, scope: :user, store: false)
ApplicationController.renderer.defaults[:http_host] = Rails.application.routes.default_url_options[:host]
renderer = ApplicationController.renderer.new(warden: proxy)
file << renderer.render(
pdf: 'report', header: { html: { template: "reports/templates/#{template}/header",
locals: { report: report, user: user },
layout: 'reports/footer_header.html.erb' } },
footer: { html: { template: "reports/templates/#{template}/footer",
locals: { report: report, user: user },
layout: 'reports/footer_header.html.erb' } },
assigns: { settings: report.settings },
locals: { report: report },
disable_javascript: false,
template: 'reports/report.pdf.erb'
)
file.rewind
file = prepend_title_page(file, template, report, renderer)
file = append_result_asset_previews(report, file) if report.settings.dig(:task, :file_results_previews)
report.pdf_file.attach(io: file, filename: 'report.pdf')
report.pdf_ready!
report_path = Rails.application.routes.url_helpers
.reports_path(team: report.team.id, preview_report_id: report.id, preview_type: :pdf)
notification = Notification.create(
type_of: :deliver,
title: I18n.t('projects.reports.index.generation.completed_pdf_notification_title'),
message: I18n.t('projects.reports.index.generation.completed_notification_message',
report_link: "#{sanitize_input(report.name)}",
team_name: sanitize_input(report.team.name))
)
notification.create_user_notification(user)
ensure
I18n.backend.date_format = Constants::DEFAULT_DATE_FORMAT
file.close(true)
end
end
private
def append_result_asset_previews(report, report_file)
Dir.mktmpdir do |tmp_dir|
report.report_elements.my_module.each do |my_module_element|
results = my_module_element.my_module.results
order_results_for_report(results, report.settings.dig(:task, :result_order)).each do |result|
next unless result.is_asset && PREVIEW_EXTENSIONS.include?(result.asset.file.blob.filename.extension)
asset = result.asset
if !asset.file_pdf_preview.attached? || (asset.file.created_at > asset.file_pdf_preview.created_at)
PdfPreviewJob.perform_now(asset.id)
asset.reload
end
asset.file_pdf_preview.open(tmpdir: tmp_dir) do |file|
report_file = merge_pdf_files(file, report_file)
end
end
end
end
report_file
end
def merge_pdf_files(file, report_file)
merged_file = Tempfile.new(['report', '.pdf'], binmode: true)
success = system(
'pdfunite', report_file.path, file.path, merged_file.path
)
unless success && File.file?(merged_file)
raise StandardError, 'There was an error merging report and PDF file preview'
end
file.close(true)
report_file.close(true)
merged_file
end
def prepend_title_page(file, template, report, renderer)
unless File.exist?(Rails.root.join('app', 'views', 'reports', 'templates', template, 'cover.html.erb'))
return file
end
total_pages = 0
IO.popen(['pdfinfo', file.path], 'r+') do |f|
total_pages = f.read.split("\n")
.find { |i| i.split(':')[0] == 'Pages' }
.gsub(/[^0-9]/, '')
end
title_page = Tempfile.new(['title_page', '.pdf'], binmode: true)
merged_file = Tempfile.new(['report', '.pdf'], binmode: true)
title_page << renderer.render(
pdf: 'report', inline: renderer.render_to_string("reports/templates/#{template}/cover.html.erb",
layout: false,
locals: { report: report, total_pages: total_pages.to_i }),
disable_javascript: false,
template: 'reports/report.pdf.erb'
)
title_page.rewind
success = system(
'pdfunite', title_page.path, file.path, merged_file.path
)
raise StandardError, 'There was an error merging report and title page' unless success && File.file?(merged_file)
file.close(true)
title_page.close(true)
merged_file
end
end
end