package com.yuanchu.mom.service.impl;

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.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.config.Configure;
import com.yuanchu.mom.dto.InternalWastesDto;
import com.yuanchu.mom.pojo.InternalWastes;
import com.yuanchu.mom.mapper.InternalWastesMapper;
import com.yuanchu.mom.pojo.InternalWastesDetail;
import com.yuanchu.mom.service.InternalWastesDetailService;
import com.yuanchu.mom.service.InternalWastesService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yuanchu.mom.utils.HackLoopTableRenderPolicy;
import com.yuanchu.mom.utils.DateImageUtil;
import com.yuanchu.mom.utils.QueryWrappers;
import lombok.AllArgsConstructor;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

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

/**
 * <p>
 * 安全内务三废登记 服务实现类
 * </p>
 *
 * @author
 * @since 2024-11-19 06:39:27
 */
@Service
@AllArgsConstructor
@Transactional(rollbackFor = Exception.class)
public class InternalWastesServiceImpl extends ServiceImpl<InternalWastesMapper, InternalWastes> implements InternalWastesService {

    private InternalWastesDetailService internalWastesDetailService;

    /**
     * 安全内务三废登记分页查询
     * @param page
     * @param internalWastes
     * @return
     */
    @Override
    public IPage<InternalWastesDto> pageInternalWastes(Page page, InternalWastes internalWastes) {
        return baseMapper.pageInternalWastes(page, QueryWrappers.queryWrappers(internalWastes));
    }

    /**
     * 安全内务三废登记新增
     * @param internalWastes
     * @return
     */
    @Override
    public boolean addInternalWastes(InternalWastesDto internalWastes) {

        baseMapper.insert(internalWastes);
        // 新增详情
        for (InternalWastesDetail internalWastesDetail : internalWastes.getWastesDetailList()) {
            internalWastesDetail.setWastesId(internalWastes.getWastesId());
        }
        internalWastesDetailService.saveBatch(internalWastes.getWastesDetailList());
        return true;
    }

    /**
     * 安全内务三废登记修改
     * @param internalWastes
     * @return
     */
    @Override
    public boolean updateInternalWastes(InternalWastesDto internalWastes) {
        baseMapper.updateById(internalWastes);

        // 删除之前的详情
        internalWastesDetailService.remove(Wrappers.<InternalWastesDetail>lambdaQuery()
                .eq(InternalWastesDetail::getWastesId, internalWastes.getWastesId()));

        // 新增详情
        for (InternalWastesDetail internalWastesDetail : internalWastes.getWastesDetailList()) {
            internalWastesDetail.setWastesId(internalWastes.getWastesId());
        }
        internalWastesDetailService.saveBatch(internalWastes.getWastesDetailList());

        return true;
    }

    /**
     * 安全内务三废登记删除
     * @param WastesId
     * @return
     */
    @Override
    public boolean delInternalWastes(Integer WastesId) {
        internalWastesDetailService.remove(Wrappers.<InternalWastesDetail>lambdaQuery()
                .eq(InternalWastesDetail::getWastesId, WastesId));
        baseMapper.deleteById(WastesId);
        return true;
    }

    /**
     * 安全内务三废登记查看详情
     * @param WastesId
     * @return
     */
    @Override
    public InternalWastesDto getInternalWastesOne(Integer WastesId) {
        InternalWastes internalWastes = baseMapper.selectById(WastesId);
        InternalWastesDto internalWastesDto = new InternalWastesDto();
        BeanUtils.copyProperties(internalWastes, internalWastesDto);

        // 查询详细信息
        internalWastesDto.setWastesDetailList(internalWastesDetailService.list(Wrappers.<InternalWastesDetail>lambdaQuery()
                .eq(InternalWastesDetail::getWastesId, WastesId)));
        return internalWastesDto;
    }

    /**
     * 导出三废处理
     * @param wastesId
     * @param response
     */
    @Override
    public void exportInternalWastes(Integer wastesId, HttpServletResponse response) {
        InternalWastes internalWastes = baseMapper.selectById(wastesId);

        List<InternalWastesDetail> wastesDetailList = internalWastesDetailService.list(Wrappers.<InternalWastesDetail>lambdaQuery()
                .eq(InternalWastesDetail::getWastesId, wastesId));

        int index = 1;
        for (InternalWastesDetail detail : wastesDetailList) {
            detail.setIndex(index);
            index++;
        }


        // 获取路径
        InputStream inputStream = this.getClass().getResourceAsStream("/static/internal-wastes.docx");
        Configure configure = Configure.builder()
                .bind("wastesDetailList", new HackLoopTableRenderPolicy())
                .build();
        XWPFTemplate template = XWPFTemplate.compile(inputStream, configure).render(
                new HashMap<String, Object>() {{
                    put("remark", internalWastes.getRemark());
                    put("wastesDetailList", wastesDetailList);
                }});

        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("导出失败");
        }

    }
}