48 lines
1.7 KiB
Java
48 lines
1.7 KiB
Java
|
package stirling.software.SPDF.controller;
|
||
|
|
|
||
|
|
import java.io.IOException;
|
||
|
|
|
||
|
|
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.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 OverlayImageController {
|
||
|
|
|
||
|
|
private static final Logger logger = LoggerFactory.getLogger(OverlayImageController.class);
|
||
|
|
|
||
|
|
@GetMapping("/add-image")
|
||
|
|
public String overlayImage() {
|
||
|
|
return "add-image";
|
||
|
|
}
|
||
|
|
|
||
|
|
@PostMapping("/add-image")
|
||
|
|
public ResponseEntity<byte[]> overlayImage(@RequestParam("fileInput") MultipartFile pdfFile,
|
||
|
|
@RequestParam("fileInput2") MultipartFile imageFile, @RequestParam("x") float x,
|
||
|
|
@RequestParam("y") float y) {
|
||
|
|
try {
|
||
|
|
byte[] pdfBytes = pdfFile.getBytes();
|
||
|
|
byte[] imageBytes = imageFile.getBytes();
|
||
|
|
byte[] result = PdfUtils.overlayImage(pdfBytes, imageBytes, x, y);
|
||
|
|
HttpHeaders headers = new HttpHeaders();
|
||
|
|
headers.setContentType(MediaType.APPLICATION_PDF);
|
||
|
|
headers.setContentDispositionFormData("attachment", "overlayed.pdf");
|
||
|
|
headers.setCacheControl("must-revalidate, post-check=0, pre-check=0");
|
||
|
|
return new ResponseEntity<>(result, headers, HttpStatus.OK);
|
||
|
|
} catch (IOException e) {
|
||
|
|
logger.error("Failed to add image to PDF", e);
|
||
|
|
return new ResponseEntity<>(HttpStatus.BAD_REQUEST);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|