mirror of
https://github.com/Stirling-Tools/Stirling-PDF.git
synced 2025-05-17 03:35:54 +00:00
5275866f09
- Support for adding and removing passwords - Support for watermarks - Dedicated page remover - Support for PDF permissions - Removed endpoint /home and replaced with / - Code cleanups - Fixed page titles
84 lines
3.3 KiB
Java
84 lines
3.3 KiB
Java
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");
|
|
}
|
|
}
|