package com.yuanchu.mom.service.impl;

import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.config.Configure;
import com.yuanchu.mom.common.GetLook;
import com.yuanchu.mom.dto.DeviceTraceabilityManagementDetailsDto;
import com.yuanchu.mom.dto.DeviceTraceabilityManagementDto;
import com.yuanchu.mom.mapper.DeviceTraceabilityManagementDetailsMapper;
import com.yuanchu.mom.mapper.DeviceTraceabilityManagementMapper;
import com.yuanchu.mom.mapper.UserMapper;
import com.yuanchu.mom.pojo.DeviceTraceabilityManagement;
import com.yuanchu.mom.pojo.DeviceTraceabilityManagementDetails;
import com.yuanchu.mom.pojo.User;
import com.yuanchu.mom.service.DeviceTraceabilityManagementDetailsService;
import com.yuanchu.mom.service.DeviceTraceabilityManagementService;
import com.yuanchu.mom.utils.HackLoopTableRenderPolicy;
import com.yuanchu.mom.utils.UserUtils;
import com.yuanchu.mom.utils.QueryWrappers;
import com.yuanchu.mom.vo.Result;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.beans.BeanUtils;
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.time.LocalDateTime;
import java.util.HashMap;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 设备量值溯源计划表 服务实现类
 * </p>
 *
 * @author 江苏鵷雏网络科技有限公司
 * @since 2024-12-20 02:27:44
 */
@Service
public class DeviceTraceabilityManagementServiceImpl extends ServiceImpl<DeviceTraceabilityManagementMapper, DeviceTraceabilityManagement> implements DeviceTraceabilityManagementService {

    @Resource
    private DeviceTraceabilityManagementDetailsService deviceTraceabilityManagementDetailsService;

    @Resource
    private DeviceTraceabilityManagementDetailsMapper deviceTraceabilityManagementDetailsMapper;

    @Resource
    private GetLook getLook;

    @Resource
    private UserMapper userMapper;

    /**
     * 分页查询设备量值溯源计划
     *
     * @param page
     * @return
     */
    @Override
    public Result<IPage<DeviceTraceabilityManagement>> selectDeviceTraceabilityManagementByPage(IPage page, DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
        IPage<DeviceTraceabilityManagement> iPage = baseMapper.selectDeviceParameterPage(page, QueryWrappers.queryWrappers(deviceTraceabilityManagementDto));
        return Result.success(iPage);
    }


    /**
     * 新增设备量值溯源计划
     *
     * @param deviceTraceabilityManagementDto 设备量值溯源计划
     */
    @Override
    public Result addTraceabilityManagement(DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
        Integer userId = getLook.selectPowerByMethodAndUserId(null).get("userId");
        User user = userMapper.selectById(userId);
        deviceTraceabilityManagementDto.setCompilerId(userId);
        deviceTraceabilityManagementDto.setCompiler(user.getName());
        deviceTraceabilityManagementDto.setDatePreparation(LocalDateTime.now());

        // 查询审核人id
        if (deviceTraceabilityManagementDto.getAuditId() != null) {
            User auditUser = userMapper.selectById(deviceTraceabilityManagementDto.getAuditId());
            deviceTraceabilityManagementDto.setAudit(auditUser.getName());
        }
        this.saveOrUpdate(deviceTraceabilityManagementDto);

        // 详情赋值并保存
        List<DeviceTraceabilityManagementDetailsDto> deviceTraceabilityManagementDetails = deviceTraceabilityManagementDto.getDeviceTraceabilityManagementDetails();
        if (CollectionUtils.isNotEmpty(deviceTraceabilityManagementDetails)) { // 详情不为空
            List<DeviceTraceabilityManagementDetails> collect = deviceTraceabilityManagementDetails.stream().map(detailsDto -> {
                detailsDto.setTraceabilityManagementId(deviceTraceabilityManagementDto.getTraceabilityManagementId()); // 量值溯源计划ID
                DeviceTraceabilityManagementDetails details = new DeviceTraceabilityManagementDetails();
                BeanUtils.copyProperties(detailsDto, details);
                return details;
            }).collect(Collectors.toList());
            deviceTraceabilityManagementDetailsService.saveBatch(collect);
        }

        return Result.success();
    }

    /**
     * 修改设备量值溯源计划
     *
     * @param deviceTraceabilityManagementDto 设备量值溯源计划
     */
    @Override
    public Result updateTraceabilityManagement(DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
        // 查询审核人id
        if (deviceTraceabilityManagementDto.getAuditId() != null) {
            User auditUser = userMapper.selectById(deviceTraceabilityManagementDto.getAuditId());
            deviceTraceabilityManagementDto.setAudit(auditUser.getName());
        }
        this.saveOrUpdate(deviceTraceabilityManagementDto);

        // 删除原本的详情
        deviceTraceabilityManagementDetailsService.remove(Wrappers.<DeviceTraceabilityManagementDetails>lambdaQuery().eq(DeviceTraceabilityManagementDetails::getTraceabilityManagementId, deviceTraceabilityManagementDto.getTraceabilityManagementId()));
        // 详情赋值并保存
        List<DeviceTraceabilityManagementDetailsDto> deviceTraceabilityManagementDetails = deviceTraceabilityManagementDto.getDeviceTraceabilityManagementDetails();
        if (CollectionUtils.isNotEmpty(deviceTraceabilityManagementDetails)) { // 详情不为空
            List<DeviceTraceabilityManagementDetails> collect = deviceTraceabilityManagementDetails.stream().map(detailsDto -> {
                detailsDto.setTraceabilityManagementId(deviceTraceabilityManagementDto.getTraceabilityManagementId()); // 量值溯源计划ID
                DeviceTraceabilityManagementDetails details = new DeviceTraceabilityManagementDetails();
                BeanUtils.copyProperties(detailsDto, details);
                return details;
            }).collect(Collectors.toList());
            deviceTraceabilityManagementDetailsService.saveBatch(collect);
        }
        return Result.success();
    }

    /**
     * 删除设备量值溯源计划
     *
     * @param deviceTraceabilityManagementDto 设备量值溯源计划
     */
    @Override
    public Result deleteTraceabilityManagement(DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
        this.removeById(deviceTraceabilityManagementDto);
        deviceTraceabilityManagementDetailsService.remove(Wrappers.<DeviceTraceabilityManagementDetails>lambdaQuery().eq(DeviceTraceabilityManagementDetails::getTraceabilityManagementId, deviceTraceabilityManagementDto.getTraceabilityManagementId()));
        return Result.success();
    }

    /**
     * 查询设备量值溯源计划详情
     *
     * @param traceabilityManagementId 设备量值溯源计划id
     */
    @Override
    public Result<DeviceTraceabilityManagementDto> getTraceabilityManagementDetail(Integer traceabilityManagementId) {
        // 查询设备量值溯源计划
        DeviceTraceabilityManagement deviceTraceabilityManagement = baseMapper.selectById(traceabilityManagementId);
        // 查询详情
        DeviceTraceabilityManagementDto deviceTraceabilityManagementDto = new DeviceTraceabilityManagementDto();
        BeanUtils.copyProperties(deviceTraceabilityManagement, deviceTraceabilityManagementDto);
        deviceTraceabilityManagementDto.setDeviceTraceabilityManagementDetails(deviceTraceabilityManagementDetailsMapper.deviceTraceabilityManagementDetailsList(traceabilityManagementId));
        return Result.success(deviceTraceabilityManagementDto);
    }

    /**
     * 审核设备量值溯源计划
     *
     * @param deviceTraceabilityManagementDto 设备量值溯源计划
     */
    @Override
    public Result reviewTraceabilityManagementStatus(DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
        Integer userId = getLook.selectPowerByMethodAndUserId(null).get("userId");
        User user = userMapper.selectById(userId);

        LambdaUpdateWrapper<DeviceTraceabilityManagement> wrapper = Wrappers.<DeviceTraceabilityManagement>lambdaUpdate()
                .eq(DeviceTraceabilityManagement::getTraceabilityManagementId, deviceTraceabilityManagementDto.getTraceabilityManagementId())
                .set(DeviceTraceabilityManagement::getAuditDate, LocalDateTime.now())
                .set(DeviceTraceabilityManagement::getAuditId, user.getId())
                .set(DeviceTraceabilityManagement::getAudit, user.getName())
                .set(DeviceTraceabilityManagement::getStatus, deviceTraceabilityManagementDto.getStatus())        // 审核状态
                .set(DeviceTraceabilityManagement::getAuditRemark, deviceTraceabilityManagementDto.getAuditRemark());// 审核备注

        // 为0清除审核人
        if (deviceTraceabilityManagementDto.getStatus().equals(0)) {
            wrapper.set(DeviceTraceabilityManagement::getAuditId, null)
                    .set(DeviceTraceabilityManagement::getAudit, null);
        }
        this.update(wrapper); // 更新
        return Result.success();
    }

    /**
     * 导出设备量值溯源计划
     *
     * @param traceabilityManagementId 设备量值溯源计划id
     */
    @Override
    public Result exportDeviceTraceabilityManagementDto(Integer traceabilityManagementId, HttpServletResponse response) {
        // 查询设备保养计划
        DeviceTraceabilityManagementDto deviceTraceabilityManagement = baseMapper.selectDeviceTraceabilityManagementById(traceabilityManagementId);

        // 查询设备保养计划详情
        List<DeviceTraceabilityManagementDetailsDto> deviceTraceabilityManagementDetailsDtoList = deviceTraceabilityManagementDetailsMapper.deviceTraceabilityManagementDetailsList(traceabilityManagementId);
        // 设置序号
        deviceTraceabilityManagementDetailsDtoList.forEach(deviceTraceabilityManagementDetails -> {
            deviceTraceabilityManagementDetails.setIndex(deviceTraceabilityManagementDetailsDtoList.indexOf(deviceTraceabilityManagementDetails) + 1);
        });

        // 获取路径
        InputStream inputStream = this.getClass().getResourceAsStream("/static/word/traceability-management-details.docx");
        Configure configure = Configure.builder()
                .bind("deviceTraceabilityManagementDetailsDtoList", new HackLoopTableRenderPolicy())
                .build();
        XWPFTemplate template = XWPFTemplate.compile(inputStream, configure).render(
                new HashMap<String, Object>() {{
                    put("deviceTraceabilityManagement", deviceTraceabilityManagement);
                    put("deviceTraceabilityManagementDetailsDtoList", deviceTraceabilityManagementDetailsDtoList);
                    // 编制人签名地址
                    put("compilerUrl", UserUtils.getFinalUserSignatureUrl(deviceTraceabilityManagement.getCompilerId()));
                    // 审核人签名地址
                    put("auditUrl", UserUtils.getFinalUserSignatureUrl(deviceTraceabilityManagement.getAuditId()));
                }});

        try {
            response.setContentType("application/msword");
            String finalFileName = deviceTraceabilityManagement.getFileName() == null ? "" : deviceTraceabilityManagement.getFileName() + "_";
            String fileName = URLEncoder.encode(
                    finalFileName+ "设备量值溯源计划", "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("导出失败");
        }
        return Result.success();
    }

}