From 48d108d4e6992a2ff63e3d305f99f4d740a31bb8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E5=93=B2=E8=B4=A4?= <1037512352@qq.com>
Date: Tue, 16 Apr 2024 22:43:43 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E7=8E=B0easyexcel=E4=B8=AD=E7=9A=84Sh?=
 =?UTF-8?q?eetWriteHandler=EF=BC=8C=E5=9C=A8afterSheetCreate=E4=B8=AD?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=B0=B4=E5=8D=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../util/excel/CustomWaterMarkHandler.java    | 105 ++++++++++++++++++
 1 file changed, 105 insertions(+)
 create mode 100644 smart-admin-api/sa-admin/src/main/java/net/lab1024/sa/admin/util/excel/CustomWaterMarkHandler.java

diff --git a/smart-admin-api/sa-admin/src/main/java/net/lab1024/sa/admin/util/excel/CustomWaterMarkHandler.java b/smart-admin-api/sa-admin/src/main/java/net/lab1024/sa/admin/util/excel/CustomWaterMarkHandler.java
new file mode 100644
index 00000000..11b6fda9
--- /dev/null
+++ b/smart-admin-api/sa-admin/src/main/java/net/lab1024/sa/admin/util/excel/CustomWaterMarkHandler.java
@@ -0,0 +1,105 @@
+package net.lab1024.sa.admin.util.excel;
+
+
+import com.alibaba.excel.write.handler.SheetWriteHandler;
+import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
+import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.poi.openxml4j.opc.PackagePartName;
+import org.apache.poi.openxml4j.opc.PackageRelationship;
+import org.apache.poi.openxml4j.opc.TargetMode;
+import org.apache.poi.xssf.usermodel.XSSFPictureData;
+import org.apache.poi.xssf.usermodel.XSSFRelation;
+import org.apache.poi.xssf.usermodel.XSSFSheet;
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
+
+import javax.imageio.ImageIO;
+import java.awt.*;
+import java.awt.geom.AffineTransform;
+import java.awt.image.BufferedImage;
+import java.io.ByteArrayOutputStream;
+
+/**
+ *
+ * @author liuzhexian
+ * @since 2024-04-16
+ * @email 1037512352@qq.com
+ */
+@Slf4j
+public class CustomWaterMarkHandler implements SheetWriteHandler {
+
+    private final Watermark watermark;
+
+    public CustomWaterMarkHandler(Watermark watermark) {
+        this.watermark = watermark;
+    }
+
+    @Override
+    public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
+        BufferedImage bufferedImage = createWatermarkImage();
+        XSSFWorkbook workbook = (XSSFWorkbook) writeSheetHolder.getParentWriteWorkbookHolder().getWorkbook();
+        try {
+            // 添加水印的具体操作
+            addWatermarkToSheet(workbook, bufferedImage);
+        } catch (Exception e) {
+            log.error("添加水印出错:", e);
+        }
+
+    }
+
+    /**
+     * 创建水印图片
+     *
+     * @return
+     */
+    private BufferedImage createWatermarkImage() {
+        // 获取水印相关参数
+        Font font = watermark.getFont();
+        int width = watermark.getWidth();
+        int height = watermark.getHeight();
+        Color color = watermark.getColor();
+        String text = watermark.getContent();
+
+        // 创建带有透明背景的 BufferedImage
+        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
+        Graphics2D g = image.createGraphics();
+
+        // 设置画笔字体、平滑、颜色
+        g.setFont(font);
+        g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
+        g.setColor(color);
+
+        // 计算水印位置和角度
+        int y = watermark.getYAxis();
+        int x = watermark.getXAxis();
+        AffineTransform transform = AffineTransform.getRotateInstance(Math.toRadians(-watermark.getAngle()), 0, y);
+        g.setTransform(transform);
+        // 绘制水印文字
+        g.drawString(text, x, y);
+
+        // 释放资源
+        g.dispose();
+
+        return image;
+    }
+
+    private void addWatermarkToSheet(XSSFWorkbook workbook, BufferedImage watermarkImage) {
+        try (ByteArrayOutputStream os = new ByteArrayOutputStream()) {
+            ImageIO.write(watermarkImage, "png", os);
+            int pictureIdx = workbook.addPicture(os.toByteArray(), XSSFWorkbook.PICTURE_TYPE_PNG);
+            XSSFPictureData pictureData = workbook.getAllPictures().get(pictureIdx);
+            for (int i = 0; i < workbook.getNumberOfSheets(); i++) {
+                // 获取每个Sheet表
+                XSSFSheet sheet = workbook.getSheetAt(i);
+                PackagePartName ppn = pictureData.getPackagePart().getPartName();
+                String relType = XSSFRelation.IMAGES.getRelation();
+                PackageRelationship pr = sheet.getPackagePart().addRelationship(ppn, TargetMode.INTERNAL, relType, null);
+                sheet.getCTWorksheet().addNewPicture().setId(pr.getId());
+            }
+        } catch (Exception e) {
+            // 处理ImageIO.write可能抛出的异常
+            log.error("添加水印图片时发生错误", e);
+        }
+    }
+}
+