9884c65b10
# Description This pull request includes several changes aimed at improving the code structure and removing redundant code. The most significant changes involve reordering methods, removing unnecessary annotations, and refactoring constructors to use dependency injection. Autowired now comes via constructor (which also doesn't need autowired annotation as its done by default for configuration) ## Checklist - [ ] I have read the [Contribution Guidelines](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/CONTRIBUTING.md) - [ ] I have performed a self-review of my own code - [ ] I have attached images of the change if it is UI based - [ ] I have commented my code, particularly in hard-to-understand areas - [ ] If my code has heavily changed functionality I have updated relevant docs on [Stirling-PDFs doc repo](https://github.com/Stirling-Tools/Stirling-Tools.github.io/blob/main/docs/) - [ ] My changes generate no new warnings - [ ] I have read the section [Add New Translation Tags](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/HowToAddNewLanguage.md#add-new-translation-tags) (for new translation tags only)
88 lines
3.6 KiB
Java
88 lines
3.6 KiB
Java
package stirling.software.SPDF.controller.api;
|
|
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.IOException;
|
|
|
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
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 io.swagger.v3.oas.annotations.Operation;
|
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
|
|
|
import stirling.software.SPDF.model.api.PDFFile;
|
|
import stirling.software.SPDF.service.CustomPDDocumentFactory;
|
|
import stirling.software.SPDF.service.PdfImageRemovalService;
|
|
import stirling.software.SPDF.utils.WebResponseUtils;
|
|
|
|
/**
|
|
* Controller class for handling PDF image removal requests. Provides an endpoint to remove images
|
|
* from a PDF file to reduce its size.
|
|
*/
|
|
@RestController
|
|
@RequestMapping("/api/v1/general")
|
|
@Tag(name = "General", description = "General APIs")
|
|
public class PdfImageRemovalController {
|
|
|
|
// Service for removing images from PDFs
|
|
private final PdfImageRemovalService pdfImageRemovalService;
|
|
|
|
private final CustomPDDocumentFactory pdfDocumentFactory;
|
|
|
|
/**
|
|
* Constructor for dependency injection of PdfImageRemovalService.
|
|
*
|
|
* @param pdfImageRemovalService The service used for removing images from PDFs.
|
|
*/
|
|
@Autowired
|
|
public PdfImageRemovalController(
|
|
PdfImageRemovalService pdfImageRemovalService,
|
|
CustomPDDocumentFactory pdfDocumentFactory) {
|
|
this.pdfImageRemovalService = pdfImageRemovalService;
|
|
this.pdfDocumentFactory = pdfDocumentFactory;
|
|
}
|
|
|
|
/**
|
|
* Endpoint to remove images from a PDF file.
|
|
*
|
|
* <p>This method processes the uploaded PDF file, removes all images, and returns the modified
|
|
* PDF file with a new name indicating that images were removed.
|
|
*
|
|
* @param file The PDF file with images to be removed.
|
|
* @return ResponseEntity containing the modified PDF file as byte array with appropriate
|
|
* content type and filename.
|
|
* @throws IOException If an error occurs while processing the PDF file.
|
|
*/
|
|
@PostMapping(consumes = "multipart/form-data", value = "/remove-image-pdf")
|
|
@Operation(
|
|
summary = "Remove images from file to reduce the file size.",
|
|
description =
|
|
"This endpoint remove images from file to reduce the file size.Input:PDF Output:PDF Type:MISO")
|
|
public ResponseEntity<byte[]> removeImages(@ModelAttribute PDFFile file) throws IOException {
|
|
// Load the PDF document
|
|
PDDocument document = pdfDocumentFactory.load(file);
|
|
|
|
// Remove images from the PDF document using the service
|
|
PDDocument modifiedDocument = pdfImageRemovalService.removeImagesFromPdf(document);
|
|
|
|
// Create a ByteArrayOutputStream to hold the modified PDF data
|
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
|
|
|
// Save the modified PDF document to the output stream
|
|
modifiedDocument.save(outputStream);
|
|
modifiedDocument.close();
|
|
|
|
// Generate a new filename for the modified PDF
|
|
String mergedFileName =
|
|
file.getFileInput().getOriginalFilename().replaceFirst("[.][^.]+$", "")
|
|
+ "_removed_images.pdf";
|
|
|
|
// Convert the byte array to a web response and return it
|
|
return WebResponseUtils.bytesToWebResponse(outputStream.toByteArray(), mergedFileName);
|
|
}
|
|
}
|