复制word文档，合并word文档

背景

现在有一个需求，需要将多个word文档合并为一个文档，并使用poi-el渲染模板生成一个新的文档。
在网上找了很多，没找到合适的，最后摸索出一个可行方案，基于XWPFDocument.getBodyElements方法的。
其实一开始考虑过这个方法，但当时对底层源码不熟悉，觉得无法分辨每个element的类型，比如chart对象其实是包含在XWPFParagraph对象中的，所以当时排除了这个方法。后来在网上找到了基于OPCPackage的，但是这种方法无法保证顺序，所以也排除了。最后想到了XWPFTemplate.render方法的底层源码，可以完美解决getBodyElements的类型问题，所以就使用了getBodyElements这个方案。
参考文档：https://codeleading.com/article/50704657377/

方案思路

使用XWPFDocument.getBodyElements()来获取整个doc的所有元素对象，元素对象供分为三类，分别为PARAGRAPH、TABLE、CONTENTCONTROL，其中，Picture、Chart均是包含在PARAGRAPH里的，关于这块可以参考底层代码TemplateResolver

完整代码

package word;

import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.xwpf.CTDrawingWrapper;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ooxml.POIXMLDocumentPart;
import org.apache.poi.util.Units;
import org.apache.poi.xwpf.usermodel.*;
import org.openxmlformats.schemas.drawingml.x2006.chart.CTPlotArea;
import org.openxmlformats.schemas.drawingml.x2006.main.CTNonVisualDrawingProps;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.*;

import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * 将多个word文档合并为一个word文件
 */
public class CopyDoc{


    static String target = "C:\\Users\\huahua\\Desktop\\template.docx";
    static String dest = "C:\\Users\\huahua\\Desktop\\out.docx";


    public static List<XWPFDocument> readDoc(List<String> paths) throws Exception {
        List<XWPFDocument> documents = new ArrayList();
        for (String path : paths) {
            documents.add(new XWPFDocument(new FileInputStream(path)));
        }
        return documents;
    }

    public static void main(String[] args) throws Exception {
        List<String> paths = new ArrayList() {{
            add("C:\\Users\\huahua\\Desktop\\模板\\11.docx");
            add("C:\\Users\\huahua\\Desktop\\模板\\12.docx");
            add("C:\\Users\\huahua\\Desktop\\模板\\14.docx");
        }};
        List<XWPFDocument> document = readDoc(paths);
        insertContext(document, new ArrayList(){{
            add("标题1");
            add("标题2");
            add("标题3");
        }});
        renderData();
    }


    public static void renderData() throws IOException {
        XWPFTemplate compile = XWPFTemplate.compile(target);
        compile.render(new HashMap() {
            {
                put("1", "ajsdkaklsjd");
                put("2", "ajsdkaklsjd");
                put("3", "ajsdkaklsjd");
            }
        }).writeToFile(dest);

    }

    /**
     * 追加内容到页面
     *
     * @throws IOException
     */
    public static void insertContext(List<XWPFDocument> documents, List<String> titles) throws Exception {
        //模板文档
        XWPFDocument document1 = new XWPFDocument();
        XWPFDocument template = new XWPFDocument(new FileInputStream("C:\\Users\\huahua\\Desktop\\标题文档.docx"));
        // 获得模板文档的整体样式
        CTStyles wordStyles = template.getStyle();
        // 获取新建文档对象的样式
        XWPFStyles newStyles = document1.createStyles();
        // 关键行// 修改设置文档样式为静态块中读取到的样式
        newStyles.setStyles(wordStyles);
        int i = 1;
        for (XWPFDocument document : documents) {

            //一级标题
            XWPFParagraph paragraph = document1.createParagraph();
            paragraph.setStyle("2");
            XWPFRun run1 = paragraph.createRun();
            run1.addBreak();
            run1.setText(i+" ."+titles.get(i-1));

            Document document2 = new XWPFDocument();

            List<IBodyElement> bodyElements = document.getBodyElements();

            int chartPos = 0;
            for (IBodyElement element : bodyElements) {
                if (element.getElementType() == BodyElementType.PARAGRAPH) {
                    XWPFParagraph sourcePara = (XWPFParagraph) element;
                    XWPFParagraph targetParagraph = document1.createParagraph();
                    copyPara(targetParagraph, sourcePara, document1, document, chartPos);
                } else if (element.getElementType() == BodyElementType.TABLE) {
                    XWPFTable sourceTable = (XWPFTable) element;
                    List<XWPFTableRow> rows = sourceTable.getRows();
                    if (null == rows) continue;
                    copyTable(sourceTable, document1);
                }
            }
            i++;

        }

        FileOutputStream fopts = new FileOutputStream(target);
        document1.write(fopts);
        document1.close();
        fopts.close();
    }

    /**
     * @param targetParagraph
     * @param sourcePara
     */
    private static void copyPara(XWPFParagraph targetParagraph, XWPFParagraph sourcePara, XWPFDocument document, XWPFDocument srcDoc, int pos) throws Exception {
        //把传过来的段落中的ppr标签设置到新的段落里
        CTP ctp = targetParagraph.getCTP();
        ctp.setPPr(sourcePara.getCTPPr());
        List<XWPFRun> runs = sourcePara.getRuns();//获取run对象
        for (int k = 0; k < runs.size(); k++) {
            XWPFRun newRun = targetParagraph.createRun();
            XWPFRun oldRun = runs.get(k);
            CTRPr oldRPr = oldRun.getCTR().getRPr();
            newRun.getCTR().setRPr(oldRPr);
            //处理图表
            if (StringUtils.isBlank(oldRun.getText(0))) {
                CTR ctr = oldRun.getCTR();
                CTDrawing ctDrawing = CollectionUtils.isNotEmpty(ctr.getDrawingList()) ? ctr.getDrawingArray(0) : null;
                if (null == ctDrawing) continue;
                CTDrawingWrapper wrapper = new CTDrawingWrapper(ctDrawing);
                String rid = wrapper.getChartId();
                if (null == rid) continue;
                POIXMLDocumentPart documentPart = oldRun.getDocument().getRelationById(rid);
                if (null == documentPart || !(documentPart instanceof XWPFChart)) continue;
                XWPFChart chart1 = srcDoc.getCharts().get(pos);
                XWPFChart chart = document.createChart(newRun, 7 * Units.EMU_PER_CENTIMETER, 7 * Units.EMU_PER_CENTIMETER);
                CTPlotArea plotArea = chart1.getCTChart().getPlotArea();
                chart.getCTChart().setPlotArea(plotArea);
                chart.getCTChart().setLegend(chart1.getCTChart().getLegend());
                chart.setWorkbook(chart1.getWorkbook());

                CTNonVisualDrawingProps docPr = newRun.getCTR().getDrawingList().get(0).getInlineList().get(0).getDocPr();
                if (StringUtils.isBlank(docPr.getTitle())) {
                    docPr.setTitle("{{test}}");
                }
                pos++;
            } else {
                newRun.setText(oldRun.getText(0));
            }
            targetParagraph.addRun(newRun);

        }

    }

    /**
     * 复制表格
     *
     * @param source
     * @param document
     */
    private static void copyTable(XWPFTable source, XWPFDocument document) {
        // 创建新的 CTTbl
        CTTbl ctTbl = CTTbl.Factory.newInstance();
        // 复制原来的CTTbl
        ctTbl.set(source.getCTTbl());
        IBody iBody = source.getBody();
        XWPFTable newTable = new XWPFTable(ctTbl, iBody);
        document.createTable();
        document.setTable(document.getTables().size()-1, newTable);
    }
}