package stirling.software.SPDF.controller.security;

import java.awt.Color;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.apache.pdfbox.pdmodel.PDPageContentStream;
import org.apache.pdfbox.pdmodel.font.PDFont;
import org.apache.pdfbox.pdmodel.font.PDType1Font;
import org.apache.pdfbox.pdmodel.graphics.state.PDExtendedGraphicsState;
import org.apache.pdfbox.util.Matrix;
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 WatermarkController {

	@GetMapping("/add-watermark")
	public String addWatermarkForm(Model model) {
		model.addAttribute("currentPage", "add-watermark");
		return "security/add-watermark";
	}

	@PostMapping("/add-watermark")
	public ResponseEntity<byte[]> addWatermark(@RequestParam("pdfFile") MultipartFile pdfFile,
			@RequestParam("watermarkText") String watermarkText,
			@RequestParam(defaultValue = "30", name = "fontSize") float fontSize,
			@RequestParam(defaultValue = "0", name = "rotation") float rotation,
			@RequestParam(defaultValue = "50", name = "widthSpacer") int widthSpacer,
			@RequestParam(defaultValue = "50", name = "heightSpacer") int heightSpacer) throws IOException {

		// Load the input PDF
		PDDocument document = PDDocument.load(pdfFile.getInputStream());

		// Create a page in the document
		for (PDPage page : document.getPages()) {
			// Get the page's content stream
			PDPageContentStream contentStream = new PDPageContentStream(document, page,
					PDPageContentStream.AppendMode.APPEND, true);

			// Set font of watermark
			PDFont font = PDType1Font.HELVETICA_BOLD;
			contentStream.beginText();
			contentStream.setFont(font, fontSize);
			contentStream.setNonStrokingColor(Color.LIGHT_GRAY);

			// Set size and location of watermark
			float pageWidth = page.getMediaBox().getWidth();
			float pageHeight = page.getMediaBox().getHeight();
			float watermarkWidth = widthSpacer + font.getStringWidth(watermarkText) * fontSize / 1000;
			float watermarkHeight = heightSpacer + fontSize;
			int watermarkRows = (int) (pageHeight / watermarkHeight + 1);
			int watermarkCols = (int) (pageWidth / watermarkWidth + 1);

			// Add the watermark text
			for (int i = 0; i < watermarkRows; i++) {
				for (int j = 0; j < watermarkCols; j++) {
					contentStream.setTextMatrix(Matrix.getRotateInstance((float) Math.toRadians(rotation),
							j * watermarkWidth, i * watermarkHeight));
					contentStream.showTextWithPositioning(new Object[] { watermarkText });
				}
			}

			contentStream.endText();

			// Close the content stream
			contentStream.close();
		}
		return PdfUtils.pdfDocToWebResponse(document, pdfFile.getName() + "_watermarked.pdf");
	}
}