<!-- iTextPDF 核心依賴 -->
<dependency>
    <groupId>com.itextpdf</groupId>
    <artifactId>itextpdf</artifactId>
    <version>5.5.13.3</version>
</dependency>

<!-- 中文字體支持 -->
<dependency>
    <groupId>com.itextpdf</groupId>
    <artifactId>itext-asian</artifactId>
    <version>5.2.0</version>
</dependency>

<!-- MinIO客戶端依賴（如果用到MinIO下載場景） -->
<dependency>
    <groupId>io.minio</groupId>
    <artifactId>minio</artifactId>
    <version>8.5.7</version>
</dependency>

<!-- Apache Commons IO 工具類 -->
<dependency>
    <groupId>commons-io</groupId>
    <artifactId>commons-io</artifactId>
    <version>2.15.1</version>
</dependency>cy>

// iTextPDF 核心
implementation 'com.itextpdf:itextpdf:5.5.13.3'
// 中文支持
implementation 'com.itextpdf:itext-asian:5.2.0'
// MinIO
implementation 'io.minio:minio:8.5.7'
// Commons IO
implementation 'commons-io:commons-io:2.15.1'

package com.tydt.util;

import com.itextpdf.text.Element;
import com.itextpdf.text.Image;
import com.itextpdf.text.pdf.*;

import java.io.File;
import java.io.FileOutputStream;

/**
 * PDF添加水印工具類
 * 支持文字水印、圖片水印兩種方式
 */
public class PDFUtil {

    /**
     * PDF添加圖片水印
     * @param srcPath 源PDF文件路徑
     * @param destPath 加水印后PDF輸出路徑
     * @param imagePath 水印圖片路徑
     * @throws Exception 處理異常
     */
    public static void addPDFImageWaterMark(String srcPath, String destPath, String imagePath) throws Exception {
        // 1. 讀取源PDF文件
        PdfReader reader = new PdfReader(srcPath);
        // 2. 創(chuàng)建PDF Stamper（用于修改PDF）
        PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(destPath));

        // 3. 加載水印圖片
        Image image = Image.getInstance(imagePath);

        // 4. 設(shè)置水印透明度
        PdfGState gs = new PdfGState();
        gs.setFillOpacity(0.2f); // 透明度0.2（0-1，值越小越淡）
        PdfContentByte content = null;

        // 5. 遍歷PDF所有頁面添加水印
        int totalPages = reader.getNumberOfPages();
        for (int i = 0; i < totalPages; i++) {
            // 獲取當(dāng)前頁面尺寸
            float pageWidth = reader.getPageSize(i + 1).getWidth();
            float pageHeight = reader.getPageSize(i + 1).getHeight();
            
            // 獲取頁面的上層內(nèi)容（水印需放在上層）
            content = stamper.getOverContent(i + 1);
            content.setGState(gs); // 應(yīng)用透明度
            
            // 循環(huán)添加水?。宽?列7行，均勻分布）
            for (int j = 0; j < 3; j++) {
                for (int k = 0; k < 7; k++) {
                    // 設(shè)置圖片水印位置
                    image.setAbsolutePosition(pageWidth / 3 * j - 30, pageHeight / 7 * k - 20);
                    content.addImage(image);
                }
            }
        }

        // 6. 關(guān)閉資源
        stamper.close();
        reader.close();
    }

    /**
     * PDF添加文字水印
     * @param srcPath 源PDF文件路徑
     * @param destPath 加水印后PDF輸出路徑
     * @param waterMarkText 水印文字內(nèi)容
     * @throws Exception 處理異常
     */
    public static void addPDFWaterMark(String srcPath, String destPath, String waterMarkText) throws Exception {
        // 1. 讀取源PDF文件
        PdfReader reader = new PdfReader(srcPath);
        PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(destPath));

        // 2. 適配Windows/Linux系統(tǒng)字體（解決中文亂碼）
        String fontPath = null;
        String osName = System.getProperties().getProperty("os.name");
        if (osName.startsWith("win") || osName.startsWith("Win")) {
            // Windows系統(tǒng)宋體
            fontPath = "C:\\Windows\\Fonts\\SIMSUN.TTC,1";
        } else {
            // Linux系統(tǒng)中文字體（需提前安裝中文字體）
            fontPath = "/usr/share/fonts/chinese/TrueType/uming.ttf";
        }

        // 3. 創(chuàng)建基礎(chǔ)字體（支持中文）
        BaseFont baseFont = BaseFont.createFont(fontPath, BaseFont.IDENTITY_H, BaseFont.NOT_EMBEDDED);

        // 4. 設(shè)置水印透明度
        PdfGState gs = new PdfGState();
        gs.setFillOpacity(0.2f); // 文字透明度
        PdfContentByte content = null;

        // 5. 遍歷所有頁面添加水印
        int totalPages = reader.getNumberOfPages();
        for (int i = 0; i < totalPages; i++) {
            float pageWidth = reader.getPageSize(i + 1).getWidth();
            float pageHeight = reader.getPageSize(i + 1).getHeight();
            
            content = stamper.getOverContent(i + 1);
            content.setGState(gs);
            content.beginText();
            content.setFontAndSize(baseFont, 20); // 設(shè)置字體和字號
            
            // 循環(huán)添加水?。宽?列7行，旋轉(zhuǎn)25度，避免遮擋內(nèi)容）
            for (int j = 0; j < 3; j++) {
                for (int k = 0; k < 7; k++) {
                    // 參數(shù)說明：對齊方式、文字內(nèi)容、X坐標(biāo)、Y坐標(biāo)、旋轉(zhuǎn)角度
                    content.showTextAligned(Element.ALIGN_CENTER, waterMarkText, 
                                           pageWidth / 3 * j + 90, pageHeight / 7 * k, 25);
                }
            }
            content.endText();
        }

        // 6. 關(guān)閉資源
        stamper.close();
        reader.close();
    }

    // 測試方法
    public static void main(String[] args) {
        // 批量給指定目錄的PDF添加文字水印
        File pdfDir = new File("D:\\wps");
        for (File pdfFile : pdfDir.listFiles()) {
            if (pdfFile.getName().endsWith(".pdf")) {
                try {
                    System.out.println("正在處理：" + pdfFile.getName());
                    String srcPath = "D:\\wps\\" + pdfFile.getName();
                    String destPath = "D:\\wps\\waterMark\\" + pdfFile.getName();
                    // 添加文字水印
                    addPDFWaterMark(srcPath, destPath, "天津港項目有限公司");
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
    }
}

import io.minio.StatObjectResponse;
import org.apache.commons.io.IOUtils;
import org.springframework.util.Assert;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.InputStream;
import java.net.URLEncoder;

/**
 * 通用文件下載接口（PDF自動加水印）
 */
@RestController
@RequestMapping("/file")
public class FileDownloadController {

    @Autowired
    private MinioTemplate minioTemplate;
    @Autowired
    private MinioConfig minioConfig;
    @Autowired
    private FileService fileService;

    // 臨時文件夾（需提前創(chuàng)建，建議配置在application.yml）
    private String destFolder = "/tmp/pdf_watermark";
    // 水印文字常量
    private static final String WATER_MARK = "天津港項目有限公司";

    /**
     * 通用附件下載
     * @param fileId 文件ID
     * @param response 響應(yīng)對象
     * @throws Exception 處理異常
     */
    @ApiOperation("通用附件下載")
    @GetMapping("/download")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "fileId", value = "文件ID", required = true)
    })
    public void download(String fileId, HttpServletResponse response) throws Exception {
        // 1. 校驗MinIO桶是否存在
        String bucketName = minioConfig.getBucketName();
        boolean bucketExists = minioTemplate.bucketExists(bucketName);
        Assert.isTrue(bucketExists, "文件存儲桶不存在！");

        // 2. 查詢文件信息
        UploadFilePO uploadFilePo = fileService.queryFileDetail(fileId);
        Assert.notNull(uploadFilePo, "文件不存在或已被刪除！");

        // 3. 獲取MinIO文件元信息
        StatObjectResponse statObject = minioTemplate.getStatObjectResponse(
                bucketName, uploadFilePo.getFilePath(), uploadFilePo.getVersionId());

        // 4. 設(shè)置響應(yīng)頭（文件下載）
        response.setContentType(statObject.contentType());
        response.setHeader("Content-Disposition", 
                "attachment;filename=" + URLEncoder.encode(uploadFilePo.getRealName(), "UTF-8"));

        // 5. 區(qū)分PDF和非PDF文件處理
        if ("pdf".equals(uploadFilePo.getFileType())) {
            // 5.1 從MinIO獲取PDF文件輸入流
            InputStream inputStream = minioTemplate.getObject(
                    bucketName, uploadFilePo.getFilePath(), uploadFilePo.getVersionId());

            // 5.2 創(chuàng)建臨時文件（避免并發(fā)沖突，使用雪花ID命名）
            String tempDir = destFolder + File.separator + IdUtils.snowflakeNewId();
            File tempDirFile = new File(tempDir);
            if (!tempDirFile.exists()) {
                tempDirFile.mkdirs(); // 創(chuàng)建多級目錄
            }
            String tempSrcPdf = tempDir + File.separator + uploadFilePo.getFileName();
            String tempDestPdf = tempDir + File.separator + uploadFilePo.getRealName();

            // 5.3 將MinIO的PDF寫入臨時文件
            org.apache.commons.io.FileUtils.copyToFile(inputStream, new File(tempSrcPdf));

            // 5.4 給PDF添加水印
            PDFUtil.addPDFWaterMark(tempSrcPdf, tempDestPdf, WATER_MARK);

            // 5.5 將加水印后的PDF寫入響應(yīng)流
            FileUtils.writeBytes(tempDestPdf, response.getOutputStream());

            // 5.6 刪除臨時文件（避免磁盤占用）
            org.apache.commons.io.FileUtils.forceDelete(tempDirFile);
        } else {
            // 非PDF文件直接下載
            InputStream inputStream = minioTemplate.getObject(
                    bucketName, uploadFilePo.getFilePath(), uploadFilePo.getVersionId());
            IOUtils.copy(inputStream, response.getOutputStream());
        }
    }
}