package com.zy.crm.manager.controller;

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 org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

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

@RestController
public class WordController {

    @GetMapping("/generateWord")
    public ResponseEntity<InputStreamResource> generateWord() throws IOException {
        // 读取 Word 模板文件
        String fileName = this.getClass().getClassLoader().getResource("word1.docx").getPath();//获取文件路径
        File file = new File(fileName);
        FileInputStream inputStream = new FileInputStream(file);
        XWPFDocument document = new XWPFDocument(inputStream);

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

        HashMap<String, Object> map = new HashMap<>();
        map.put("${table}", "");
        map.put("${name}", "张三");
        processParagraphs(document.getParagraphs(), 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 paragraphList 段落列表
     * @param param 需要替换的变量及变量值
     * @param doc 需要替换的DOC
     */
    public static void processParagraphs(List<XWPFParagraph> paragraphList, HashMap<String, Object> param, XWPFDocument doc) {
        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) {
                        boolean isSetText = false;
                        for (Map.Entry<String, Object> entry : param.entrySet()) {
                            String key = entry.getKey();
                            if (text.indexOf(key) != -1) {
                                isSetText = true;
                                Object value = entry.getValue();
                                if (value instanceof String) {// 文本替换
                                    text = text.replace(key, value.toString());
                                }
                            }
                        }
                        if (isSetText) {
                            run.setText(text, 0);
                        }
                    }
                }
            }
        }
    }

    /**
     * 在定位的位置插入表格
     */
    public static void insertTab(String key, XWPFDocument doc2) {
        List<XWPFParagraph> paragraphList = doc2.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 = doc2.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 = list.size();
                            int cols = list.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(list.get(i).get(j));
                                    }else {
                                        XWPFTableCell cell = tableRow.getCell(j);
                                        if (cell == null) {
                                            tableRow.addNewTableCell().setText(list.get(i).get(j));
                                        }else {
                                            cell.setText(list.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);
                        }
                    }
                }
            }
        }
    }

}