package com.estate.expert.admin.stat.report.render;

import com.alibaba.fastjson.JSONObject;
import com.deepoove.poi.policy.DynamicTableRenderPolicy;
import com.deepoove.poi.util.TableTools;
import org.apache.poi.xwpf.usermodel.ParagraphAlignment;
import org.apache.poi.xwpf.usermodel.XWPFTable;
import org.apache.poi.xwpf.usermodel.XWPFTableCell;
import org.apache.poi.xwpf.usermodel.XWPFTableRow;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.STBorder;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.STTblWidth;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.STVerticalJc;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;

import java.math.BigInteger;
import java.util.List;

/**
 * 设备分组表格渲染策略：同一设备类型内合并类型和数量，设备类型变更时重新分组。
 */
public class DeviceGroupTablePolicy extends DynamicTableRenderPolicy {

    private static final int TYPE_COL = 0;
    private static final int COUNT_COL = 1;
    private static final int NUMBER_COL = 2;

    @Override
    public void render(XWPFTable table, Object data) {
        var groups = (List<Object>) data;
        int templateRowIndex = findTemplateRowIndex(table);
        if (templateRowIndex < 0) {
            return;
        }

        table.removeRow(templateRowIndex);
        if (CollectionUtils.isEmpty(groups)) {
            return;
        }

        int insertIndex = templateRowIndex;
        int groupStart = templateRowIndex;
        int groupSize = 0;

        for (Object item : groups) {
            if (!(item instanceof JSONObject group)) {
                continue;
            }

            boolean firstRowOfGroup = StringUtils.hasText(group.getString("设备类型"));
            XWPFTableRow row = table.insertNewTableRow(insertIndex++);
            renderRow(row, group, firstRowOfGroup);

            if (firstRowOfGroup) {
                mergeGroup(table, groupStart, groupSize);
                groupStart = insertIndex - 1;
                groupSize = 1;
            } else {
                groupSize++;
            }
        }

        mergeGroup(table, groupStart, groupSize);
    }

    private int findTemplateRowIndex(XWPFTable table) {
        for (int i = 0; i < table.getRows().size(); i++) {
            if (rowText(table.getRow(i)).contains("[设备编号]")) {
                return i;
            }
        }
        return -1;
    }

    private String rowText(XWPFTableRow row) {
        if (row == null) {
            return "";
        }
        return row.getTableCells().stream()
                .map(XWPFTableCell::getText)
                .reduce("", String::concat);
    }

    private void renderRow(XWPFTableRow row, JSONObject group, boolean firstRowOfGroup) {
        renderCell(row, firstRowOfGroup ? group.getString("设备类型") : "", 2972);
        renderCell(row, firstRowOfGroup ? group.getString("设备数量") : "", 1531);
        renderCell(row, group.getString("设备编号"), 4706);
    }

    private void renderCell(XWPFTableRow row, String text, int width) {
        XWPFTableCell cell = row.createCell();
        var tcPr = cell.getCTTc().isSetTcPr() ? cell.getCTTc().getTcPr() : cell.getCTTc().addNewTcPr();
        tcPr.addNewTcW().setW(BigInteger.valueOf(width));
        tcPr.getTcW().setType(STTblWidth.DXA);
        tcPr.addNewGridSpan().setVal(BigInteger.valueOf(2));
        tcPr.addNewVAlign().setVal(STVerticalJc.CENTER);

        var borders = tcPr.addNewTcBorders();
        borders.addNewTop().setVal(STBorder.SINGLE);
        borders.addNewBottom().setVal(STBorder.SINGLE);
        borders.addNewLeft().setVal(STBorder.SINGLE);
        borders.addNewRight().setVal(STBorder.SINGLE);
        borders.getTop().setColor("000000");
        borders.getBottom().setColor("000000");
        borders.getLeft().setColor("000000");
        borders.getRight().setColor("000000");
        borders.getTop().setSz(BigInteger.valueOf(4));
        borders.getBottom().setSz(BigInteger.valueOf(4));
        borders.getLeft().setSz(BigInteger.valueOf(4));
        borders.getRight().setSz(BigInteger.valueOf(4));

        var paragraph = cell.getParagraphs().get(0);
        paragraph.setAlignment(ParagraphAlignment.CENTER);
        var run = paragraph.createRun();
        run.setFontFamily("黑体");
        run.setFontSize(12);
        run.setText(text == null ? "" : text);
        cell.setVerticalAlignment(XWPFTableCell.XWPFVertAlign.CENTER);
    }

    private void mergeGroup(XWPFTable table, int groupStart, int groupSize) {
        if (groupSize <= 1) {
            return;
        }
        TableTools.mergeCellsVertically(table, TYPE_COL, groupStart, groupStart + groupSize - 1);
        TableTools.mergeCellsVertically(table, COUNT_COL, groupStart, groupStart + groupSize - 1);
    }
}