Stirling-PDF/src/main/java/stirling/software/SPDF/utils/WebResponseUtils.java

package stirling.software.SPDF.utils;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;

import org.apache.pdfbox.pdmodel.PDDocument;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;

public class WebResponseUtils {

	public static ResponseEntity<byte[]> boasToWebResponse(ByteArrayOutputStream baos, String docName) throws IOException {
	    return WebResponseUtils.bytesToWebResponse(baos.toByteArray(), docName);
	}

	public static ResponseEntity<byte[]> boasToWebResponse(ByteArrayOutputStream baos, String docName, MediaType mediaType) throws IOException {
	    return WebResponseUtils.bytesToWebResponse(baos.toByteArray(), docName, mediaType);
	}

	public static ResponseEntity<byte[]> bytesToWebResponse(byte[] bytes, String docName, MediaType mediaType) throws IOException {
	
	    // Return the PDF as a response
	    HttpHeaders headers = new HttpHeaders();
	    headers.setContentType(mediaType);
	    headers.setContentLength(bytes.length);
	    String encodedDocName = URLEncoder.encode(docName, StandardCharsets.UTF_8.toString()).replaceAll("\\+", "%20");
	    headers.setContentDispositionFormData("attachment", encodedDocName);
	    return new ResponseEntity<>(bytes, headers, HttpStatus.OK);
	}

	public static ResponseEntity<byte[]> bytesToWebResponse(byte[] bytes, String docName) throws IOException {
	    return bytesToWebResponse(bytes, docName, MediaType.APPLICATION_PDF);
	}

	public static ResponseEntity<byte[]> pdfDocToWebResponse(PDDocument document, String docName) throws IOException {
	
	    // Open Byte Array and save document to it
	    ByteArrayOutputStream baos = new ByteArrayOutputStream();
	    document.save(baos);
	    // Close the document
	    document.close();
	
	    return boasToWebResponse(baos, docName);
	}

}