package com.ruoyi.common.easyExcel;

import cn.hutool.core.util.NumberUtil;
import com.alibaba.excel.metadata.Head;
import com.alibaba.excel.metadata.data.WriteCellData;
import com.alibaba.excel.write.handler.CellWriteHandler;
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
import com.alibaba.excel.write.metadata.holder.WriteTableHolder;
import com.alibaba.excel.write.metadata.style.WriteCellStyle;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFClientAnchor;
import org.apache.poi.xssf.usermodel.XSSFRichTextString;

import java.lang.reflect.Field;
import java.util.List;
import java.util.Map;

public class CommentWriteHandler<T> implements CellWriteHandler {

    private final List<T> dataList;
    private final Map<String, String> warnToTargetFieldMap;
    private final Map<String, String> warnMessages;

    public CommentWriteHandler(List<T> dataList,
                               Map<String, String> warnToTargetFieldMap,
                               Map<String, String> warnMessages) {
        this.dataList = dataList;
        this.warnToTargetFieldMap = warnToTargetFieldMap;
        this.warnMessages = warnMessages;
    }

    @Override
    public void afterCellDispose(WriteSheetHolder writeSheetHolder,
                                 WriteTableHolder writeTableHolder,
                                 List<WriteCellData<?>> cellDataList,
                                 Cell cell,
                                 Head head,
                                 Integer relativeRowIndex,
                                 Boolean isHead) {
        if (isHead) return;

        String fieldName = head.getFieldName();
        T rowData = dataList.get(relativeRowIndex);

        for (Map.Entry<String, String> entry : warnToTargetFieldMap.entrySet()) {
            String warnField = entry.getKey();
            String targetField = entry.getValue();

            if (!fieldName.equals(targetField)) continue;

            try {
                Field warn = rowData.getClass().getDeclaredField(warnField);
                warn.setAccessible(true);
                Object warnValue = warn.get(rowData);
                if (NumberUtil.parseInt(warnValue.toString()) == 1) {

                    // ✅ 设置背景色通过 WriteCellData 实现
                    if (cellDataList != null && !cellDataList.isEmpty()) {
                        WriteCellStyle style = new WriteCellStyle();
                        style.setFillForegroundColor(IndexedColors.RED.getIndex());
                        style.setFillPatternType(FillPatternType.SOLID_FOREGROUND);
                        cellDataList.get(0).setWriteCellStyle(style);
                    }

                    // 获取现有的批注
                    Comment existingComment = cell.getCellComment();
                    String existingText = "";
                    if (existingComment != null) {
                        // 获取已有批注的内容
                        existingText = existingComment.getString().getString();
                    }

                    // 追加新的批注内容
                    String newCommentText = warnMessages.getOrDefault(warnField, "数据异常，请核查");
                    String updatedCommentText = existingText + "\n" + newCommentText;

                    // 如果已有批注，则更新批注内容；如果没有，则创建一个新的批注
                    if (existingComment != null) {
                        existingComment.setString(new XSSFRichTextString(updatedCommentText));
                    } else {
                        // 创建新的批注并设置内容
                        Drawing<?> drawing = cell.getSheet().createDrawingPatriarch();
                        Comment comment = drawing.createCellComment(new XSSFClientAnchor(
                                0, 0, 0, 0,
                                cell.getColumnIndex(), cell.getRowIndex(),
                                cell.getColumnIndex() + 3, cell.getRowIndex() + 2
                        ));
                        comment.setString(new XSSFRichTextString(updatedCommentText));
                        comment.setAuthor("系统提示");
                        cell.setCellComment(comment);
                    }
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}