package com.yuanchu.mom.service.impl;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.config.Configure;
import com.deepoove.poi.config.ConfigureBuilder;
import com.deepoove.poi.data.Pictures;
import com.yuanchu.mom.dto.PersonSupervisionControlSheetExportDto;
import com.yuanchu.mom.exception.ErrorException;
import com.yuanchu.mom.mapper.UserMapper;
import com.yuanchu.mom.pojo.PersonSupervisionControlSheet;
import com.yuanchu.mom.mapper.PersonSupervisionControlSheetMapper;
import com.yuanchu.mom.service.PersonSupervisionControlSheetService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.List;

/**
 * <p>
 * 监督记录 - 控制单 服务实现类
 * </p>
 *
 * @author
 * @since 2024-10-10 05:18:57
 */
@Service
public class PersonSupervisionControlSheetServiceImpl extends ServiceImpl<PersonSupervisionControlSheetMapper, PersonSupervisionControlSheet> implements PersonSupervisionControlSheetService {

    @Resource
    private UserMapper userMapper;

    @Value("${file.path}")
    private String imgUrl;

    /**
     * 导出人员监督记录纠正控制单
     * @param supervisionRecordId
     * @param response
     */
    @Override
    public void exportSupervisionControlSheet(Integer supervisionRecordId, HttpServletResponse response) {
        PersonSupervisionControlSheetExportDto controlSheetExportDto = baseMapper.selectSupervisionControl(supervisionRecordId);

        // 判断不符合工作发现途径是否为空
        if (StringUtils.isNotBlank(controlSheetExportDto.getDiscoveryApproach())) {
            List<String> splitList = StrUtil.split(controlSheetExportDto.getDiscoveryApproach(), ',');
            for (String s : splitList) {
                switch (s) {
                    case "0":
                        controlSheetExportDto.setDiscoveryApproach0("☑");
                        break;
                    case "1":
                        controlSheetExportDto.setDiscoveryApproach1("☑");
                        break;
                    case "2":
                        controlSheetExportDto.setDiscoveryApproach2("☑");
                        break;
                    case "3":
                        controlSheetExportDto.setDiscoveryApproach3("☑");
                        break;
                    case "4":
                        controlSheetExportDto.setDiscoveryApproach4("☑");
                        break;
                    case "5":
                        controlSheetExportDto.setDiscoveryApproach5("☑");
                        break;
                    case "6":
                        controlSheetExportDto.setDiscoveryApproach6("☑");
                        break;
                    case "7":
                        controlSheetExportDto.setDiscoveryApproach7("☑");
                        break;
                }
            }
        }
        // 发现部门
        String discovererUrl = null;
        if (controlSheetExportDto.getDepartmentHeadId() != null) {
            discovererUrl = userMapper.selectById(controlSheetExportDto.getDepartmentHeadId()).getSignatureUrl();
            if (StringUtils.isBlank(discovererUrl)) {
                throw new ErrorException("找不到发现部门人的签名");
            }
        }

        // 处理措施负责人
        String responsibleUrl = null;
        if (controlSheetExportDto.getResponsibleDepartmentPersonId() != null) {
            responsibleUrl = userMapper.selectById(controlSheetExportDto.getResponsibleDepartmentPersonId()).getSignatureUrl();
            if (StringUtils.isBlank(responsibleUrl)) {
                throw new ErrorException("找不到处理措施负责人的签名");
            }
        }

        // 技术负责人
        String correctiveUrl = null;
        if (controlSheetExportDto.getResponsibleDepartmentPersonId() != null) {
            correctiveUrl = userMapper.selectById(controlSheetExportDto.getResponsibleDepartmentPersonId()).getSignatureUrl();
            if (StringUtils.isBlank(correctiveUrl)) {
                throw new ErrorException("找不到技术负责人的签名");
            }
        }

        // 质量负责人
        String qualityUrl = null;
        if (controlSheetExportDto.getQualitySupervisorId() != null) {
            qualityUrl = userMapper.selectById(controlSheetExportDto.getQualitySupervisorId()).getSignatureUrl();
            if (StringUtils.isBlank(qualityUrl)) {
                throw new ErrorException("找不到质量负责人的签名");
            }
        }

        // 获取路径
        InputStream inputStream = this.getClass().getResourceAsStream("/static/supervision-control-sheet.docx");
        ConfigureBuilder builder = Configure.builder();
        builder.useSpringEL(true);
        String finalDiscovererUrl = discovererUrl;
        String finalResponsibleUrl = responsibleUrl;
        String finalCorrectiveUrl = correctiveUrl;
        String finalQualityUrl = qualityUrl;
        XWPFTemplate template = XWPFTemplate.compile(inputStream, builder.build()).render(
                new HashMap<String, Object>() {{
                    put("control", controlSheetExportDto);
                    put("discovererUrl", StringUtils.isNotBlank(finalDiscovererUrl) ? Pictures.ofLocal(imgUrl + "/" + finalDiscovererUrl).create() : null);
                    put("responsibleUrl", StringUtils.isNotBlank(finalResponsibleUrl) ? Pictures.ofLocal(imgUrl + "/" + finalResponsibleUrl).create() : null);
                    put("correctiveUrl", StringUtils.isNotBlank(finalCorrectiveUrl) ? Pictures.ofLocal(imgUrl + "/" + finalCorrectiveUrl).create() : null);
                    put("qualityUrl", StringUtils.isNotBlank(finalQualityUrl) ? Pictures.ofLocal(imgUrl + "/" + finalQualityUrl).create() : null);
                }});

        try {
            response.setContentType("application/msword");
            String fileName = URLEncoder.encode(
                    "人员监督记录控制单", "UTF-8");
            response.setHeader("Content-disposition",
                    "attachment;filename=" + fileName + ".docx");
            OutputStream os = response.getOutputStream();
            template.write(os);
            os.flush();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("导出失败");
        }


    }
}