feat: 文件预览接口加水印

laws-module-system/pom.xml

@@ -72,6 +72,11 @@
 			<artifactId>spire.doc.free</artifactId>
 			<version>5.2.0</version>
 		</dependency>
+		<dependency>
+			<groupId>org.apache.pdfbox</groupId>
+			<artifactId>pdfbox</artifactId>
+			<version>2.0.6</version>
+		</dependency>
 	</dependencies>
 
 

laws-module-system/src/main/java/com/jero/modules/system/controller/CommonController.java

@@ -1,5 +1,6 @@
 package com.jero.modules.system.controller;
 
+import cn.hutool.core.io.FileUtil;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
@@ -10,6 +11,7 @@ import com.jero.common.system.api.ISysBaseAPI;
 import com.jero.common.util.*;
 import com.jero.modules.oss.entity.OSSFile;
 import com.jero.modules.oss.service.IOSSFileService;
+import com.jero.modules.system.util.SysWaterMarkUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -50,6 +52,9 @@ public class CommonController {
     @Resource
     private  IOSSFileService ossFileService;
 
+    @Resource
+    private SysWaterMarkUtil sysWaterMarkUtil;
+
     @Value(value = "${jero.path.upload}")
     private String uploadpath;
 
@@ -157,7 +162,7 @@ public class CommonController {
      */
     @GetMapping(value = "/view/{id}")
     public void view(@PathVariable String id,HttpServletRequest request, HttpServletResponse response) {
-        downloadAndView(id, response);
+        downloadAndViewWithWaterMark(id, response);
     }
 
     private void downloadAndView(@PathVariable String id, HttpServletResponse response) {
@@ -224,6 +229,69 @@ public class CommonController {
         }
     }
 
+    private void downloadAndViewWithWaterMark(@PathVariable String id, HttpServletResponse response) {
+        // 查询数据表数据是否存在
+        LambdaQueryWrapper<OSSFile> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.eq(OSSFile::getId, id);
+        OSSFile ossFile = ossFileService.getOne(queryWrapper);
+        if (null == ossFile) {
+            throw new JeroBootException("文件不存在..");
+        }
+        String fileUrl = ossFile.getUrl();
+        // minio 下载
+        // 通过MinioUtil查询时 只需要桶后面的路径
+        String minioUrl = MinioUtil.getMinioUrl();
+        // Linux/unix 系统下文件路径分隔符为"/" 获取minio与存储桶的路径
+        minioUrl = minioUrl + MinioUtil.getBucketName() + "/";
+        String url = fileUrl.replace(minioUrl, "");
+        // 文件名称
+        String fileName = ossFile.getFileName();
+        response.addHeader("Content-Disposition", "attachment;fileName=" + new String(fileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
+        // 设置强制下载不打开
+        response.setContentType("application/force-download");
+        // 然后在您的controller方法中：
+        if ("pdf".equalsIgnoreCase(FileUtil.extName(fileName))) {
+            try (InputStream originalInputStream = MinioUtil.getMinioFile(MinioUtil.getBucketName(), url);
+                 InputStream watermarkedInputStream = sysWaterMarkUtil.addWatermarkToPdf(originalInputStream);
+                 OutputStream outputStream = response.getOutputStream()) {
+
+                if (originalInputStream == null) {
+                    return;
+                }
+
+                byte[] buf = new byte[1024];
+                int len;
+                while ((len = watermarkedInputStream.read(buf)) > 0) {
+                    outputStream.write(buf, 0, len);
+                }
+                response.flushBuffer();
+
+            } catch (Exception e) {
+                log.error(e.getMessage());
+                response.setStatus(404);
+                e.printStackTrace();
+            }
+        } else {
+            try (InputStream inputStream = MinioUtil.getMinioFile(MinioUtil.getBucketName(), url);
+                 OutputStream outputStream = response.getOutputStream()
+            ) {
+                byte[] buf = new byte[1024];
+                int len;
+                if(inputStream==null) {
+                    return;
+                }
+                while ((len = inputStream.read(buf)) > 0) {
+                    outputStream.write(buf, 0, len);
+                }
+                response.flushBuffer();
+            } catch (Exception e){
+                log.error(e.getMessage());
+                response.setStatus(404);
+                e.printStackTrace();
+            }
+        }
+    }
+
     private void setResponse(HttpServletResponse response, String fileName) {
         response.addHeader("Content-Disposition", "attachment;fileName=" + new String(fileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
         // 设置强制下载不打开

laws-module-system/src/main/java/com/jero/modules/system/util/SysWaterMarkUtil.java

@@ -0,0 +1,94 @@
+package com.jero.modules.system.util;
+
+import com.jero.modules.system.entity.SysUser;
+import com.jero.modules.system.service.ISysUserService;
+import org.apache.pdfbox.pdmodel.PDDocument;
+import org.apache.pdfbox.pdmodel.PDPage;
+import org.apache.pdfbox.pdmodel.PDPageContentStream;
+import org.apache.pdfbox.pdmodel.common.PDRectangle;
+import org.apache.pdfbox.pdmodel.font.PDType0Font;
+import org.apache.pdfbox.pdmodel.graphics.state.PDExtendedGraphicsState;
+import org.apache.pdfbox.pdmodel.graphics.state.RenderingMode;
+import org.apache.pdfbox.util.Matrix;
+import org.springframework.core.io.ClassPathResource;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+import java.awt.*;
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.time.LocalDate;
+import java.time.format.DateTimeFormatter;
+
+/**
+ * 水印工具类
+ * @author CaiHaohan
+ */
+@Component
+public class SysWaterMarkUtil {
+    
+    @Resource
+    private ISysUserService userEoService;
+
+    public String getWaterMarkConfig() {
+        LocalDate currentDate = LocalDate.now();
+        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy/MM/dd");
+        String formattedDate = currentDate.format(formatter);
+
+        SysUser user = userEoService.getById(UserUtils.getUserId());
+        String userName = user.calcNameWorkNo();
+
+        return userName + " " + formattedDate;
+    }
+
+    public synchronized InputStream addWatermarkToPdf(InputStream inputStream) throws IOException {
+        String watermarkText = getWaterMarkConfig();
+        PDDocument document = PDDocument.load(inputStream);
+
+        // 设置水印字体、字体大小、颜色和透明度
+        ClassPathResource fontResource = new ClassPathResource("fonts/SourceHanSerif-VF.ttf");
+        PDType0Font font = PDType0Font.load(document, fontResource.getInputStream());
+        float fontSize = 13.0f;
+        Color color = new Color(100, 100, 100, 0);
+        PDExtendedGraphicsState graphicsState = new PDExtendedGraphicsState();
+        graphicsState.setNonStrokingAlphaConstant(0.4f);
+        graphicsState.setStrokingAlphaConstant(0.4f);
+
+        // 遍历每一页并添加水印
+        for (PDPage page : document.getPages()) {
+            // 获取页面尺寸以计算水印位置
+            PDRectangle pageSize = page.getMediaBox();
+            float xStep = pageSize.getWidth() / 4;
+            float yStep = pageSize.getHeight() / 4;
+            float rotationInRadians = (float) Math.toRadians(20);
+
+            for (float xPosition = pageSize.getLowerLeftX(); xPosition <= pageSize.getWidth(); xPosition += xStep) {
+                for (float yPosition = pageSize.getLowerLeftY(); yPosition <= pageSize.getHeight(); yPosition += yStep) {
+                    // 创建内容流并设置图形状态参数
+                    try (PDPageContentStream contentStream = new PDPageContentStream(document, page, PDPageContentStream.AppendMode.APPEND, true, true)) {
+                        contentStream.setGraphicsStateParameters(graphicsState);
+                        contentStream.setFont(font, fontSize);
+                        contentStream.setNonStrokingColor(color);
+                        contentStream.beginText();
+                        contentStream.setRenderingMode(RenderingMode.FILL);
+                        // 设置水印文本的旋转和位置
+                        contentStream.setTextMatrix(Matrix.getRotateInstance(rotationInRadians, xPosition, yPosition));
+                        contentStream.showText(watermarkText);
+                        contentStream.endText();
+                    }
+                }
+            }
+        }
+
+        // 创建一个临时的ByteArrayOutputStream保存修改过的PDF
+        ByteArrayOutputStream baos = new ByteArrayOutputStream();
+        document.save(baos);
+        document.close();
+
+        // 将ByteArrayOutputStream转换为ByteArrayInputStream以供返回
+        return new ByteArrayInputStream(baos.toByteArray());
+    }
+
+}