changed response type, updated tests

common/src/main/java/stirling/software/common/util/EmlToPdf.java

@@ -1104,15 +1104,13 @@ public class EmlToPdf {
                             new PDEmbeddedFile(document, new ByteArrayInputStream(attachment.getData()));
                     embeddedFile.setSize(attachment.getData().length);
                     embeddedFile.setCreationDate(new GregorianCalendar());
-                    if (attachment.getContentType() != null) {
-                        embeddedFile.setSubtype(attachment.getContentType());
-                    }
 
                     // Create file specification
                     PDComplexFileSpecification fileSpec = new PDComplexFileSpecification();
                     fileSpec.setFile(uniqueFilename);
                     fileSpec.setEmbeddedFile(embeddedFile);
                     if (attachment.getContentType() != null) {
+                        embeddedFile.setSubtype(attachment.getContentType());
                         fileSpec.setFileDescription("Email attachment: " + uniqueFilename);
                     }
 

stirling-pdf/src/main/java/stirling/software/SPDF/controller/api/misc/AttachmentsController.java

@@ -60,10 +60,11 @@ public class AttachmentsController {
 
         // Add attachments
         catalog.setNames(documentNames);
+        byte[] output =
                 pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
 
-        return WebResponseUtils.pdfDocToWebResponse(
+        return WebResponseUtils.bytesToWebResponse(
-                document,
+                output,
                 Filenames.toSimpleFileName(pdfFile.getOriginalFilename())
                                 .replaceFirst("[.][^.]+$", "")
                         + "_with_attachments.pdf");

stirling-pdf/src/main/java/stirling/software/SPDF/service/PDFAttachmentService.java

@@ -1,5 +1,6 @@
 package stirling.software.SPDF.service;
 
+import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.util.GregorianCalendar;
 import java.util.HashMap;
@@ -25,7 +26,7 @@ import stirling.software.common.util.PDFAttachmentUtils;
 public class PDFAttachmentService implements PDFAttachmentServiceInterface {
 
     @Override
-    public void addAttachment(
+    public byte[] addAttachment(
             PDDocument document,
             PDEmbeddedFilesNameTreeNode embeddedFilesTree,
             List<MultipartFile> attachments)
@@ -48,52 +49,47 @@ public class PDFAttachmentService implements PDFAttachmentServiceInterface {
         final Map<String, PDComplexFileSpecification> existingEmbeddedFiles = existingNames;
         attachments.forEach(
                 attachment -> {
-                    // Create attachments specification
+                    String filename = attachment.getOriginalFilename();
-                    PDComplexFileSpecification fileSpecification = new PDComplexFileSpecification();
-                    fileSpecification.setFile(attachment.getOriginalFilename());
-                    fileSpecification.setFileUnicode(attachment.getOriginalFilename());
-                    fileSpecification.setFileDescription(
-                            "Embedded attachment: " + attachment.getOriginalFilename());
 
                     try {
-                        // Create embedded attachment
                         PDEmbeddedFile embeddedFile =
                                 new PDEmbeddedFile(document, attachment.getInputStream());
                         embeddedFile.setSize((int) attachment.getSize());
                         embeddedFile.setCreationDate(new GregorianCalendar());
                         embeddedFile.setModDate(new GregorianCalendar());
-
-                        // Set MIME type if available
                         String contentType = attachment.getContentType();
+
                         if (StringUtils.isNotBlank(contentType)) {
                             embeddedFile.setSubtype(contentType);
                         }
 
-                        // Associate embedded attachment with file specification
+                        // Create attachments specification and associate embedded attachment with
+                        // file
+                        PDComplexFileSpecification fileSpecification =
+                                new PDComplexFileSpecification();
+                        fileSpecification.setFile(filename);
+                        fileSpecification.setFileUnicode(filename);
+                        fileSpecification.setFileDescription("Embedded attachment: " + filename);
                         embeddedFile.setFile(fileSpecification);
                         fileSpecification.setEmbeddedFile(embeddedFile);
                         fileSpecification.setEmbeddedFileUnicode(embeddedFile);
 
                         // Add to the existing files map
-                        existingEmbeddedFiles.put(
+                        existingEmbeddedFiles.put(filename, fileSpecification);
-                                attachment.getOriginalFilename(), fileSpecification);
 
-                        log.info(
+                        log.info("Added attachment: {} ({} bytes)", filename, attachment.getSize());
-                                "Added attachment: {} ({} bytes)",
-                                attachment.getOriginalFilename(),
-                                attachment.getSize());
                     } catch (IOException e) {
-                        log.warn(
+                        log.warn("Failed to create embedded file for attachment: {}", filename, e);
-                                "Failed to create embedded file for attachment: {}",
-                                attachment.getOriginalFilename(),
-                                e);
                     }
                 });
 
         embeddedFilesTree.setNames(existingNames);
-
         grantAccessPermissions(document);
         PDFAttachmentUtils.setCatalogViewerPreferences(document, PageMode.USE_ATTACHMENTS);
+        ByteArrayOutputStream output = new ByteArrayOutputStream();
+        document.save(output);
+
+        return output.toByteArray();
     }
 
     private void grantAccessPermissions(PDDocument document) {

stirling-pdf/src/main/java/stirling/software/SPDF/service/PDFAttachmentServiceInterface.java

@@ -9,7 +9,7 @@ import org.springframework.web.multipart.MultipartFile;
 
 public interface PDFAttachmentServiceInterface {
 
-    void addAttachment(
+    byte[] addAttachment(
             PDDocument document,
             PDEmbeddedFilesNameTreeNode efTree,
             List<MultipartFile> attachments)

stirling-pdf/src/test/java/stirling/software/SPDF/controller/api/misc/AttachmentsControllerTest.java

@@ -61,16 +61,19 @@ class AttachmentsControllerTest {
     @Test
     void addAttachments_WithExistingNames() throws IOException {
         List<MultipartFile> attachments = List.of(attachment1, attachment2);
+        byte[] expectedOutput = "modified PDF content".getBytes();
 
         when(pdfDocumentFactory.load(pdfFile, false)).thenReturn(mockDocument);
         when(mockDocument.getDocumentCatalog()).thenReturn(mockCatalog);
         when(mockCatalog.getNames()).thenReturn(mockNameDict);
         when(mockNameDict.getEmbeddedFiles()).thenReturn(mockEmbeddedFilesTree);
+        when(pdfAttachmentService.addAttachment(mockDocument, mockEmbeddedFilesTree, attachments)).thenReturn(expectedOutput);
 
         ResponseEntity<byte[]> response = attachmentsController.addAttachments(pdfFile, attachments);
 
         assertNotNull(response);
         assertEquals(HttpStatus.OK, response.getStatusCode());
+        assertNotNull(response.getBody());
         verify(pdfDocumentFactory).load(pdfFile, false);
         verify(mockCatalog).setNames(mockNameDict);
         verify(pdfAttachmentService).addAttachment(mockDocument, mockEmbeddedFilesTree, attachments);
@@ -79,14 +82,17 @@ class AttachmentsControllerTest {
     @Test
     void addAttachments_WithoutExistingNames() throws IOException {
         List<MultipartFile> attachments = List.of(attachment1);
+        byte[] expectedOutput = "modified PDF content".getBytes();
 
         try (PDDocument realDocument = new PDDocument()) {
             when(pdfDocumentFactory.load(pdfFile, false)).thenReturn(realDocument);
+            when(pdfAttachmentService.addAttachment(eq(realDocument), any(PDEmbeddedFilesNameTreeNode.class), eq(attachments))).thenReturn(expectedOutput);
 
             ResponseEntity<byte[]> response = attachmentsController.addAttachments(pdfFile, attachments);
 
             assertNotNull(response);
             assertEquals(HttpStatus.OK, response.getStatusCode());
+            assertNotNull(response.getBody());
             verify(pdfDocumentFactory).load(pdfFile, false);
             verify(pdfAttachmentService).addAttachment(eq(realDocument), any(PDEmbeddedFilesNameTreeNode.class), eq(attachments));
         }
@@ -113,7 +119,7 @@ class AttachmentsControllerTest {
         when(mockDocument.getDocumentCatalog()).thenReturn(mockCatalog);
         when(mockCatalog.getNames()).thenReturn(mockNameDict);
         when(mockNameDict.getEmbeddedFiles()).thenReturn(mockEmbeddedFilesTree);
-        doThrow(ioException).when(pdfAttachmentService).addAttachment(mockDocument, mockEmbeddedFilesTree, attachments);
+        when(pdfAttachmentService.addAttachment(mockDocument, mockEmbeddedFilesTree, attachments)).thenThrow(ioException);
 
         assertThrows(IOException.class, () -> attachmentsController.addAttachments(pdfFile, attachments));
         verify(pdfAttachmentService).addAttachment(mockDocument, mockEmbeddedFilesTree, attachments);
@@ -162,16 +168,19 @@ class AttachmentsControllerTest {
     @Test
     void addAttachments_EmptyAttachmentsList() throws IOException {
         List<MultipartFile> emptyAttachments = List.of();
+        byte[] expectedOutput = "PDF content without new attachments".getBytes();
 
         when(pdfDocumentFactory.load(pdfFile, false)).thenReturn(mockDocument);
         when(mockDocument.getDocumentCatalog()).thenReturn(mockCatalog);
         when(mockCatalog.getNames()).thenReturn(mockNameDict);
         when(mockNameDict.getEmbeddedFiles()).thenReturn(mockEmbeddedFilesTree);
+        when(pdfAttachmentService.addAttachment(mockDocument, mockEmbeddedFilesTree, emptyAttachments)).thenReturn(expectedOutput);
 
         ResponseEntity<byte[]> response = attachmentsController.addAttachments(pdfFile, emptyAttachments);
 
         assertNotNull(response);
         assertEquals(HttpStatus.OK, response.getStatusCode());
+        assertNotNull(response.getBody());
         verify(pdfAttachmentService).addAttachment(mockDocument, mockEmbeddedFilesTree, emptyAttachments);
     }
 
@@ -179,16 +188,19 @@ class AttachmentsControllerTest {
     void addAttachments_NullFilename() throws IOException {
         MockMultipartFile attachmentWithNullName = new MockMultipartFile("attachment", null, "text/plain", "content".getBytes());
         List<MultipartFile> attachments = List.of(attachmentWithNullName);
+        byte[] expectedOutput = "PDF with null filename attachment".getBytes();
 
         when(pdfDocumentFactory.load(pdfFile, false)).thenReturn(mockDocument);
         when(mockDocument.getDocumentCatalog()).thenReturn(mockCatalog);
         when(mockCatalog.getNames()).thenReturn(mockNameDict);
         when(mockNameDict.getEmbeddedFiles()).thenReturn(mockEmbeddedFilesTree);
+        when(pdfAttachmentService.addAttachment(mockDocument, mockEmbeddedFilesTree, attachments)).thenReturn(expectedOutput);
 
         ResponseEntity<byte[]> response = attachmentsController.addAttachments(pdfFile, attachments);
 
         assertNotNull(response);
         assertEquals(HttpStatus.OK, response.getStatusCode());
+        assertNotNull(response.getBody());
         verify(pdfAttachmentService).addAttachment(mockDocument, mockEmbeddedFilesTree, attachments);
     }
 

stirling-pdf/src/test/java/stirling/software/SPDF/service/PDFAttachmentServiceTest.java

@@ -10,7 +10,9 @@ import org.apache.pdfbox.pdmodel.common.filespecification.PDComplexFileSpecifica
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.springframework.web.multipart.MultipartFile;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.mockito.ArgumentMatchers.anyMap;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
@@ -39,8 +41,10 @@ class PDFAttachmentServiceTest {
             when(attachments.get(0).getSize()).thenReturn(12L);
             when(attachments.get(0).getContentType()).thenReturn("text/plain");
 
-            pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
+            byte[] result = pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
 
+            assertNotNull(result);
+            assertTrue(result.length > 0);
             verify(embeddedFilesTree).setNames(anyMap());
         }
     }
@@ -58,8 +62,10 @@ class PDFAttachmentServiceTest {
             when(attachments.get(0).getSize()).thenReturn(15L);
             when(attachments.get(0).getContentType()).thenReturn("application/pdf");
 
-            pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
+            byte[] result = pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
 
+            assertNotNull(result);
+            assertTrue(result.length > 0);
             verify(embeddedFilesTree).setNames(anyMap());
         }
     }
@@ -78,8 +84,10 @@ class PDFAttachmentServiceTest {
             when(attachments.get(0).getSize()).thenReturn(25L);
             when(attachments.get(0).getContentType()).thenReturn("");
 
-            pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
+            byte[] result = pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
 
+            assertNotNull(result);
+            assertTrue(result.length > 0);
             verify(embeddedFilesTree).setNames(anyMap());
         }
     }
@@ -111,8 +119,10 @@ class PDFAttachmentServiceTest {
             when(attachments.get(0).getInputStream()).thenThrow(ioException);
             when(attachments.get(0).getSize()).thenReturn(10L);
 
-            pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
+            byte[] result = pdfAttachmentService.addAttachment(document, embeddedFilesTree, attachments);
 
+            assertNotNull(result);
+            assertTrue(result.length > 0);
             verify(embeddedFilesTree).setNames(anyMap());
         }
     }