package com.zy.crm.manager.utils;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import org.apache.poi.xwpf.usermodel.*;
import org.apache.xmlbeans.XmlCursor;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.STMerge;
import org.springframework.core.io.InputStreamResource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;

import java.io.*;
import java.text.DecimalFormat;
import java.text.NumberFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

//Word操作工具类
public class WordUtils {

    //生成word合同
    public static ResponseEntity<InputStreamResource> generate(String fileName, HashMap<String, Object> map, List<List<String>> tabParam) throws IOException {
        // 读取 Word 模板文件
        File file = new File(fileName);
        FileInputStream inputStream = new FileInputStream(file);
        XWPFDocument document = new XWPFDocument(inputStream);

        insertTab("{{table}}", document, tabParam);

        map.put("{{table}}", "");
        processParagraphs(map, document);

        // 将文档写入输出流
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        document.write(outputStream);
        outputStream.close();

        // 创建响应实体，将输出流作为文件内容返回
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
        headers.setContentDispositionFormData("attachment", "output.docx");
        InputStreamResource resource = new InputStreamResource(new ByteArrayInputStream(outputStream.toByteArray()));
        return ResponseEntity.ok()
                .headers(headers)
                .body(resource);
    }

    /**
     * 处理段落中文本，替换文本中定义的变量；
     * @param param 需要替换的变量及变量值
     * @param document 需要替换的DOC
     */
    public static void processParagraphs(HashMap<String, Object> param, XWPFDocument document) {
        // 遍历所有段落
        for (XWPFParagraph paragraph : document.getParagraphs()) {
            for (Map.Entry<String, Object> entry : param.entrySet()) {
                String key = entry.getKey();
                String value = entry.getValue().toString();
                // 替换段落文本
                replaceTextInParagraph(paragraph, key, value);
            }
        }

        // 遍历所有表格
        for (XWPFTable table : document.getTables()) {
            // 遍历表格中的每个单元格
            for (XWPFTableRow row : table.getRows()) {
                for (XWPFTableCell cell : row.getTableCells()) {
                    for (Map.Entry<String, Object> entry : param.entrySet()) {
                        String key = entry.getKey();
                        String value = entry.getValue().toString();
                        // 替换表格单元格的文本
                        replaceTextInParagraph(cell.getParagraphs().get(0), key, value);
                    }
                }
            }
        }
    }

    private static void replaceTextInParagraph(XWPFParagraph paragraph, String searchString, String replacementString) {
        for (XWPFRun run : paragraph.getRuns()) {
            // 获取段落中的文本
            String text = run.getText(0);
            if (text != null && text.contains(searchString)) {
                // 使用正则表达式进行替换
                Pattern pattern = Pattern.compile(Pattern.quote(searchString));
                Matcher matcher = pattern.matcher(text);
                if (matcher.find()) {
                    String replacedText = matcher.replaceAll(replacementString);
                    // 替换文本
                    run.setText(replacedText, 0);
                }
            }
        }
    }

    /**
     * 在定位的位置插入表格
     */
    public static void insertTab(String key, XWPFDocument document, List<List<String>> param) {
        List<XWPFParagraph> paragraphList = document.getParagraphs();
        if (paragraphList != null && paragraphList.size() > 0) {
            for (XWPFParagraph paragraph : paragraphList) {
                List<XWPFRun> runs = paragraph.getRuns();
                for (XWPFRun run : runs) {
                    String text = run.getText(0);
                    if (text != null) {
                        if (text.indexOf(key) >= 0) {
                            XmlCursor cursor = paragraph.getCTP().newCursor();
                            XWPFTable tableOne = document.insertNewTbl(cursor);// ---这个是关键
//                            String test = "[\n" +
//                                    "  [\"序号\",\"品名\",\"数量\",\"单位\",\"单价(元)\",\"合计(元)\",\"备注\"],\n" +
//                                    "  [\"1\",\"自动化立体仓库设备\",\"1\",\"套\",\"130000\",\"130000\",\"备注\"],\n" +
//                                    "  [\"合计（大写金额）：\",\"\",\"\",\"\",\"合计：\",\"\",\"\"]\n" +
//                                    "]";
//
//                            JSONArray row = JSON.parseArray(test);
//                            ArrayList<List<String>> list = new ArrayList<>();
//                            for (Object o : row) {
//                                ArrayList<String> arrayList = new ArrayList<>();
//                                JSONArray objects = JSON.parseArray(o.toString());
//                                for (Object object : objects) {
//                                    arrayList.add(object.toString());
//                                }
//                                list.add(arrayList);
//                            }

                            int rows = param.size();
                            int cols = param.get(0).size();
                            for (int i = 0; i < rows; i++) {
                                XWPFTableRow tableRow = null;
                                if (i == 0) {
                                    tableRow = tableOne.getRow(i);
                                } else {
                                    tableRow = tableOne.createRow();
                                }

                                for (int j = 0; j < cols; j++) {
                                    if (i == 0 && j == 0) {
                                        tableRow.getCell(j).setText(param.get(i).get(j));
                                    } else {
                                        XWPFTableCell cell = tableRow.getCell(j);
                                        if (cell == null) {
                                            tableRow.addNewTableCell().setText(param.get(i).get(j));
                                        } else {
                                            cell.setText(param.get(i).get(j));
                                        }
                                    }
                                }
                            }

                            //合并最后一行单元格
                            tableOne.getRow(rows - 1).getCell(0).getCTTc().addNewTcPr().addNewHMerge().setVal(STMerge.RESTART);
                            tableOne.getRow(rows - 1).getCell(1).getCTTc().addNewTcPr().addNewHMerge().setVal(STMerge.CONTINUE);
                            tableOne.getRow(rows - 1).getCell(2).getCTTc().addNewTcPr().addNewHMerge().setVal(STMerge.CONTINUE);
                            tableOne.getRow(rows - 1).getCell(4).getCTTc().addNewTcPr().addNewHMerge().setVal(STMerge.RESTART);
                            tableOne.getRow(rows - 1).getCell(5).getCTTc().addNewTcPr().addNewHMerge().setVal(STMerge.CONTINUE);
                        }
                    }
                }
            }
        }
    }

    public static String formatNumberForAccounting(double number) {
        if (number == 0) {
            return "0.00";
        }
        NumberFormat formatter = new DecimalFormat("#,###.00");
        return formatter.format(number);
    }

}