reapir

src/main/java/stirling/software/SPDF/controller/api/other/RepairController.java

@@ -0,0 +1,59 @@
+package stirling.software.SPDF.controller.api.other;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RequestPart;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+
+import stirling.software.SPDF.utils.PdfUtils;
+import stirling.software.SPDF.utils.ProcessExecutor;
+
+@RestController
+public class RepairController {
+
+    private static final Logger logger = LoggerFactory.getLogger(RepairController.class);
+
+    @PostMapping(consumes = "multipart/form-data", value = "/repair")
+    public ResponseEntity<byte[]> repairPdf(@RequestPart(required = true, value = "fileInput") MultipartFile inputFile)
+            throws IOException, InterruptedException {
+
+        // Save the uploaded file to a temporary location
+        Path tempInputFile = Files.createTempFile("input_", ".pdf");
+        inputFile.transferTo(tempInputFile.toFile());
+
+        // Prepare the output file path
+        Path tempOutputFile = Files.createTempFile("output_", ".pdf");
+
+        List<String> command = new ArrayList<>();
+        command.add("gs");
+        command.add("-o");
+        command.add(tempOutputFile.toString());
+        command.add("-sDEVICE=pdfwrite");
+        command.add(tempInputFile.toString());
+
+
+        int returnCode = ProcessExecutor.getInstance(ProcessExecutor.Processes.GHOSTSCRIPT).runCommandWithOutputHandling(command);
+
+        // Read the optimized PDF file
+        byte[] pdfBytes = Files.readAllBytes(tempOutputFile);
+
+        // Clean up the temporary files
+        Files.delete(tempInputFile);
+        Files.delete(tempOutputFile);
+
+        // Return the optimized PDF as a response
+        String outputFilename = inputFile.getOriginalFilename().replaceFirst("[.][^.]+$", "") + "_repaired.pdf";
+        return PdfUtils.bytesToWebResponse(pdfBytes, outputFilename);
+    }
+
+}

src/main/java/stirling/software/SPDF/controller/web/OtherWebController.java

@@ -88,5 +88,11 @@ public class OtherWebController {
         return "other/adjust-contrast";
     }
     
+    @GetMapping("/repair")
+    @Hidden
+    public String repairForm(Model model) {
+        model.addAttribute("currentPage", "repair");
+        return "other/repair";
+    }
     
 }