import org.apache.poi.xwpf.usermodel.*;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTBody;

import java.io.*;
import java.util.HashMap;
import java.util.Map;

public class ComplexFormDocGenerator {

    public static void main(String[] args) {
        String[] templatePaths = {"complex_template.docx"};
        String outputPath = "complex_output.docx";
        Map<String, String> values = new HashMap<>();
        values.put("${name}", "李四");
        values.put("${department}", "技術部");

        try {
            generateDocument(templatePaths, outputPath, values);
            System.out.println("文檔生成成功，路徑為: " + outputPath);
        } catch (IOException e) {
            System.err.println("生成文檔時出現(xiàn)錯誤: " + e.getMessage());
            e.printStackTrace();
        }
    }

    public static void generateDocument(String[] templatePaths, String outputPath, Map<String, String> values) throws IOException {
        XWPFDocument finalDocument = new XWPFDocument();

        for (String templatePath : templatePaths) {
            try (FileInputStream templateStream = new FileInputStream(templatePath);
                 XWPFDocument templateDocument = new XWPFDocument(templateStream)) {

                replacePlaceholders(templateDocument, values);
                appendDocument(finalDocument, templateDocument);
            }
        }

        try (FileOutputStream outputStream = new FileOutputStream(outputPath)) {
            finalDocument.write(outputStream);
        }
    }

    private static void replacePlaceholders(XWPFDocument document, Map<String, String> values) {
        // 處理段落中的占位符
        for (XWPFParagraph paragraph : document.getParagraphs()) {
            for (XWPFRun run : paragraph.getRuns()) {
                String text = run.getText(0);
                if (text != null) {
                    for (Map.Entry<String, String> entry : values.entrySet()) {
                        text = text.replace(entry.getKey(), entry.getValue());
                    }
                    run.setText(text, 0);
                }
            }
        }

        // 處理表格中的占位符
        for (XWPFTable table : document.getTables()) {
            for (XWPFTableRow row : table.getRows()) {
                for (XWPFTableCell cell : row.getTableCells()) {
                    for (XWPFParagraph paragraph : cell.getParagraphs()) {
                        for (XWPFRun run : paragraph.getRuns()) {
                            String text = run.getText(0);
                            if (text != null) {
                                for (Map.Entry<String, String> entry : values.entrySet()) {
                                    text = text.replace(entry.getKey(), entry.getValue());
                                }
                                run.setText(text, 0);
                            }
                        }
                    }
                }
            }
        }

        // 處理文本框中的占位符（POI 對文本框處理有限，可結合其他庫）
        // 這里僅簡單示例，實際可能需要更復雜邏輯
        // 可以使用 docx4j 等庫來更好地處理文本框
        for (IBodyElement element : document.getBodyElements()) {
            if (element instanceof XWPFSDT) {
                XWPFSDT sdt = (XWPFSDT) element;
                for (XWPFParagraph paragraph : sdt.getContent().getParagraphs()) {
                    for (XWPFRun run : paragraph.getRuns()) {
                        String text = run.getText(0);
                        if (text != null) {
                            for (Map.Entry<String, String> entry : values.entrySet()) {
                                text = text.replace(entry.getKey(), entry.getValue());
                            }
                            run.setText(text, 0);
                        }
                    }
                }
            }
        }
    }

    private static void appendDocument(XWPFDocument mainDocument, XWPFDocument appendDocument) throws IOException {
        CTBody mainBody = mainDocument.getDocument().getBody();
        CTBody appendBody = appendDocument.getDocument().getBody();

        for (Object object : appendBody.getContent()) {
            mainBody.addNewP().set(object);
        }
    }
}