saulteafarmer/Stirling-PDF
1
0
Fork 0
mirror of https://github.com/Stirling-Tools/Stirling-PDF.git synced 2025-06-23 16:05:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix nullpointerexception

Browse Source
This commit is contained in:
Dario Ghunney Ware 2025-06-23 16:22:41 +01:00
parent 689cc3ebd0
commit 0986a5d5b1
1 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
stirling-pdf/src/main/java/stirling/software/SPDF/service/AttachmentService.java
View File

@ -28,28 +28,17 @@ public class AttachmentService implements AttachmentServiceInterface {
@Override @Override
public PDDocument addAttachment(PDDocument document, List<MultipartFile> attachments) public PDDocument addAttachment(PDDocument document, List<MultipartFile> attachments)
throws IOException { throws IOException {
PDDocumentCatalog catalog = document.getDocumentCatalog(); PDEmbeddedFilesNameTreeNode embeddedFilesTree = getEmbeddedFilesTree(document);
PDDocumentNameDictionary documentNames = catalog.getNames();
PDEmbeddedFilesNameTreeNode embeddedFilesTree = new PDEmbeddedFilesNameTreeNode();
if (documentNames != null) {
embeddedFilesTree = documentNames.getEmbeddedFiles();
} else {
documentNames = new PDDocumentNameDictionary(catalog);
documentNames.setEmbeddedFiles(embeddedFilesTree);
}
catalog.setNames(documentNames);
Map<String, PDComplexFileSpecification> existingNames; Map<String, PDComplexFileSpecification> existingNames;
try { try {
Map<String, PDComplexFileSpecification> originalNames = embeddedFilesTree.getNames(); Map<String, PDComplexFileSpecification> names = embeddedFilesTree.getNames();
if (originalNames == null) { if (names == null) {
log.debug("No existing embedded files found, creating new names map."); log.debug("No existing embedded files found, creating new names map.");
existingNames = new HashMap<>(); existingNames = new HashMap<>();
} else { } else {
existingNames = new HashMap<>(originalNames); existingNames = new HashMap<>(names);
log.debug("Embedded files: {}", existingNames.keySet()); log.debug("Embedded files: {}", existingNames.keySet());
} }
} catch (IOException e) { } catch (IOException e) {
@ -95,4 +84,22 @@ public class AttachmentService implements AttachmentServiceInterface {
return document; return document;
} }
private PDEmbeddedFilesNameTreeNode getEmbeddedFilesTree(PDDocument document) {
PDDocumentCatalog catalog = document.getDocumentCatalog();
PDDocumentNameDictionary documentNames = catalog.getNames();
if (documentNames == null) {
documentNames = new PDDocumentNameDictionary(catalog);
}
catalog.setNames(documentNames);
PDEmbeddedFilesNameTreeNode embeddedFilesTree = documentNames.getEmbeddedFiles();
if (embeddedFilesTree == null) {
embeddedFilesTree = new PDEmbeddedFilesNameTreeNode();
documentNames.setEmbeddedFiles(embeddedFilesTree);
}
return embeddedFilesTree;
}
} }