package com.ots.project.monitor.job.task;

import com.ots.common.enums.LangTypeEnum;
import com.ots.common.enums.ReportTypeEnum;
import com.ots.common.enums.SenderCode;
import com.ots.common.utils.StringUtils;
import com.ots.common.utils.poi.WordUtil;
import com.ots.framework.config.EssConfig;
import com.ots.project.exam.domain.TExamReport;
import com.ots.project.exam.domain.TReportTemplate;
import com.ots.project.exam.mapper.TExamReportMapper;
import com.ots.project.exam.mapper.TReportTemplateMapper;
import com.ots.project.exam.service.IEntTestSendtaskMailService;
import com.ots.project.exam.service.ITExamReportService;
import com.ots.project.exam.service.ITReportTemplateService;
import com.ots.project.exam.service.impl.EntTestMemberServiceImpl;
import com.ots.project.exam.service.impl.EntTestSendtaskServiceImpl;
import com.ots.project.tool.email.EmailMessage;
import com.ots.project.tool.exam.ExamUtil;
import com.ots.project.tool.report.MAQ.MAQReport;
import com.ots.project.tool.report.ReportResultData;
import com.ots.project.tool.sms.ShortMessage;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

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

/**
 * 邮件短信发送任务
 */
@Component("SenderEmailReportTask")
@Slf4j
public class SenderEmailReportTask {

    private static final String SMS_ZJCM = "SMS_ZJCM";
    private static final String SMS_ZJCM_INTERNATIONAL = "SMS_ZJCM_INTERNATIONAL";
    private static final String EMAIL_126 = "EMAIL_126";
    @Autowired
    EntTestSendtaskServiceImpl entTestSendtaskService;
    @Autowired
    EntTestMemberServiceImpl entTestMemberService;
    @Autowired
    ShortMessage shortMessage;
    @Autowired
    EmailMessage emailMessage;
    @Autowired
    IEntTestSendtaskMailService entTestSendtaskMailService;
    @Autowired
    TExamReportMapper tExamReportMapper;
    @Autowired
    private ITExamReportService tExamReportService;
    @Autowired
    private TReportTemplateMapper tReportTemplateMapper;
    @Autowired
    private ITReportTemplateService itReportTemplateService;

    public void senderTask() {
        // 查找report表得出要发送报告的用户
        TExamReport tExamReportTemp = new TExamReport();
        tExamReportTemp.setFinish(1);
        tExamReportTemp.setSendTesterStatus(0);
        List<TExamReport> tExamReportList = tExamReportMapper.selectTExamReportList(tExamReportTemp);
        if (ExamUtil.isListEmpty(tExamReportList)) {
            return;
        }

        // textMapList的
        List<Map<String, Object>> textMapList = new ArrayList<>();
        List<String> deleteFileStrList = new ArrayList<>();
        // 语言类型
        Map<String, LangTypeEnum> langTypeEnumMap = getLangTypeEnumMap();
        // 报告类型
        Map<String, ReportTypeEnum> reportTypeEnumMap = getStringReportTypeEnumMap();
        // 查出全部的报告，减少消耗的性能
        Map<Long, TReportTemplate> tReportTemplateMap = getAllTReportTemplateMap();

        // 遍历每个report表
        // 给测试者发送邮件
        List<String> allReportList = new ArrayList<>();

        // hr 邮箱
        Map<String, List<String>> hrEmail = new HashMap<>();

        // 循环每个个报告记录
        for (TExamReport tExamReport : tExamReportList) {
            String[] reportIds = tExamReport.getReportTemplateId().split(",");

            // 遍历出模版生成文件
            List<String> reprotList =  saveReport(textMapList, deleteFileStrList, langTypeEnumMap, reportTypeEnumMap, tReportTemplateMap, allReportList, tExamReport, reportIds);

            // 调用发送程序发给用户
            if (emailMessage.sendEmailwithAttachment(tExamReport.getMemberEmail(), "测试报告", "您已完成测试", reprotList.stream().toArray(String[]::new)).isResult()) {
                tExamReport.setSendTesterStatus(SenderCode.success.getCode());
            } else { // 发送成功，发送失败处理
                tExamReport.setSendTesterStatus(SenderCode.fail.getCode());
            }

            // 发送给Hr
            if (emailMessage.sendEmailwithAttachment(tExamReport.getTestEmail(), "测试报告", "您已完成测试", reprotList.stream().toArray(String[]::new)).isResult()) {
                tExamReport.setSendHrStatus(SenderCode.success.getCode());
            } else { // 发送成功，发送失败处理
                tExamReport.setSendHrStatus(SenderCode.fail.getCode());
            }

            tExamReportService.updateTExamReportStatus(tExamReport);
        }
    }

    private void setHrFiles(Map<String, List<String>> hrEmail, TExamReport tExamReport, List<String> reprotList) {
        // 保留发给Hr的邮箱
        List<String> hrEmailList =  hrEmail.get(tExamReport.getTestEmail());
        if (ExamUtil.isListEmpty(hrEmailList)) {
            List<String> hrFileList = new ArrayList<>();
            hrFileList.addAll(reprotList);
            hrEmail.put(tExamReport.getTestEmail(), hrFileList);
        } else {
            hrEmailList.addAll(reprotList);
        }
    }

    private List<String> saveReport(List<Map<String, Object>> textMapList, List<String> deleteFileStrList, Map<String, LangTypeEnum> langTypeEnumMap, Map<String, ReportTypeEnum> reportTypeEnumMap, Map<Long, TReportTemplate> tReportTemplateMap, List<String> allReportList, TExamReport tExamReport, String[] reportIds) {
        List<String> sendReportList = new ArrayList<>();
        for (String reportId : reportIds) {
            // 模拟调用接口
            Map<String, String> thaiCalculationMap = new HashMap<>();
            MAQReport report = new MAQReport();
            WordUtil util = new WordUtil();
            // 导出基础版本还是详细版
            String reportType = tReportTemplateMap.get(Long.valueOf(reportId)).getReportType();// "Brief";
            String templateType = tReportTemplateMap.get(Long.valueOf(reportId)).getTemplateType();// "Brief";
            String path = "";
            Map<String, Object> textMap = new HashMap<>();
            TReportTemplate tReportTemplate = tReportTemplateMap.get(Long.valueOf(reportId));
            try {
                ReportResultData reportResultData = report.getTemplateParameters(thaiCalculationMap, reportTypeEnumMap.get(templateType), langTypeEnumMap.get(tExamReport.getLangType()));
                tExamReportService.setTextPicture(tExamReport, reportResultData, deleteFileStrList, langTypeEnumMap.get(tExamReport.getLangType()), textMap);
                path = itReportTemplateService.selectTReportTemplateFilePath(tReportTemplate);
                if (StringUtils.isEmpty(path)) {
                    continue;
                }
            } catch (Exception e) {
                e.printStackTrace();
                continue;
            }

            // 生成报告文件
            String zipFileName = util.makeReportFile(path, tReportTemplate, textMap, new HashMap<>(), deleteFileStrList);
            allReportList.add(zipFileName);
            for (String fileName : allReportList) {
                sendReportList.add(EssConfig.getProfile() + "/" + fileName);
            }
            // 清空，给下一个用户用
            textMapList.clear();
            textMap.clear();
            // 删除临时文件
            util.deleteFileByStr(deleteFileStrList);
        }

        return sendReportList;
    }

    private Map<Long, TReportTemplate> getAllTReportTemplateMap() {
        List<TReportTemplate> tReportTemplateList = tReportTemplateMapper.selectTReportTemplateList(new TReportTemplate());
        Map<Long, TReportTemplate> tReportTemplateMap = new HashMap<>();
        for (TReportTemplate tReportTemplate : tReportTemplateList) {
            tReportTemplateMap.put(tReportTemplate.getId(), tReportTemplate);
        }
        return tReportTemplateMap;
    }

    private Map<String, ReportTypeEnum> getStringReportTypeEnumMap() {
        Map<String, ReportTypeEnum> reportTypeEnumMap = new HashMap<>();
        for (ReportTypeEnum reportTypeEnumTemp : ReportTypeEnum.values()) {
            reportTypeEnumMap.put(reportTypeEnumTemp.getCode(), reportTypeEnumTemp);
        }
        return reportTypeEnumMap;
    }

    private Map<String, LangTypeEnum> getLangTypeEnumMap() {
        Map<String, LangTypeEnum> langTypeEnumMap = new HashMap<>();
        for (LangTypeEnum langTypeEnumTemp : LangTypeEnum.values()) {
            langTypeEnumMap.put(langTypeEnumTemp.getCode(), langTypeEnumTemp);
        }
        return langTypeEnumMap;
    }
}