98 lines
4.2 KiB
Java
98 lines
4.2 KiB
Java
package stirling.software.SPDF.controller.converters;
|
|
|
|
import java.io.IOException;
|
|
|
|
import org.apache.pdfbox.rendering.ImageType;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.core.io.ByteArrayResource;
|
|
import org.springframework.core.io.Resource;
|
|
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 ConvertImgPDFController {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(ConvertImgPDFController.class);
|
|
|
|
@GetMapping("/img-to-pdf")
|
|
public String convertToPdfForm(Model model) {
|
|
model.addAttribute("currentPage", "img-to-pdf");
|
|
return "convert/img-to-pdf";
|
|
}
|
|
|
|
@GetMapping("/pdf-to-img")
|
|
public String pdfToimgForm(Model model) {
|
|
model.addAttribute("currentPage", "pdf-to-img");
|
|
return "convert/pdf-to-img";
|
|
}
|
|
|
|
@PostMapping("/img-to-pdf")
|
|
public ResponseEntity<byte[]> convertToPdf(@RequestParam("fileInput") MultipartFile file) throws IOException {
|
|
// Convert the file to PDF and get the resulting bytes
|
|
byte[] bytes = PdfUtils.convertToPdf(file.getInputStream());
|
|
logger.info("File {} successfully converted to pdf", file.getOriginalFilename());
|
|
|
|
return PdfUtils.bytesToWebResponse(bytes, file.getName() + "_coverted.pdf");
|
|
}
|
|
|
|
@PostMapping("/pdf-to-img")
|
|
public ResponseEntity<Resource> convertToImage(@RequestParam("fileInput") MultipartFile file, @RequestParam("imageFormat") String imageFormat,
|
|
@RequestParam("singleOrMultiple") String singleOrMultiple, @RequestParam("colorType") String colorType, @RequestParam("dpi") String dpi) throws IOException {
|
|
|
|
byte[] pdfBytes = file.getBytes();
|
|
ImageType colorTypeResult = ImageType.RGB;
|
|
if ("greyscale".equals(colorType)) {
|
|
colorTypeResult = ImageType.GRAY;
|
|
} else if ("blackwhite".equals(colorType)) {
|
|
colorTypeResult = ImageType.BINARY;
|
|
}
|
|
// returns bytes for image
|
|
boolean singleImage = singleOrMultiple.equals("single");
|
|
byte[] result = null;
|
|
try {
|
|
result = PdfUtils.convertFromPdf(pdfBytes, imageFormat.toLowerCase(), colorTypeResult, singleImage, Integer.valueOf(dpi));
|
|
} catch (IOException e) {
|
|
// TODO Auto-generated catch block
|
|
e.printStackTrace();
|
|
} catch (Exception e) {
|
|
// TODO Auto-generated catch block
|
|
e.printStackTrace();
|
|
}
|
|
if (singleImage) {
|
|
HttpHeaders headers = new HttpHeaders();
|
|
headers.setContentType(MediaType.parseMediaType(getMediaType(imageFormat)));
|
|
headers.setCacheControl("must-revalidate, post-check=0, pre-check=0");
|
|
ResponseEntity<Resource> response = new ResponseEntity<>(new ByteArrayResource(result), headers, HttpStatus.OK);
|
|
return response;
|
|
} else {
|
|
ByteArrayResource resource = new ByteArrayResource(result);
|
|
// return the Resource in the response
|
|
return ResponseEntity.ok().header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=converted_documents.zip").contentType(MediaType.APPLICATION_OCTET_STREAM)
|
|
.contentLength(resource.contentLength()).body(resource);
|
|
}
|
|
}
|
|
|
|
private String getMediaType(String imageFormat) {
|
|
if (imageFormat.equalsIgnoreCase("PNG"))
|
|
return "image/png";
|
|
else if (imageFormat.equalsIgnoreCase("JPEG") || imageFormat.equalsIgnoreCase("JPG"))
|
|
return "image/jpeg";
|
|
else if (imageFormat.equalsIgnoreCase("GIF"))
|
|
return "image/gif";
|
|
else
|
|
return "application/octet-stream";
|
|
}
|
|
|
|
}
|