Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix/full invert crash 2942 #2957

Merged
merged 4 commits into from
Feb 17, 2025
Merged

Fix/full invert crash 2942 #2957

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
a15ffa9
fix: Resolve Full-Invert crash caused by file permissions
Abdurrahman-shaikh Feb 16, 2025
2b6a2df
Clean up temporary file with deleteOnExit
Abdurrahman-shaikh Feb 16, 2025
b106498
Clean up temporary files immediately with try-finally
Abdurrahman-shaikh Feb 16, 2025
5e65ad7
Clean up temporary files immediately with try-finally
Abdurrahman-shaikh Feb 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
109 changes: 63 additions & 46 deletions src/main/java/stirling/software/SPDF/utils/misc/InvertFullColorStrategy.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,52 +30,69 @@ public InvertFullColorStrategy(MultipartFile file, ReplaceAndInvert replaceAndIn
@Override
public InputStreamResource replace() throws IOException {

// Create a temporary file, with the original filename from the multipart file
File file = Files.createTempFile("temp", getFileInput().getOriginalFilename()).toFile();

// Transfer the content of the multipart file to the file
getFileInput().transferTo(file);

// Load the uploaded PDF
PDDocument document = Loader.loadPDF(file);

// Render each page and invert colors
PDFRenderer pdfRenderer = new PDFRenderer(document);
for (int page = 0; page < document.getNumberOfPages(); page++) {
BufferedImage image =
pdfRenderer.renderImageWithDPI(page, 300); // Render page at 300 DPI

// Invert the colors
invertImageColors(image);

// Create a new PDPage from the inverted image
PDPage pdPage = document.getPage(page);
PDImageXObject pdImage =
PDImageXObject.createFromFileByContent(
convertToBufferedImageTpFile(image), document);

PDPageContentStream contentStream =
new PDPageContentStream(
document, pdPage, PDPageContentStream.AppendMode.OVERWRITE, true);
contentStream.drawImage(
pdImage,
0,
0,
pdPage.getMediaBox().getWidth(),
pdPage.getMediaBox().getHeight());
contentStream.close();
}

// Save the modified PDF to a ByteArrayOutputStream
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
document.save(byteArrayOutputStream);
document.close();
File file = null;
try {
// Create a temporary file, with the original filename from the multipart file
file = Files.createTempFile("temp", getFileInput().getOriginalFilename()).toFile();

// Transfer the content of the multipart file to the file
getFileInput().transferTo(file);

// Load the uploaded PDF
PDDocument document = Loader.loadPDF(file);

// Render each page and invert colors
PDFRenderer pdfRenderer = new PDFRenderer(document);
for (int page = 0; page < document.getNumberOfPages(); page++) {
BufferedImage image =
pdfRenderer.renderImageWithDPI(page, 300); // Render page at 300 DPI

// Invert the colors
invertImageColors(image);

// Create a new PDPage from the inverted image
PDPage pdPage = document.getPage(page);
File tempImageFile = null;
try {
tempImageFile = convertToBufferedImageTpFile(image);
PDImageXObject pdImage =
PDImageXObject.createFromFileByContent(tempImageFile, document);

PDPageContentStream contentStream =
new PDPageContentStream(
document,
pdPage,
PDPageContentStream.AppendMode.OVERWRITE,
true);
contentStream.drawImage(
pdImage,
0,
0,
pdPage.getMediaBox().getWidth(),
pdPage.getMediaBox().getHeight());
contentStream.close();
} finally {
if (tempImageFile != null && tempImageFile.exists()) {
Files.delete(tempImageFile.toPath());
}
}
}

// Prepare the modified PDF for download
ByteArrayInputStream inputStream =
new ByteArrayInputStream(byteArrayOutputStream.toByteArray());
InputStreamResource resource = new InputStreamResource(inputStream);
return resource;
// Save the modified PDF to a ByteArrayOutputStream
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
document.save(byteArrayOutputStream);
document.close();

// Prepare the modified PDF for download
ByteArrayInputStream inputStream =
new ByteArrayInputStream(byteArrayOutputStream.toByteArray());
InputStreamResource resource = new InputStreamResource(inputStream);
return resource;
} finally {
if (file != null && file.exists()) {
Files.delete(file.toPath());
}
}
}

// Method to invert image colors
Expand All @@ -98,7 +115,7 @@ private void invertImageColors(BufferedImage image) {

// Helper method to convert BufferedImage to InputStream
private File convertToBufferedImageTpFile(BufferedImage image) throws IOException {
File file = new File("image.png");
File file = File.createTempFile("image", ".png");
ImageIO.write(image, "png", file);
return file;
}
Expand Down
Loading