package stirling.software.SPDF.controller.api;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.pdfbox.cos.COSName;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.apache.pdfbox.pdmodel.PDPageContentStream.AppendMode;
import org.apache.pdfbox.pdmodel.PDResources;
import org.apache.pdfbox.util.Matrix;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
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.RestController;
import org.springframework.web.multipart.MultipartFile;

import org.apache.pdfbox.pdmodel.common.PDRectangle;
import org.apache.pdfbox.pdmodel.graphics.PDXObject;
import org.apache.pdfbox.pdmodel.graphics.form.PDFormXObject;

import io.swagger.v3.oas.annotations.Hidden;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.tags.Tag;
import stirling.software.SPDF.utils.WebResponseUtils;
import org.apache.pdfbox.pdmodel.PDPageContentStream;
import org.apache.pdfbox.multipdf.LayerUtility;
@RestController
@Tag(name = "General", description = "General APIs")
public class ScalePagesController {

	private static final Logger logger = LoggerFactory.getLogger(ScalePagesController.class);

	@PostMapping(value = "/scale-pages", consumes = "multipart/form-data")
	@Operation(summary = "Change the size of a PDF page/document", description = "This operation takes an input PDF file and the size to scale the pages to in the output PDF file. Input:PDF Output:PDF Type:SISO")
	public ResponseEntity<byte[]> scalePages(
			@Parameter(description = "The input PDF file", required = true) @RequestParam("fileInput") MultipartFile file,
			@Parameter(description = "The scale of pages in the output PDF. Acceptable values are A0-A10, B0-B9, LETTER, TABLOID, LEDGER, LEGAL, EXECUTIVE.", required = true, schema = @Schema(type = "string", allowableValues = {
					"A0", "A1", "A2", "A3", "A4", "A5", "A6", "A7", "A8", "A9", "A10", "B0", "B1", "B2", "B3", "B4",
					"B5", "B6", "B7", "B8", "B9", "LETTER", "TABLOID", "LEDGER", "LEGAL",
					"EXECUTIVE" })) @RequestParam("pageSize") String targetPDRectangle,
			@Parameter(description = "The scale of the content on the pages of the output PDF. Acceptable values are floats.", required = true, schema = @Schema(type = "integer")) @RequestParam("scaleFactor") float scaleFactor)
			throws IOException {

		Map<String, PDRectangle> sizeMap = new HashMap<>();
		// Add A0 - A10
		sizeMap.put("A0", PDRectangle.A0);
		sizeMap.put("A1", PDRectangle.A1);
		sizeMap.put("A2", PDRectangle.A2);
		sizeMap.put("A3", PDRectangle.A3);
		sizeMap.put("A4", PDRectangle.A4);
		sizeMap.put("A5", PDRectangle.A5);
		sizeMap.put("A6", PDRectangle.A6);

		// Add other sizes
		sizeMap.put("LETTER", PDRectangle.LETTER);
		sizeMap.put("LEGAL", PDRectangle.LEGAL);

		if (!sizeMap.containsKey(targetPDRectangle)) {
			throw new IllegalArgumentException(
					"Invalid PDRectangle. It must be one of the following: A0, A1, A2, A3, A4, A5, A6, A7, A8, A9, A10");
		}

		PDRectangle targetSize = sizeMap.get(targetPDRectangle);

	    PDDocument sourceDocument = PDDocument.load(file.getBytes());
	    PDDocument outputDocument = new PDDocument();

	    int totalPages = sourceDocument.getNumberOfPages();
	    for (int i = 0; i < totalPages; i++) {
	        PDPage sourcePage = sourceDocument.getPage(i);
	        PDRectangle sourceSize = sourcePage.getMediaBox();
	        
	        float scaleWidth = targetSize.getWidth() / sourceSize.getWidth();
	        float scaleHeight = targetSize.getHeight() / sourceSize.getHeight();
	        float scale = Math.min(scaleWidth, scaleHeight) * scaleFactor;
	        
	        PDPage newPage = new PDPage(targetSize);
	        outputDocument.addPage(newPage);
	        
	        PDPageContentStream contentStream = new PDPageContentStream(outputDocument, newPage, PDPageContentStream.AppendMode.APPEND, true);
	        
	        float x = (targetSize.getWidth() - sourceSize.getWidth() * scale) / 2;
	        float y = (targetSize.getHeight() - sourceSize.getHeight() * scale) / 2;
	        
	        contentStream.saveGraphicsState();
	        contentStream.transform(Matrix.getTranslateInstance(x, y));
	        contentStream.transform(Matrix.getScaleInstance(scale, scale));
	        
	        LayerUtility layerUtility = new LayerUtility(outputDocument);
	        PDFormXObject form = layerUtility.importPageAsForm(sourceDocument, i);
	        contentStream.drawForm(form);

	        contentStream.restoreGraphicsState();
	        contentStream.close();
	    }





	    ByteArrayOutputStream baos = new ByteArrayOutputStream();
	    outputDocument.save(baos);
	    outputDocument.close();
	    sourceDocument.close();
	    
	    
		return WebResponseUtils.bytesToWebResponse(baos.toByteArray(),
				file.getOriginalFilename().replaceFirst("[.][^.]+$", "") + "_scaled.pdf");
	}


}