package org.dromara.eims.listener;

import cn.hutool.core.util.ObjectUtil;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.exception.ExcelDataConvertException;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.dromara.common.core.service.DictService;
import org.dromara.common.core.utils.SpringUtils;
import org.dromara.common.excel.core.ExcelListener;
import org.dromara.common.excel.core.ExcelResult;
import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.eims.domain.vo.EimsEquVo;
import org.dromara.eims.domain.vo.InspectCheckItemVo;
import org.dromara.eims.service.IEimsEquService;

import java.util.ArrayList;
import java.util.List;

/**
 * 点检项目导入监听器
 *
 * @author zhuguifei
 */
@Data
@Slf4j
public class InspectCheckItemImportListener extends AnalysisEventListener<InspectCheckItemVo> implements ExcelListener<InspectCheckItemVo> {

    private final IEimsEquService equService;
    private final DictService dictService;
    private final Long operUserId;
    private final Boolean isUpdateSupport;

    /**
     * 成功条数
     */
    private Integer successCount = 0;

    /**
     * 失败条数
     */
    private Integer failureCount = 0;

    /**
     * 导入成功数据列表
     */
    private List<InspectCheckItemVo> successList = new ArrayList<>();

    /**
     * 导入失败数据列表
     */
    private List<InspectCheckItemVo> failureList = new ArrayList<>();

    private final StringBuilder successMsg = new StringBuilder();
    private final StringBuilder failureMsg = new StringBuilder();

    public InspectCheckItemImportListener(Boolean isUpdateSupport) {
        this.equService = SpringUtils.getBean(IEimsEquService.class);
        this.dictService = SpringUtils.getBean(DictService.class);
        this.isUpdateSupport = isUpdateSupport;
        this.operUserId = LoginHelper.getUserId();
    }

    @Override
    public void invoke(InspectCheckItemVo data, AnalysisContext context) {
        try {
            // 数据校验
            if (!validateData(data)) {
                failureList.add(data);
                failureCount++;
                failureMsg.append("<br/>").append(failureCount).append("、点检项 ")
                    .append(data.getItemName()).append(" 数据校验失败");
                return;
            }


            successList.add(data);
            successCount++;
            successMsg.append("<br/>").append(successCount).append("、点检项 ")
                .append(data.getItemName()).append(" 导入成功");

        } catch (Exception e) {
            failureList.add(data);
            failureCount++;
            String msg = "<br/>" + failureCount + "、点检项 " + data.getItemName() + " 导入失败：";
            failureMsg.append(msg).append(e.getMessage());
            log.error("导入点检项目失败：", e);
        }
    }

    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        log.info("点检项目导入完成，成功：{}条，失败：{}条", successCount, failureCount);
    }

    @Override
    public void onException(Exception exception, AnalysisContext context) throws Exception {
        log.error("导入异常", exception);
        if (exception instanceof ExcelDataConvertException) {
            ExcelDataConvertException e = (ExcelDataConvertException) exception;
            failureMsg.append("<br/>").append("第").append(e.getRowIndex()).append("行，第").append(e.getColumnIndex())
                .append("列解析异常：").append(e.getCause().getMessage());
        } else {
            failureMsg.append(exception.getMessage());
        }
    }

    @Override
    public ExcelResult<InspectCheckItemVo> getExcelResult() {
        return new ExcelResult<InspectCheckItemVo>() {
            @Override
            public List<InspectCheckItemVo> getList() {
                return successList;
            }

            @Override
            public List<String> getErrorList() {
                return List.of();
            }

            @Override
            public String getAnalysis() {
                if (successCount > 0) {
                    successMsg.insert(0, "恭喜您，共 " + successCount + " 条数据导入成功！");
                }
                if (failureCount > 0) {
                    failureMsg.insert(0, "很抱歉，共 " + failureCount + " 条数据导入失败！");
                }
                return successMsg + failureMsg.toString();
            }
        };
    }

    /**
     * 校验数据
     *
     * @param data 待校验数据
     * @return 校验结果
     */
    private boolean validateData(InspectCheckItemVo data) {
        if (data.getItemName() == null || data.getItemName().trim().isEmpty()) {
            return false;
        }
        return true;
    }
}