<dependency>
      <groupId>org.apache.commons</groupId>
      <artifactId>commons-compress</artifactId>
      <version>1.21</version>
  </dependency>
  <!-- Apache POI for Excel -->
  <dependency>
      <groupId>org.apache.poi</groupId>
      <artifactId>poi-ooxml</artifactId>
      <version>4.1.2</version>
  </dependency>
  <!-- Apache POI for Word -->
  <dependency>
      <groupId>org.apache.poi</groupId>
      <artifactId>poi-ooxml-schemas</artifactId>
      <version>4.1.2</version>
  </dependency>
  <!-- Apache POI dependencies -->
  <dependency>
      <groupId>org.apache.xmlbeans</groupId>
      <artifactId>xmlbeans</artifactId>
      <version>3.1.0</version>
  </dependency>

package com.meritdata.ddc.common.util;


import org.apache.poi.xwpf.usermodel.*;

import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

/**
 * @author dongxiajun
 * @since 2024-11-14 15:15
 */
public class WordMerger {
    /**
     * 合并兩個(gè)Word文檔，將第二個(gè)文件合并到第一個(gè)文件中
     *
     * @param firstFile    第一個(gè)文件
     * @param secondStream 上傳的文件
     * @param outputFile   輸出文件
     * @throws IOException IOException
     */
    public static void mergeWordDocuments(String firstFile, InputStream secondStream, String outputFile) throws IOException {
        // 使用 try-with-resources 自動(dòng)關(guān)閉資源
        try (FileInputStream fis1 = new FileInputStream(firstFile);
             FileOutputStream out = new FileOutputStream(outputFile)) {
            // 加載第一個(gè)文檔
            XWPFDocument doc1 = new XWPFDocument(fis1);
            // 加載第二個(gè)文檔
            XWPFDocument doc2 = new XWPFDocument(secondStream);
            // 合并文檔
            appendDocument(doc1, doc2);
            // 保存合并后的文檔
            doc1.write(out);
        }
    }

    /**
     * 合并兩個(gè)Word文檔，將第二個(gè)文件合并到第一個(gè)文件中
     *
     * @param firstFile  第一個(gè)文件
     * @param secondFile 第二個(gè)文件
     * @param outputFile 輸出文件
     * @throws IOException IOException
     */
    public static void mergeWordDocuments(String firstFile, String secondFile, String outputFile) throws IOException {
        // 使用 try-with-resources 自動(dòng)關(guān)閉資源
        try (FileInputStream fis1 = new FileInputStream(firstFile);
             FileInputStream fis2 = new FileInputStream(secondFile);
             FileOutputStream out = new FileOutputStream(outputFile)) {
            // 加載第一個(gè)文檔
            XWPFDocument doc1 = new XWPFDocument(fis1);
            // 加載第二個(gè)文檔
            XWPFDocument doc2 = new XWPFDocument(fis2);
            // 合并文檔
            appendDocument(doc1, doc2);
            // 保存合并后的文檔
            doc1.write(out);
        }
    }

    public static void appendDocument(XWPFDocument doc1, XWPFDocument doc2) {
        // 獲取第二個(gè)文檔的所有部分，保持順序合并
        for (IBodyElement element : doc2.getBodyElements()) {
            // 根據(jù)元素的類型進(jìn)行不同的處理
            if (element instanceof XWPFParagraph) {
                XWPFParagraph paragraph = (XWPFParagraph) element;
                XWPFParagraph newParagraph = doc1.createParagraph();
                copyParagraph(paragraph, newParagraph);
            } else if (element instanceof XWPFTable) {
                XWPFTable table = (XWPFTable) element;
                XWPFTable newTable = doc1.createTable();
                copyTable(table, newTable);
            }
        }
    }

    private static void copyParagraph(XWPFParagraph source, XWPFParagraph target) {
        // 只復(fù)制段落的布局屬性（例如對(duì)齊、縮進(jìn)、行距等），但不包括段落樣式（避免改變標(biāo)題級(jí)別）
        target.getCTP().setPPr(source.getCTP().getPPr());

        // 復(fù)制對(duì)齊方式、縮進(jìn)、行間距等樣式
        target.setAlignment(source.getAlignment());
        target.setVerticalAlignment(source.getVerticalAlignment());
        target.setIndentationLeft(source.getIndentationLeft());
        target.setIndentationRight(source.getIndentationRight());
        target.setIndentationFirstLine(source.getIndentationFirstLine());
        target.setSpacingBefore(source.getSpacingBefore());
        target.setSpacingAfter(source.getSpacingAfter());
        target.setSpacingBetween(source.getSpacingBetween());

        // 復(fù)制段落中的每個(gè)run
        for (XWPFRun run : source.getRuns()) {
            XWPFRun newRun = target.createRun();
            newRun.setText(run.text());
            // 復(fù)制格式
            newRun.setBold(run.isBold());
            newRun.setItalic(run.isItalic());
            newRun.setUnderline(run.getUnderline());
            newRun.setColor(run.getColor());
            newRun.setFontSize(run.getFontSize());
            newRun.setFontFamily(run.getFontFamily());
        }
        if (source.getStyle() != null) {
            target.setStyle(source.getStyle());
        }
    }

    // 復(fù)制源表格到目標(biāo)表格，包括其行和單元格的樣式和內(nèi)容
    private static void copyTable(XWPFTable source, XWPFTable target) {
        // 刪除目標(biāo)表格中的默認(rèn)創(chuàng)建的第一行，確保從空狀態(tài)開始復(fù)制
        target.removeRow(0);

        // 遍歷源表格的每一行
        for (XWPFTableRow sourceRow : source.getRows()) {
            // 在目標(biāo)表格中創(chuàng)建新行
            XWPFTableRow newTableRow = target.createRow();
            // 復(fù)制源行的樣式和內(nèi)容到目標(biāo)新行
            copyTableRow(sourceRow, newTableRow);
        }
    }

    // 將源表格行的樣式和內(nèi)容復(fù)制到目標(biāo)表格行
    private static void copyTableRow(XWPFTableRow source, XWPFTableRow target) {
        // 復(fù)制行的樣式屬性
        target.getCtRow().setTrPr(source.getCtRow().getTrPr());

        // 遍歷源行的每一個(gè)單元格
        for (int i = 0; i < source.getTableCells().size(); i++) {
            XWPFTableCell sourceCell = source.getCell(i);
            XWPFTableCell targetCell;

            // 如果目標(biāo)行的單元格還不夠，則創(chuàng)建新單元格
            if (i < target.getTableCells().size()) {
                targetCell = target.getCell(i);
            } else {
                targetCell = target.addNewTableCell();
            }
            String text = source.getCell(0).getText();
            if (text.contains("{{$fe")) {
                // 復(fù)制單元格的樣式屬性
                targetCell.getCTTc().setTcPr(sourceCell.getCTTc().getTcPr());
                targetCell.setText(sourceCell.getText());
            } else {
                // 復(fù)制單元格的樣式和內(nèi)容
                copyTableCell(sourceCell, targetCell);
            }
        }
    }

    // 將源單元格的樣式和內(nèi)容復(fù)制到目標(biāo)單元格
    private static void copyTableCell(XWPFTableCell source, XWPFTableCell target) {
        // 復(fù)制單元格的樣式屬性
        target.getCTTc().setTcPr(source.getCTTc().getTcPr());
        // 清除目標(biāo)單元格的段落
        target.getParagraphs().clear();
        // 復(fù)制每個(gè)段落
        for (XWPFParagraph sourceParagraph : source.getParagraphs()) {
            XWPFParagraph targetParagraph = target.addParagraph();
            copyParagraph(sourceParagraph, targetParagraph);
        }
    }

}