package com.yuanchu.mom.service.impl;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.support.ExcelTypeEnum;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yuanchu.mom.mapper.MbomModelMapper;
import com.yuanchu.mom.pojo.MbomModel;
import com.yuanchu.mom.pojo.dto.MbomModelDto;
import com.yuanchu.mom.pojo.dto.MbomModelDto2;
import com.yuanchu.mom.service.MbomModelService;
import com.yuanchu.mom.service.TechnicalModelService;
import com.yuanchu.mom.utils.StringUtils;
import com.yuanchu.mom.utils.easyexcel.CustemHandler;
import com.yuanchu.mom.utils.easyexcel.EasyExcelUtils;
import com.yuanchu.mom.utils.easyexcel.ExcelMergeStrategy;
import com.yuanchu.mom.utils.easyexcel.converter.LocalDateConverter;
import com.yuanchu.mom.utils.easyexcel.template.ExcelMBomTemplate;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.collections4.map.HashedMap;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 物料清单维护表(MbomModel)表服务实现类
 *
 * @author zss
 * @since 2023-08-30 09:17:26
 */
@Service
public class MbomModelServiceImpl extends ServiceImpl<MbomModelMapper, MbomModel> implements MbomModelService {

    @Resource
    MbomModelMapper mbomModelMapper;

    @Autowired
    private TechnicalModelService technicalModelService;


    //查询物料清单维护列表-->右边展示该工艺所需要的原材料
    @Override
    public List<Map<String, Object>> selectAllMbom(Integer id) {
        return mbomModelMapper.selectAllMbom(id);
    }

    //新增物料清单维护表
    @Override
    public void addMbom(MbomModelDto mbomModelDto) {
        List<MbomModelDto2> mbomModelDto2List = mbomModelDto.getMbomModelDto2List();
        List<MbomModel> mbomModelList = mbomModelDto2List.stream().map(mbomModelDto2 -> {
            MbomModel mbomModel = new MbomModel();
            mbomModel.setTechTemId(mbomModelDto.getTechTemId());
            BeanUtils.copyProperties(mbomModelDto2, mbomModel);
            return mbomModel;
        }).collect(Collectors.toList());
        //批量添加
        saveBatch(mbomModelList);
    }

    //删除
    @Override
    public void delMbomById(Integer id) {
        MbomModel mbomModel = new MbomModel();
        mbomModel.setId(id);
        mbomModel.setState(0);
        mbomModelMapper.updateById(mbomModel);
    }

    //批量删除
    @Override
    public void delAllMbom(String ids) {
        mbomModelMapper.delAllMbom(ids);
    }

    @Override
    public void downloadTemplate(Map<String, Object> map,HttpServletResponse response) {
        try{
            List<String> typeList = getTypeList(map);
            ExcelWriter excelWriter = EasyExcel.write(EasyExcelUtils.getResponse(response,"物料清单导入模板").getOutputStream())
                    .excelType(ExcelTypeEnum.XLS)
                    .build();
            // 导出多个sheet页
            for (int i = 0; i < typeList.size(); i++) {
                WriteSheet sheet = EasyExcel.writerSheet(i, typeList.get(i))
                        .head(ExcelMBomTemplate.class)
                        .registerConverter(new LocalDateConverter())
                        .registerWriteHandler(new CustemHandler())
                        .registerWriteHandler(EasyExcelUtils.getStyleStrategy())
                        .build();
                excelWriter.write(null, sheet);
            }
            // 关闭 ExcelWriter，完成导出
            excelWriter.finish();
        }catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                response.getOutputStream().close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    @Override
    public void exportMBom(Map<String, Object> map,HttpServletResponse response) {
        try{
            //类型列表
            List<String> typeList = getTypeList(map);
            //sheet数据
            Map<String,List<ExcelMBomTemplate>> exportListByType = new HashedMap<>();
            typeList.forEach((type)->{
                List<Map<String,Object>> techTemList =
                        technicalModelService.selectAllTechTem(type, "");
                List<ExcelMBomTemplate> list = new ArrayList<>();
                techTemList.forEach(tech -> {
                    String father = "";
                    if(StringUtils.isNotEmpty(tech.get("name").toString())){
                        father = tech.get("name").toString();
                    }
                    if(StringUtils.isNotEmpty(tech.get("children").toString())){
                        List<Map<String,Object>> childrenList = (List<Map<String, Object>>) tech.get("children");
                        String finalFather = father;
                        childrenList.forEach(c -> {
                            String name = c.get("name").toString();
                            Integer id = Integer.parseInt(c.get("id").toString());
                            List<Map<String, Object>> mbom = mbomModelMapper.selectAllMbom(id);
                            if(CollectionUtils.isNotEmpty(mbom)){
                                mbom.forEach(m->{
                                    ExcelMBomTemplate temp = new ExcelMBomTemplate();
                                    temp.setFather(finalFather);
                                    temp.setName(name);
                                    temp.setMaterialName(m.get("name").toString());
                                    temp.setUnit(m.get("unit").toString());
                                    list.add(temp);
                                });
                            }
                        });
                    }
                });
                exportListByType.put(type,list);
            });
            ExcelWriter excelWriter = EasyExcel.write(
                        EasyExcelUtils.getResponse(response, "物料清单导出").getOutputStream()
                    )
                    .excelType(ExcelTypeEnum.XLS)
                    .build();
            // 导出多个sheet页
            for (int i = 0; i < typeList.size(); i++) {
                String type = typeList.get(i);
                List<String> fathers = exportListByType.get(type).stream()
                        .map(ExcelMBomTemplate::getFather).collect(Collectors.toList());
                List<String> names = exportListByType.get(type).stream()
                        .map(ExcelMBomTemplate::getName).collect(Collectors.toList());
                WriteSheet sheet = EasyExcel.writerSheet(i, typeList.get(i))
                        .head(ExcelMBomTemplate.class)
                        .registerConverter(new LocalDateConverter())
                        .registerWriteHandler(new CustemHandler())
                        .registerWriteHandler(EasyExcelUtils.getStyleStrategy())
                        .registerWriteHandler(new ExcelMergeStrategy(fathers,0))
                        .registerWriteHandler(new ExcelMergeStrategy(names,1))
                        .build();
                excelWriter.write(exportListByType.get(type), sheet);
            }
            // 关闭 ExcelWriter，完成导出
            excelWriter.finish();
        }catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                response.getOutputStream().close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

    }

    /**
     * 获取类型列表
     * @param map
     * @return
     */
    private List<String> getTypeList(Map<String, Object> map){
        if(!map.isEmpty()){
            String arrStr = String.valueOf(map.get("typeList"));
            String[] strings = arrStr.split(",");
            return Arrays.asList(strings);
        }
        return null;
    }

}