75 lines
2.4 KiB
Java
75 lines
2.4 KiB
Java
|
package stirling.software.SPDF.controller;
|
||
|
|
|
||
|
|
import java.io.ByteArrayOutputStream;
|
||
|
|
import java.io.FileOutputStream;
|
||
|
|
import java.io.IOException;
|
||
|
|
import java.util.Iterator;
|
||
|
|
import java.util.ListIterator;
|
||
|
|
|
||
|
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
||
|
|
import org.apache.pdfbox.pdmodel.PDPage;
|
||
|
|
import org.apache.pdfbox.pdmodel.PDPageTree;
|
||
|
|
import org.slf4j.Logger;
|
||
|
|
import org.slf4j.LoggerFactory;
|
||
|
|
import org.springframework.http.HttpHeaders;
|
||
|
|
import org.springframework.http.HttpStatus;
|
||
|
|
import org.springframework.http.MediaType;
|
||
|
|
import org.springframework.http.ResponseEntity;
|
||
|
|
import org.springframework.stereotype.Controller;
|
||
|
|
import org.springframework.ui.Model;
|
||
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
||
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
||
|
|
import org.springframework.web.bind.annotation.RequestParam;
|
||
|
|
import org.springframework.web.multipart.MultipartFile;
|
||
|
|
|
||
|
|
import stirling.software.SPDF.utils.PdfUtils;
|
||
|
|
|
||
|
|
@Controller
|
||
|
|
public class RotationController {
|
||
|
|
|
||
|
|
private static final Logger logger = LoggerFactory.getLogger(RotationController.class);
|
||
|
|
|
||
|
|
@GetMapping("/rotate-pdf")
|
||
|
|
public String rotatePdfForm(Model model) {
|
||
|
|
model.addAttribute("currentPage", "rotate-pdf");
|
||
|
|
return "rotate-pdf";
|
||
|
|
}
|
||
|
|
|
||
|
|
@PostMapping("/rotate-pdf")
|
||
|
|
public ResponseEntity<byte[]> rotatePDF(@RequestParam("fileInput") MultipartFile pdfFile,
|
||
|
|
@RequestParam("angle") String angle) throws IOException {
|
||
|
|
|
||
|
|
// Load the PDF document
|
||
|
|
PDDocument document = PDDocument.load(pdfFile.getBytes());
|
||
|
|
|
||
|
|
// Get the list of pages in the document
|
||
|
|
PDPageTree pages = document.getPages();
|
||
|
|
|
||
|
|
// Rotate all pages by the specified angle
|
||
|
|
Iterator<PDPage> iterPage = pages.iterator();
|
||
|
|
|
||
|
|
while (iterPage.hasNext()) {
|
||
|
|
PDPage page = iterPage.next();
|
||
|
|
page.setRotation(Integer.valueOf(angle));
|
||
|
|
}
|
||
|
|
|
||
|
|
// Save the rearranged PDF to a ByteArrayOutputStream
|
||
|
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
||
|
|
document.save(outputStream);
|
||
|
|
|
||
|
|
// Close the document
|
||
|
|
document.close();
|
||
|
|
|
||
|
|
// Prepare the response headers
|
||
|
|
HttpHeaders headers = new HttpHeaders();
|
||
|
|
headers.setContentType(MediaType.APPLICATION_PDF);
|
||
|
|
headers.setContentDispositionFormData("attachment", "output.pdf");
|
||
|
|
headers.setContentLength(outputStream.size());
|
||
|
|
|
||
|
|
// Return the response with the PDF data and headers
|
||
|
|
return new ResponseEntity<>(outputStream.toByteArray(), headers, HttpStatus.OK);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|