feat: 带水印下载

laws-modules/src/main/java/com/jero/modules/laws/common/controller/LawsCommonController.java

@@ -1,27 +1,39 @@
 package com.jero.modules.laws.common.controller;
 
+import cn.hutool.core.io.FileUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.jero.common.api.vo.Result;
 import com.jero.common.exception.JeroBootException;
 import com.jero.common.util.MessageUtils;
+import com.jero.common.util.MinioUtil;
 import com.jero.modules.laws.common.constant.ResultCommon;
 import com.jero.modules.laws.common.service.ILawsCommonService;
 import com.jero.modules.laws.common.vo.ManyStandardSelectionBox;
 import com.jero.modules.laws.common.vo.ManyStandardSelectionBoxVO;
+import com.jero.modules.laws.enterprise.util.WaterMarkUtil;
+import com.jero.modules.oss.entity.OSSFile;
+import com.jero.modules.oss.service.IOSSFileService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.*;
-import org.springframework.web.bind.annotation.RequestMapping;
+
-import org.springframework.web.bind.annotation.RequestParam;
+import javax.annotation.Resource;
-import org.springframework.web.bind.annotation.RestController;
+import javax.servlet.http.HttpServletResponse;
+import java.io.ByteArrayOutputStream;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.nio.charset.StandardCharsets;
 
 /**
  * @Author: liao
  * @Date: 2023/9/11 15:40
  * @Description: 法规通用操作
  */
+@Slf4j
 @Api(tags="法规通用操作")
 @RestController
 @RequestMapping("/laws/standard/common")
@@ -30,6 +42,12 @@ public class LawsCommonController {
     @Autowired
     private ILawsCommonService lawsCommonService;
 
+    @Resource
+    private WaterMarkUtil waterMarkUtil;
+
+    @Resource
+    private IOSSFileService ossFileService;
+
     @ApiOperation(value="多选标准弹框", notes="多选标准弹框")
     @GetMapping(value = "/getManyStandardSelectionBox")
     public Result<IPage<ManyStandardSelectionBox>> getManyStandardSelectionBox(ManyStandardSelectionBoxVO selectionBoxVO,
@@ -52,4 +70,75 @@ public class LawsCommonController {
         return Result.OK(lawsCommonService.getManyStandardSelectionBox(selectionBoxVO, pageNo, pageSize));
     }
 
+
+    /**
+     * 下载带水印的PDF文件
+     *
+     * @param id 传入文件id
+     * @param response
+     */
+    @ApiOperation(value="PDF带水印下载", notes="PDF带水印下载")
+    @GetMapping(value = "/downloadWithWaterMark/{id}")
+    public void downloadWithWaterMark(@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".equals(FileUtil.extName(fileName))) {
+            try (InputStream originalInputStream = MinioUtil.getMinioFile(MinioUtil.getBucketName(), url);
+                 InputStream watermarkedInputStream = waterMarkUtil.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();
+            }
+        }
+    }
 }

laws-modules/src/main/java/com/jero/modules/laws/enterprise/util/WaterMarkUtil.java

@@ -0,0 +1,92 @@
+package com.jero.modules.laws.enterprise.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 com.jero.modules.sys.utils.UserUtils;
+
+import javax.annotation.Resource;
+import java.awt.*;
+import java.io.*;
+import java.time.LocalDate;
+import java.time.format.DateTimeFormatter;
+
+/**
+ * 水印工具类
+ * @author CaiHaohan
+ */
+@Component
+public class WaterMarkUtil {
+    
+    @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.getNameWorkNo();
+
+        return userName + " " + formattedDate;
+    }
+
+    public 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());
+    }
+
+}

laws-modules/src/main/java/com/jero/modules/sys/utils/UserUtils.java

@@ -0,0 +1,20 @@
+package com.jero.modules.sys.utils;
+
+import com.jero.common.system.vo.LoginUser;
+import org.apache.shiro.SecurityUtils;
+
+/**
+ * @author: Mzaxd
+ * @Date: 2023/7/7 14:54
+ */
+public class UserUtils {
+
+    public static String getUserId() {
+        LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
+        return sysUser.getId();
+    }
+
+    public static LoginUser getCurrentUser() {
+        return (LoginUser) SecurityUtils.getSubject().getPrincipal();
+    }
+}