认证-word工具类-批量插入图片

jero-boot/jero-boot-modules/src/main/java/com/jero/modules/project/util/WordUtil.java

@@ -20,14 +20,14 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.util.FileCopyUtils;
 
+import javax.xml.bind.JAXBElement;
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.InputStream;
 import java.io.OutputStream;
-import java.util.HashMap;
+import java.util.*;
-import java.util.Iterator;
+import java.util.regex.Matcher;
-import java.util.List;
+import java.util.regex.Pattern;
-import java.util.Map;
 
 public class WordUtil {
     private static final Logger log = LoggerFactory.getLogger(WordUtil.class);
@@ -118,7 +118,16 @@ public class WordUtil {
         table.getContent().remove(rowNum - 1);
     }
 
-    public static void replacePicture(WordprocessingMLPackage wordMLPackage, String bookmarkName, Object text, byte[] bytes) throws Exception {
+    /**
+     * 通过书签识别需要插入的图片位置，并替换书签对应的文本为图片
+     * @param wordMLPackage
+     * @param bookmarkName 书签名称
+     * @param text 图片前需要加的文本
+     * @param bytes 图片
+     * @throws Exception
+     * @author liyawei
+     */
+    public static void replacePictureBM(WordprocessingMLPackage wordMLPackage, String bookmarkName, Object text, byte[] bytes) throws Exception {
         try {
             MainDocumentPart mainDocumentPart = wordMLPackage.getMainDocumentPart();
             Document wmlDoc = (Document)mainDocumentPart.getJaxbElement();
@@ -147,6 +156,125 @@ public class WordUtil {
 
     }
 
+    /**
+     * 通过 占位符的key 识别需要插入的图片位置，并替换占位符为图片
+     * @param wordMLPackage
+     * @param key 占位符的key ${key}
+     * @param text 图片前需要加的文本
+     * @param bytes 图片
+     * @throws Exception
+     */
+    public static void replacePicture(WordprocessingMLPackage wordMLPackage, String key, Object text, byte[] bytes) throws Exception {
+        try {
+            MainDocumentPart mainDocumentPart = wordMLPackage.getMainDocumentPart();
+            Docx4jUtils.cleanDocumentPart(mainDocumentPart);
+            Document wmlDoc = (Document)mainDocumentPart.getJaxbElement();
+            Body body = wmlDoc.getBody();
+            Iterator var8 = getAllPlaceholderElementFromObject(body).iterator();
+
+            while(var8.hasNext()) {
+                Text bm = (Text) var8.next();
+                if (bm.getValue().equals("${" + key + "}")) {
+                    BinaryPartAbstractImage imagePart = BinaryPartAbstractImage.createImagePart(wordMLPackage, bytes);
+                    Inline inline = imagePart.createImageInline("", "", 0, 1, true);
+                    ObjectFactory factory = new ObjectFactory();
+                    R run = factory.createR();
+                    Drawing drawing = factory.createDrawing();
+                    drawing.getAnchorOrInline().add(inline);
+                    run.getContent().add(drawing);
+                    replaceTextToImage(bm, text, wordMLPackage, bytes);
+                }
+            }
+        } catch (Exception var17) {
+            log.error(var17.getMessage(), var17);
+        }
+
+    }
+
+    /**
+     * 通过 占位符的key 识别需要插入的图片位置，并替换占位符为图片
+     * 批量替换
+     * @param wordMLPackage
+     * @param picList
+     * @throws Exception
+     */
+    public static void replacePictureBatch(WordprocessingMLPackage wordMLPackage, List<Map<String, Object>> picList) throws Exception {
+        try {
+            MainDocumentPart mainDocumentPart = wordMLPackage.getMainDocumentPart();
+            Docx4jUtils.cleanDocumentPart(mainDocumentPart);
+            Document wmlDoc = (Document)mainDocumentPart.getJaxbElement();
+            Body body = wmlDoc.getBody();
+            List<Text> textList = getAllPlaceholderElementFromObject(body); // 获取文档中所有占位符
+
+            for (Map<String, Object> picMap :picList) {
+                String key = (String) picMap.get("key");
+                Object text = picMap.get("text");
+                byte[] bytes = (byte[]) picMap.get("bytes");
+
+                for(Text bm : textList) {
+                    if (bm.getValue().equals("${" + key + "}")) {
+                        BinaryPartAbstractImage imagePart = BinaryPartAbstractImage.createImagePart(wordMLPackage, bytes);
+                        Inline inline = imagePart.createImageInline("", "", 0, 1, true);
+                        ObjectFactory factory = new ObjectFactory();
+                        R run = factory.createR();
+                        Drawing drawing = factory.createDrawing();
+                        drawing.getAnchorOrInline().add(inline);
+                        run.getContent().add(drawing);
+                        replaceTextToImage(bm, text, wordMLPackage, bytes);
+                    }
+                }
+            }
+        } catch (Exception var17) {
+            log.error(var17.getMessage(), var17);
+        }
+
+    }
+
+    /**
+     * 发现docx文档包含占位符的文本节点
+     * 未解决问题：两个连着的相同占位符会识别成一个。如 原文：${key1}${key1}
+     * @param obj
+     * @return
+     * @author liyawei
+     */
+    private static List<Text> getAllPlaceholderElementFromObject(Object obj) {
+        List<Text> result = new ArrayList<>();
+        Class<Text> toSearch = Text.class;
+        Text textPlaceholder;
+        if (obj instanceof JAXBElement) {
+            obj = ((JAXBElement<?>) obj).getValue();
+        }
+        if (obj.getClass().equals(toSearch)) {
+            textPlaceholder = (Text) obj;
+            if (isPlaceholder(textPlaceholder.getValue())) {
+                result.add((Text) obj);
+            }
+        } else if (obj instanceof ContentAccessor) {
+            List<?> children = ((ContentAccessor) obj).getContent();
+            for (Object child : children) {
+                result.addAll(getAllPlaceholderElementFromObject(child));
+            }
+        }
+        return result;
+
+    }
+
+    /**
+     * 判断字符串是否有${}占位符
+     *
+     * @param str 需要判断的字符串
+     * @return 是否字符串是否有${}占位符
+     * @author liyawei
+     */
+    private static boolean isPlaceholder(String str) {
+        if (str != null && !str.isEmpty()) {
+            Pattern pattern = Pattern.compile("([$]\\{[A-Za-z0-9./ -]+\\})"); // 识别范围 认证模块专用
+            Matcher m = pattern.matcher(str);
+            return m.find();
+        }
+        return false;
+    }
+
     /**
      * 在标签处插入替换内容
      *
@@ -217,6 +345,72 @@ public class WordUtil {
         }
     }
 
+
+    /**
+     * 在占位符处插入替换内容
+     *
+     * @param bm
+     * @param object
+     * @throws Exception
+     * @author liyawei
+     */
+    public static void replaceTextToImage(Text bm, Object object, WordprocessingMLPackage wordMLPackage , byte[] bytes) throws Exception {
+        if (wordMLPackage == null) {
+            return;
+        }
+
+        String text = object.toString();
+        try {
+            List<Object> theList = null;
+            ParaRPr rpr = null;
+            if (bm.getParent() instanceof P) {
+                PPr pprTemp = ((P) (bm.getParent())).getPPr();
+                if (pprTemp == null) {
+                    rpr = null;
+                } else {
+                    rpr = ((P) (bm.getParent())).getPPr().getRPr();
+                }
+                theList = ((ContentAccessor) (bm.getParent())).getContent();
+            } else {
+                R r = (R) bm.getParent();
+                if (r.getParent() instanceof P) {
+                    theList = ((ContentAccessor) (r.getParent())).getContent();
+
+                    int rangeStart = -1; // 记录占位符run位置
+                    int i = 0;
+                    for (Object ox : theList) {
+                        Object listEntry = XmlUtils.unwrap(ox);
+                        if (listEntry.equals(r)) {
+                            rangeStart = i;
+                        }
+                        i++;
+                    }
+                    theList.remove(rangeStart); // 移除原占位符
+
+                    BinaryPartAbstractImage imagePart = BinaryPartAbstractImage.createImagePart(wordMLPackage, bytes);
+                    Inline inline = imagePart.createImageInline("", "", 0, 1, true);
+                    ObjectFactory factory = Context.getWmlObjectFactory();
+                    R run = factory.createR();
+                    Drawing drawing = factory.createDrawing();
+                    if(text != null){
+                        Text txt = factory.createText();
+                        txt.setValue(text);
+                        run.getContent().add(txt);
+                    }
+                    run.getContent().add(drawing);
+                    drawing.getAnchorOrInline().add(inline);
+                    theList.add(rangeStart, run); // 添加图片到原占位符位置
+
+                } else {
+                    return;
+                }
+            }
+
+        } catch (ClassCastException cce) {
+            log.error(cce.getMessage(), cce);
+        }
+    }
+
     public static void replaceForTemplate(String templatePath, Map<String, String> map, int tableNum, List<Map<String, Object>> dataList, String outPath) throws Exception {
         WordprocessingMLPackage wordMLPackage = WordprocessingMLPackage.load(new File(templatePath));
         MainDocumentPart mainDocumentPart = wordMLPackage.getMainDocumentPart();