package com.yuanchu.mom.service.impl;


import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.metadata.data.ImageData;
import com.alibaba.excel.metadata.data.WriteCellData;
import com.alibaba.excel.util.FileUtils;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yuanchu.mom.dto.ProcessMethodSearchNewDto;
import com.yuanchu.mom.mapper.ProcessMethodSearchNewMapper;
import com.yuanchu.mom.pojo.ProcessMethodSearchNew;
import com.yuanchu.mom.service.ProcessMethodSearchNewService;
import com.yuanchu.mom.utils.QueryWrappers;
import com.yuanchu.mom.utils.UserUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 标准查新
 *
 * @author zhuo
 * @since 2024-11-04
 */
@Service
public class ProcessMethodSearchNewServiceImpl extends ServiceImpl<ProcessMethodSearchNewMapper, ProcessMethodSearchNew> implements ProcessMethodSearchNewService {

    /**
     * 新增标准查新
     * @param processMethodSearchNewList
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean addMethodSearchNew(List<ProcessMethodSearchNew> processMethodSearchNewList) {
        for (ProcessMethodSearchNew processMethodSearchNew : processMethodSearchNewList) {
            processMethodSearchNew.setIsNewStandard(1);
        }
        // 修改之前所有没有变新的
        List<ProcessMethodSearchNew> processMethodSearchNews = baseMapper.selectList(Wrappers.<ProcessMethodSearchNew>lambdaQuery()
                .eq(ProcessMethodSearchNew::getIsNewStandard, 1));
        for (ProcessMethodSearchNew processMethodSearchNew : processMethodSearchNews) {
            baseMapper.update(null, Wrappers.<ProcessMethodSearchNew>lambdaUpdate()
                    .set(ProcessMethodSearchNew::getMethodName, processMethodSearchNew.getNewMethodName())
                    .set(ProcessMethodSearchNew::getStandardNo, processMethodSearchNew.getNewStandardNo())
                    .set(ProcessMethodSearchNew::getFileNo, processMethodSearchNew.getNewStandardNo())
                    .set(ProcessMethodSearchNew::getIsNewStandard, 0)
                    .set(ProcessMethodSearchNew::getNewMethodName, null)
                    .set(ProcessMethodSearchNew::getNewStandardNo, null)
                    .eq(ProcessMethodSearchNew::getMethodSearchNewId, processMethodSearchNew.getMethodSearchNewId()));
        }
        // 新增
        this.saveBatch(processMethodSearchNewList);
        return true;
    }

    /**
     * 标准查新列表
     * @param processMethodSearchNew
     * @return
     */
    @Override
    public IPage<ProcessMethodSearchNew> pageMethodSearchNew(Page page, ProcessMethodSearchNewDto processMethodSearchNew) {
        String beginDate = processMethodSearchNew.getBeginDate();
        String endDate = processMethodSearchNew.getEndDate();
        processMethodSearchNew.setBeginDate(null);
        processMethodSearchNew.setEndDate(null);
        return baseMapper.pageMethodSearchNew(page, QueryWrappers.queryWrappers(processMethodSearchNew), beginDate, endDate);
    }

    /**
     * 标准查新导出
     * @param processMethodSearchNew
     * @param response
     */
    @Override
    public void exportMethodSearchNew(ProcessMethodSearchNewDto processMethodSearchNew, HttpServletResponse response) {
        // 查询标准查新
        String beginDate = processMethodSearchNew.getBeginDate();
        String endDate = processMethodSearchNew.getEndDate();
        String writeUserId = processMethodSearchNew.getWriteUserId().toString();
        String ratifyUserId = processMethodSearchNew.getRatifyUserId().toString();
        processMethodSearchNew.setBeginDate(null);
        processMethodSearchNew.setEndDate(null);
        processMethodSearchNew.setWriteUserId(null);
        processMethodSearchNew.setRatifyUserId(null);
        List<ProcessMethodSearchNewDto> methodSearchNews = baseMapper.selectMethodSearchNew(QueryWrappers.queryWrappers(processMethodSearchNew), beginDate, endDate);

        int index = 1;
        // 格式化参数
        for (ProcessMethodSearchNewDto methodSearchNew : methodSearchNews) {
            // 是否是跟新标准
            if (methodSearchNew.getIsNewStandard().equals(1)) {
                methodSearchNew.setIsNewStandardString("是");
            } else {
                methodSearchNew.setIsNewStandardString("否");
            }

            // 备注
            if (methodSearchNew.getRemark() != null) {
                if (methodSearchNew.getRemark().equals(1)) {
                    methodSearchNew.setIsNewStandardString("替换");
                } else {
                    methodSearchNew.setIsNewStandardString("作废");
                }
            }

            if (methodSearchNew.getRemark() != null) {
                if (methodSearchNew.getRemark().equals(1)) {
                    methodSearchNew.setRemarkString("替换");
                } else {
                    methodSearchNew.setRemarkString("作废");
                }
            }

            switch (methodSearchNew.getSearchNewSource()) {
                case 0:
                    methodSearchNew.setStandardNet("√");
                    break;
                case 1:
                    methodSearchNew.setInformationOffices("√");
                    break;
                case 2:
                    methodSearchNew.setStandardBookstore("√");
                    break;
                case 3:
                    methodSearchNew.setOther("√");
                    break;
            }
            methodSearchNew.setIndex(index);
            index++;

        }
        String writeUserUrl = UserUtils.getUserSignatureUrl(Integer.valueOf(writeUserId));
        String ratifyUserUrl = UserUtils.getUserSignatureUrl(Integer.valueOf(ratifyUserId));

        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("UTF-8");
        try {
            // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
            String fileName = URLEncoder.encode("原材料检测信息导出", "UTF-8");
            response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");

            // 创建 ExcelWriter
            ServletOutputStream outputStream = response.getOutputStream();
            ExcelWriter excelWriter = EasyExcel.write(outputStream).registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).build();

            // 获取 sheet0 对象
            WriteSheet mainSheet = EasyExcel.writerSheet(0, "标准查新导出").head(ProcessMethodSearchNewDto.class).build();

            // 图片对象
            ImageData imageData = new ImageData();
            ImageData imageData1 = new ImageData();
            // 设置图片数据
            imageData.setImage(FileUtils.readFileToByteArray(new File(writeUserUrl))); // 设置编制人图片的文件路径
            imageData1.setImage(FileUtils.readFileToByteArray(new File(ratifyUserUrl))); // 设置批准人图片的文件路径

            // 设置编制人图片位置
            imageData.setRelativeFirstRowIndex(1);
            imageData.setRelativeFirstColumnIndex(2);
            imageData.setRelativeLastRowIndex(1);
            imageData.setRelativeLastColumnIndex(2);
            // 设置批准人图片位置
            imageData1.setRelativeFirstRowIndex(1);
            imageData1.setRelativeFirstColumnIndex(4);
            imageData1.setRelativeLastRowIndex(1);
            imageData1.setRelativeLastColumnIndex(4);


            List<ImageData> list = new ArrayList<>();
            // 放入列表中
            list.add(imageData);
            list.add(imageData1);

//            // 设置额外的文字（可选）
//            WriteCellData<Void> writeCellData = new WriteCellData<>();
////            imageDemoData.setWriteCellDataFile(writeCellData);
//            // 这里可以设置为 EMPTY 则代表不需要其他数据了
//            writeCellData.setType(CellDataTypeEnum.STRING);
//            writeCellData.setStringValue("额外的放一些文字");


            // 将图片数据设置到单元格中
            WriteCellData<ImageData> writeCellData = new WriteCellData<>();
            writeCellData.setImageDataList(list);

            // 创建数据列表
            List<List<WriteCellData>> dataList = new ArrayList<>();
            List<WriteCellData> rowData = new ArrayList<>();
            rowData.add(writeCellData); // 把图片放入一行
            dataList.add(rowData);


            // 向 sheet0 写入数据（传入空list这样只导出表头）
            excelWriter.write(methodSearchNews, mainSheet);
            // 写入 图片
            excelWriter.write(dataList,mainSheet);

            //关闭流
            excelWriter.finish();

        } catch (IOException e) {
            throw new RuntimeException("导出失败");
        }
    }







    /**
     * 导入标准查新
     * @param file
     * @return
     */
    @Override
    public boolean importMethodSearchNew(MultipartFile file) {

        List<ProcessMethodSearchNewDto> searchNewDtoList = new ArrayList<>();

        try {
            // excel解析
            EasyExcel.read(file.getInputStream(), ProcessMethodSearchNewDto.class, new AnalysisEventListener<ProcessMethodSearchNewDto>() {
                @Override
                public void invoke(ProcessMethodSearchNewDto searchNewDto, AnalysisContext analysisContext) {
                    searchNewDtoList.add(searchNewDto);
                }

                @Override
                public void doAfterAllAnalysed(AnalysisContext analysisContext) {

                }
            }).sheet().doRead();
            // 格式化查新对象
            List<ProcessMethodSearchNew> collect = searchNewDtoList.stream().map(searchNewDto -> {
                ProcessMethodSearchNew searchNew = new ProcessMethodSearchNew();
                BeanUtils.copyProperties(searchNewDto, searchNew);
                // 是否是更新
                if (StringUtils.isNotBlank(searchNewDto.getIsNewStandardString())) {
                    if (searchNewDto.getIsNewStandardString().equals("是")) {
                        searchNew.setIsNewStandard(1);
                    } else {
                        searchNew.setIsNewStandard(0);
                    }
                }
                // 查新记录
                if (StringUtils.isNotBlank(searchNewDto.getStandardNet())) {
                    searchNew.setSearchNewSource(0);
                } else if (StringUtils.isNotBlank(searchNewDto.getInformationOffices())) {
                    searchNew.setSearchNewSource(1);
                } else if (StringUtils.isNotBlank(searchNewDto.getIsNewStandardString())) {
                    searchNew.setSearchNewSource(2);
                } else if (StringUtils.isNotBlank(searchNewDto.getOther())) {
                    searchNew.setSearchNewSource(3);
                }

                // 备注
                if (StringUtils.isNotBlank(searchNewDto.getRemarkString())) {
                    if (searchNewDto.getRemarkString().equals("替换")) {
                        searchNew.setRemark(1);
                    } else if (searchNewDto.getRemarkString().equals("作废")) {
                        searchNew.setRemark(0);
                    }
                }

                return searchNew;
            }).collect(Collectors.toList());
            this.saveBatch(collect);

        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        return true;
    }
}