# frozen_string_literal: true module Templates module CreateAttachments PDF_CONTENT_TYPE = 'application/pdf' InvalidFileType = Class.new(StandardError) module_function def call(template, params) find_or_create_blobs(params).map do |blob| document_data = blob.download if !blob.image? && blob.content_type != PDF_CONTENT_TYPE blob, document_data = handle_file_types(blob, document_data) end document = template.documents.create!(blob:) if blob.content_type == PDF_CONTENT_TYPE && blob.metadata['pdf'].nil? blob.metadata['pdf'] = { 'annotations' => Templates::BuildAnnotations.call(document_data) } blob.metadata['sha256'] = Base64.urlsafe_encode64(Digest::SHA256.digest(document_data)) end blob.save! Templates::ProcessDocument.call(document, document_data) end end def find_or_create_blobs(params) blobs = params[:blobs]&.map do |attrs| ActiveStorage::Blob.find_signed(attrs[:signed_id]) end blobs || params[:files].map do |file| data = file.read if file.content_type == PDF_CONTENT_TYPE metadata = { 'identified' => true, 'analyzed' => true, 'sha256' => Base64.urlsafe_encode64(Digest::SHA256.digest(data)), 'pdf' => { 'annotations' => Templates::BuildAnnotations.call(data) } } end ActiveStorage::Blob.create_and_upload!( io: StringIO.new(data), filename: file.original_filename, metadata:, content_type: file.content_type ) end end def handle_file_types(_document_data, blob) raise InvalidFileType, blob.content_type end end end