# 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