142 lines
6 KiB
Java
142 lines
6 KiB
Java
package stirling.software.SPDF.controller.api;
|
|
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Paths;
|
|
import java.nio.file.attribute.BasicFileAttributes;
|
|
import java.util.ArrayList;
|
|
import java.util.Arrays;
|
|
import java.util.Comparator;
|
|
import java.util.List;
|
|
|
|
import org.apache.pdfbox.Loader;
|
|
import org.apache.pdfbox.multipdf.PDFMergerUtility;
|
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
|
import org.apache.pdfbox.pdmodel.PDPage;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.web.bind.annotation.ModelAttribute;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import io.swagger.v3.oas.annotations.Operation;
|
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
|
|
|
import stirling.software.SPDF.model.api.general.MergePdfsRequest;
|
|
import stirling.software.SPDF.utils.GeneralUtils;
|
|
import stirling.software.SPDF.utils.WebResponseUtils;
|
|
|
|
@RestController
|
|
@RequestMapping("/api/v1/general")
|
|
@Tag(name = "General", description = "General APIs")
|
|
public class MergeController {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(MergeController.class);
|
|
|
|
private PDDocument mergeDocuments(List<PDDocument> documents) throws IOException {
|
|
PDDocument mergedDoc = new PDDocument();
|
|
for (PDDocument doc : documents) {
|
|
for (PDPage page : doc.getPages()) {
|
|
mergedDoc.addPage(page);
|
|
}
|
|
}
|
|
return mergedDoc;
|
|
}
|
|
|
|
private Comparator<MultipartFile> getSortComparator(String sortType) {
|
|
switch (sortType) {
|
|
case "byFileName":
|
|
return Comparator.comparing(MultipartFile::getOriginalFilename);
|
|
case "byDateModified":
|
|
return (file1, file2) -> {
|
|
try {
|
|
BasicFileAttributes attr1 =
|
|
Files.readAttributes(
|
|
Paths.get(file1.getOriginalFilename()),
|
|
BasicFileAttributes.class);
|
|
BasicFileAttributes attr2 =
|
|
Files.readAttributes(
|
|
Paths.get(file2.getOriginalFilename()),
|
|
BasicFileAttributes.class);
|
|
return attr1.lastModifiedTime().compareTo(attr2.lastModifiedTime());
|
|
} catch (IOException e) {
|
|
return 0; // If there's an error, treat them as equal
|
|
}
|
|
};
|
|
case "byDateCreated":
|
|
return (file1, file2) -> {
|
|
try {
|
|
BasicFileAttributes attr1 =
|
|
Files.readAttributes(
|
|
Paths.get(file1.getOriginalFilename()),
|
|
BasicFileAttributes.class);
|
|
BasicFileAttributes attr2 =
|
|
Files.readAttributes(
|
|
Paths.get(file2.getOriginalFilename()),
|
|
BasicFileAttributes.class);
|
|
return attr1.creationTime().compareTo(attr2.creationTime());
|
|
} catch (IOException e) {
|
|
return 0; // If there's an error, treat them as equal
|
|
}
|
|
};
|
|
case "byPDFTitle":
|
|
return (file1, file2) -> {
|
|
try (PDDocument doc1 = Loader.loadPDF(file1.getBytes());
|
|
PDDocument doc2 = Loader.loadPDF(file2.getBytes())) {
|
|
String title1 = doc1.getDocumentInformation().getTitle();
|
|
String title2 = doc2.getDocumentInformation().getTitle();
|
|
return title1.compareTo(title2);
|
|
} catch (IOException e) {
|
|
return 0;
|
|
}
|
|
};
|
|
case "orderProvided":
|
|
default:
|
|
return (file1, file2) -> 0; // Default is the order provided
|
|
}
|
|
}
|
|
|
|
@PostMapping(consumes = "multipart/form-data", value = "/merge-pdfs")
|
|
@Operation(
|
|
summary = "Merge multiple PDF files into one",
|
|
description =
|
|
"This endpoint merges multiple PDF files into a single PDF file. The merged file will contain all pages from the input files in the order they were provided. Input:PDF Output:PDF Type:MISO")
|
|
public ResponseEntity<byte[]> mergePdfs(@ModelAttribute MergePdfsRequest form)
|
|
throws IOException {
|
|
List<File> filesToDelete = new ArrayList<File>();
|
|
try {
|
|
MultipartFile[] files = form.getFileInput();
|
|
Arrays.sort(files, getSortComparator(form.getSortType()));
|
|
|
|
PDFMergerUtility mergedDoc = new PDFMergerUtility();
|
|
ByteArrayOutputStream docOutputstream = new ByteArrayOutputStream();
|
|
|
|
for (MultipartFile multipartFile : files) {
|
|
File tempFile = GeneralUtils.convertMultipartFileToFile(multipartFile);
|
|
filesToDelete.add(tempFile);
|
|
mergedDoc.addSource(tempFile);
|
|
}
|
|
|
|
mergedDoc.setDestinationFileName(
|
|
files[0].getOriginalFilename().replaceFirst("[.][^.]+$", "") + "_merged.pdf");
|
|
mergedDoc.setDestinationStream(docOutputstream);
|
|
|
|
mergedDoc.mergeDocuments(null);
|
|
|
|
return WebResponseUtils.bytesToWebResponse(
|
|
docOutputstream.toByteArray(), mergedDoc.getDestinationFileName());
|
|
} catch (Exception ex) {
|
|
logger.error("Error in merge pdf process", ex);
|
|
throw ex;
|
|
} finally {
|
|
for (File file : filesToDelete) {
|
|
file.delete();
|
|
}
|
|
}
|
|
}
|
|
}
|