package com.yuanchu.mom.utils.easyexcel;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import lombok.Getter;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;

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

/**
 * excel数据导入数据处理
 *
 * @param <T>
 * @author Zou, Yu
 */
@Slf4j
@Getter
@Setter
public class ExcelListener<T> extends AnalysisEventListener<T> {

    private List<T> dataList = new ArrayList<>();

    @Override
    public void invoke(T classType, AnalysisContext analysisContext) {
        dataList.add(classType);
    }

    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
    }

    @Override
    public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {
        String templateHead = "";

        List<String> headList = new ArrayList<>();
        for (Integer i : headMap.keySet()) {
            headList.add(headMap.get(i));
        }
        String importHead = headMap.values().toString();
        log.info("importHead:{}", importHead);
        String templateName = context.getCurrentSheet().getClazz().getSimpleName();
        log.info("templateName:", templateName);
//        if ("CostModel".equals(templateName)) {
//            templateHead = templateHeadCost;
//        } else if ("StaffInfoTemplate".equals(templateName)) {
//            templateHead = templateHeadStaff;
//        } else if ("OrganizationTemplate".equals(templateName)) {
//            templateHead = templateHeadOrganization;
//        }else if ("StaffDictionaryTemplate".equals(templateName)) {
//            templateHead = templateHeadDictionary;
//        }else if ("StaffCtfTemplate".equals(templateName)){
//            templateHead = templateHeadCtf;
//        }else if ("IndustryLineTemplate".equals(templateName)){
//            templateHead = templateHeadIndustryLine;
//        }else if ("SubServiceLineTemplate".equals(templateName)){
//            templateHead = templateHeadSubServiceLine;
//        }else if ("StaffHealthTemplate".equals(templateName)){
//            templateHead = templateHeadHealth;
//        }else if ("AccountDataTemplate".equals(templateName)){
//            templateHead = templateAccountData;
//        }else if ("PlInfoTemplate".equals(templateName)){
//            templateHead = templatePlInfo;
//        }else if ("CodeDataTemplate".equals(templateName)){
//            templateHead = templateCodeData;
//        }else if ("CodeTbdDataTemplate".equals(templateName)){
//            templateHead = templateCodeTbdData;
//        }

        if (!templateHead.equals(importHead)) {
            throw new RuntimeException(ResponseResultEnum.EXCEL_MODEL_ERROR.getMessage());
        }

    }

}