# frozen_string_literal: true class RepositoriesExportJob < ApplicationJob include StringUtility def perform(repository_ids, user, team) @user = user @team = team @repositories = Repository.viewable_by_user(@user, @team).where(id: repository_ids).order(:id) zip_input_dir = FileUtils.mkdir_p(Rails.root.join("tmp/temp_zip_#{Time.now.to_i}")).first zip_dir = FileUtils.mkdir_p(Rails.root.join('tmp/zip-ready')).first zip_name = "inventories_export_#{Time.now.utc.strftime('%F_%H-%M-%S_UTC')}.zip" full_zip_name = File.join(zip_dir, zip_name) fill_content(zip_input_dir) ZipExport.transaction do @zip_export = ZipExport.create!(user: @user) @zip_export.zip!(zip_input_dir, full_zip_name) @zip_export.zip_file.attach(io: File.open(full_zip_name), filename: zip_name) generate_notification end ensure FileUtils.rm_rf([zip_input_dir, full_zip_name], secure: true) end private def fill_content(tmp_dir) # Create team dir team_path = "#{tmp_dir}/#{to_filesystem_name(@team.name)}" FileUtils.mkdir_p(team_path) @repositories.each_with_index do |repository, idx| save_repository_to_csv(team_path, repository, idx) end end def save_repository_to_csv(path, repository, idx) repository_name = "#{to_filesystem_name(repository.name)} (#{idx})" # Attachments dir relative_attachments_path = "#{repository_name} attachments" attachments_path = "#{path}/#{relative_attachments_path}" FileUtils.mkdir_p(attachments_path) # CSV file csv_file = FileUtils.touch("#{path}/#{repository_name}.csv").first # Define headers and columns IDs col_ids = [-3, -4, -5, -6, -7, -8] + repository.repository_columns.map(&:id) # Define callback function for file name assets = {} asset_counter = 0 handle_name_func = lambda do |asset| file_name = append_file_suffix(asset.file_name, "_#{asset_counter}").to_s # Save pair for downloading it later assets[asset] = "#{attachments_path}/#{file_name}" asset_counter += 1 relative_path = "#{relative_attachments_path}/#{file_name}" return "=HYPERLINK(\"#{relative_path}\", \"#{relative_path}\")" end # Generate CSV csv_data = RepositoryZipExport.to_csv(repository.repository_rows, col_ids, @user, repository, handle_name_func) File.binwrite(csv_file, csv_data) # Save all attachments (it doesn't work directly in callback function assets.each do |asset, asset_path| asset.file.open do |file| FileUtils.mv(file.path, asset_path) end end end def append_file_suffix(file_name, suffix) file_name = to_filesystem_name(file_name) ext = File.extname(file_name) File.basename(file_name, ext) + suffix + ext end def generate_notification notification = Notification.create!( type_of: :deliver, title: I18n.t('zip_export.notification_title'), message: "" \ "#{@zip_export.zip_file_name}" ) UserNotification.create!(notification: notification, user: @user) end end