package cn.iocoder.yudao.module.mes.service.pd.documentaudit;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.module.mes.controller.admin.pd.documentaudit.vo.MesPdDocumentAuditPageReqVO;
import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.module.mes.controller.admin.pd.documentaudit.vo.MesPdDocumentAuditSaveReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pd.document.MesPdDocumentDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pd.documentaudit.MesPdDocumentAuditDO;
import cn.iocoder.yudao.module.mes.dal.mysql.pd.documentaudit.MesPdDocumentAuditMapper;
import cn.iocoder.yudao.module.mes.enums.MesPdStatusConstants;
import cn.iocoder.yudao.module.mes.service.pd.document.MesPdDocumentService;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.PD_DOCUMENT_AUDIT_NOT_EXISTS;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.PD_DOCUMENT_AUDIT_STATUS_INVALID;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.PD_DOCUMENT_AUDIT_STATUS_REQUIRED;

/**
 * MES 资料审核 Service 实现类
 *
 * @author 超级管理员
 */
@Service
@Validated
public class MesPdDocumentAuditServiceImpl implements MesPdDocumentAuditService {

    @Resource
    private MesPdDocumentAuditMapper documentAuditMapper;

    @Resource
    private MesPdDocumentService documentService;

    @Override
    public Long createDocumentAudit(MesPdDocumentAuditSaveReqVO createReqVO) {
        // 校验资料存在
        documentService.getDocument(createReqVO.getDocumentId());
        // 插入
        MesPdDocumentAuditDO documentAudit = BeanUtils.toBean(createReqVO, MesPdDocumentAuditDO.class);
        documentAuditMapper.insert(documentAudit);
        return documentAudit.getId();
    }

    @Override
    public void updateDocumentAudit(MesPdDocumentAuditSaveReqVO updateReqVO) {
        // 校验存在
        validateDocumentAuditExists(updateReqVO.getId());
        // 校验审核结果：自动审批模式下前端会隐藏「审核结果」字段导致漏传，必须在此显式拦截，避免拆箱 NPE
        Integer auditStatus = updateReqVO.getAuditStatus();
        if (auditStatus == null) {
            throw exception(PD_DOCUMENT_AUDIT_STATUS_REQUIRED);
        }
        if (!Objects.equals(auditStatus, MesPdStatusConstants.DOCUMENT_AUDIT_PASS)
                && !Objects.equals(auditStatus, MesPdStatusConstants.DOCUMENT_AUDIT_REJECT)) {
            throw exception(PD_DOCUMENT_AUDIT_STATUS_INVALID);
        }
        // 审核人与审核时间统一由后端记录，不采用前端传值
        updateReqVO.setReviewer(SecurityFrameworkUtils.getLoginUserId());
        updateReqVO.setAuditTime(LocalDateTime.now());
        // 更新
        MesPdDocumentAuditDO updateObj = BeanUtils.toBean(updateReqVO, MesPdDocumentAuditDO.class);
        documentAuditMapper.updateById(updateObj);
    }

    @Override
    public void deleteDocumentAudit(Long id) {
        // 校验存在
        validateDocumentAuditExists(id);
        // 删除
        documentAuditMapper.deleteById(id);
    }

    @Override
    public MesPdDocumentAuditDO getDocumentAudit(Long id) {
        return documentAuditMapper.selectById(id);
    }

    @Override
    public PageResult<MesPdDocumentAuditDO> getDocumentAuditPage(MesPdDocumentAuditPageReqVO pageReqVO) {
        // 按项目筛选：先查出项目下所有资料，再按资料 ID 过滤审核记录
        if (pageReqVO.getProjectId() != null) {
            List<MesPdDocumentDO> documents = documentService.getDocumentListByProjectId(
                    pageReqVO.getProjectId());
            if (CollUtil.isEmpty(documents)) {
                return PageResult.empty();
            }
            List<Long> documentIds = documents.stream()
                    .map(MesPdDocumentDO::getId).collect(Collectors.toList());
            return documentAuditMapper.selectPage(pageReqVO, documentIds);
        }
        return documentAuditMapper.selectPage(pageReqVO);
    }

    @Override
    public List<MesPdDocumentAuditDO> getDocumentAuditListByDocumentId(Long documentId) {
        return documentAuditMapper.selectListByDocumentId(documentId);
    }

    private MesPdDocumentAuditDO validateDocumentAuditExists(Long id) {
        MesPdDocumentAuditDO documentAudit = documentAuditMapper.selectById(id);
        if (documentAudit == null) {
            throw exception(PD_DOCUMENT_AUDIT_NOT_EXISTS);
        }
        return documentAudit;
    }

}